From f54eb9222090caea24bfabbdf6652a9b14bf97cf Mon Sep 17 00:00:00 2001
From: Ben Jackson <b@benjackson.email>
Date: Sun, 17 Dec 2023 09:37:35 +1100
Subject: [PATCH] feat: enable retention policies

---
 internal/harbor/harbor22x.go      | 50 +++++++++++++++++++++++
 internal/harbor/harbor_helpers.go | 67 +++++++++++++++++++++++++++++++
 2 files changed, 117 insertions(+)

diff --git a/internal/harbor/harbor22x.go b/internal/harbor/harbor22x.go
index bea48adb..9a2804ac 100644
--- a/internal/harbor/harbor22x.go
+++ b/internal/harbor/harbor22x.go
@@ -64,6 +64,32 @@ func (h *Harbor) CreateProjectV2(ctx context.Context, projectName string) (*harb
 	// 	fmt.Println(x)
 	// }
 
+	retentionPolicyEnable := true
+	if retentionPolicyEnable {
+		// always set the retention policy to what it should be
+		retention := h.generateRetentionPolicy(int64(project.ProjectID), "", 5, 1)
+		policy, err := h.ClientV5.GetRetentionPolicyByProject(ctx, projectName)
+		if err != nil {
+			h.Log.Info(fmt.Sprintf("Error creating retention policy %s: %v", project.Name, err))
+		}
+		fmt.Println(policy)
+		if policy == nil {
+			err := h.ClientV5.NewRetentionPolicy(ctx, retention)
+			if err != nil {
+				f, _ := json.Marshal(err)
+				h.Log.Info(fmt.Sprintf("Error creating retention policy %s: %v", project.Name, string(f)))
+			}
+		} else {
+			policy.Rules = retention.Rules
+			policy.Trigger = retention.Trigger
+			err = h.ClientV5.UpdateRetentionPolicy(ctx, retention)
+			if err != nil {
+				f, _ := json.Marshal(err)
+				h.Log.Info(fmt.Sprintf("Error updating retention policy %s: %v", project.Name, string(f)))
+			}
+		}
+	}
+
 	if h.WebhookAddition {
 		wps, err := h.ClientV5.ListProjectWebhookPolicies(ctx, int(project.ProjectID))
 		if err != nil {
@@ -278,6 +304,14 @@ func (h *Harbor) DeleteRepository(ctx context.Context, projectName, branch strin
 			if err != nil {
 				h.Log.Info(fmt.Sprintf("Error deleting harbor repository %s", repo.Name))
 			}
+			h.Log.Info(
+				fmt.Sprintf(
+					"Deleted harbor repository %s in  project %s, environment %s",
+					repo.Name,
+					projectName,
+					environmentName,
+				),
+			)
 		}
 	}
 	if len(listRepositories) > 100 {
@@ -293,6 +327,14 @@ func (h *Harbor) DeleteRepository(ctx context.Context, projectName, branch strin
 					if err != nil {
 						h.Log.Info(fmt.Sprintf("Error deleting harbor repository %s", repo.Name))
 					}
+					h.Log.Info(
+						fmt.Sprintf(
+							"Deleted harbor repository %s in  project %s, environment %s",
+							repo.Name,
+							projectName,
+							environmentName,
+						),
+					)
 				}
 			}
 		}
@@ -321,6 +363,14 @@ func (h *Harbor) DeleteRobotAccount(ctx context.Context, projectName, branch str
 				h.Log.Info(fmt.Sprintf("Error deleting project %s robot account %s", projectName, robot.Name))
 				return
 			}
+			h.Log.Info(
+				fmt.Sprintf(
+					"Deleted harbor robot account %s in  project %s, environment %s",
+					robot.Name,
+					projectName,
+					environmentName,
+				),
+			)
 		}
 	}
 }
diff --git a/internal/harbor/harbor_helpers.go b/internal/harbor/harbor_helpers.go
index 0acd1287..db8c4382 100644
--- a/internal/harbor/harbor_helpers.go
+++ b/internal/harbor/harbor_helpers.go
@@ -10,6 +10,7 @@ import (
 	"encoding/json"
 	"time"
 
+	harborclientv5model "github.com/mittwald/goharbor-client/v5/apiv2/model"
 	"github.com/uselagoon/remote-controller/internal/helpers"
 
 	corev1 "k8s.io/api/core/v1"
@@ -204,3 +205,69 @@ func (h *Harbor) UpsertHarborSecret(ctx context.Context, cl client.Client, ns, n
 	}
 	return false, nil
 }
+
+func (h *Harbor) generateRetentionPolicy(projectID int64, schedule string, branchRetention, prRetention int) *harborclientv5model.RetentionPolicy {
+	return &harborclientv5model.RetentionPolicy{
+		Algorithm: "or",
+		Rules: []*harborclientv5model.RetentionRule{
+			{ // create a retention policy for all images
+				Action: "retain",
+				Params: map[string]interface{}{
+					"latestPulledN": branchRetention,
+				},
+				ScopeSelectors: map[string][]harborclientv5model.RetentionSelector{
+					"repository": {
+						{
+							Decoration: "repoMatches",
+							Kind:       "doublestar",
+							Pattern:    "[^pr\\-]*/*", // exclude pullrequest repository images https://github.com/bmatcuk/doublestar#patterns
+						},
+					},
+				},
+				TagSelectors: []*harborclientv5model.RetentionSelector{
+					{
+						Decoration: "matches",
+						Extras:     "{\"untagged\":true}",
+						Kind:       "doublestar",
+						Pattern:    "**",
+					},
+				},
+				Template: "latestPulledN",
+			},
+			{ // create a retention policy specifically for pullrequests
+				Action: "retain",
+				Params: map[string]interface{}{
+					"latestPulledN": prRetention,
+				},
+				ScopeSelectors: map[string][]harborclientv5model.RetentionSelector{
+					"repository": {
+						{
+							Decoration: "repoMatches",
+							Kind:       "doublestar",
+							Pattern:    "pr-*",
+						},
+					},
+				},
+				TagSelectors: []*harborclientv5model.RetentionSelector{
+					{
+						Decoration: "matches",
+						Extras:     "{\"untagged\":true}",
+						Kind:       "doublestar",
+						Pattern:    "**",
+					},
+				},
+				Template: "latestPulledN",
+			},
+		},
+		Scope: &harborclientv5model.RetentionPolicyScope{
+			Level: "project",
+			Ref:   projectID,
+		},
+		Trigger: &harborclientv5model.RetentionRuleTrigger{
+			Kind: "schedule",
+			Settings: map[string]string{
+				"cron": schedule,
+			},
+		},
+	}
+}