Introduce annotations relevant to resource view during deploy

lib/utilities/manifest-utilities.js

@@ -252,7 +252,7 @@ function updateImagePullSecretsInManifestFiles(filePaths, imagePullSecrets) {
                 }
             });
         });
-        core.debug('New K8s objects after addin imagePullSecrets are :' + JSON.stringify(newObjectsList));
+        core.debug('New K8s objects after adding imagePullSecrets are :' + JSON.stringify(newObjectsList));
         const newFilePaths = fileHelper.writeObjectsToFile(newObjectsList);
         return newFilePaths;
     }

lib/utilities/resource-object-utility.js

@@ -1,6 +1,6 @@
 'use strict';
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getResources = exports.updateSelectorLabels = exports.updateSpecLabels = exports.updateImagePullSecrets = exports.updateObjectLabels = exports.getReplicaCount = exports.isIngressEntity = exports.isServiceEntity = exports.isWorkloadEntity = exports.isDeploymentEntity = void 0;
+exports.getResources = exports.updateSelectorLabels = exports.updateSpecLabels = exports.updateImagePullSecrets = exports.updateObjectAnnotations = exports.updateObjectLabels = exports.getReplicaCount = exports.isIngressEntity = exports.isServiceEntity = exports.isWorkloadEntity = exports.isDeploymentEntity = void 0;
 const fs = require("fs");
 const core = require("@actions/core");
 const yaml = require("js-yaml");
@@ -78,6 +78,31 @@ function updateObjectLabels(inputObject, newLabels, override) {
     }
 }
 exports.updateObjectLabels = updateObjectLabels;
+function updateObjectAnnotations(inputObject, newAnnotations, override) {
+    if (!inputObject) {
+        throw ('NullInputObject');
+    }
+    if (!inputObject.metadata) {
+        throw ('NullInputObjectMetadata');
+    }
+    if (!newAnnotations) {
+        return;
+    }
+    if (override) {
+        inputObject.metadata.annotations = newAnnotations;
+    }
+    else {
+        let existingAnnotations = inputObject.metadata.annotations;
+        if (!existingAnnotations) {
+            existingAnnotations = new Map();
+        }
+        Object.keys(newAnnotations).forEach(function (key) {
+            existingAnnotations[key] = newAnnotations[key];
+        });
+        inputObject.metadata.annotations = existingAnnotations;
+    }
+}
+exports.updateObjectAnnotations = updateObjectAnnotations;
 function updateImagePullSecrets(inputObject, newImagePullSecrets, override) {
     if (!inputObject || !inputObject.spec || !newImagePullSecrets) {
         return;

lib/utilities/strategy-helpers/canary-deployment-helper.js

@@ -155,6 +155,7 @@ function addCanaryLabelsAndAnnotations(inputObject, type) {
     const newLabels = new Map();
     newLabels[exports.CANARY_VERSION_LABEL] = type;
     helper.updateObjectLabels(inputObject, newLabels, false);
+    helper.updateObjectAnnotations(inputObject, newLabels, false);
     helper.updateSelectorLabels(inputObject, newLabels, false);
     if (!helper.isServiceEntity(inputObject.kind)) {
         helper.updateSpecLabels(inputObject, newLabels, false);

lib/utilities/strategy-helpers/deployment-helper.js

@@ -11,6 +11,7 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.getManifestFiles = exports.deploy = void 0;
 const fs = require("fs");
+const core = require("@actions/core");
 const yaml = require("js-yaml");
 const canaryDeploymentHelper = require("./canary-deployment-helper");
 const KubernetesObjectUtility = require("../resource-object-utility");
@@ -46,6 +47,15 @@ function deploy(kubectl, manifestFilePaths, deploymentStrategy) {
         ingressResources.forEach(ingressResource => {
             kubectl.getResource(KubernetesConstants.DiscoveryAndLoadBalancerResource.ingress, ingressResource.name);
         });
+        // annotate resources
+        let allPods;
+        try {
+            allPods = JSON.parse((kubectl.getAllPods()).stdout);
+        }
+        catch (e) {
+            core.debug("Unable to parse pods; Error: " + e);
+        }
+        annotateResources(deployedManifestFiles, kubectl, resourceTypes, allPods);
     });
 }
 exports.deploy = deploy;
@@ -121,6 +131,18 @@ function checkManifestStability(kubectl, resources) {
         yield KubernetesManifestUtility.checkManifestStability(kubectl, resources);
     });
 }
+function annotateResources(files, kubectl, resourceTypes, allPods) {
+    const annotateResults = [];
+    annotateResults.push(utility_1.annotateNamespace(kubectl, TaskInputParameters.namespace));
+    annotateResults.push(kubectl.annotateFiles(files, models.workflowAnnotations, true));
+    resourceTypes.forEach(resource => {
+        if (resource.type.toUpperCase() !== models.KubernetesWorkload.pod.toUpperCase()) {
+            utility_1.annotateChildPods(kubectl, resource.type, resource.name, allPods)
+                .forEach(execResult => annotateResults.push(execResult));
+        }
+    });
+    utility_1.checkForErrors(annotateResults, true);
+}
 function isCanaryDeploymentStrategy(deploymentStrategy) {
     return deploymentStrategy != null && deploymentStrategy.toUpperCase() === canaryDeploymentHelper.CANARY_DEPLOYMENT_STRATEGY.toUpperCase();
 }

lib/utilities/utility.js

@@ -1,8 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getCurrentTime = exports.getRandomInt = exports.sleep = exports.checkForErrors = exports.isEqual = exports.getExecutableExtension = void 0;
+exports.getCurrentTime = exports.getRandomInt = exports.sleep = exports.annotateNamespace = exports.annotateChildPods = exports.checkForErrors = exports.isEqual = exports.getExecutableExtension = void 0;
 const os = require("os");
 const core = require("@actions/core");
+const constants_1 = require("../constants");
 function getExecutableExtension() {
     if (os.type().match(/^Win/)) {
         return '.exe';
@@ -49,6 +50,41 @@ function checkForErrors(execResults, warnIfError) {
     }
 }
 exports.checkForErrors = checkForErrors;
+function annotateChildPods(kubectl, resourceType, resourceName, allPods) {
+    const commandExecutionResults = [];
+    let owner = resourceName;
+    if (resourceType.toLowerCase().indexOf('deployment') > -1) {
+        owner = kubectl.getNewReplicaSet(resourceName);
+    }
+    if (!!allPods && !!allPods.items && allPods.items.length > 0) {
+        allPods.items.forEach((pod) => {
+            const owners = pod.metadata.ownerReferences;
+            if (!!owners) {
+                owners.forEach(ownerRef => {
+                    if (ownerRef.name === owner) {
+                        commandExecutionResults.push(kubectl.annotate('pod', pod.metadata.name, constants_1.workflowAnnotations, true));
+                    }
+                });
+            }
+        });
+    }
+    return commandExecutionResults;
+}
+exports.annotateChildPods = annotateChildPods;
+function annotateNamespace(kubectl, namespaceName) {
+    let annotate = true;
+    const result = kubectl.getResource('namespace', namespaceName);
+    this.checkForErrors([result]);
+    const annotationsSet = JSON.parse(result.stdout).metadata.annotations;
+    if (!!annotationsSet && !!annotationsSet.runUri && annotationsSet.runUri.indexOf(process.env['GITHUB_REPOSITORY']) == -1) {
+        annotate = false;
+        core.debug(`Skipping 'annotate namespace' as namespace annotated by other workflow`);
+    }
+    if (annotate) {
+        return kubectl.annotate('namespace', namespaceName, constants_1.workflowAnnotations, true);
+    }
+}
+exports.annotateNamespace = annotateNamespace;
 function sleep(timeout) {
     return new Promise(resolve => setTimeout(resolve, timeout));
 }