feat: add handler function for the commands

Signed-off-by: amands98 <amandeepsm.in@gmail.com>
goharbor · May 26, 2024 · 887d550 · 887d550
1 parent a502930
commit 887d550
Show file tree

Hide file tree

Showing 30 changed files with 637 additions and 520 deletions.
diff --git a/cmd/harbor/root/artifact/cmd.go b/cmd/harbor/root/artifact/cmd.go
@@ -1,7 +1,6 @@
 package artifact
 
 import (
-	"github.com/goharbor/harbor-cli/cmd/harbor/root/artifact/tags"
 	"github.com/spf13/cobra"
 )
 
@@ -19,7 +18,7 @@ func Artifact() *cobra.Command {
 		InfoArtifactCommmand(),
 		DeleteArtifactCommand(),
 		ScanArtifactCommand(),
-		tags.ArtifactTagsCmd(),
+		ArtifactTagsCmd(),
 	)
 
 	return cmd

diff --git a/cmd/harbor/root/artifact/delete.go b/cmd/harbor/root/artifact/delete.go
@@ -1,17 +1,13 @@
 package artifact
 
 import (
-	"context"
-
-	"github.com/goharbor/go-client/pkg/sdk/v2.0/client/artifact"
+	"github.com/goharbor/harbor-cli/pkg/api"
 	"github.com/goharbor/harbor-cli/pkg/utils"
 	log "github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
-	"github.com/spf13/viper"
 )
 
 func DeleteArtifactCommand() *cobra.Command {
-
 	cmd := &cobra.Command{
 		Use:   "delete",
 		Short: "delete an artifact",
@@ -21,12 +17,12 @@ func DeleteArtifactCommand() *cobra.Command {
 
 			if len(args) > 0 {
 				projectName, repoName, reference := utils.ParseProjectRepoReference(args[0])
-				err = runDeleteArtifact(projectName, repoName, reference)
+				err = api.RunDeleteArtifact(projectName, repoName, reference)
 			} else {
 				projectName := utils.GetProjectNameFromUser()
 				repoName := utils.GetRepoNameFromUser(projectName)
 				reference := utils.GetReferenceFromUser(repoName, projectName)
-				err = runDeleteArtifact(projectName, repoName, reference)
+				err = api.RunDeleteArtifact(projectName, repoName, reference)
 			}
 
 			if err != nil {
@@ -37,19 +33,3 @@ func DeleteArtifactCommand() *cobra.Command {
 
 	return cmd
 }
-
-func runDeleteArtifact(projectName, repoName, reference string) error {
-	credentialName := viper.GetString("current-credential-name")
-	client := utils.GetClientByCredentialName(credentialName)
-	ctx := context.Background()
-
-	_, err := client.Artifact.DeleteArtifact(ctx, &artifact.DeleteArtifactParams{ProjectName: projectName, RepositoryName: repoName, Reference: reference})
-
-	if err != nil {
-		return err
-	}
-
-	log.Infof("Artifact deleted successfully")
-
-	return nil
-}
diff --git a/cmd/harbor/root/artifact/info.go b/cmd/harbor/root/artifact/info.go
@@ -1,13 +1,10 @@
 package artifact
 
 import (
-	"context"
-
-	"github.com/goharbor/go-client/pkg/sdk/v2.0/client/artifact"
+	"github.com/goharbor/harbor-cli/pkg/api"
 	"github.com/goharbor/harbor-cli/pkg/utils"
 	log "github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
-	"github.com/spf13/viper"
 )
 
 func InfoArtifactCommmand() *cobra.Command {
@@ -22,12 +19,12 @@ func InfoArtifactCommmand() *cobra.Command {
 
 			if len(args) > 0 {
 				projectName, repoName, reference := utils.ParseProjectRepoReference(args[0])
-				err = runInfoArtifact(projectName, repoName, reference)
+				err = api.RunInfoArtifact(projectName, repoName, reference)
 			} else {
 				projectName := utils.GetProjectNameFromUser()
 				repoName := utils.GetRepoNameFromUser(projectName)
 				reference := utils.GetReferenceFromUser(repoName, projectName)
-				err = runInfoArtifact(projectName, repoName, reference)
+				err = api.RunInfoArtifact(projectName, repoName, reference)
 			}
 
 			if err != nil {
@@ -39,19 +36,3 @@ func InfoArtifactCommmand() *cobra.Command {
 
 	return cmd
 }
-
-func runInfoArtifact(projectName, repoName, reference string) error {
-	credentialName := viper.GetString("current-credential-name")
-	client := utils.GetClientByCredentialName(credentialName)
-	ctx := context.Background()
-
-	response, err := client.Artifact.GetArtifact(ctx, &artifact.GetArtifactParams{ProjectName: projectName, RepositoryName: repoName, Reference: reference})
-
-	if err != nil {
-		return err
-	}
-
-	utils.PrintPayloadInJSONFormat(response.Payload)
-
-	return nil
-}
diff --git a/cmd/harbor/root/artifact/list.go b/cmd/harbor/root/artifact/list.go
@@ -1,14 +1,10 @@
 package artifact
 
 import (
-	"context"
-	"fmt"
-
-	"github.com/goharbor/go-client/pkg/sdk/v2.0/client/artifact"
+	"github.com/goharbor/harbor-cli/pkg/api"
 	"github.com/goharbor/harbor-cli/pkg/utils"
 	log "github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
-	"github.com/spf13/viper"
 )
 
 func ListArtifactCommand() *cobra.Command {
@@ -21,11 +17,11 @@ func ListArtifactCommand() *cobra.Command {
 
 			if len(args) > 0 {
 				projectName, repoName := utils.ParseProjectRepo(args[0])
-				err = runListArtifact(projectName, repoName)
+				err = api.RunListArtifact(projectName, repoName)
 			} else {
 				projectName := utils.GetProjectNameFromUser()
 				repoName := utils.GetRepoNameFromUser(projectName)
-				err = runListArtifact(projectName, repoName)
+				err = api.RunListArtifact(projectName, repoName)
 			}
 
 			if err != nil {
@@ -37,20 +33,3 @@ func ListArtifactCommand() *cobra.Command {
 
 	return cmd
 }
-
-func runListArtifact(projectName, repoName string) error {
-	credentialName := viper.GetString("current-credential-name")
-	client := utils.GetClientByCredentialName(credentialName)
-	ctx := context.Background()
-
-	response, err := client.Artifact.ListArtifacts(ctx, &artifact.ListArtifactsParams{ProjectName: projectName, RepositoryName: repoName})
-
-	if err != nil {
-		return err
-	}
-
-	fmt.Println(response.Payload)
-
-	return nil
-
-}
diff --git a/cmd/harbor/root/artifact/scan.go b/cmd/harbor/root/artifact/scan.go
@@ -1,13 +1,10 @@
 package artifact
 
 import (
-	"context"
-
-	"github.com/goharbor/go-client/pkg/sdk/v2.0/client/scan"
+	"github.com/goharbor/harbor-cli/pkg/api"
 	"github.com/goharbor/harbor-cli/pkg/utils"
 	log "github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
-	"github.com/spf13/viper"
 )
 
 func ScanArtifactCommand() *cobra.Command {
@@ -38,12 +35,12 @@ func StartScanArtifactCommand() *cobra.Command {
 
 			if len(args) > 0 {
 				projectName, repoName, reference := utils.ParseProjectRepoReference(args[0])
-				err = runStartScanArtifact(projectName, repoName, reference)
+				err = api.StartScanArtifact(projectName, repoName, reference)
 			} else {
 				projectName := utils.GetProjectNameFromUser()
 				repoName := utils.GetRepoNameFromUser(projectName)
 				reference := utils.GetReferenceFromUser(repoName, projectName)
-				err = runStartScanArtifact(projectName, repoName, reference)
+				err = api.StartScanArtifact(projectName, repoName, reference)
 			}
 
 			if err != nil {
@@ -54,38 +51,6 @@ func StartScanArtifactCommand() *cobra.Command {
 	return cmd
 }
 
-func runStartScanArtifact(projectName, repoName, reference string) error {
-	credentialName := viper.GetString("current-credential-name")
-	client := utils.GetClientByCredentialName(credentialName)
-	ctx := context.Background()
-
-	_, err := client.Scan.ScanArtifact(ctx, &scan.ScanArtifactParams{ProjectName: projectName, RepositoryName: repoName, Reference: reference})
-
-	if err != nil {
-		return err
-	}
-
-	log.Infof("Scan started successfully")
-
-	return nil
-}
-
-func runStopScanArtifact(projectName, repoName, reference string) error {
-	credentialName := viper.GetString("current-credential-name")
-	client := utils.GetClientByCredentialName(credentialName)
-	ctx := context.Background()
-
-	_, err := client.Scan.StopScanArtifact(ctx, &scan.StopScanArtifactParams{ProjectName: projectName, RepositoryName: repoName, Reference: reference})
-
-	if err != nil {
-		return err
-	}
-
-	log.Infof("Scan stopped successfully")
-
-	return nil
-}
-
 func StopScanArtifactCommand() *cobra.Command {
 	cmd := &cobra.Command{
 		Use:     "stop",
@@ -97,12 +62,12 @@ func StopScanArtifactCommand() *cobra.Command {
 
 			if len(args) > 0 {
 				projectName, repoName, reference := utils.ParseProjectRepoReference(args[0])
-				err = runStopScanArtifact(projectName, repoName, reference)
+				err = api.StopScanArtifact(projectName, repoName, reference)
 			} else {
 				projectName := utils.GetProjectNameFromUser()
 				repoName := utils.GetRepoNameFromUser(projectName)
 				reference := utils.GetReferenceFromUser(repoName, projectName)
-				err = runStopScanArtifact(projectName, repoName, reference)
+				err = api.StopScanArtifact(projectName, repoName, reference)
 			}
 
 			if err != nil {

diff --git a/cmd/harbor/root/artifact/tags.go b/cmd/harbor/root/artifact/tags.go
@@ -0,0 +1,77 @@
+package artifact
+
+import (
+	"github.com/goharbor/harbor-cli/pkg/api"
+	"github.com/goharbor/harbor-cli/pkg/utils"
+	log "github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+)
+
+func ArtifactTagsCmd() *cobra.Command {
+
+	cmd := &cobra.Command{
+		Use:     "tags",
+		Short:   "Manage tags of an artifact",
+		Example: ` harbor artifact tags list <project>/<repository>/<reference>`,
+	}
+
+	cmd.AddCommand(
+		ListTagsCommand(),
+		DeleteTagsCmd(),
+	)
+
+	return cmd
+}
+
+func ListTagsCommand() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:     "list",
+		Short:   "List tags of an artifact",
+		Example: `harbor artifact tags list <project>/<repository>/<reference>`,
+		Run: func(cmd *cobra.Command, args []string) {
+			var err error
+			if len(args) > 0 {
+				projectName, repoName, reference := utils.ParseProjectRepoReference(args[0])
+				err = api.ListTags(projectName, repoName, reference)
+			} else {
+				projectName := utils.GetProjectNameFromUser()
+				repoName := utils.GetRepoNameFromUser(projectName)
+				reference := utils.GetReferenceFromUser(repoName, projectName)
+				err = api.ListTags(projectName, repoName, reference)
+			}
+			if err != nil {
+				log.Errorf("failed to list tags of an artifact: %v", err)
+			}
+
+		},
+	}
+
+	return cmd
+}
+
+func DeleteTagsCmd() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:     "delete",
+		Short:   "Delete a tag of an artifact",
+		Example: `harbor artifact tags delete <project>/<repository>/<reference> <tag>`,
+		Run: func(cmd *cobra.Command, args []string) {
+			var err error
+			if len(args) > 0 {
+				projectName, repoName, reference := utils.ParseProjectRepoReference(args[0])
+				tag := args[1]
+				err = api.DeleteTag(projectName, repoName, reference, tag)
+			} else {
+				projectName := utils.GetProjectNameFromUser()
+				repoName := utils.GetRepoNameFromUser(projectName)
+				reference := utils.GetReferenceFromUser(repoName, projectName)
+				tag := utils.GetTagFromUser(repoName, projectName, reference)
+				err = api.DeleteTag(projectName, repoName, reference, tag)
+			}
+			if err != nil {
+				log.Errorf("failed to delete a tag of an artifact: %v", err)
+			}
+		},
+	}
+
+	return cmd
+}
diff --git a/cmd/harbor/root/artifact/tags/cmd.go b/cmd/harbor/root/artifact/tags/cmd.go
diff --git a/cmd/harbor/root/artifact/tags/create.go b/cmd/harbor/root/artifact/tags/create.go