transitioned ipfs2 -> ipfs

core/commands/block.go

@@ -0,0 +1,131 @@
+package commands
+
+import (
+	"bytes"
+	"io/ioutil"
+	"time"
+
+	"github.com/jbenet/go-ipfs/Godeps/_workspace/src/code.google.com/p/go.net/context"
+
+	mh "github.com/jbenet/go-ipfs/Godeps/_workspace/src/github.com/jbenet/go-multihash"
+	"github.com/jbenet/go-ipfs/blocks"
+	cmds "github.com/jbenet/go-ipfs/commands"
+	u "github.com/jbenet/go-ipfs/util"
+)
+
+type Block struct {
+	Key    string
+	Length int
+}
+
+var blockCmd = &cmds.Command{
+	Helptext: cmds.HelpText{
+		Tagline: "Manipulate raw IPFS blocks",
+		ShortDescription: `
+'ipfs block' is a plumbing command used to manipulate raw ipfs blocks.
+Reads from stdin or writes to stdout, and <key> is a base58 encoded
+multihash.
+`,
+	},
+
+	Subcommands: map[string]*cmds.Command{
+		"get": blockGetCmd,
+		"put": blockPutCmd,
+	},
+}
+
+var blockGetCmd = &cmds.Command{
+	Helptext: cmds.HelpText{
+		Tagline: "Get a raw IPFS block",
+		ShortDescription: `
+'ipfs block get' is a plumbing command for retreiving raw ipfs blocks.
+It outputs to stdout, and <key> is a base58 encoded multihash.
+`,
+	},
+
+	Arguments: []cmds.Argument{
+		cmds.StringArg("key", true, false, "The base58 multihash of an existing block to get"),
+	},
+	Run: func(req cmds.Request) (interface{}, error) {
+		n, err := req.Context().GetNode()
+		if err != nil {
+			return nil, err
+		}
+
+		key := req.Arguments()[0]
+
+		if !u.IsValidHash(key) {
+			return nil, cmds.Error{"Not a valid hash", cmds.ErrClient}
+		}
+
+		h, err := mh.FromB58String(key)
+		if err != nil {
+			return nil, err
+		}
+
+		k := u.Key(h)
+		ctx, _ := context.WithTimeout(context.TODO(), time.Second*5)
+		b, err := n.Blocks.GetBlock(ctx, k)
+		if err != nil {
+			return nil, err
+		}
+		log.Debugf("BlockGet key: '%q'", b.Key())
+
+		return bytes.NewReader(b.Data), nil
+	},
+}
+
+var blockPutCmd = &cmds.Command{
+	Helptext: cmds.HelpText{
+		Tagline: "Stores input as an IPFS block",
+		ShortDescription: `
+ipfs block put is a plumbing command for storing raw ipfs blocks.
+It reads from stdin, and <key> is a base58 encoded multihash.
+`,
+	},
+
+	Arguments: []cmds.Argument{
+		cmds.FileArg("data", true, false, "The data to be stored as an IPFS block").EnableStdin(),
+	},
+	Run: func(req cmds.Request) (interface{}, error) {
+		n, err := req.Context().GetNode()
+		if err != nil {
+			return nil, err
+		}
+
+		file, err := req.Files().NextFile()
+		if err != nil {
+			return nil, err
+		}
+
+		data, err := ioutil.ReadAll(file)
+		if err != nil {
+			return nil, err
+		}
+
+		err = file.Close()
+		if err != nil {
+			return nil, err
+		}
+
+		b := blocks.NewBlock(data)
+		log.Debugf("BlockPut key: '%q'", b.Key())
+
+		k, err := n.Blocks.AddBlock(b)
+		if err != nil {
+			return nil, err
+		}
+
+		return &Block{
+			Key:    k.String(),
+			Length: len(data),
+		}, nil
+	},
+	Type: &Block{},
+	Marshalers: cmds.MarshalerMap{
+		cmds.Text: func(res cmds.Response) ([]byte, error) {
+			block := res.Output().(*Block)
+			return []byte(block.Key + "\n"), nil
+		},
+	},
+}