cleaning up customer license

Author: nandajavarma

.werft/installer-tests.ts

@@ -3,6 +3,7 @@ import * as https from "https";
 import { join } from "path";
 import { exec } from "./util/shell";
 import { Werft } from "./util/werft";
+import { deleteReplicatedLicense } from "./jobs/build/self-hosted-upgrade-tests";
 
 const context: any = JSON.parse(fs.readFileSync("context.json").toString());
 
@@ -13,6 +14,7 @@ const testConfig: string = process.argv.length > 2 ? process.argv[2] : "STANDARD
 const channel: string = annotations.channel || "unstable";
 const version: string = annotations.version || "-";
 const preview: string = annotations.preview || "false"; // setting to true will not destroy the setup
+const customerID: string = annotations.customerID || "";
 const upgrade: string = annotations.upgrade || "false"; // setting to true will not KOTS upgrade to the latest version. Set the channel to beta or stable in this case.
 const skipTests: string = annotations.skipTests || "false"; // setting to true skips the integration tests
 const deps: string = annotations.deps || ""; // options: ["external", "internal"] setting to `external` will ensure that all resource dependencies(storage, db, registry) will be external. if unset, a random selection will be used
@@ -261,6 +263,7 @@ if (config === undefined) {
 
 installerTests(TEST_CONFIGURATIONS[testConfig]).catch((err) => {
     cleanup();
+    deleteReplicatedLicense(werft, customerID);
     console.error(err);
     process.exit(1);
 });
@@ -341,6 +344,9 @@ export async function installerTests(config: TestConfig) {
         // if we are not doing preview, we delete the infrastructure
         cleanup();
     }
+
+    deleteReplicatedLicense(werft, customerID)
+
 }
 
 function runIntegrationTests() {

.werft/jobs/build/self-hosted-upgrade-tests.ts

@@ -1,6 +1,7 @@
 import { exec } from "../../util/shell";
 import { Werft } from "../../util/werft";
 import { JobConfig } from "./job-config";
+import * as https from "https";
 
 interface config {
     phase: string;
@@ -71,16 +72,21 @@ export async function triggerUpgradeTests(werft: Werft, config: JobConfig, usern
 export async function triggerSelfHostedPreview(werft: Werft, config: JobConfig, username: string) {
     const replicatedChannel =  config.replicatedChannel || config.repository.branch;
     const cluster =  config.cluster || "k3s";
+    const subdomain =  `${replicatedChannel.replace("/", "-").slice(0,10)}-${cluster}`
 
     var licenseFlag: string = ""
+    var customerID: string = ""
+    var annotation: string = ""
+
+
 
     if(!["stable", "unstable", "beta"].includes(replicatedChannel.toLowerCase())){
         werft.phase("get-replicated-license", `Create and download replicated license for ${replicatedChannel}`);
 
-        exec(`replicated customer create --channel ${replicatedChannel} --name ${replicatedChannel}`,
+        exec(`replicated customer create --channel ${replicatedChannel} --name ${config.version}`,
             { slice: "get-replicated-license"})
 
-        exec(`replicated customer download-license --customer ${replicatedChannel} > license.yaml`,
+        exec(`replicated customer download-license --customer ${config.version} > license.yaml`,
             { slice: "get-replicated-license", dontCheckRc: true})
 
         exec(`install -D license.yaml install/licenses/${replicatedChannel}.yaml`,
@@ -89,16 +95,26 @@ export async function triggerSelfHostedPreview(werft: Werft, config: JobConfig,
         werft.done("get-replicated-license");
 
         licenseFlag = `-s install/licenses/${replicatedChannel}.yaml`
+
+        const ret = exec(`replicated customer ls | grep ${config.version} | awk '{print $1}'`,
+                         { slice: "get-replicated-license", dontCheckRc: true})
+
+        const customerIDS = ret.stdout.split("\n").filter(item => item);
+        if(customerIDS.length > 0) {
+            customerID = customerIDS[0].trim()
+            annotation = `-a customerID=${customerID}`
+        }
     }
 
+    exec(`cat install/licenses/${replicatedChannel}.yaml`)
 
     exec(`git config --global user.name "${username}"`);
 
-    var annotation = `-a channel=${replicatedChannel} -a preview=true -a skipTests=true -a deps=external`;
+    annotation = `${annotation} -a channel=${replicatedChannel} -a preview=true -a skipTests=true -a deps=external`;
 
     werft.phase("self-hosted-preview", `Create self-hosted preview in ${cluster}`);
 
-    annotation = `${annotation} -a cluster=${cluster} -a updateGitHubStatus=gitpod-io/gitpod`
+    annotation = `${annotation} -a cluster=${cluster} -a updateGitHubStatus=gitpod-io/gitpod -a subdomain=${subdomain}`
 
     const testFile: string = ".werft/self-hosted-installer-tests.yaml";
 
@@ -115,6 +131,44 @@ export async function triggerSelfHostedPreview(werft: Werft, config: JobConfig,
         if (!config.mainBuild) {
             werft.fail("self-hosted-preview", err);
         }
+        deleteReplicatedLicense(werft, customerID)
         exec("exit 0");
     }
 }
+
+export async function deleteReplicatedLicense(werft: Werft, customerID: string) {
+    if(customerID == "") {
+        console.log("No customerID found, skipping replicated license cleanup")
+        return
+    }
+
+    console.log("trying to cleanup replicated license")
+    werft.phase("delete-replicated-license", "Deletes the replicated license created")
+    const http = require('https');
+
+    const options = {
+        method: 'POST',
+        hostname: 'api.replicated.com',
+        port: null,
+        path: `/vendor/v3/customer/${customerID}/archive`,
+        headers: {
+            Authorization: process.env.REPLICATED_API_TOKEN
+        }
+    };
+
+    const req = http.request(options, function (res) {
+        const chunks = [];
+
+        res.on('data', function (chunk) {
+            chunks.push(chunk);
+        });
+
+        res.on('end', function () {
+            const body = Buffer.concat(chunks);
+            console.log(body.toString());
+        });
+    });
+
+    req.end();
+    werft.done("delete-replicated-license")
+}

.werft/self-hosted-installer-tests.yaml

@@ -135,6 +135,11 @@ pod:
         secretKeyRef:
           name: slack-webhook-urls
           key: self_hosted_jobs
+    - name: REPLICATED_API_TOKEN
+      valueFrom:
+        secretKeyRef:
+          name: replicated
+          key: token
     command:
       - bash
       - -c