refactor(artifact): refactoring scan and tag subcommands under artifacts (#700)

Co-authored-by: Prasanth Baskar <bupdprasanth@gmail.com>

Author: NucleoFusion

cmd/harbor/root/artifact/cmd.go

@@ -15,6 +15,8 @@ package artifact
 
 import (
 	"github.com/goharbor/harbor-cli/cmd/harbor/root/artifact/label"
+	artifactscan "github.com/goharbor/harbor-cli/cmd/harbor/root/artifact/scan"
+	artifacttags "github.com/goharbor/harbor-cli/cmd/harbor/root/artifact/tags"
 	"github.com/spf13/cobra"
 )
 
@@ -30,8 +32,8 @@ func Artifact() *cobra.Command {
 		ListArtifactCommand(),
 		ViewArtifactCommmand(),
 		DeleteArtifactCommand(),
-		ScanArtifactCommand(),
-		ArtifactTagsCmd(),
+		artifactscan.ScanArtifactCommand(),
+		artifacttags.ArtifactTagsCmd(),
 		label.LabelsArtifactCommmand(),
 	)
 

cmd/harbor/root/artifact/scan.go

@@ -0,0 +1,52 @@
+// Copyright Project Harbor Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+package artifactscan
+
+import (
+	"github.com/goharbor/harbor-cli/pkg/prompt"
+	"github.com/goharbor/harbor-cli/pkg/utils"
+	"github.com/spf13/cobra"
+)
+
+func ScanArtifactCommand() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:     "scan",
+		Short:   "Scan an artifact",
+		Long:    `Scan an artifact in Harbor Repository`,
+		Example: `harbor artifact scan start <project>/<repository>/<reference>`,
+	}
+
+	cmd.AddCommand(
+		StartScanArtifactCommand(),
+		StopScanArtifactCommand(),
+		// LogScanArtifactCommand(),
+	)
+
+	return cmd
+}
+
+func parseArgs(args []string) (string, string, string, error) {
+	if len(args) > 0 {
+		return utils.ParseProjectRepoReference(args[0])
+	} else {
+		projectName, err := prompt.GetProjectNameFromUser()
+		if err != nil {
+			return "", "", "", err
+		}
+		repoName := prompt.GetRepoNameFromUser(projectName)
+		reference := prompt.GetReferenceFromUser(repoName, projectName)
+
+		return projectName, repoName, reference, nil
+	}
+}

cmd/harbor/root/artifact/scan/scan.go

@@ -0,0 +1,43 @@
+// Copyright Project Harbor Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+package artifactscan
+
+import (
+	"fmt"
+
+	"github.com/goharbor/harbor-cli/pkg/api"
+	"github.com/spf13/cobra"
+)
+
+func StartScanArtifactCommand() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:     "start",
+		Short:   "Start a scan of an artifact",
+		Long:    `Start a scan of an artifact in Harbor Repository`,
+		Example: `harbor artifact scan start <project>/<repository>/<reference>`,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			projectName, repoName, reference, err := parseArgs(args)
+			if err != nil {
+				return err
+			}
+
+			err = api.StartScanArtifact(projectName, repoName, reference)
+			if err != nil {
+				return fmt.Errorf("failed to start scan of artifact: %v", err)
+			}
+			return nil
+		},
+	}
+	return cmd
+}

cmd/harbor/root/artifact/scan/start.go

@@ -0,0 +1,44 @@
+// Copyright Project Harbor Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+package artifactscan
+
+import (
+	"fmt"
+
+	"github.com/goharbor/harbor-cli/pkg/api"
+	"github.com/spf13/cobra"
+)
+
+func StopScanArtifactCommand() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:     "stop",
+		Short:   "Stop a scan of an artifact",
+		Long:    `Stop a scan of an artifact in Harbor Repository`,
+		Example: `harbor artifact scan stop <project>/<repository>/<reference>`,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			projectName, repoName, reference, err := parseArgs(args)
+			if err != nil {
+				return err
+			}
+
+			err = api.StopScanArtifact(projectName, repoName, reference)
+			if err != nil {
+				return fmt.Errorf("failed to stop scan of artifact: %v", err)
+			}
+
+			return nil
+		},
+	}
+	return cmd
+}