Merge pull request #432 from hacdias/no-wrap

feat(cmd/car): add '--no-wrap' option to 'create' command
ipld · Jun 3, 2023 · 67cf6ad · 67cf6ad
2 parents 3f39582 + c9fd84c
commit 67cf6ad
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/cmd/car/car.go b/cmd/car/car.go
@@ -40,6 +40,10 @@ func main1() int {
 						Usage:     "The car file to write to",
 						TakesFile: true,
 					},
+					&cli.BoolFlag{
+						Name:  "no-wrap",
+						Usage: "Do not wrap the files in a directory",
+					},
 					&cli.IntFlag{
 						Name:  "version",
 						Value: 2,

diff --git a/cmd/car/create.go b/cmd/car/create.go
@@ -31,6 +31,10 @@ func CreateCar(c *cli.Context) error {
 		return fmt.Errorf("a file destination must be specified")
 	}
 
+	if c.Bool("no-wrap") && c.Args().Len() > 1 {
+		return fmt.Errorf("no-wrap cannot be set with multiple source locations")
+	}
+
 	// make a cid with the right length that we eventually will patch with the root.
 	hasher, err := multihash.GetHasher(multihash.SHA2_256)
 	if err != nil {
@@ -59,7 +63,7 @@ func CreateCar(c *cli.Context) error {
 	}
 
 	// Write the unixfs blocks into the store.
-	root, err := writeFiles(c.Context, cdest, c.Args().Slice()...)
+	root, err := writeFiles(c.Context, c.Bool("no-wrap"), cdest, c.Args().Slice()...)
 	if err != nil {
 		return err
 	}
@@ -71,7 +75,7 @@ func CreateCar(c *cli.Context) error {
 	return car.ReplaceRootsInFile(c.String("file"), []cid.Cid{root})
 }
 
-func writeFiles(ctx context.Context, bs *blockstore.ReadWrite, paths ...string) (cid.Cid, error) {
+func writeFiles(ctx context.Context, noWrap bool, bs *blockstore.ReadWrite, paths ...string) (cid.Cid, error) {
 	ls := cidlink.DefaultLinkSystem()
 	ls.TrustedStorage = true
 	ls.StorageReadOpener = func(_ ipld.LinkContext, l ipld.Link) (io.Reader, error) {
@@ -107,6 +111,13 @@ func writeFiles(ctx context.Context, bs *blockstore.ReadWrite, paths ...string)
 		if err != nil {
 			return cid.Undef, err
 		}
+		if noWrap {
+			rcl, ok := l.(cidlink.Link)
+			if !ok {
+				return cid.Undef, fmt.Errorf("could not interpret %s", l)
+			}
+			return rcl.Cid, nil
+		}
 		name := path.Base(p)
 		entry, err := builder.BuildUnixFSDirectoryEntry(name, int64(size), l)
 		if err != nil {