add: quota update command

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

Author: bupd

cmd/harbor/root/quota/cmd.go

@@ -14,6 +14,7 @@ func Quota() *cobra.Command {
 	cmd.AddCommand(
 		ListQuotaCommand(),
 		ViewQuotaCommand(),
+		UpdateQuotaCommand(),
 	)
 
 	return cmd

cmd/harbor/root/quota/update.go

@@ -0,0 +1,116 @@
+package quota
+
+import (
+	"errors"
+	"os"
+	"regexp"
+	"strconv"
+
+	"github.com/goharbor/go-client/pkg/sdk/v2.0/models"
+	"github.com/goharbor/harbor-cli/pkg/api"
+	"github.com/goharbor/harbor-cli/pkg/prompt"
+	"github.com/goharbor/harbor-cli/pkg/views/quota/update"
+	log "github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+)
+
+type QuotaUpdateReq struct {
+	// The new hard limits for the quota
+	Hard ResourceList `json:"hard,omitempty"`
+}
+
+type ResourceList map[string]int64
+
+// UpdateQuotaCommand updates the quota
+func UpdateQuotaCommand() *cobra.Command {
+	var (
+		quotaID int64
+		storage string
+	)
+
+	cmd := &cobra.Command{
+		Use:   "update [QuotaID]",
+		Short: "update project quotas for projects",
+		Args:  cobra.MaximumNArgs(1),
+		Run: func(cmd *cobra.Command, args []string) {
+			var err error
+			var storageValue int64
+
+			if len(args) > 0 {
+				quotaID, err = strconv.ParseInt(args[0], 10, 64)
+			} else {
+				quotaID = prompt.GetQuotaIDFromUser()
+			}
+
+			if err != nil {
+				log.Errorf("failed to parse registry id: %v", err)
+			}
+
+			if storage != "" {
+				if storage == "-1" {
+					storageValue = -1
+				} else {
+					storageValue, err = storageStringToBytes(storage)
+					if err != nil {
+						log.Errorf("failed to parse storage: %v", err)
+						os.Exit(1)
+					}
+				}
+			} else {
+				storage = update.UpdateQuotaView()
+				storageValue, err = storageStringToBytes(storage)
+			}
+
+			hardlimit := &models.QuotaUpdateReq{
+				Hard: models.ResourceList{"storage": storageValue},
+			}
+
+			err = api.UpdateQuota(quotaID, hardlimit)
+			if err != nil {
+				log.Errorf("failed to update quota: %v", err)
+				os.Exit(1)
+			}
+
+			log.Infof("quota updated successfully!")
+		},
+	}
+
+	flags := cmd.Flags()
+	flags.StringVarP(&storage, "storage", "", "", "Enter storage size (e.g., 50GiB, 200MiB, 4TiB)")
+
+	return cmd
+}
+
+func storageStringToBytes(storage string) (int64, error) {
+	// Define the conversion multipliers
+	multipliers := map[string]int64{
+		"MiB": 1024 * 1024,
+		"GiB": 1024 * 1024 * 1024,
+		"TiB": 1024 * 1024 * 1024 * 1024,
+	}
+
+	// Define the regex to parse the input string
+	re := regexp.MustCompile(`^(\d+)(MiB|GiB|TiB)$`)
+	matches := re.FindStringSubmatch(storage)
+	if matches == nil {
+		return 0, errors.New("invalid storage format")
+	}
+
+	// Extract the value and unit from the matches
+	valueStr, unit := matches[1], matches[2]
+	value, err := strconv.ParseInt(valueStr, 10, 64)
+	if err != nil {
+		return 0, err
+	}
+
+	// Calculate the value in bytes
+	bytes := value * multipliers[unit]
+
+	// Check if the value exceeds 1024 TB
+	maxBytes := 1024 * 1024 * 1024 * 1024 * 1024
+	if bytes > int64(maxBytes) {
+		return 0, errors.New("value exceeds 1024 TB")
+	}
+
+	return bytes, nil
+}

pkg/api/quota_handler.go

@@ -2,6 +2,7 @@ package api
 
 import (
 	"github.com/goharbor/go-client/pkg/sdk/v2.0/client/quota"
+	"github.com/goharbor/go-client/pkg/sdk/v2.0/models"
 	"github.com/goharbor/harbor-cli/pkg/utils"
 )
 
@@ -27,16 +28,33 @@ func ListQuota(opts ListQuotaFlags) (*quota.ListQuotasOK, error) {
 	return response, nil
 }
 
-func GetQuota(QuotaID int64) (*quota.GetQuotaOK, error) {
+func GetQuota(quotaID int64) (*quota.GetQuotaOK, error) {
 	ctx, client, err := utils.ContextWithClient()
 	if err != nil {
 		return nil, err
 	}
 
-	response, err := client.Quota.GetQuota(ctx, &quota.GetQuotaParams{ID: QuotaID})
+	response, err := client.Quota.GetQuota(ctx, &quota.GetQuotaParams{ID: quotaID})
 	if err != nil {
 		return nil, err
 	}
 
 	return response, nil
 }
+
+func UpdateQuota(quotaID int64, hardlimit *models.QuotaUpdateReq) error {
+	ctx, client, err := utils.ContextWithClient()
+	if err != nil {
+		return err
+	}
+
+	_, err = client.Quota.UpdateQuota(
+		ctx,
+		&quota.UpdateQuotaParams{ID: quotaID, Hard: hardlimit},
+	)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}

pkg/views/quota/update/view.go

@@ -0,0 +1,71 @@
+package update
+
+import (
+	"errors"
+	"fmt"
+	"strconv"
+
+	"github.com/charmbracelet/huh"
+	log "github.com/sirupsen/logrus"
+)
+
+type CreateView struct {
+	StorageUnit string
+	Value       int64
+}
+
+func UpdateQuotaView() string {
+	var (
+		value      string
+		createView CreateView
+	)
+
+	storageUnits := []string{"MiB", "GiB", "TiB"}
+
+	// Initialize a slice to hold select options
+	var storageUnitSelectOptions []huh.Option[string]
+
+	// Iterate over registryOptions to populate registrySelectOptions
+	for _, option := range storageUnits {
+		storageUnitSelectOptions = append(
+			storageUnitSelectOptions,
+			huh.NewOption(option, option),
+		)
+	}
+
+	theme := huh.ThemeCharm()
+	err := huh.NewForm(
+		huh.NewGroup(
+			huh.NewSelect[string]().
+				Title("Select a Storage Unit").
+				Value(&createView.StorageUnit).
+				Options(storageUnitSelectOptions...).
+				Validate(func(str string) error {
+					if str == "" {
+						return errors.New("Storage Type cannot be empty.")
+					}
+					return nil
+				}),
+
+			huh.NewInput().
+				Title("Quota Limit").
+				Value(&value).
+				Validate(func(str string) error {
+					if str == "" {
+						return errors.New("Quota Limits cannot be empty")
+					}
+					_, err := strconv.ParseInt(str, 10, 64)
+					if err != nil {
+						return errors.New("Quota limit must be a valid integer")
+					}
+					createView.Value, _ = strconv.ParseInt(value, 10, 64)
+					return nil
+				}),
+		),
+	).WithTheme(theme).Run()
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	return fmt.Sprintf("%v%v", createView.Value, createView.StorageUnit)
+}