dag: add dag-pb support to put

License: MIT
Signed-off-by: Łukasz Magiera <magik6k@gmail.com>

core/coredag/dagpb.go

@@ -0,0 +1,66 @@
+package coredag
+
+import (
+	"io"
+	"io/ioutil"
+	"math"
+
+	"github.com/ipfs/go-ipfs/merkledag"
+
+	cid "gx/ipfs/QmTprEaAA2A9bst5XH7exuyi5KzNMK3SEDNN8rBDnKWcUS/go-cid"
+	mh "gx/ipfs/QmU9a9NV9RdPNwZQDYd5uKsm6N6LJLSvLbywDDYFbaaC6P/go-multihash"
+	node "gx/ipfs/QmYNyRZJBUYPNrLszFmrBrPJbsBh2vMsefz5gnDpB5M1P6/go-ipld-format"
+)
+
+func dagpbJSONParser(r io.Reader, mhType uint64, mhLen int) ([]node.Node, error) {
+	data, err := ioutil.ReadAll(r)
+	if err != nil {
+		return nil, err
+	}
+
+	nd := &merkledag.ProtoNode{}
+
+	err = nd.UnmarshalJSON(data)
+	if err != nil {
+		return nil, err
+	}
+
+	nd.SetPrefix(cidPrefix(mhType, mhLen))
+
+	return []node.Node{nd}, nil
+}
+
+func dagpbRawParser(r io.Reader, mhType uint64, mhLen int) ([]node.Node, error) {
+	data, err := ioutil.ReadAll(r)
+	if err != nil {
+		return nil, err
+	}
+
+	nd, err := merkledag.DecodeProtobuf(data)
+	if err != nil {
+		return nil, err
+	}
+
+	nd.SetPrefix(cidPrefix(mhType, mhLen))
+
+	return []node.Node{nd}, nil
+}
+
+func cidPrefix(mhType uint64, mhLen int) *cid.Prefix {
+	if mhType == math.MaxUint64 {
+		mhType = mh.SHA2_256
+	}
+
+	prefix := &cid.Prefix{
+		MhType:   mhType,
+		MhLength: mhLen,
+		Version:  1,
+		Codec:    cid.DagProtobuf,
+	}
+
+	if mhType == mh.SHA2_256 {
+		prefix.Version = 0
+	}
+
+	return prefix
+}

core/coredag/dagtransl.go

@@ -26,11 +26,17 @@ var DefaultInputEncParsers = InputEncParsers{
 var defaultJSONParsers = FormatParsers{
 	"cbor":     cborJSONParser,
 	"dag-cbor": cborJSONParser,
+
+	"protobuf": dagpbJSONParser,
+	"dag-pb":   dagpbJSONParser,
 }
 
 var defaultRawParsers = FormatParsers{
 	"cbor":     cborRawParser,
 	"dag-cbor": cborRawParser,
+
+	"protobuf": dagpbRawParser,
+	"dag-pb":   dagpbRawParser,
 }
 
 var defaultCborParsers = FormatParsers{