Feature/andrii/saas 6026 set helm version (#417)

* set helm version

* bump version

* set helm config, tests

Author: pasha-codefresh

lib/interface/cli/commands/helm/helm.sdk.spec.js

@@ -1,15 +1,32 @@
-let setHelmVersionCmd = require('./set-helm-version.cmd');
+let setHelmConfigCmd = require('./set-helm-config.cmd');
 
-setHelmVersionCmd.requiresAuthentication = false;
+setHelmConfigCmd.requiresAuthentication = false;
 
-setHelmVersionCmd = setHelmVersionCmd.toCommand();
+setHelmConfigCmd = setHelmConfigCmd.toCommand();
 
 jest.mock('../../helpers/helm');
 jest.mock('../../helpers/logs');
 jest.spyOn(process, 'exit').mockImplementation();
 
 const request = require('requestretry');
 
+jest.mock('../../../../logic', () => {
+    const mockPipelinesList = jest.fn();
+    mockPipelinesList.mockReturnValue(Promise.resolve([{
+        metadata: {
+            name: 'pip1',
+            id: 'pip1',
+        },
+    }]));
+
+    // eslint-disable-next-line global-require
+    const sdk = require('../../../../logic/sdk');
+    sdk.pipelines = {
+        getNames: mockPipelinesList,
+    };
+    return { sdk };
+});
+
 const DEFAULT_RESPONSE = request.__defaultResponse();
 
 describe('helm commands', () => {
@@ -19,27 +36,47 @@ describe('helm commands', () => {
         await configureSdk(); // eslint-disable-line
     });
 
-    describe('set-helm-version', () => {
+    describe('set-helm-config', () => {
 
-        it('should set version', async () => {
+        it('should set config', async () => {
             const argv = {
                 cluster: 'selector',
-                version: '3',
+                version: 'helm3',
+                testReleasePipeline: 'pip1',
+                rollbackPipeline: 'pip1',
+                deleteReleasePipeline: 'pip1',
+                tillerNamespace: 'kube-system',
+                releaseNamespaces: ['ns1', 'ns2'],
             };
-            await setHelmVersionCmd.handler(argv);
+            await setHelmConfigCmd.handler(argv);
             await verifyResponsesReturned([DEFAULT_RESPONSE]); // eslint-disable-line
         });
 
         it('should throw error for wrong version', async () => {
             const argv = {
                 cluster: 'selector',
-                version: '1',
+                version: 'helm1',
             };
             try {
-                await setHelmVersionCmd.handler(argv);
+                await setHelmConfigCmd.handler(argv);
             } catch (error) {
                 expect(error.message).toBe('Wrong version value');
             }
         });
+
+        it('should throw error for wrong pipeline name', async () => {
+            const argv = {
+                cluster: 'selector',
+                version: 'helm3',
+                testReleasePipeline: 'pip2',
+            };
+            try {
+                await setHelmConfigCmd.handler(argv);
+            } catch (error) {
+                expect(error.message).toBe('Pipeline pip2 not found');
+                return;
+            }
+            throw new Error('should throw error for wrong pipeline name');
+        });
     });
 });

lib/interface/cli/commands/helm/set-helm-config.cmd.js

@@ -0,0 +1,93 @@
+const Command = require('../../Command');
+const { sdk } = require('../../../../logic');
+const Output = require('../../../../output/Output');
+const _ = require('lodash');
+
+const install = new Command({
+    root: true,
+    command: 'set-helm-config <cluster>',
+    description: 'Set Helm config',
+    webDocs: {
+        category: 'Predefined Pipelines',
+        title: 'Set Helm config',
+        weight: 20,
+    },
+    builder: (yargs) => {
+        return yargs
+            .positional('cluster', {
+                describe: 'Cluster name on integrations page',
+                required: true,
+            })
+            .option('version', {
+                describe: 'Major part of helm version (helm2 | helm3)',
+                alias: 'v',
+            })
+            .option('test-release-pipeline', {
+                describe: 'Pipeline for testing release',
+                alias: 't',
+            })
+            .option('rollback-pipeline', {
+                describe: 'Pipeline for rollback',
+                alias: 'r',
+            })
+            .option('delete-release-pipeline', {
+                describe: 'Pipeline for deleting release',
+                alias: 'd',
+            })
+            .option('tiller-namespace', {
+                describe: 'Tiller namespace for helm2',
+            })
+            .option('release-namespaces', {
+                describe: 'List of namespaces for using',
+            })
+            .example('codefresh set-helm-config cluster -v 2 -t pip1 -r pip2 -d pip3 --tiller-namespace kube-system', 'Helm2 config')
+            .example('codefresh set-helm-config cluster -v 3 -t pip1 -r pip2 -d pip3 --release-namespaces ns1 --release-namespaces ns2', 'Helm3 config');
+    },
+    handler: async (argv) => {
+        try {
+            if (!['helm2', 'helm3'].includes(argv.version)) {
+                throw new Error('Wrong version value');
+            }
+            let options = {};
+            options.selector = argv.cluster;
+            options.helmVersion = argv.version;
+
+            const pipelines = await sdk.pipelines.getNames({});
+
+            if (argv.deleteReleasePipeline) {
+                const pipeline = _.find(pipelines, p => p.metadata.name === argv.deleteReleasePipeline);
+                options.deleteReleasePipeline = _.get(pipeline, 'metadata.id');
+                if (!options.deleteReleasePipeline) {
+                    throw new Error(`Pipeline ${argv.deleteReleasePipeline} not found`);
+                }
+            }
+
+            if (argv.testReleasePipeline) {
+                const pipeline = _.find(pipelines, p => p.metadata.name === argv.testReleasePipeline);
+                options.testReleasePipeline = _.get(pipeline, 'metadata.id');
+                if (!options.testReleasePipeline) {
+                    throw new Error(`Pipeline ${argv.testReleasePipeline} not found`);
+                }
+            }
+
+            if (argv.rollbackPipeline) {
+                const pipeline = _.find(pipelines, p => p.metadata.name === argv.rollbackPipeline);
+                options.rollbackPipeline = _.get(pipeline, 'metadata.id');
+                if (!options.rollbackPipeline) {
+                    throw new Error(`Pipeline ${argv.rollbackPipeline} not found`);
+                }
+            }
+
+            options = _.merge(options, _.pick(argv, ['releaseNamespaces', 'tillerNamespace']));
+
+            await sdk.helm['cluster-config'].store(options);
+            console.log('Helm config was set successfully');
+            process.exit();
+        } catch (err) {
+            Output.printError(err);
+            process.exit(1);
+        }
+    },
+});
+
+module.exports = install;