add raw support to the dag put command.

`dag get` still doesn't properly support raw output but that's a bigger issue. License: MIT Signed-off-by: Steven Allen <[email protected]>
ipfs · Oct 11, 2017 · 613508a · 613508a
1 parent 1e73c3f
commit 613508a
Show file tree

Hide file tree

Showing 3 changed files with 45 additions and 0 deletions.
diff --git a/core/coredag/dagtransl.go b/core/coredag/dagtransl.go
@@ -37,6 +37,8 @@ var defaultRawParsers = FormatParsers{
 
 	"protobuf": dagpbRawParser,
 	"dag-pb":   dagpbRawParser,
+
+	"raw": rawRawParser,
 }
 
 var defaultCborParsers = FormatParsers{

diff --git a/core/coredag/raw.go b/core/coredag/raw.go
@@ -0,0 +1,37 @@
+package coredag
+
+import (
+	"io"
+	"io/ioutil"
+	"math"
+
+	"github.com/ipfs/go-ipfs/merkledag"
+
+	cid "gx/ipfs/QmNp85zy9RLrQ5oQD4hPyS39ezrrXpcaa7R4Y9kxdWQLLQ/go-cid"
+	node "gx/ipfs/QmPN7cwmpcc4DWXb4KTB9dNAJgjuPY69h3npsMfhRrQL9c/go-ipld-format"
+	block "gx/ipfs/QmSn9Td7xgxm9EV7iEjTckpUWmWApggzPxu7eFGWkkpwin/go-block-format"
+	mh "gx/ipfs/QmU9a9NV9RdPNwZQDYd5uKsm6N6LJLSvLbywDDYFbaaC6P/go-multihash"
+)
+
+func rawRawParser(r io.Reader, mhType uint64, mhLen int) ([]node.Node, error) {
+	if mhType == math.MaxUint64 {
+		mhType = mh.SHA2_256
+	}
+
+	data, err := ioutil.ReadAll(r)
+	if err != nil {
+		return nil, err
+	}
+
+	h, err := mh.Sum(data, mhType, mhLen)
+	if err != nil {
+		return nil, err
+	}
+	c := cid.NewCidV1(cid.Raw, h)
+	blk, err := block.NewBlockWithCid(data, c)
+	if err != nil {
+		return nil, err
+	}
+	nd := &merkledag.RawNode{Block: blk}
+	return []node.Node{nd}, nil
+}
diff --git a/test/sharness/t0053-dag.sh b/test/sharness/t0053-dag.sh
@@ -167,6 +167,12 @@ test_dag_cmd() {
     test_cmp dag_put_exp dag_put_out
   '
 
+  test_expect_success "dag put with raw node works" '
+    echo "foo bar" > raw_node_in &&
+    HASH=$(ipfs dag put --format=raw --input-enc=raw -- raw_node_in) &&
+    ipfs block get "$HASH" > raw_node_out &&
+    test_cmp raw_node_in raw_node_out'
+
   test_expect_success "dag put multiple files" '
     printf {\"foo\":\"bar\"} > a.json &&
     printf {\"foo\":\"baz\"} > b.json &&