add: quota list command

Signed-off-by: bupd <bupdprasanth@gmail.com>

Author: bupd

cmd/harbor/root/cmd.go

@@ -7,6 +7,7 @@ import (
 
 	"github.com/goharbor/harbor-cli/cmd/harbor/root/artifact"
 	"github.com/goharbor/harbor-cli/cmd/harbor/root/project"
+	"github.com/goharbor/harbor-cli/cmd/harbor/root/quota"
 	"github.com/goharbor/harbor-cli/cmd/harbor/root/registry"
 	repositry "github.com/goharbor/harbor-cli/cmd/harbor/root/repository"
 	"github.com/goharbor/harbor-cli/cmd/harbor/root/user"
@@ -106,6 +107,7 @@ harbor help
 		repositry.Repository(),
 		user.User(),
 		artifact.Artifact(),
+		quota.Quota(),
 	)
 
 	return root

cmd/harbor/root/quota/cmd.go

@@ -0,0 +1,19 @@
+package quota
+
+import (
+	"github.com/spf13/cobra"
+)
+
+func Quota() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:     "quota",
+		Short:   "Manage quotas",
+		Long:    `Manage quotas`,
+		Example: `  harbor quota list`,
+	}
+	cmd.AddCommand(
+		ListQuotaCommand(),
+	)
+
+	return cmd
+}

cmd/harbor/root/quota/list.go

@@ -0,0 +1,48 @@
+package quota
+
+import (
+	"github.com/goharbor/harbor-cli/pkg/api"
+	"github.com/goharbor/harbor-cli/pkg/utils"
+	"github.com/goharbor/harbor-cli/pkg/views/quota/list"
+	log "github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+)
+
+// Lists the Quotas specified for each project
+func ListQuotaCommand() *cobra.Command {
+	var opts api.ListQuotaFlags
+
+	cmd := &cobra.Command{
+		Use:   "list",
+		Short: "list quota",
+		Run: func(cmd *cobra.Command, args []string) {
+			quota, err := api.ListQuota(opts)
+			if err != nil {
+				log.Fatalf("failed to get projects list: %v", err)
+			}
+			FormatFlag := viper.GetString("output-format")
+			if FormatFlag != "" {
+				utils.PrintPayloadInJSONFormat(quota)
+				return
+			}
+
+			list.ListQuotas(quota.Payload)
+		},
+	}
+
+	flags := cmd.Flags()
+	flags.Int64VarP(&opts.Page, "page", "", 1, "Page number")
+	flags.Int64VarP(&opts.PageSize, "page-size", "", 10, "Size of per page")
+	flags.StringVarP(&opts.Reference, "ref", "", "", "Reference type of quota")
+	flags.StringVarP(&opts.ReferenceID, "refid", "", "", "Reference ID of quota")
+	flags.StringVarP(
+		&opts.Sort,
+		"sort",
+		"",
+		"",
+		"Sort the resource list in ascending or descending order",
+	)
+
+	return cmd
+}

pkg/api/quota_handler.go

@@ -0,0 +1,28 @@
+package api
+
+import (
+	"github.com/goharbor/go-client/pkg/sdk/v2.0/client/quota"
+	"github.com/goharbor/harbor-cli/pkg/utils"
+)
+
+func ListQuota(opts ListQuotaFlags) (quota.ListQuotasOK, error) {
+	ctx, client, err := utils.ContextWithClient()
+	if err != nil {
+		return quota.ListQuotasOK{}, err
+	}
+
+	response, err := client.Quota.ListQuotas(
+		ctx,
+		&quota.ListQuotasParams{
+			Page:        &opts.Page,
+			PageSize:    &opts.PageSize,
+			Reference:   &opts.Reference,
+			ReferenceID: &opts.ReferenceID,
+			Sort:        &opts.Sort,
+		},
+	)
+	if err != nil {
+		return quota.ListQuotasOK{}, err
+	}
+	return *response, nil
+}

pkg/api/types.go

@@ -8,3 +8,11 @@ type ListFlags struct {
 	Sort     string
 	Public   bool
 }
+
+type ListQuotaFlags struct {
+	PageSize    int64
+	Page        int64
+	Sort        string
+	Reference   string
+	ReferenceID string
+}

pkg/views/quota/list/view.go

@@ -0,0 +1,103 @@
+package list
+
+import (
+	"fmt"
+	"os"
+	"strconv"
+
+	"github.com/charmbracelet/bubbles/table"
+	tea "github.com/charmbracelet/bubbletea"
+	"github.com/goharbor/go-client/pkg/sdk/v2.0/models"
+	"github.com/goharbor/harbor-cli/pkg/utils"
+	"github.com/goharbor/harbor-cli/pkg/views/base/tablelist"
+)
+
+var columns = []table.Column{
+	{Title: "ID", Width: 4},
+	{Title: "Project", Width: 12},
+	{Title: "Owner Name", Width: 12},
+	{Title: "Storage", Width: 32},
+	{Title: "Creation Time", Width: 20},
+}
+
+// Function to get project details
+func GetProjectDetails(ref models.QuotaRefObject) (string, string, error) {
+	if refMap, ok := ref.(map[string]interface{}); ok {
+		projectName, _ := refMap["name"].(string)
+		ownerName, _ := refMap["owner_name"].(string)
+		return projectName, ownerName, nil
+	}
+	return "", "", fmt.Errorf("Error: Ref is not of expected type")
+}
+
+// Function to convert bytes to human-readable storage
+func BytesToStorageString(bytes int64) string {
+	const (
+		mebibyte = 1024 * 1024
+		gibibyte = 1024 * mebibyte
+	)
+
+	mib := float64(bytes) / float64(mebibyte)
+
+	if mib >= 1024 {
+		gib := mib / 1024
+		return fmt.Sprintf("%.1f GiB", gib)
+	}
+
+	return fmt.Sprintf("%.2f MiB", mib)
+}
+
+// Function to calculate storage
+func CalculateStorage(hard models.ResourceList, used models.ResourceList) (string, string) {
+	var storageUsed, storageGiven string
+
+	if hard["storage"] == -1 {
+		storageGiven = "Unlimited"
+	} else {
+		storageGiven = BytesToStorageString(hard["storage"])
+	}
+
+	if used["storage"] == 0 {
+		storageUsed = "0 MiB"
+	} else {
+		storageUsed = BytesToStorageString(used["storage"])
+	}
+
+	return storageUsed, storageGiven
+}
+
+// Function to format storage
+func FormatStorage(hard models.ResourceList, used models.ResourceList) string {
+	storageUsed, storageGiven := CalculateStorage(hard, used)
+	return fmt.Sprintf("%v of %v", storageUsed, storageGiven)
+}
+
+// ListQuotas in table format
+func ListQuotas(quotas []*models.Quota) {
+	var rows []table.Row
+	for _, quota := range quotas {
+		projectName, ownerName, err := GetProjectDetails(quota.Ref)
+		if err != nil {
+			fmt.Println(err)
+			continue
+		}
+
+		storage := FormatStorage(quota.Hard, quota.Used)
+
+		createdTime, _ := utils.FormatCreatedTime(quota.CreationTime.String())
+		rows = append(rows, table.Row{
+			strconv.FormatInt(quota.ID, 10),
+			projectName,
+			ownerName,
+			storage,
+			createdTime,
+		})
+	}
+
+	m := tablelist.NewModel(columns, rows, len(rows))
+
+	if _, err := tea.NewProgram(m).Run(); err != nil {
+		fmt.Println("Error running program:", err)
+		os.Exit(1)
+	}
+}