forked from TrueCloudLab/frostfs-node
Ekaterina Lebedeva
cc3f762cf2
Add usage replacement for `container list -g` and verbose warning when using `-g` without `--owner`. Signed-off-by: Ekaterina Lebedeva <ekaterina.lebedeva@yadro.com>
115 lines
3.4 KiB
Go
115 lines
3.4 KiB
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
|
|
}
|