Feat/project preheat execution

cmd/harbor/root/project/preheat.go

@@ -28,6 +28,7 @@ func Preheat() *cobra.Command {
 	}
 	cmd.AddCommand(
 		preheat.PolicyCommand(),
+		preheat.ExecutionCommand(),
 	)
 
 	return cmd

cmd/harbor/root/project/preheat/execution.go

@@ -0,0 +1,37 @@
+// 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 preheat
+
+import (
+	"github.com/goharbor/harbor-cli/cmd/harbor/root/project/preheat/execution"
+	"github.com/spf13/cobra"
+)
+
+func ExecutionCommand() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:     "execution",
+		Aliases: []string{"exec"},
+		Short:   "Manage preheat executions",
+		Long:    "Manage P2P preheat executions under a project",
+		Example: `  harbor-cli project preheat execution list [NAME|ID] [POLICY_NAME]`,
+	}
+
+	cmd.AddCommand(
+		execution.ListExecutionCommand(),
+		execution.ViewExecutionCommand(),
+		execution.StopExecutionCommand(),
+	)
+
+	return cmd
+}

cmd/harbor/root/project/preheat/execution/list.go

@@ -0,0 +1,118 @@
+// 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 execution
+
+import (
+	"fmt"
+
+	"github.com/goharbor/harbor-cli/pkg/api"
+	"github.com/goharbor/harbor-cli/pkg/prompt"
+	"github.com/goharbor/harbor-cli/pkg/utils"
+	"github.com/goharbor/harbor-cli/pkg/views/preheat/execution/list"
+	log "github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+)
+
+func ListExecutionCommand() *cobra.Command {
+	var opts api.ListFlags
+	var isID bool
+
+	cmd := &cobra.Command{
+		Use:     "list",
+		Short:   "List preheat executions",
+		Long:    "List preheat executions under a project",
+		Example: `  harbor-cli project preheat execution list [NAME|ID] [POLICY_NAME]`,
+		Args:    cobra.MaximumNArgs(2),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			var err error
+			var projectName, policyName string
+
+			if opts.Page < 1 {
+				return fmt.Errorf("page number must be greater than or equal to 1")
+			}
+			if opts.PageSize <= 0 || opts.PageSize > 100 {
+				return fmt.Errorf("page size must be greater than 0 and less than or equal to 100")
+			}
+
+			if isID && len(args) == 0 {
+				return fmt.Errorf("project ID must be provided when using --id")
+			}
+
+			if len(args) >= 1 {
+				log.Debugf("Project name provided: %s", args[0])
+				projectName = args[0]
+			} else {
+				log.Debug("No project name provided, prompting user")
+				projectName, err = prompt.GetProjectNameFromUser()
+				if err != nil {
+					return fmt.Errorf("failed to get project name: %v", utils.ParseHarborErrorMsg(err))
+				}
+			}
+
+			if isID {
+				project, err := api.GetProject(projectName, true)
+				if err != nil {
+					return fmt.Errorf("failed to resolve project ID: %v", utils.ParseHarborErrorMsg(err))
+				}
+				projectName = project.Payload.Name
+			}
+
+			if len(args) >= 2 {
+				log.Debugf("Policy name provided: %s", args[1])
+				policyName = args[1]
+			} else {
+				log.Debug("No policy name provided, prompting user")
+				policyName, err = prompt.GetPreheatPolicyNameFromUser(projectName)
+				if err != nil {
+					return fmt.Errorf("failed to get policy name: %v", utils.ParseHarborErrorMsg(err))
+				}
+			}
+
+			log.Debug("Fetching preheat policy executions...")
+			resp, err := api.ListPreheatExecutions(projectName, policyName, opts)
+			if err != nil {
+				if utils.ParseHarborErrorCode(err) == "404" {
+					return fmt.Errorf("no executions found for policy %s in project %s", policyName, projectName)
+				}
+				return fmt.Errorf("failed to list preheat executions: %v", utils.ParseHarborErrorMsg(err))
+			}
+
+			if len(resp.Payload) == 0 {
+				fmt.Println("No executions found")
+				return nil
+			}
+
+			FormatFlag := viper.GetString("output-format")
+			if FormatFlag != "" {
+				err = utils.PrintFormat(resp.Payload, FormatFlag)
+				if err != nil {
+					return err
+				}
+			} else {
+				list.ListExecutions(resp.Payload)
+			}
+			return nil
+		},
+	}
+
+	flags := cmd.Flags()
+	flags.BoolVar(&isID, "id", false, "Get preheat executions by project id")
+	flags.Int64VarP(&opts.Page, "page", "", 1, "Page number")
+	flags.Int64VarP(&opts.PageSize, "page-size", "", 10, "Size of per page")
+	flags.StringVarP(&opts.Q, "query", "q", "", "Query string to query resources")
+	flags.StringVarP(&opts.Sort, "sort", "", "", "Sort the resource list in ascending or descending order")
+
+	return cmd
+}

cmd/harbor/root/project/preheat/execution/stop.go

@@ -0,0 +1,106 @@
+// 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 execution
+
+import (
+	"fmt"
+	"strconv"
+
+	"github.com/goharbor/harbor-cli/pkg/api"
+	"github.com/goharbor/harbor-cli/pkg/prompt"
+	"github.com/goharbor/harbor-cli/pkg/utils"
+	log "github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+)
+
+func StopExecutionCommand() *cobra.Command {
+	var isID bool
+
+	cmd := &cobra.Command{
+		Use:     "stop [PROJECT_NAME|ID] [POLICY_NAME] [EXECUTION_ID]",
+		Short:   "Stop preheat execution",
+		Long:    "Stop a specific P2P preheat execution of a policy under a project",
+		Example: `  harbor-cli project preheat execution stop [NAME|ID] [POLICY_NAME] [EXECUTION_ID]`,
+		Args:    cobra.MaximumNArgs(3),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			var err error
+			var projectName, policyName string
+			var executionID int64
+
+			if isID && len(args) == 0 {
+				return fmt.Errorf("project ID must be provided when using --id")
+			}
+
+			if len(args) >= 1 {
+				log.Debugf("Project name provided: %s", args[0])
+				projectName = args[0]
+			} else {
+				log.Debug("No project name provided, prompting user")
+				projectName, err = prompt.GetProjectNameFromUser()
+				if err != nil {
+					return fmt.Errorf("failed to get project name: %v", utils.ParseHarborErrorMsg(err))
+				}
+			}
+			if isID {
+				project, err := api.GetProject(projectName, true)
+				if err != nil {
+					return fmt.Errorf("failed to resolve project ID: %v", utils.ParseHarborErrorMsg(err))
+				}
+				projectName = project.Payload.Name
+			}
+
+			if len(args) >= 2 {
+				log.Debugf("Policy name provided: %s", args[1])
+				policyName = args[1]
+			} else {
+				log.Debug("No policy name provided, prompting user")
+				policyName, err = prompt.GetPreheatPolicyNameFromUser(projectName)
+				if err != nil {
+					return fmt.Errorf("failed to get policy name: %v", utils.ParseHarborErrorMsg(err))
+				}
+			}
+
+			if len(args) >= 3 {
+				log.Debugf("Execution ID provided: %s", args[2])
+				executionID, err = strconv.ParseInt(args[2], 10, 64)
+				if err != nil {
+					return fmt.Errorf("invalid execution ID %q: %v", args[2], err)
+				}
+			} else {
+				log.Debug("No execution ID provided, prompting user")
+				executionID, err = prompt.GetPreheatPolicyExecIDFromUser(projectName, policyName)
+				if err != nil {
+					return fmt.Errorf("failed to get execution id: %v", utils.ParseHarborErrorMsg(err))
+				}
+			}
+
+			log.Debug("Stopping preheat execution...")
+			err = api.StopPreheatExecution(projectName, policyName, executionID)
+			if err != nil {
+				if utils.ParseHarborErrorCode(err) == "404" {
+					return fmt.Errorf("preheat execution %d not found for policy %s in project %s", executionID, policyName, projectName)
+				}
+				return fmt.Errorf("failed to stop preheat execution: %v", utils.ParseHarborErrorMsg(err))
+			}
+
+			fmt.Printf("Preheat execution %d stopped successfully for policy '%s' in project '%s'\n", executionID, policyName, projectName)
+			return nil
+		},
+	}
+
+	flags := cmd.Flags()
+	flags.BoolVar(&isID, "id", false, "Get preheat policy execution by project id")
+
+	return cmd
+}

cmd/harbor/root/project/preheat/execution/view.go

@@ -0,0 +1,116 @@
+// 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 execution
+
+import (
+	"fmt"
+	"strconv"
+
+	"github.com/goharbor/harbor-cli/pkg/api"
+	"github.com/goharbor/harbor-cli/pkg/prompt"
+	"github.com/goharbor/harbor-cli/pkg/utils"
+	"github.com/goharbor/harbor-cli/pkg/views/preheat/execution/view"
+	log "github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+)
+
+func ViewExecutionCommand() *cobra.Command {
+	var isID bool
+
+	cmd := &cobra.Command{
+		Use:     "view [PROJECT_NAME|ID] [POLICY_NAME] [EXECUTION_ID]",
+		Short:   "View preheat execution details",
+		Long:    "Get details of a specific P2P preheat execution under a project",
+		Example: `  harbor-cli project preheat execution view [NAME|ID] [POLICY_NAME] [EXECUTION_ID]`,
+		Args:    cobra.MaximumNArgs(3),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			var err error
+			var projectName, policyName string
+			var executionID int64
+
+			if isID && len(args) == 0 {
+				return fmt.Errorf("project ID must be provided when using --id")
+			}
+
+			if len(args) >= 1 {
+				log.Debugf("Project name provided: %s", args[0])
+				projectName = args[0]
+			} else {
+				log.Debug("No project name provided, prompting user")
+				projectName, err = prompt.GetProjectNameFromUser()
+				if err != nil {
+					return fmt.Errorf("failed to get project name: %v", utils.ParseHarborErrorMsg(err))
+				}
+			}
+			if isID {
+				project, err := api.GetProject(projectName, true)
+				if err != nil {
+					return fmt.Errorf("failed to resolve project ID: %v", utils.ParseHarborErrorMsg(err))
+				}
+				projectName = project.Payload.Name
+			}
+
+			if len(args) >= 2 {
+				log.Debugf("Policy name provided: %s", args[1])
+				policyName = args[1]
+			} else {
+				log.Debug("No policy name provided, prompting user")
+				policyName, err = prompt.GetPreheatPolicyNameFromUser(projectName)
+				if err != nil {
+					return fmt.Errorf("failed to get policy name: %v", utils.ParseHarborErrorMsg(err))
+				}
+			}
+
+			if len(args) >= 3 {
+				log.Debugf("Execution ID provided: %s", args[2])
+				executionID, err = strconv.ParseInt(args[2], 10, 64)
+				if err != nil {
+					return fmt.Errorf("invalid execution ID %q: %v", args[2], err)
+				}
+			} else {
+				log.Debug("No execution ID provided, prompting user")
+				executionID, err = prompt.GetPreheatPolicyExecIDFromUser(projectName, policyName)
+				if err != nil {
+					return fmt.Errorf("failed to get execution id: %v", utils.ParseHarborErrorMsg(err))
+				}
+			}
+
+			log.Debug("Fetching preheat execution details...")
+			resp, err := api.GetPreheatExecution(projectName, policyName, executionID)
+			if err != nil {
+				if utils.ParseHarborErrorCode(err) == "404" {
+					return fmt.Errorf("no execution found for execution ID %d in policy %s in project %s", executionID, policyName, projectName)
+				}
+				return fmt.Errorf("failed to view preheat execution: %v", utils.ParseHarborErrorMsg(err))
+			}
+
+			FormatFlag := viper.GetString("output-format")
+			if FormatFlag != "" {
+				err = utils.PrintFormat(resp.Payload, FormatFlag)
+				if err != nil {
+					return err
+				}
+			} else {
+				view.ViewExecution(resp.Payload)
+			}
+			return nil
+		},
+	}
+
+	flags := cmd.Flags()
+	flags.BoolVar(&isID, "id", false, "Get preheat policy execution by project id")
+
+	return cmd
+}