frostfs-node/cmd/frostfs-cli/modules/container/list.go

package container

import (
	"strings"

	"git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/container"
	internalclient "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-cli/internal/client"
	"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-cli/internal/commonflags"
	"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-cli/internal/key"
	commonCmd "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/internal/common"
	containerSDK "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container"
	"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/user"
	"github.com/spf13/cobra"
)

// flags of list command.
const (
	flagListPrintAttr      = "with-attr"
	flagListContainerOwner = "owner"
	flagListName           = "name"

	generateKeyContainerUsage = commonflags.GenerateKeyUsage + ", should be used with --owner flag"
)

// flag vars of list command.
var (
	flagVarListPrintAttr      bool
	flagVarListContainerOwner string
	flagVarListName           string
)

var listContainersCmd = &cobra.Command{
	Use:   "list",
	Short: "List all created containers",
	Long:  "List all created containers",
	Run: func(cmd *cobra.Command, _ []string) {
		var idUser user.ID

		generateKey, _ := cmd.Flags().GetBool(commonflags.GenerateKey)
		if flagVarListContainerOwner == "" && generateKey {
			cmd.PrintErrln("WARN: using -g without --owner - output will be empty")
		}

		key := key.GetOrGenerate(cmd)

		if flagVarListContainerOwner == "" {
			user.IDFromKey(&idUser, key.PublicKey)
		} else {
			err := idUser.DecodeString(flagVarListContainerOwner)
			commonCmd.ExitOnErr(cmd, "invalid user ID: %w", err)
		}

		cli := internalclient.GetSDKClientByFlag(cmd, key, commonflags.RPC)

		var prm internalclient.ListContainersPrm
		prm.SetClient(cli)
		prm.Account = idUser

		res, err := internalclient.ListContainers(cmd.Context(), prm)
		commonCmd.ExitOnErr(cmd, "rpc error: %w", err)

		prmGet := internalclient.GetContainerPrm{
			Client: cli,
		}

		containerIDs := res.SortedIDList()
		for _, cnrID := range containerIDs {
			if flagVarListName == "" && !flagVarListPrintAttr {
				cmd.Println(cnrID.String())
				continue
			}

			cnrID := cnrID
			prmGet.ClientParams.ContainerID = &cnrID
			res, err := internalclient.GetContainer(cmd.Context(), prmGet)
			if err != nil {
				cmd.Printf("  failed to read attributes: %v\n", err)
				continue
			}

			cnr := res.Container()
			if cnrName := containerSDK.Name(cnr); flagVarListName != "" && cnrName != flagVarListName {
				continue
			}
			cmd.Println(cnrID.String())

			if flagVarListPrintAttr {
				cnr.IterateAttributes(func(key, val string) {
					if !strings.HasPrefix(key, container.SysAttributePrefix) && !strings.HasPrefix(key, container.SysAttributePrefixNeoFS) {
						// FIXME(@cthulhu-rider): https://git.frostfs.info/TrueCloudLab/frostfs-sdk-go/issues/97
						//	Use dedicated method to skip system attributes.
						cmd.Printf("  %s: %s\n", key, val)
					}
				})
			}
		}
	},
}

func initContainerListContainersCmd() {
	commonflags.Init(listContainersCmd)

	flags := listContainersCmd.Flags()

	flags.StringVar(&flagVarListName, flagListName, "",
		"List containers by the attribute name",
	)
	flags.StringVar(&flagVarListContainerOwner, flagListContainerOwner, "",
		"Owner of containers (omit to use owner from private key)",
	)
	flags.BoolVar(&flagVarListPrintAttr, flagListPrintAttr, false,
		"Request and print attributes of each container",
	)
	flags.Lookup(commonflags.GenerateKey).Usage = generateKeyContainerUsage
}