Auto creation cluster and re (#234)

ziv-codefresh · web-flow · commit c89f745a3357 · 2018-11-28T12:08:24.000+02:00
diff --git a/lib/interface/cli/commands/cluster/create.cmd.js b/lib/interface/cli/commands/cluster/create.cmd.js
@@ -0,0 +1,36 @@
+const debug = require('debug')('codefresh:cli:create:context');
+const Command = require('../../Command');
+const { cluster } = require('../../../../logic').api;
+const createRoot = require('../root/create.cmd');
+const authManager = require('../../../../logic/auth').manager; // eslint-disable-line
+
+const command = new Command({
+    command: 'clusters [name]',
+    aliases: ['cluster'],
+    parent: createRoot,
+    description: 'Create a cluster',
+    webDocs: {
+        category: 'Clusters',
+        title: 'Create Cluster',
+        weight: 100,
+    },
+    builder: (yargs) => {
+        return yargs
+            .positional('name', {
+                describe: 'cluster name',
+                required: true,
+            })
+            .example('codefresh create cluster [name]', 'Creating a cluster');
+    },
+    handler: async (argv) => {
+        const context = authManager.getCurrentContext();
+        const { name } = argv;
+        await cluster.createCluster({
+            name,
+            context,
+        });
+    },
+});
+
+
+module.exports = command;
diff --git a/lib/interface/cli/commands/cluster/delete.cmd.js b/lib/interface/cli/commands/cluster/delete.cmd.js
@@ -0,0 +1,29 @@
+const debug = require('debug')('codefresh:cli:create:context');
+const Command = require('../../Command');
+const { cluster } = require('../../../../logic').api;
+const deleteRoot = require('../root/delete.cmd');
+
+const command = new Command({
+    command: 'cluster <name>',
+    aliases: ['clusters'],
+    description: 'Delete a cluster',
+    parent: deleteRoot,
+    webDocs: {
+        category: 'Clusters',
+        title: 'Delete Cluster',
+    },
+    builder: (yargs) => {
+        return yargs
+            .positional('name', {
+                describe: 'Cluster name',
+                required: true,
+            })
+            .example('codefresh delete cluster NAME', 'Delete cluster NAME');
+    },
+    handler: async (argv) => {
+        await cluster.deleteCluster(argv.name);
+        console.log(`Cluster: ${argv.name} deleted`);
+    },
+});
+
+module.exports = command;
diff --git a/lib/interface/cli/commands/cluster/get.cmd.js b/lib/interface/cli/commands/cluster/get.cmd.js
@@ -0,0 +1,29 @@
+const debug = require('debug')('codefresh:cli:get:cluster');
+const Command = require('../../Command');
+const { cluster } = require('../../../../logic').api;
+const { specifyOutputForArray } = require('../../helpers/get');
+const getRoot = require('../root/get.cmd');
+
+
+const command = new Command({
+    command: 'clusters',
+    aliases: ['cluster'],
+    category: 'Clusters',
+    parent: getRoot,
+    description: 'Get an array of clusters',
+    webDocs: {
+        category: 'Clusters',
+        title: 'Get Clusters',
+    },
+    builder: (yargs) => {
+        return yargs
+            .example('codefresh get clusters', 'Get all clusters');
+    },
+    handler: async (argv) => {
+        const clusters = await cluster.getAllClusters();
+        specifyOutputForArray(argv.output, clusters);
+    },
+});
+
+module.exports = command;
+
diff --git a/lib/interface/cli/commands/runtimeEnvironments/create.cmd.js b/lib/interface/cli/commands/runtimeEnvironments/create.cmd.js
@@ -0,0 +1,90 @@
+const debug = require('debug')('codefresh:cli:create:context');
+const Command = require('../../Command');
+const fs = require('fs');
+const { spawn } = require('child_process');
+const { homedir } = require('os');
+const rp = require('request-promise');
+const createRoot = require('../root/create.cmd');
+const authManager = require('../../../../logic/auth').manager; // eslint-disable-line
+
+const scriptUrl = 'https://raw.githubusercontent.com/codefresh-io/k8s-dind-config/master/codefresh-k8s-configure.sh';
+let filePath = `${homedir()}/.Codefresh/runtime/codefresh-k8s-configure.sh`;
+const dirPath = `${homedir()}/.Codefresh/runtime`;
+
+
+const callToScript = (k8sScript) =>{
+    k8sScript.stdout.pipe(process.stdout);
+    k8sScript.stderr.pipe(process.stderr);
+    process.stdin.pipe(k8sScript.stdin);
+    k8sScript.on('exit', (code) => {
+        fs.unlink('filePath', (error) => {
+            if (error) {
+                throw error;
+            }
+            console.log('finish');
+        });
+        process.exit(code);
+    });
+};
+
+
+const command = new Command({
+    command: 'runtime-environments [cluster]',
+    aliases: ['re', 'runtime-environment'],
+    parent: createRoot,
+    description: 'Create a runtime environment',
+    webDocs: {
+        category: 'Runtime-Environments (On Prem)',
+        title: 'Create Runtime-Environments',
+        weight: 100,
+    },
+    builder: (yargs) => {
+        return yargs
+            .positional('cluster', {
+                describe: 'cluster name',
+                required: true,
+            })
+            .option('namespace', {
+                describe: 'namespace',
+            })
+            .option('local', {
+                describe: 'set if run the script from local file system',
+                type: 'boolean',
+                default: false,
+            })
+            .option('context', {
+                describe: 'set your kubectl context',
+                default: '',
+            })
+            .example('codefresh create re [cluster] --namespace codefresh --context kubeCodefresh', 'Creating a runtime environment');
+    },
+    handler: async (argv) => {
+        const currentContext = authManager.getCurrentContext();
+        const { local, namespace, cluster, context } = argv;
+        if (!local) {
+            if (!fs.existsSync(dirPath)) {
+                fs.mkdirSync(dirPath);
+            }
+            const options = {
+                url: scriptUrl,
+                method: 'GET',
+            };
+            const response = await rp(options);
+            fs.writeFile(filePath, response, (err) => {
+                if (err) {
+                    throw err;
+                }
+                fs.chmodSync(filePath, '644');
+                const k8sScript = spawn('bash', [filePath, '--api-token', currentContext.token, '--api-host', currentContext.url, '--namespace', namespace, '--context', context, cluster]);
+                callToScript(k8sScript);
+            });
+        } else {
+            filePath = './codefresh-k8s-configure.sh';
+            const k8sScript = spawn('bash', [filePath, '--api-token', currentContext.token, '--api-host', currentContext.url, '--namespace', namespace, '--local','--context', context, cluster]);
+            callToScript(k8sScript);
+        }
+    },
+});
+
+
+module.exports = command;
diff --git a/lib/logic/api/cluster.js b/lib/logic/api/cluster.js
@@ -0,0 +1,125 @@
+const _ = require('lodash');
+const { sendHttpRequest } = require('./helper');
+const Cluster = require('../entities/Cluster');
+const rp = require('request-promise');
+const fs = require('fs');
+const { spawn } = require('child_process');
+const { homedir, arch } = require('os');
+const request = require('request');
+const decompress = require('decompress');
+const decompressTargz = require('decompress-targz');
+const compareVersions = require('compare-versions');
+
+const _createClusterScript = (info, filePath) => {
+    const { name, context } = info;
+    fs.chmodSync(filePath, '755');
+    const clusterScript = spawn(filePath, ['create', '--c', name, '--token', context.token, '--api-host', `${context.url}/`] );
+    clusterScript.stdout.pipe(process.stdout);
+    clusterScript.stderr.pipe(process.stderr);
+    process.stdin.pipe(clusterScript.stdin);
+    clusterScript.on('exit', (code) => {
+        process.exit(code);
+    });
+};
+
+const _extractFieldsForTeamEntity = cluster => ({
+    id: cluster._id,
+    name: cluster.selector,
+    provider: cluster.provider,
+    providerAgent: cluster.providerAgent,
+});
+
+const getAllClusters = async () => {
+    const userOptions = {
+        url: '/api/clusters',
+        method: 'GET',
+    };
+
+    const result = await sendHttpRequest(userOptions);
+    const clusters = [];
+    let data = {};
+    _.forEach(result, (cluster) => {
+        data = _extractFieldsForTeamEntity(cluster);
+        clusters.push(new Cluster(data));
+    });
+    return clusters;
+};
+
+const createCluster = async (info) => {
+    const dirPath = `${homedir()}/.Codefresh/cluster`;
+    const filePath = `${homedir()}/.Codefresh/cluster/stevedore`;
+    const versionPath = `${homedir()}/.Codefresh/cluster/version.txt`;
+    const versionUrl = 'https://raw.githubusercontent.com/codefresh-io/Stevedore/master/VERSION';
+    let zipPath = `${homedir()}/.Codefresh/cluster/data`;
+    let shouldUpdate = true;
+    const options = {
+        url: versionUrl,
+        method: 'GET',
+        headers: {
+            'User-Agent': 'codefresh',
+        },
+    };
+    const version = await rp(options);
+    if (!fs.existsSync(dirPath)) {
+        fs.mkdirSync(dirPath);
+    } else if (fs.existsSync(versionPath)) {
+        const currVersion = fs.readFileSync(versionPath, { encoding: 'UTF8' }).trim();
+        if (compareVersions(currVersion, version) >= 0) {
+            shouldUpdate = false;
+        }
+    }
+    if (shouldUpdate) {
+        let osType;
+        const { platform } = process;
+        if (_.isEqual(platform, 'darwin')) {
+            osType = _.isEqual(arch(), 'x32') ? 'Darwin_i386.tar.gz' : 'Darwin_x86_64.tar.gz';
+            zipPath = `${zipPath}.tar.gz`;
+        } else if (_.isEqual(platform, 'linux')) {
+            osType = _.isEqual(arch(), 'x32') ? 'Linux_i386.tar.gz' : 'Linux_x86_64.tar.gz';
+            zipPath = `${zipPath}.tar.gz`;
+        } else if (_.isEqual(platform, 'ein32')) {
+            osType = _.isEqual(arch(), 'x32') ? 'Windows_i386.zip' : 'Windows_x86_64.zip';
+            zipPath = `${zipPath}.zip`;
+        }
+        const assetUrl = `https://github.com/codefresh-io/Stevedore/releases/download/v${version}/stevedore_${version}_${osType}`;
+        const req = request(assetUrl);
+        req.pipe(fs.createWriteStream(zipPath));
+        req.on('end', () => {
+            console.log('File written!');
+            decompress(zipPath, `${homedir()}/.Codefresh/cluster`, {
+                plugins: [
+                    decompressTargz(),
+                ],
+            }).then(() => {
+                console.log('Files decompressed');
+                fs.writeFile(versionPath, version, (err) => {
+                    if (err) {
+                        throw err;
+                    }
+                });
+                _createClusterScript(info, filePath);
+            });
+        });
+    } else {
+        _createClusterScript(info, filePath);
+    }
+};
+
+const deleteCluster = async (clusterName) => {
+    const clusters = await getAllClusters();
+    const cluster = _.find(clusters, (curr) => {
+        return _.isEqual(curr.info.name, clusterName);
+    });
+    const options = {
+        url: `/api/clusters/${cluster.info.provider}/cluster/${cluster.info.id}`,
+        method: 'DELETE',
+    };
+
+    return sendHttpRequest(options);
+};
+
+module.exports = {
+    createCluster,
+    getAllClusters,
+    deleteCluster,
+};
diff --git a/lib/logic/api/index.js b/lib/logic/api/index.js
@@ -13,6 +13,7 @@ const runtimeEnvironments = require('./runtimeEnvironments');
 const team = require('./team');
 const board = require('./board');
 const section = require('./section');
+const cluster = require('./cluster');
 
 module.exports = {
     user,
@@ -30,4 +31,5 @@ module.exports = {
     team,
     board,
     section,
+    cluster,
 };
diff --git a/lib/logic/entities/Cluster.js b/lib/logic/entities/Cluster.js
@@ -0,0 +1,13 @@
+const Entity = require('./Entity');
+
+class Cluster extends Entity {
+    constructor(data) {
+        super();
+        this.entityType = 'cluster';
+        this.info = data;
+        this.defaultColumns = ['id', 'name'];
+        this.wideColumns = this.defaultColumns.concat(['provider', 'providerAgent']);
+    }
+}
+
+module.exports = Cluster;
diff --git a/package.json b/package.json
@@ -32,8 +32,11 @@
     "cli-progress": "^1.6.1",
     "colors": "^1.1.2",
     "columnify": "^1.5.4",
+    "compare-versions": "^3.4.0",
     "copy-dir": "^0.3.0",
     "debug": "^3.1.0",
+    "decompress": "^4.2.0",
+    "decompress-targz": "^4.1.1",
     "diff": "^3.5.0",
     "draftlog": "^1.0.12",
     "filesize": "^3.5.11",
@@ -47,7 +50,7 @@
     "mongodb": "^3.0.1",
     "prettyjson": "^1.2.1",
     "recursive-readdir": "^2.2.1",
-    "request": "^2.83.0",
+    "request": "^2.88.0",
     "request-promise": "^4.2.2",
     "rimraf": "^2.6.2",
     "tty-table": "^2.5.5",
diff --git a/yarn.lock b/yarn.lock
