feat: enable retention policies

go.mod

@@ -5,6 +5,7 @@ go 1.20
 require (
 	github.com/cheshir/go-mq/v2 v2.0.1
 	github.com/coreos/go-semver v0.3.1
+	github.com/cxmcc/unixsums v0.0.0-20131125091133-89564297d82f
 	github.com/go-logr/logr v1.2.4
 	github.com/google/go-cmp v0.5.9
 	github.com/hashicorp/go-version v1.6.0

go.sum

@@ -334,6 +334,8 @@ github.com/cpuguy83/go-md2man/v2 v2.0.0/go.mod h1:maD7wRr/U5Z6m/iR4s+kqSMx2CaBsr
 github.com/creack/pty v1.1.7/go.mod h1:lj5s0c3V2DBrqTV7llrYr5NG6My20zk30Fl46Y7DoTY=
 github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=
 github.com/creack/pty v1.1.11/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=
+github.com/cxmcc/unixsums v0.0.0-20131125091133-89564297d82f h1:PkAFGgVtJnasAxOaiEY1RYPx8W+7X7l66vi8T2apKCM=
+github.com/cxmcc/unixsums v0.0.0-20131125091133-89564297d82f/go.mod h1:XJq7OckzkOtlgeEKFwkH2gFbc1+1WRFUBf7QnvfyrzQ=
 github.com/cyphar/filepath-securejoin v0.2.2/go.mod h1:FpkQEhXnPnOthhzymB7CGsFk2G9VLXONKD9G7QGMM+4=
 github.com/cyphar/filepath-securejoin v0.2.3/go.mod h1:aPGpWjXOXUn2NCNjFvBE6aRxGGx79pTxQpKOJNYHHl4=
 github.com/d2g/dhcp4 v0.0.0-20170904100407-a1d1b6c41b1c/go.mod h1:Ct2BUK8SB0YC1SMSibvLzxjeJLnrYEVLULFNiHY9YfQ=

internal/harbor/harbor.go

@@ -35,6 +35,14 @@ type Harbor struct {
 	WebhookEventTypes     []string
 	LagoonTargetName      string
 	Config                *config.Options
+	TagRetention          TagRetention
+}
+
+type TagRetention struct {
+	Enabled              bool
+	PullRequestRetention int
+	BranchRetention      int
+	Schedule             string
 }
 
 // New create a new harbor connection.

internal/harbor/harbor22x.go

@@ -64,6 +64,51 @@ func (h *Harbor) CreateProjectV2(ctx context.Context, projectName string) (*harb
 	// 	fmt.Println(x)
 	// }
 
+	// handle the creation and updating of retention policies as required
+	if h.TagRetention.Enabled {
+		// generate a somewhat random schedule from the retention schedule template, using the harbor projectname as the seed
+		schedule, err := helpers.ConvertCrontab(projectName, h.TagRetention.Schedule)
+		if err != nil {
+			h.Log.Info(fmt.Sprintf("Error generating retention schedule %s: %v", project.Name, err))
+		}
+		// create the retention policy as required
+		retention := h.generateRetentionPolicy(int64(project.ProjectID), schedule, h.TagRetention.BranchRetention, h.TagRetention.PullRequestRetention)
+		// get the existing one if one exists
+		policy, err := h.ClientV5.GetRetentionPolicyByProject(ctx, projectName)
+		if err != nil {
+			h.Log.Info(fmt.Sprintf("Error getting retention policy %s: %v", project.Name, err))
+		}
+		if policy != nil {
+			needNewPolicy := false
+			r1, _ := json.Marshal(policy.Rules)
+			r2, _ := json.Marshal(retention.Rules)
+			t1, _ := json.Marshal(policy.Trigger)
+			t2, _ := json.Marshal(retention.Trigger)
+			fmt.Println(string(r1) != string(r2), string(t1) != string(t2))
+			if string(r1) != string(r2) {
+				needNewPolicy = true
+				policy.Rules = retention.Rules
+			}
+			if string(t1) != string(t2) {
+				needNewPolicy = true
+				policy.Trigger = retention.Trigger
+			}
+			if needNewPolicy {
+				// delete any existing retention policies
+				err := h.ClientV5.UpdateRetentionPolicy(ctx, policy)
+				if err != nil {
+					f, _ := json.Marshal(err)
+					h.Log.Info(fmt.Sprintf("Error updating retention policy %s: %v", project.Name, string(f)))
+				}
+			}
+		} else {
+			// create it if it doesn't
+			if err := h.ClientV5.NewRetentionPolicy(ctx, retention); err != nil {
+				h.Log.Info(fmt.Sprintf("Error creating retention policy %s: %v", project.Name, err))
+			}
+		}
+	}
+
 	if h.WebhookAddition {
 		wps, err := h.ClientV5.ListProjectWebhookPolicies(ctx, int(project.ProjectID))
 		if err != nil {
@@ -278,6 +323,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 +346,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 +382,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,
+				),
+			)
 		}
 	}
 }

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,
+			},
+		},
+	}
+}