go-pulse/cmd/swarm/mru.go

// Copyright 2016 The go-ethereum Authors
// This file is part of go-ethereum.
//
// go-ethereum is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// go-ethereum is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with go-ethereum. If not, see <http://www.gnu.org/licenses/>.

// Command resource allows the user to create and update signed mutable resource updates
package main

import (
	"fmt"
	"strconv"
	"strings"

	"github.com/ethereum/go-ethereum/common/hexutil"

	"github.com/ethereum/go-ethereum/cmd/utils"
	swarm "github.com/ethereum/go-ethereum/swarm/api/client"
	"github.com/ethereum/go-ethereum/swarm/storage/mru"
	"gopkg.in/urfave/cli.v1"
)

func NewGenericSigner(ctx *cli.Context) mru.Signer {
	return mru.NewGenericSigner(getPrivKey(ctx))
}

// swarm resource create <frequency> [--name <name>] [--data <0x Hexdata> [--multihash=false]]
// swarm resource update <Manifest Address or ENS domain> <0x Hexdata> [--multihash=false]
// swarm resource info <Manifest Address or ENS domain>

func resourceCreate(ctx *cli.Context) {
	args := ctx.Args()

	var (
		bzzapi      = strings.TrimRight(ctx.GlobalString(SwarmApiFlag.Name), "/")
		client      = swarm.NewClient(bzzapi)
		multihash   = ctx.Bool(SwarmResourceMultihashFlag.Name)
		initialData = ctx.String(SwarmResourceDataOnCreateFlag.Name)
		name        = ctx.String(SwarmResourceNameFlag.Name)
	)

	if len(args) < 1 {
		fmt.Println("Incorrect number of arguments")
		cli.ShowCommandHelpAndExit(ctx, "create", 1)
		return
	}
	signer := NewGenericSigner(ctx)
	frequency, err := strconv.ParseUint(args[0], 10, 64)
	if err != nil {
		fmt.Printf("Frequency formatting error: %s\n", err.Error())
		cli.ShowCommandHelpAndExit(ctx, "create", 1)
		return
	}

	metadata := mru.ResourceMetadata{
		Name:      name,
		Frequency: frequency,
		Owner:     signer.Address(),
	}

	var newResourceRequest *mru.Request
	if initialData != "" {
		initialDataBytes, err := hexutil.Decode(initialData)
		if err != nil {
			fmt.Printf("Error parsing data: %s\n", err.Error())
			cli.ShowCommandHelpAndExit(ctx, "create", 1)
			return
		}
		newResourceRequest, err = mru.NewCreateUpdateRequest(&metadata)
		if err != nil {
			utils.Fatalf("Error creating new resource request: %s", err)
		}
		newResourceRequest.SetData(initialDataBytes, multihash)
		if err = newResourceRequest.Sign(signer); err != nil {
			utils.Fatalf("Error signing resource update: %s", err.Error())
		}
	} else {
		newResourceRequest, err = mru.NewCreateRequest(&metadata)
		if err != nil {
			utils.Fatalf("Error creating new resource request: %s", err)
		}
	}

	manifestAddress, err := client.CreateResource(newResourceRequest)
	if err != nil {
		utils.Fatalf("Error creating resource: %s", err.Error())
		return
	}
	fmt.Println(manifestAddress) // output manifest address to the user in a single line (useful for other commands to pick up)

}

func resourceUpdate(ctx *cli.Context) {
	args := ctx.Args()

	var (
		bzzapi    = strings.TrimRight(ctx.GlobalString(SwarmApiFlag.Name), "/")
		client    = swarm.NewClient(bzzapi)
		multihash = ctx.Bool(SwarmResourceMultihashFlag.Name)
	)

	if len(args) < 2 {
		fmt.Println("Incorrect number of arguments")
		cli.ShowCommandHelpAndExit(ctx, "update", 1)
		return
	}
	signer := NewGenericSigner(ctx)
	manifestAddressOrDomain := args[0]
	data, err := hexutil.Decode(args[1])
	if err != nil {
		utils.Fatalf("Error parsing data: %s", err.Error())
		return
	}

	// Retrieve resource status and metadata out of the manifest
	updateRequest, err := client.GetResourceMetadata(manifestAddressOrDomain)
	if err != nil {
		utils.Fatalf("Error retrieving resource status: %s", err.Error())
	}

	// set the new data
	updateRequest.SetData(data, multihash)

	// sign update
	if err = updateRequest.Sign(signer); err != nil {
		utils.Fatalf("Error signing resource update: %s", err.Error())
	}

	// post update
	err = client.UpdateResource(updateRequest)
	if err != nil {
		utils.Fatalf("Error updating resource: %s", err.Error())
		return
	}
}

func resourceInfo(ctx *cli.Context) {
	var (
		bzzapi = strings.TrimRight(ctx.GlobalString(SwarmApiFlag.Name), "/")
		client = swarm.NewClient(bzzapi)
	)
	args := ctx.Args()
	if len(args) < 1 {
		fmt.Println("Incorrect number of arguments.")
		cli.ShowCommandHelpAndExit(ctx, "info", 1)
		return
	}
	manifestAddressOrDomain := args[0]
	metadata, err := client.GetResourceMetadata(manifestAddressOrDomain)
	if err != nil {
		utils.Fatalf("Error retrieving resource metadata: %s", err.Error())
		return
	}
	encodedMetadata, err := metadata.MarshalJSON()
	if err != nil {
		utils.Fatalf("Error encoding metadata to JSON for display:%s", err)
	}
	fmt.Println(string(encodedMetadata))
}