feat add security command

cmd/harbor/root/cmd.go

@@ -8,6 +8,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/registry"
+	"github.com/goharbor/harbor-cli/cmd/harbor/root/security"
 	repositry "github.com/goharbor/harbor-cli/cmd/harbor/root/repository"
 	"github.com/goharbor/harbor-cli/cmd/harbor/root/user"
 	"github.com/goharbor/harbor-cli/pkg/utils"
@@ -109,6 +110,7 @@ harbor help
 		repositry.Repository(),
 		user.User(),
 		artifact.Artifact(),
+		security.Security(),
 	)
 
 	return root

cmd/harbor/root/security/cmd.go

@@ -0,0 +1,17 @@
+package security
+
+import "github.com/spf13/cobra"
+
+func Security() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "security",
+		Short: "Manage security-related operations in Harbor",
+	}
+
+	cmd.AddCommand(
+		getSecuritySummaryCommand(),
+		listVulnerabilitiesCommand(),
+	)
+
+	return cmd
+}

cmd/harbor/root/security/summary.go

@@ -0,0 +1,23 @@
+package security
+
+import (
+	"fmt"
+	"github.com/goharbor/harbor-cli/pkg/api"
+	"github.com/spf13/cobra"
+)
+
+func getSecuritySummaryCommand() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "summary",
+		Short: "Get the security summary of the system",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			err := api.GetSecuritySummary()
+			if err != nil {
+				return fmt.Errorf("error getting security summary: %w", err)
+			}
+			return nil
+		},
+	}
+
+	return cmd
+}

cmd/harbor/root/security/vulnerabilities.go

@@ -0,0 +1,27 @@
+package security
+
+import (
+	"fmt"
+	"github.com/goharbor/harbor-cli/pkg/api"
+	"github.com/spf13/cobra"
+)
+
+func listVulnerabilitiesCommand() *cobra.Command {
+	var query string
+
+	cmd := &cobra.Command{
+		Use:   "vulnerabilities",
+		Short: "List vulnerabilities in the system",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			err := api.ListVulnerabilities(query)
+			if err != nil {
+				return fmt.Errorf("error listing vulnerabilities: %w", err)
+			}
+			return nil
+		},
+	}
+
+	cmd.Flags().StringVarP(&query, "query", "q", "", "Query condition for filtering vulnerabilities")
+
+	return cmd
+}

pkg/api/security_handler.go

@@ -0,0 +1,41 @@
+package api
+
+import (
+	"github.com/goharbor/go-client/pkg/sdk/v2.0/client/securityhub"
+	"github.com/goharbor/harbor-cli/pkg/utils"
+)
+
+func GetSecuritySummary() error {
+	ctx, client, err := utils.ContextWithClient()
+	if err != nil {
+		return err
+	}
+
+	params := &securityhub.GetSecuritySummaryParams{}
+	response, err := client.Securityhub.GetSecuritySummary(ctx, params)
+	if err != nil {
+		return err
+	}
+
+	utils.PrintPayloadInJSONFormat(response.Payload)
+	return nil
+}
+
+func ListVulnerabilities(query string) error {
+	ctx, client, err := utils.ContextWithClient()
+	if err != nil {
+		return err
+	}
+
+	params := &securityhub.ListVulnerabilitiesParams{
+		Q: &query,
+	}
+
+	response, err := client.Securityhub.ListVulnerabilities(ctx, params)
+	if err != nil {
+		return err
+	}
+
+	utils.PrintPayloadInJSONFormat(response.Payload)
+	return nil
+}