PLT-1670 - Propagate the labels to Pod for the argo workflow templates

README.md

@@ -115,3 +115,4 @@ The project is licensed under the MIT License, see the [LICENSE](LICENSE) file f
 
 -   Q&A: [Github Discussions](https://github.com/atlanhq/argopm/discussions)
 -   You can also reach out to engineering@atlan.com
+

lib/k8s.js

@@ -78,6 +78,17 @@ function enableClusterScope(yamlObject) {
     }
 }
 
+function copyLabelsToPodMetaData(yamlObject, kind) {
+    if (kind === constants.ARGO_WORKFLOW_TEMPLATES_KIND || kind === constants.ARGO_CLUSTER_WORKFLOW_TEMPLATES_KIND) {
+        let metadata = yamlObject.metadata;
+        let spec = yamlObject.spec;
+        // Copy labels from metadata to spec.podMetadata.labels
+        spec.podMetadata = spec.podMetadata || {};
+        spec.podMetadata.labels = Object.assign({}, spec.podMetadata.labels, metadata.labels);
+    }
+    return yamlObject;
+}
+
 class K8sInstaller {
     /**
      * Installs the given package to Argo K8s deployment
@@ -261,7 +272,7 @@ class K8sInstaller {
                         if (preProcessFn) {
                             yamlData = preProcessFn(yamlData);
                         }
-                        const apmYAML = mainThis.addAPMLabels(yamlData, folder, fileName);
+                        const apmYAML = mainThis.addAPMLabels(yamlData, folder, fileName, kind);
                         return fn(
                             mainThis.package.name,
                             mainThis.namespace,
@@ -275,11 +286,13 @@ class K8sInstaller {
             });
         });
     }
+
     /**
      * @param {Object} yamlObject YAML object
      */
     convertS3ArtifactsToAzureBlob(yamlObject) {
         const searchKey = "s3";
+
         // Recursive function to search for the key
         function searchForKey(obj) {
             if (typeof obj !== "object" || obj === null) {
@@ -297,6 +310,7 @@ class K8sInstaller {
                 searchForKey(obj[prop]);
             }
         }
+
         // Start the search
         searchForKey(yamlObject);
 
@@ -307,7 +321,7 @@ class K8sInstaller {
      * Installs the given package to Argo K8s deployment
      * @param {Object} yamlObject YAML object
      */
-    addAPMLabels(yamlObject, folder, fileName) {
+    addAPMLabels(yamlObject, folder, fileName, kind) {
         let metadata = yamlObject.metadata;
         if (!metadata.name) {
             metadata.name = this.package.name.replace(/@/g, "-").replace(/\//g, "-").replace(/:/g, "-");
@@ -338,7 +352,7 @@ class K8sInstaller {
         });
 
         yamlObject.metadata = metadata;
-        return yamlObject;
+        return copyLabelsToPodMetaData(yamlObject, kind);
     }
 }