kubo/core/coredag/dagpb.go
Steven Allen ec9fac7379 gx: update go-ipld-cbor
(might as well do this at the same time)

License: MIT
Signed-off-by: Steven Allen <steven@stebalien.com>
2018-11-02 13:17:44 -07:00

67 lines
1.3 KiB
Go

package coredag
import (
"io"
"io/ioutil"
"math"
"gx/ipfs/QmXyuFW7at4r9dxRAbrPU9JpHW5aqngAFyxvyhvYjzxRKS/go-merkledag"
cid "gx/ipfs/QmPSQnBKM9g7BaUcZCvswUJVscQ1ipjmwxN5PXCjkp9EQ7/go-cid"
mh "gx/ipfs/QmPnFwZ2JXKnXgMw8CdBPxn7FWh6LLdjUjxV1fKHuJnkr8/go-multihash"
ipld "gx/ipfs/QmR7TcHkR9nxkUorfi8XMTAMLUK7GiP64TWWBzY3aacc1o/go-ipld-format"
)
func dagpbJSONParser(r io.Reader, mhType uint64, mhLen int) ([]ipld.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.SetCidBuilder(cidPrefix(mhType, mhLen))
return []ipld.Node{nd}, nil
}
func dagpbRawParser(r io.Reader, mhType uint64, mhLen int) ([]ipld.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.SetCidBuilder(cidPrefix(mhType, mhLen))
return []ipld.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
}