package tree import ( "crypto/sha256" "github.com/TrueCloudLab/frostfs-node/cmd/frostfs-cli/internal/commonflags" "github.com/TrueCloudLab/frostfs-node/cmd/frostfs-cli/internal/key" commonCmd "github.com/TrueCloudLab/frostfs-node/cmd/internal/common" "github.com/TrueCloudLab/frostfs-node/pkg/services/tree" cid "github.com/TrueCloudLab/frostfs-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(commonflags.CIDFlag, "", commonflags.CIDFlagUsage) _ = listCmd.MarkFlagRequired(commonflags.CIDFlag) _ = cobra.MarkFlagRequired(ff, commonflags.RPC) } func list(cmd *cobra.Command, _ []string) { pk := key.GetOrGenerate(cmd) cidString, _ := cmd.Flags().GetString(commonflags.CIDFlag) var cnr cid.ID err := cnr.DecodeString(cidString) commonCmd.ExitOnErr(cmd, "decode container ID string: %w", err) ctx := cmd.Context() cli, err := _client(ctx) commonCmd.ExitOnErr(cmd, "client: %w", err) rawCID := make([]byte, sha256.Size) cnr.Encode(rawCID) req := &tree.TreeListRequest{ Body: &tree.TreeListRequest_Body{ ContainerId: rawCID, }, } commonCmd.ExitOnErr(cmd, "message signing: %w", tree.SignMessage(req, pk)) resp, err := cli.TreeList(ctx, req) commonCmd.ExitOnErr(cmd, "rpc call: %w", err) for _, treeID := range resp.GetBody().GetIds() { cmd.Println(treeID) } }