From 27023d504adb858031c2dfbf564cbe88a4254bb4 Mon Sep 17 00:00:00 2001
From: Colin Walker <colinwalker270@gmail.com>
Date: Sat, 31 Aug 2019 15:22:52 +0000
Subject: [PATCH] feat: Add canary support built in

Canary support configured the built in helm chart and release names for
deployment as canaries.
---
 index.js | 56 ++++++++++++++++++++++++++++++++++++++++++++++----------
 1 file changed, 46 insertions(+), 10 deletions(-)

diff --git a/index.js b/index.js
index 38edf027..28da139f 100644
--- a/index.js
+++ b/index.js
@@ -39,6 +39,34 @@ async function status(state) {
   }
 }
 
+function releaseName(name, track) {
+  if (track !== "stable") {
+    return `${name}-${track}`
+  }
+  return name
+}
+
+function chartName(name) {
+  if (name === "app") {
+    return "./usr/src/charts/app"
+  }
+  return name
+}
+
+function getInput(name, options) {
+  const context = github.context;
+  const deployment = context.payload.deployment;
+  const val = core.getInput(name, { ...opts, required: false })
+  if (deployment) {
+    if (deployment[name]) return deployment[name];
+    if (deployment.payload[name]) return deployment.payload[name];
+  }
+  if (options && options.required && !val) {
+    throw new Error(`Input required and not supplied: ${name}`);
+  }
+  return val
+}
+
 /**
  * Run executes the helm deployment.
  */
@@ -46,16 +74,14 @@ async function run() {
   try {
     await status("pending");
 
-    const release = core.getInput("release", required);
-    const namespace = core.getInput("namespace", required);
-    const chart = core.getInput("chart", required);
-    const values = core.getInput("values") || "{}";
-    const dryRun = core.getInput("dry-run");
-
-    // Load in the github context and deployment event.
-    const context = github.context;
-    const task = (context.payload.deployment &&
-                  context.payload.deployment.task) || "deploy";
+    const track = getInput("track") || "stable";
+    const release = releaseName(getInput("release", required), track);
+    const namespace = getInput("namespace", required);
+    const chart = chartName(getInput("chart", required));
+    const values = getInput("values") || "{}";
+    const dryRun = getInput("dry-run");
+    const task = getInput("task");
+    const version = getInput("version");
 
     // Setup command options and arguments.
     const opts = { env: {} };
@@ -66,6 +92,16 @@ async function run() {
       "--values", "./values.yml",
     ];
     if (dryRun) args.push("--dry-run");
+    if (version) args.push(`--set=version=${version}`)
+
+    // Stable track only deploys service and ingress resources. Any other track
+    // name can be treated like a canary deployment.
+    if (track !== "stable") {
+      args.push(
+        "--set=service.enabled=false",
+        "--set=ingress.enabled=false",
+      )
+    }
 
     // Setup necessary files.
     if (process.env.KUBECONFIG_FILE) {