[#1332] cli: Add tree list

Signed-off-by: Pavel Karpy <carpawell@nspcc.ru>
This commit is contained in:
Pavel Karpy 2022-10-18 16:41:10 +03:00 committed by Pavel Karpy
parent 6d4beea187
commit 1805c6606d
2 changed files with 65 additions and 0 deletions

View file

@ -0,0 +1,63 @@
package tree
import (
"crypto/sha256"
"github.com/nspcc-dev/neofs-node/cmd/neofs-cli/internal/common"
"github.com/nspcc-dev/neofs-node/cmd/neofs-cli/internal/commonflags"
"github.com/nspcc-dev/neofs-node/cmd/neofs-cli/internal/key"
"github.com/nspcc-dev/neofs-node/pkg/services/tree"
cid "github.com/nspcc-dev/neofs-sdk-go/container/id"
"github.com/spf13/cobra"
)
var listCmd = &cobra.Command{
Use: "list",
Short: "Get tree IDs",
Run: list,
PersistentPreRun: func(cmd *cobra.Command, _ []string) {
commonflags.Bind(cmd)
},
}
func initListCmd() {
commonflags.Init(listCmd)
ff := listCmd.Flags()
ff.String(containerIDFlagKey, "", "Container ID")
_ = listCmd.MarkFlagRequired(containerIDFlagKey)
_ = cobra.MarkFlagRequired(ff, commonflags.RPC)
}
func list(cmd *cobra.Command, _ []string) {
pk := key.GetOrGenerate(cmd)
cidString, _ := cmd.Flags().GetString(containerIDFlagKey)
var cnr cid.ID
err := cnr.DecodeString(cidString)
common.ExitOnErr(cmd, "decode container ID string: %w", err)
ctx := cmd.Context()
cli, err := _client(ctx)
common.ExitOnErr(cmd, "client: %w", err)
rawCID := make([]byte, sha256.Size)
cnr.Encode(rawCID)
req := &tree.TreeListRequest{
Body: &tree.TreeListRequest_Body{
ContainerId: rawCID,
},
}
common.ExitOnErr(cmd, "message signing: %w", tree.SignMessage(req, pk))
resp, err := cli.TreeList(ctx, req)
common.ExitOnErr(cmd, "rpc call: %w", err)
for _, treeID := range resp.GetBody().GetIds() {
cmd.Println(treeID)
}
}

View file

@ -13,10 +13,12 @@ func init() {
Cmd.AddCommand(addCmd) Cmd.AddCommand(addCmd)
Cmd.AddCommand(getByPathCmd) Cmd.AddCommand(getByPathCmd)
Cmd.AddCommand(addByPathCmd) Cmd.AddCommand(addByPathCmd)
Cmd.AddCommand(listCmd)
initAddCmd() initAddCmd()
initGetByPathCmd() initGetByPathCmd()
initAddByPathCmd() initAddByPathCmd()
initListCmd()
} }
const ( const (