JCS-pub/jcsctl/cmd/userspace/ls.go

package userspace

import (
	"fmt"
	"strconv"

	"github.com/jedib0t/go-pretty/v6/table"
	"github.com/spf13/cobra"
	cliapi "gitlink.org.cn/cloudream/jcs-pub/client/sdk/api/v1"
	clitypes "gitlink.org.cn/cloudream/jcs-pub/client/types"
	"gitlink.org.cn/cloudream/jcs-pub/jcsctl/cmd"
)

func init() {
	var opt lsOpt
	cmd := cobra.Command{
		Use:  "ls",
		Args: cobra.MaximumNArgs(1),
		Run: func(c *cobra.Command, args []string) {
			ctx := cmd.GetCmdCtx(c)
			ls(c, ctx, opt, args)
		},
	}

	cmd.Flags().BoolVarP(&opt.ByID, "id", "", false, "按id查询")
	cmd.Flags().BoolVarP(&opt.ShowPassword, "password", "p", false, "显示密码信息，请在安全环境下使用")
	UserSpaceCmd.AddCommand(&cmd)
}

type lsOpt struct {
	ByID         bool
	ShowPassword bool
}

func ls(c *cobra.Command, ctx *cmd.CommandContext, opt lsOpt, args []string) {
	// 仅ls无参数
	if len(args) == 0 {
		resp, err := ctx.Client.UserSpaceGetAll()
		if err != nil {
			cmd.ErrorExitln(err.Error())
			return
		}

		fmt.Printf("total: %d\n", len(resp.UserSpaces))
		tb := table.NewWriter()
		tb.AppendHeader(table.Row{"ID", "Name", "StorageType"})
		for _, userSpace := range resp.UserSpaces {
			tb.AppendRow(table.Row{userSpace.UserSpaceID, userSpace.Name, userSpace.Storage.GetStorageType()})
		}
		fmt.Println(tb.Render())
		return
	}

	searchKey := args[0]
	var userSpace *clitypes.UserSpace
	if opt.ByID {
		id, err := strconv.Atoi(searchKey)
		if err != nil {
			cmd.ErrorExitln("ID必须是数字")
			return
		}

		result, err := ctx.Client.UserSpaceGet(cliapi.UserSpaceGet{
			UserSpaceID: clitypes.UserSpaceID(id),
		})
		if err != nil {
			cmd.ErrorExitln(err.Error())
			return
		}
		userSpace = &result.UserSpace

	} else {
		result, err := ctx.Client.UserSpaceGetByName(cliapi.UserSpaceGetByName{
			Name: searchKey,
		})
		if err != nil {
			cmd.ErrorExitln(err.Error())
			return
		}
		userSpace = &result.UserSpace
	}

	if userSpace == nil {
		cmd.ErrorExitln(fmt.Sprintf("未找到匹配的云存储: %s", searchKey))
		return
	}

	fmt.Println("\n\033[1;36m云存储详情\033[0m")
	fmt.Println("----------------------------------")
	fmt.Printf("\033[1m%-8s\033[0m %d\n", "ID:", userSpace.UserSpaceID)
	fmt.Printf("\033[1m%-8s\033[0m %s\n", "名称:", userSpace.Name)
	fmt.Printf("\033[1m%-8s\033[0m %s\n", "类型:", userSpace.Storage.GetStorageType())
	fmt.Printf("\033[1m%-8s\033[0m %s\n", "Location:", userSpace.Storage.GetLocation().Location)
	if opt.ShowPassword {
		fmt.Printf("\033[1m%-8s\033[0m %s\n", "Credential:", userSpace.Credential.String(true))
	} else {
		fmt.Printf("\033[1m%-8s\033[0m %s\n", "Credential:", userSpace.Credential.String(false))
	}

	if len(userSpace.Features) > 0 {
		fmt.Printf("\033[1m%-8s\033[0m %s\n", "Features:", userSpace.Features[0].GetFeatureType())
	}
	fmt.Printf("\033[1m%-8s\033[0m %s\n", "WorkingDir:", userSpace.WorkingDir)
	fmt.Println("----------------------------------")

}