feat: add exclusions for cleanup policy

cmd/gitlabUpdateCleanUpPolicyCmd.go

@@ -27,8 +27,6 @@ var gitlabUpdateCleanUpPolicyCmd = &cobra.Command{
 			fmt.Println(err)
 			os.Exit(1)
 		}
-
-		fmt.Println("Updated cleanup policy of the project with ID", projectID)
 	},
 }
 

cmd/root.go

@@ -87,6 +87,7 @@ func initConfig() {
 		mainConfig.Gitlab.ApiURL,
 		mainConfig.Gitlab.Token,
 		mainConfig.Gitlab.Mirror,
+		mainConfig.Gitlab.Exclusions,
 	)
 
 	onepassword = op.NewOnePassword(mainConfig.OnePassword.Address)

config/defs.go

@@ -9,9 +9,10 @@ type ConfigPostamark struct {
 }
 
 type ConfigGitlab struct {
-	Token  string                     `mapstructure:"token"`
-	ApiURL string                     `mapstructure:"api_url"`
-	Mirror gitlab.GitlabMirrorOptions `mapstructure:"mirror"`
+	Token      string                        `mapstructure:"token"`
+	ApiURL     string                        `mapstructure:"api_url"`
+	Exclusions gitlab.GitlabExclusionsConfig `mapstructure:"exclusions"`
+	Mirror     gitlab.GitlabMirrorOptions    `mapstructure:"mirror"`
 }
 
 type ConfigOnePassword struct {

config/template.yml

@@ -1,6 +1,8 @@
 gitlab:
   api_url: "https://company.gitlab.com/api/v4"
   token: "<GITLAB_TOKEN>"
+  exclusions:
+    cleanup_policies: [<PROJECT_IDS_LIST>]
   mirror:
     url: "https://gitlab.com/api/v4"
     group_id: "<GITLAB_MIRROR_GROUP_ID>"

scopes/gitlab/defs.go

@@ -9,9 +9,10 @@ const gitlabDefaultGroupMemberOwner string = "default_group_member_owner"
 const gitlabDefaultGroupMember string = "default_group_member"
 
 type gitlab struct {
-	token  string
-	apiURL string
-	mirror GitlabMirrorOptions
+	token      string
+	apiURL     string
+	mirror     GitlabMirrorOptions
+	exclusions GitlabExclusionsConfig
 }
 
 type Gitlab interface {
@@ -36,6 +37,10 @@ type GitlabMirrorOptions struct {
 	GroupID   int    `mapstructure:"group_id"`
 }
 
+type GitlabExclusionsConfig struct {
+	CleanupPolicies []int `mapstructure:"cleanup_policies"`
+}
+
 type gitlabCreateMirrorRequest struct {
 	Enabled               bool   `json:"enabled"`
 	URL                   string `json:"url"`

scopes/gitlab/main.go

@@ -341,9 +341,25 @@ func (g *gitlab) UpdateCleanUpPolicy(projectID string) error {
 	return err
 }
 
+// Function to check if an array contains an element
+func (g *gitlab) contains(array []int, item int) (bool, error) {
+	for i := 0; i < len(array); i++ {
+		// check
+		if array[i] == item {
+			return true, nil
+		}
+	}
+	return false, nil
+}
+
 // Apply a cleanUP policy on gitlab project.
 func (g *gitlab) applyCleanUpPolicy(projectID int) error {
-	_, err := g.request("PUT", fmt.Sprintf("/projects/%d", projectID), defaultCleanUpPolicy, nil)
+	isExcluded, err := g.contains(g.exclusions.CleanupPolicies, projectID)
+	if !isExcluded {
+		_, err = g.request("PUT", fmt.Sprintf("/projects/%d", projectID), defaultCleanUpPolicy, nil)
+	} else {
+		fmt.Println("Cleanup policy not updated for the project with ID", projectID)
+	}
 
 	return err
 }
@@ -903,10 +919,11 @@ func (g *gitlab) Deprovionioning(username string) error {
 	return nil
 }
 
-func NewGitlab(apiURL string, token string, mirror GitlabMirrorOptions) Gitlab {
+func NewGitlab(apiURL string, token string, mirror GitlabMirrorOptions, exclusions GitlabExclusionsConfig) Gitlab {
 	return &gitlab{
-		apiURL: apiURL,
-		token:  token,
-		mirror: mirror,
+		apiURL:     apiURL,
+		token:      token,
+		mirror:     mirror,
+		exclusions: exclusions,
 	}
 }