feat(engagement): add close engagement in cli

Author: pngouin

client/engagement.go

@@ -2,7 +2,9 @@ package client
 
 import (
 	"encoding/json"
+	"errors"
 	"fmt"
+	"net/http"
 
 	"github.com/pngouin/defectdojo-cli/config"
 )
@@ -28,6 +30,10 @@ type EngagementResponse struct {
 	Engagement
 }
 
+var (
+	ErrCannotCloseEngagement = errors.New("cannot close engagement")
+)
+
 const (
 	NotStarted         EngagementStatus = "Not Started"
 	Blocked            EngagementStatus = "Blocked"
@@ -80,3 +86,15 @@ func (ec EngagementClient) Get(id string) (EngagementResponse, error) {
 	err = json.NewDecoder(resp.Body).Decode(&engagementResponse)
 	return engagementResponse, err
 }
+
+func (ec EngagementClient) Close(id string) (error) {
+	path := fmt.Sprint(ec.baseEndpoint, id, "/close/")
+	resp, err := ec.http.Post(path, nil)
+	if err != nil {
+		return err
+	}
+	if resp.StatusCode != http.StatusOK {
+		return ErrCannotCloseEngagement
+	}
+	return nil
+}

cmd/engagement.go

@@ -24,6 +24,13 @@ var createEngagement = &cobra.Command{
 	Run:   create,
 }
 
+var closeEngagement = &cobra.Command{
+	Use:   "close",
+	Short: "close an engagement",
+	Long:  "close an engagement for a product",
+	Run:   close,
+}
+
 func init() {
 	createEngagement.PersistentFlags().StringP("name", "n", "CI/CD", "engagement name")
 	createEngagement.PersistentFlags().StringP("description", "d", "CI/CD engagement", "engagement description")
@@ -33,7 +40,10 @@ func init() {
 	createEngagement.PersistentFlags().StringP("branch", "b", "", "branch of the code repository")
 	createEngagement.PersistentFlags().IntP("product", "p", -1, "product id")
 
+	closeEngagement.PersistentFlags().StringP("engagement", "e", "", "engagement id")
+
 	rootEngagement.AddCommand(createEngagement)
+	rootEngagement.AddCommand(closeEngagement)
 	cli.AddCommand(rootEngagement)
 }
 
@@ -62,3 +72,13 @@ func create(cmd *cobra.Command, args []string) {
 	data, _ := json.Marshal(resp)
 	fmt.Println(string(data))
 }
+
+func close(cmd *cobra.Command, args []string) {
+	client := client.NewEngagementClient(config.Configuration)
+	engagement := getFlagS(cmd, "engagement")
+	err := client.Close(engagement)
+	if err != nil {
+		log.Fatalf("cannot close engagement %s: %v", engagement, err)
+	}
+	fmt.Println("deleted")
+}