feat: label step pods

Author: John Dreyer

packages/k8s/src/k8s/index.ts

@@ -90,6 +90,26 @@ function sanitizeLabel(label: string): string {
   return sluggedLabel
 }
 
+function getArcContextLabels(): { [key: string]: string } {
+  const { GITHUB_RUN_ID, GITHUB_RUN_NUMBER, GITHUB_RUN_ATTEMPT } = process.env
+  return Object.fromEntries(
+    Object.entries({
+      'arc-context-event-name': jobContext.eventName,
+      'arc-context-sha': jobContext.sha,
+      'arc-context-workflow': jobContext.workflow,
+      'arc-context-actor': jobContext.actor,
+      'arc-context-job': jobContext.job,
+      'arc-context-repository': jobContext.repo.repo,
+      'arc-context-repository-owner': jobContext.repo.owner,
+      'arc-context-run-id': GITHUB_RUN_ID || '',
+      'arc-context-run-number': GITHUB_RUN_NUMBER || '',
+      'arc-context-run-attempt': GITHUB_RUN_ATTEMPT || ''
+    })
+      .map(([key, value]) => [key, sanitizeLabel(value)])
+      .filter(([, value]) => value !== '')
+  )
+}
+
 export async function createPod(
   jobContainer?: k8s.V1Container,
   services?: k8s.V1Container[],
@@ -112,22 +132,7 @@ export async function createPod(
   appPod.metadata = new k8s.V1ObjectMeta()
   appPod.metadata.name = getJobPodName()
 
-  const { GITHUB_RUN_ID, GITHUB_RUN_NUMBER, GITHUB_RUN_ATTEMPT } = process.env
-  const arcLabels = Object.fromEntries(
-    Object.entries({
-      'arc-context-event-name': jobContext.eventName,
-      'arc-context-sha': jobContext.sha,
-      'arc-context-workflow': jobContext.workflow,
-      'arc-context-actor': jobContext.actor,
-      'arc-context-job': jobContext.job,
-      'arc-context-repository': jobContext.repo.repo,
-      'arc-context-repository-owner': jobContext.repo.owner,
-      'arc-context-run-id': GITHUB_RUN_ID || '',
-      'arc-context-run-number': GITHUB_RUN_NUMBER || '',
-      'arc-context-run-attempt': GITHUB_RUN_ATTEMPT || ''
-    }).map(([key, value]) => [key, sanitizeLabel(value)])
-  )
-
+  const arcLabels = getArcContextLabels()
   const instanceLabel = new RunnerInstanceLabel()
   appPod.metadata.labels = {
     [instanceLabel.key]: instanceLabel.value,
@@ -197,9 +202,10 @@ export async function createJob(
   job.spec.backoffLimit = 0
   job.spec.template = new k8s.V1PodTemplateSpec()
 
+  const arcLabels = getArcContextLabels()
   job.spec.template.spec = new k8s.V1PodSpec()
   job.spec.template.metadata = new k8s.V1ObjectMeta()
-  job.spec.template.metadata.labels = {}
+  job.spec.template.metadata.labels = arcLabels
   job.spec.template.metadata.annotations = {}
   job.spec.template.spec.containers = [container]
   job.spec.template.spec.restartPolicy = 'Never'