132 lines
2.8 KiB
Go
132 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/base64"
|
|
"encoding/hex"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
|
|
"github.com/filecoin-project/lotus/blockstore"
|
|
"github.com/ipld/go-car"
|
|
"golang.org/x/net/context"
|
|
|
|
"github.com/filecoin-project/go-state-types/abi"
|
|
"github.com/ipfs/go-cid"
|
|
mh "github.com/multiformats/go-multihash"
|
|
"github.com/urfave/cli/v2"
|
|
"golang.org/x/xerrors"
|
|
)
|
|
|
|
var cidCmd = &cli.Command{
|
|
Name: "cid",
|
|
Usage: "Cid command",
|
|
Subcommands: cli.Commands{
|
|
cidIdCmd,
|
|
cidFromCarCmd,
|
|
},
|
|
}
|
|
|
|
var cidIdCmd = &cli.Command{
|
|
Name: "id",
|
|
Usage: "Create identity CID from hex or base64 data",
|
|
ArgsUsage: "[data]",
|
|
Flags: []cli.Flag{
|
|
&cli.StringFlag{
|
|
Name: "encoding",
|
|
Aliases: []string{"e"},
|
|
Value: "base64",
|
|
Usage: "specify input encoding to parse",
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "codec",
|
|
Value: "id",
|
|
Usage: "multicodec-packed content types: abi or id",
|
|
},
|
|
},
|
|
Action: func(cctx *cli.Context) error {
|
|
if !cctx.Args().Present() {
|
|
return fmt.Errorf("must specify data")
|
|
}
|
|
|
|
var dec []byte
|
|
switch cctx.String("encoding") {
|
|
case "base64":
|
|
data, err := base64.StdEncoding.DecodeString(cctx.Args().First())
|
|
if err != nil {
|
|
return xerrors.Errorf("decoding base64 value: %w", err)
|
|
}
|
|
dec = data
|
|
case "hex", "x":
|
|
data, err := hex.DecodeString(cctx.Args().First())
|
|
if err != nil {
|
|
return xerrors.Errorf("decoding hex value: %w", err)
|
|
}
|
|
dec = data
|
|
case "raw", "r":
|
|
dec = []byte(cctx.Args().First())
|
|
default:
|
|
return xerrors.Errorf("unrecognized encoding: %s", cctx.String("encoding"))
|
|
}
|
|
|
|
switch cctx.String("codec") {
|
|
case "abi":
|
|
aCid, err := abi.CidBuilder.Sum(dec)
|
|
if err != nil {
|
|
return xerrors.Errorf("cidBuilder abi: %w", err)
|
|
}
|
|
fmt.Println(aCid)
|
|
case "id":
|
|
builder := cid.V1Builder{Codec: cid.Raw, MhType: mh.IDENTITY}
|
|
rCid, err := builder.Sum(dec)
|
|
if err != nil {
|
|
return xerrors.Errorf("cidBuilder raw: %w", err)
|
|
}
|
|
fmt.Println(rCid)
|
|
default:
|
|
return xerrors.Errorf("unrecognized codec: %s", cctx.String("codec"))
|
|
}
|
|
|
|
return nil
|
|
},
|
|
}
|
|
|
|
var cidFromCarCmd = &cli.Command{
|
|
Name: "manifest-cid-from-car",
|
|
Usage: "Get the manifest CID from a car file",
|
|
ArgsUsage: "[path]",
|
|
Action: func(cctx *cli.Context) error {
|
|
ctx := context.TODO()
|
|
|
|
cf := cctx.Args().Get(0)
|
|
f, err := os.OpenFile(cf, os.O_RDONLY, 0664)
|
|
if err != nil {
|
|
return xerrors.Errorf("opening the car file: %w", err)
|
|
}
|
|
|
|
bs := blockstore.NewMemory()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
data, err := io.ReadAll(f)
|
|
if err != nil {
|
|
return xerrors.Errorf("error reading car file: %w", err)
|
|
}
|
|
|
|
blobr := bytes.NewReader(data)
|
|
|
|
hdr, err := car.LoadCar(ctx, bs, blobr)
|
|
if err != nil {
|
|
return xerrors.Errorf("error loading car file: %w", err)
|
|
}
|
|
|
|
manifestCid := hdr.Roots[0]
|
|
|
|
fmt.Printf("Manifest CID: %s\n", manifestCid.String())
|
|
|
|
return nil
|
|
},
|
|
}
|