feat: add option to add additional custom pod labels

Signed-off-by: Patrick von Steht <[email protected]>
eraser-dev · Feb 26, 2024 · b7a87ef · b7a87ef
1 parent 9917f5d
commit b7a87ef
Show file tree

Hide file tree

Showing 22 changed files with 93 additions and 38 deletions.
diff --git a/api/unversioned/config/config.go b/api/unversioned/config/config.go
@@ -106,6 +106,7 @@ func Default() *unversioned.EraserConfig {
 					"eraser.sh/cleanup.filter",
 				},
 			},
+			AdditionalPodLabels: map[string]string{},
 		},
 		Components: unversioned.Components{
 			Collector: unversioned.OptionalContainerConfig{

diff --git a/api/unversioned/eraserconfig_types.go b/api/unversioned/eraserconfig_types.go
@@ -156,15 +156,16 @@ type ContainerConfig struct {
 }
 
 type ManagerConfig struct {
-	Runtime           RuntimeSpec      `json:"runtime,omitempty"`
-	OTLPEndpoint      string           `json:"otlpEndpoint,omitempty"`
-	LogLevel          string           `json:"logLevel,omitempty"`
-	Scheduling        ScheduleConfig   `json:"scheduling,omitempty"`
-	Profile           ProfileConfig    `json:"profile,omitempty"`
-	ImageJob          ImageJobConfig   `json:"imageJob,omitempty"`
-	PullSecrets       []string         `json:"pullSecrets,omitempty"`
-	NodeFilter        NodeFilterConfig `json:"nodeFilter,omitempty"`
-	PriorityClassName string           `json:"priorityClassName,omitempty"`
+	Runtime             RuntimeSpec       `json:"runtime,omitempty"`
+	OTLPEndpoint        string            `json:"otlpEndpoint,omitempty"`
+	LogLevel            string            `json:"logLevel,omitempty"`
+	Scheduling          ScheduleConfig    `json:"scheduling,omitempty"`
+	Profile             ProfileConfig     `json:"profile,omitempty"`
+	ImageJob            ImageJobConfig    `json:"imageJob,omitempty"`
+	PullSecrets         []string          `json:"pullSecrets,omitempty"`
+	NodeFilter          NodeFilterConfig  `json:"nodeFilter,omitempty"`
+	PriorityClassName   string            `json:"priorityClassName,omitempty"`
+	AdditionalPodLabels map[string]string `json:"additionalPodLabels,omitempty"`
 }
 
 type ScheduleConfig struct {

diff --git a/api/unversioned/zz_generated.deepcopy.go b/api/unversioned/zz_generated.deepcopy.go
diff --git a/api/v1alpha1/config/config.go b/api/v1alpha1/config/config.go
@@ -61,6 +61,7 @@ func Default() *v1alpha1.EraserConfig {
 					"eraser.sh/cleanup.filter",
 				},
 			},
+			AdditionalPodLabels: map[string]string{},
 		},
 		Components: v1alpha1.Components{
 			Collector: v1alpha1.OptionalContainerConfig{

diff --git a/api/v1alpha1/eraserconfig_types.go b/api/v1alpha1/eraserconfig_types.go
@@ -91,15 +91,16 @@ type ContainerConfig struct {
 }
 
 type ManagerConfig struct {
-	Runtime           Runtime          `json:"runtime,omitempty"`
-	OTLPEndpoint      string           `json:"otlpEndpoint,omitempty"`
-	LogLevel          string           `json:"logLevel,omitempty"`
-	Scheduling        ScheduleConfig   `json:"scheduling,omitempty"`
-	Profile           ProfileConfig    `json:"profile,omitempty"`
-	ImageJob          ImageJobConfig   `json:"imageJob,omitempty"`
-	PullSecrets       []string         `json:"pullSecrets,omitempty"`
-	NodeFilter        NodeFilterConfig `json:"nodeFilter,omitempty"`
-	PriorityClassName string           `json:"priorityClassName,omitempty"`
+	Runtime             Runtime           `json:"runtime,omitempty"`
+	OTLPEndpoint        string            `json:"otlpEndpoint,omitempty"`
+	LogLevel            string            `json:"logLevel,omitempty"`
+	Scheduling          ScheduleConfig    `json:"scheduling,omitempty"`
+	Profile             ProfileConfig     `json:"profile,omitempty"`
+	ImageJob            ImageJobConfig    `json:"imageJob,omitempty"`
+	PullSecrets         []string          `json:"pullSecrets,omitempty"`
+	NodeFilter          NodeFilterConfig  `json:"nodeFilter,omitempty"`
+	PriorityClassName   string            `json:"priorityClassName,omitempty"`
+	AdditionalPodLabels map[string]string `json:"additionalPodLabels,omitempty"`
 }
 
 type ScheduleConfig struct {

diff --git a/api/v1alpha1/zz_generated.conversion.go b/api/v1alpha1/zz_generated.conversion.go
diff --git a/api/v1alpha1/zz_generated.deepcopy.go b/api/v1alpha1/zz_generated.deepcopy.go
diff --git a/api/v1alpha2/config/config.go b/api/v1alpha2/config/config.go
@@ -61,6 +61,7 @@ func Default() *v1alpha2.EraserConfig {
 					"eraser.sh/cleanup.filter",
 				},
 			},
+			AdditionalPodLabels: map[string]string{},
 		},
 		Components: v1alpha2.Components{
 			Collector: v1alpha2.OptionalContainerConfig{

diff --git a/api/v1alpha2/eraserconfig_types.go b/api/v1alpha2/eraserconfig_types.go
@@ -89,15 +89,16 @@ type ContainerConfig struct {
 }
 
 type ManagerConfig struct {
-	Runtime           Runtime          `json:"runtime,omitempty"`
-	OTLPEndpoint      string           `json:"otlpEndpoint,omitempty"`
-	LogLevel          string           `json:"logLevel,omitempty"`
-	Scheduling        ScheduleConfig   `json:"scheduling,omitempty"`
-	Profile           ProfileConfig    `json:"profile,omitempty"`
-	ImageJob          ImageJobConfig   `json:"imageJob,omitempty"`
-	PullSecrets       []string         `json:"pullSecrets,omitempty"`
-	NodeFilter        NodeFilterConfig `json:"nodeFilter,omitempty"`
-	PriorityClassName string           `json:"priorityClassName,omitempty"`
+	Runtime             Runtime           `json:"runtime,omitempty"`
+	OTLPEndpoint        string            `json:"otlpEndpoint,omitempty"`
+	LogLevel            string            `json:"logLevel,omitempty"`
+	Scheduling          ScheduleConfig    `json:"scheduling,omitempty"`
+	Profile             ProfileConfig     `json:"profile,omitempty"`
+	ImageJob            ImageJobConfig    `json:"imageJob,omitempty"`
+	PullSecrets         []string          `json:"pullSecrets,omitempty"`
+	NodeFilter          NodeFilterConfig  `json:"nodeFilter,omitempty"`
+	PriorityClassName   string            `json:"priorityClassName,omitempty"`
+	AdditionalPodLabels map[string]string `json:"additionalPodLabels,omitempty"`
 }
 
 type ScheduleConfig struct {

diff --git a/api/v1alpha2/zz_generated.conversion.go b/api/v1alpha2/zz_generated.conversion.go
diff --git a/api/v1alpha2/zz_generated.deepcopy.go b/api/v1alpha2/zz_generated.deepcopy.go
diff --git a/api/v1alpha3/config/config.go b/api/v1alpha3/config/config.go
@@ -64,6 +64,7 @@ func Default() *v1alpha3.EraserConfig {
 					"eraser.sh/cleanup.filter",
 				},
 			},
+			AdditionalPodLabels: map[string]string{},
 		},
 		Components: v1alpha3.Components{
 			Collector: v1alpha3.OptionalContainerConfig{

diff --git a/api/v1alpha3/eraserconfig_types.go b/api/v1alpha3/eraserconfig_types.go
@@ -156,15 +156,16 @@ type ContainerConfig struct {
 }
 
 type ManagerConfig struct {
-	Runtime           RuntimeSpec      `json:"runtime,omitempty"`
-	OTLPEndpoint      string           `json:"otlpEndpoint,omitempty"`
-	LogLevel          string           `json:"logLevel,omitempty"`
-	Scheduling        ScheduleConfig   `json:"scheduling,omitempty"`
-	Profile           ProfileConfig    `json:"profile,omitempty"`
-	ImageJob          ImageJobConfig   `json:"imageJob,omitempty"`
-	PullSecrets       []string         `json:"pullSecrets,omitempty"`
-	NodeFilter        NodeFilterConfig `json:"nodeFilter,omitempty"`
-	PriorityClassName string           `json:"priorityClassName,omitempty"`
+	Runtime             RuntimeSpec       `json:"runtime,omitempty"`
+	OTLPEndpoint        string            `json:"otlpEndpoint,omitempty"`
+	LogLevel            string            `json:"logLevel,omitempty"`
+	Scheduling          ScheduleConfig    `json:"scheduling,omitempty"`
+	Profile             ProfileConfig     `json:"profile,omitempty"`
+	ImageJob            ImageJobConfig    `json:"imageJob,omitempty"`
+	PullSecrets         []string          `json:"pullSecrets,omitempty"`
+	NodeFilter          NodeFilterConfig  `json:"nodeFilter,omitempty"`
+	PriorityClassName   string            `json:"priorityClassName,omitempty"`
+	AdditionalPodLabels map[string]string `json:"additionalPodLabels,omitempty"`
 }
 
 type ScheduleConfig struct {

diff --git a/api/v1alpha3/zz_generated.conversion.go b/api/v1alpha3/zz_generated.conversion.go
diff --git a/api/v1alpha3/zz_generated.deepcopy.go b/api/v1alpha3/zz_generated.deepcopy.go
diff --git a/config/manager/controller_manager_config.yaml b/config/manager/controller_manager_config.yaml
@@ -19,6 +19,7 @@ manager:
       delayOnFailure: 24h
   pullSecrets: [] # image pull secrets for collector/scanner/eraser
   priorityClassName: "" # priority class name for collector/scanner/eraser
+  additionalPodLabels: {}
   nodeFilter:
     type: exclude # must be either exclude|include
     selectors:

diff --git a/controllers/imagejob/imagejob_controller.go b/controllers/imagejob/imagejob_controller.go
@@ -409,10 +409,12 @@ func (r *Reconciler) handleNewJob(ctx context.Context, imageJob *eraserv1.ImageJ
 			},
 		}
 
+		pod.Labels = eraserConfig.Manager.AdditionalPodLabels
+
 		if containerName == removerContainer {
-			pod.Labels = map[string]string{imageJobTypeLabelKey: manualJobType}
+			pod.Labels[imageJobTypeLabelKey] = manualJobType
 		} else {
-			pod.Labels = map[string]string{imageJobTypeLabelKey: collectorJobType}
+			pod.Labels[imageJobTypeLabelKey] = collectorJobType
 		}
 
 		fitness := checkNodeFitness(pod, &nodeList[i])

diff --git a/docs/docs/customization.md b/docs/docs/customization.md
@@ -102,6 +102,7 @@ manager:
       delayOnFailure: 24h
   pullSecrets: [] # image pull secrets for collector/scanner/remover
   priorityClassName: "" # priority class name for collector/scanner/remover
+  additionalPodLabels: {}
   nodeFilter:
     type: exclude # must be either exclude|include
     selectors:
@@ -207,6 +208,7 @@ timeout:
 | manager.imageJob.cleanup.delayOnFailure | The amount of time to wait after a failed image job before performing cleanup. | 24h |
 | manager.pullSecrets | The image pull secrets to use for collector, scanner, and remover containers. | [] |
 | manager.priorityClassName | The priority class to use for collector, scanner, and remover containers. | "" |
+| manager.additionalPodLabels | Additional labels for all pods that the controller creates at runtime. | `{}` |
 | manager.nodeFilter.type | The type of node filter to use. Must be either "exclude" or "include". | exclude |
 | manager.nodeFilter.selectors | A list of selectors used to filter nodes. | [] |
 | components.collector.enabled | Whether to enable the collector component. | true |

diff --git a/third_party/open-policy-agent/gatekeeper/helmify/kustomize-for-helm.yaml b/third_party/open-policy-agent/gatekeeper/helmify/kustomize-for-helm.yaml
@@ -5,6 +5,9 @@ metadata:
   namespace: eraser-system
 spec:
   template:
+    metadata:
+      labels:
+        HELMSUBST_DEPLOYMENT_CONTROLLER_MANAGER_ADDITIONALPODLABELS: ""
     spec:
       HELMSUBST_DEPLOYMENT_CONTROLLER_MANAGER_PULL_SECRETS: ""
       volumes:

diff --git a/third_party/open-policy-agent/gatekeeper/helmify/replacements.go b/third_party/open-policy-agent/gatekeeper/helmify/replacements.go
@@ -7,6 +7,7 @@ var replacements = map[string]string{
 	`HELMSUBST_DEPLOYMENT_CONTROLLER_MANAGER_AFFINITY: ""`:            `{{- toYaml .Values.deploy.affinity | nindent 8 }}`,
 	`- HELMSUBST_DEPLOYMENT_CONTROLLER_MANAGER_ADDITIONAL_ARGS`:       `{{- if .Values.deploy.additionalArgs }}{{- range .Values.deploy.additionalArgs }}{{ nindent 8 "- " }}{{ . }}{{- end -}}{{ end }}`,
 	`HELMSUBST_CONTROLLER_MANAGER_CONFIG_YAML`:                        `{{- toYaml .Values.runtimeConfig | nindent 4 }}`,
+	`HELMSUBST_DEPLOYMENT_CONTROLLER_MANAGER_ADDITIONALPODLABELS: ""`: `{{- if .Values.deploy.additionalPodLabels }}{{- toYaml .Values.deploy.additionalPodLabels | nindent 8 }}{{end}}`,
 
 	`HELMSUBST_DEPLOYMENT_CONTROLLER_MANAGER_PULL_SECRETS: ""`: `{{- if .Values.runtimeConfig.manager.pullSecrets }}
       imagePullSecrets:

diff --git a/third_party/open-policy-agent/gatekeeper/helmify/static/README.md b/third_party/open-policy-agent/gatekeeper/helmify/static/README.md
@@ -45,6 +45,7 @@ _See [helm install](https://helm.sh/docs/helm/helm_install/) for command documen
 | runtimeConfig.manager.imageJob.cleanup          | Settings for image job cleanup.                                                                      | `{}`                           |
 | runtimeConfig.manager.pullSecrets               | Image pull secrets for collector/scanner/eraser.                                                     | `[]`                           |
 | runtimeConfig.manager.priorityClassName         | Priority class name for collector/scanner/eraser.                                                    | `""`                           |
+| runtimeConfig.manager.additionalPodLabels       | Additional labels for all pods that the controller creates at runtime.                               | `{}`                           |
 | runtimeConfig.manager.nodeFilter                | Filter for nodes.                                                                                    | `{}`                           |
 | runtimeConfig.components.collector              | Settings for the collector component.                                                                | `{ enabled: false }`           |
 | runtimeConfig.components.scanner                | Settings for the scanner component.                                                                  | `{ enabled: false }`           |
@@ -54,6 +55,7 @@ _See [helm install](https://helm.sh/docs/helm/helm_install/) for command documen
 | deploy.image.tag                                | Overrides the default image tag.                                                                     | `""`                           |
 | deploy.additionalArgs                           | Additional arguments to pass to the command.                                                         | `[]`                           |
 | deploy.priorityClassName                        | Priority class name.                                                                                 | `""`                           |
+| deploy.additionalPodLabels                      | Additional labels for the controller pod.                                                            | `{}`                           |
 | deploy.securityContext.allowPrivilegeEscalation | Whether to allow privilege escalation.                                                               | `false`                        |
 | deploy.resources.limits.memory                  | Memory limit for the resources.                                                                      | `30Mi`                         |
 | deploy.resources.requests.cpu                   | CPU request for the resources.                                                                       | `100m`                         |

diff --git a/third_party/open-policy-agent/gatekeeper/helmify/static/values.yaml b/third_party/open-policy-agent/gatekeeper/helmify/static/values.yaml
@@ -29,6 +29,7 @@ runtimeConfig:
         # delayOnFailure: ""
     pullSecrets: [] # image pull secrets for collector/scanner/eraser
     priorityClassName: "" # priority class name for collector/scanner/eraser
+    additionalPodLabels: {}
     nodeFilter:
       type: exclude # must be either exclude|include
       selectors:
@@ -97,6 +98,7 @@ deploy:
     tag: "v1.4.0-beta.0"
   additionalArgs: []
   priorityClassName: ""
+  additionalPodLabels: {}
 
   securityContext:
     allowPrivilegeEscalation: false