Update run.ts

src/run.ts

@@ -1 +1,202 @@
+// Copyright (c) Microsoft Corporation.
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT license.
 
+import * as os from 'os';
+import * as path from 'path';
+import * as util from 'util';
+import * as fs from 'fs';
+import * as semver from 'semver';
+
+import * as toolCache from '@actions/tool-cache';
+import * as core from '@actions/core';
+import { graphql } from '@octokit/graphql';
+
+const helmToolName = 'helm';
+const stableHelmVersion = 'v3.2.1';
+const stableHelm3Version = 'v3.5.3';
+const stableHelm2Version = 'v2.17.0';
+const LATEST_HELM2_VERSION = '2.*';
+const LATEST_HELM3_VERSION = '3.*';
+const helmAllReleasesUrl = 'https://api.github.com/repos/helm/helm/releases';
+
+export function getExecutableExtension(): string {
+    if (os.type().match(/^Win/)) {
+        return '.exe';
+    }
+    return '';
+}
+
+export function getHelmDownloadURL(version: string): string {
+    switch (os.type()) {
+        case 'Linux':
+            return util.format('https://get.helm.sh/helm-%s-linux-amd64.zip', version);
+
+        case 'Darwin':
+            return util.format('https://get.helm.sh/helm-%s-darwin-amd64.zip', version);
+
+        case 'Windows_NT':
+        default:
+            return util.format('https://get.helm.sh/helm-%s-windows-amd64.zip', version);
+    }
+}
+
+export async function getStableHelmVersion(): Promise<string> {
+    try {
+        const downloadPath = await toolCache.downloadTool(helmAllReleasesUrl);
+        const responseArray = JSON.parse(fs.readFileSync(downloadPath, 'utf8').toString().trim());
+        let latestHelmVersion = semver.clean(stableHelmVersion);
+        responseArray.forEach(response => {
+            if (response && response.tag_name) {
+                let currentHelmVerison = semver.clean(response.tag_name.toString());
+                if (currentHelmVerison) {
+                    if (currentHelmVerison.toString().indexOf('rc') == -1 && semver.gt(currentHelmVerison, latestHelmVersion)) {
+                        //If current helm version is not a pre release and is greater than latest helm version
+                        latestHelmVersion = currentHelmVerison;
+                    }
+                }
+            }
+        });
+        latestHelmVersion = "v" + latestHelmVersion;
+        return latestHelmVersion;
+    } catch (error) {
+        core.warning(util.format("Cannot get the latest Helm info from %s. Error %s. Using default Helm version %s.", helmAllReleasesUrl, error, stableHelmVersion));
+    }
+
+    return stableHelmVersion;
+}
+
+export var walkSync = function (dir, filelist, fileToFind) {
+    var files = fs.readdirSync(dir);
+    filelist = filelist || [];
+    files.forEach(function (file) {
+        if (fs.statSync(path.join(dir, file)).isDirectory()) {
+            filelist = walkSync(path.join(dir, file), filelist, fileToFind);
+        }
+        else {
+            core.debug(file);
+            if (file == fileToFind) {
+                filelist.push(path.join(dir, file));
+            }
+        }
+    });
+    return filelist;
+};
+
+export async function downloadHelm(version: string): Promise<string> {
+    if (!version) { version = await getStableHelmVersion(); }
+    let cachedToolpath = toolCache.find(helmToolName, version);
+    if (!cachedToolpath) {
+        let helmDownloadPath;
+        try {
+            helmDownloadPath = await toolCache.downloadTool(getHelmDownloadURL(version));
+        } catch (exception) {
+            throw new Error(util.format("Failed to download Helm from location", getHelmDownloadURL(version)));
+        }
+
+        fs.chmodSync(helmDownloadPath, '777');
+        const unzipedHelmPath = await toolCache.extractZip(helmDownloadPath);
+        cachedToolpath = await toolCache.cacheDir(unzipedHelmPath, helmToolName, version);
+    }
+
+    const helmpath = findHelm(cachedToolpath);
+    if (!helmpath) {
+        throw new Error(util.format("Helm executable not found in path", cachedToolpath));
+    }
+
+    fs.chmodSync(helmpath, '777');
+    return helmpath;
+}
+
+async function getLatestHelmVersionFor(type: string): Promise<string> {
+    const token = core.getInput('token', { 'required': true });
+    try {
+        const { repository } = await graphql(
+            `{
+                repository(name: "helm", owner: "helm") {
+                    releases(last: 100)  {
+                        nodes {
+                            tagName
+                        }
+                    }
+                }
+            }`,
+            {
+                headers: {
+                    authorization: `token ${token}`
+                }
+            }
+        );
+
+        const releases = repository.releases.nodes.reverse();
+        let latestValidRelease = releases.find(release => isValidVersion(release.tagName, type));
+        if (latestValidRelease)
+            return latestValidRelease.tagName;
+    } catch (err) {
+        core.warning(util.format("Error while fetching the latest Helm %s release. Error: %s. Using default Helm version %s.", type, err.toString(), getStableHelmVersionFor(type)));
+        return getStableHelmVersionFor(type);
+    }
+    core.warning(util.format("Could not find stable release for Helm %s. Using default Helm version %s.", type, getStableHelmVersionFor(type)));
+    return getStableHelmVersionFor(type);
+}
+
+function getStableHelmVersionFor(type: string) {
+    return type === "v2" ? stableHelm2Version : stableHelm3Version;
+}
+
+// isValidVersion checks if verison matches the specified type and is a stable release
+function isValidVersion(version: string, type: string): boolean {
+    if (!version.toLocaleLowerCase().startsWith(type))
+        return false;
+    return version.indexOf('rc') == -1;
+}
+
+export function findHelm(rootFolder: string): string {
+    fs.chmodSync(rootFolder, '777');
+    var filelist: string[] = [];
+    walkSync(rootFolder, filelist, helmToolName + getExecutableExtension());
+    if (!filelist || filelist.length == 0) {
+        throw new Error(util.format("Helm executable not found in path", rootFolder));
+    }
+    else {
+        return filelist[0];
+    }
+}
+
+export async function run() {
+    let version = core.getInput('version', { 'required': true });
+
+    if (process.env['HELM_INSTALLER_LEGACY_VERSIONING'] == 'true') {
+        if (version.toLocaleLowerCase() === 'latest') {
+            version = await getStableHelmVersion();
+        } else if (!version.toLocaleLowerCase().startsWith('v')) {
+            version = 'v' + version;
+        }
+    } else {
+        if (version.toLocaleLowerCase() === 'latest' || version === LATEST_HELM3_VERSION) {
+            version = await getLatestHelmVersionFor("v3");
+        } else if (version === LATEST_HELM2_VERSION) {
+            version = await getLatestHelmVersionFor("v2");
+        } else if (!version.toLocaleLowerCase().startsWith('v')) {
+            version = 'v' + version;
+        }
+    }
+
+    core.debug(util.format("Downloading %s", version));
+    let cachedPath = await downloadHelm(version);
+
+    try {
+
+        if (!process.env['PATH'].startsWith(path.dirname(cachedPath))) {
+            core.addPath(path.dirname(cachedPath));
+        }
+    }
+    catch {
+        //do nothing, set as output variable
+    }
+
+    console.log(`Helm tool version: '${version}' has been cached at ${cachedPath}`);
+    core.setOutput('helm-path', cachedPath);
+}
+
+run().catch(core.setFailed);