Add way to generate deployment token, addresses #33

Author: yamalight

src/commands/token.js

@@ -0,0 +1,46 @@
+// npm packages
+const got = require('got');
+const chalk = require('chalk');
+
+// our packages
+const {userConfig, isLoggedIn, logout} = require('../config');
+
+exports.command = ['token'];
+exports.describe = 'generate new deployment token';
+exports.builder = {};
+exports.handler = async () => {
+  if (!isLoggedIn()) {
+    return;
+  }
+
+  console.log(chalk.bold('Generating new deployment token for:'), userConfig.endpoint);
+
+  // services request url
+  const remoteUrl = `${userConfig.endpoint}/deployToken`;
+  // construct shared request params
+  const options = {
+    headers: {
+      Authorization: `Bearer ${userConfig.token}`,
+    },
+    json: true,
+  };
+  // try sending request
+  try {
+    const {body} = await got(remoteUrl, options);
+    const {token} = body;
+    console.log(chalk.bold('New token generated:'));
+    console.log('');
+    console.log(token);
+    console.log('');
+    console.log(chalk.yellow('WARNING!'), 'Make sure to write it down, you will not be able to get it again!');
+  } catch (e) {
+    // if authorization is expired/broken/etc
+    if (e.statusCode === 401) {
+      logout(userConfig);
+      console.log(chalk.red('Error: authorization expired!'), 'Please, relogin and try again.');
+      return;
+    }
+
+    console.log(chalk.red('Error generating deployment token:'), e.toString());
+  }
+};

src/index.js

@@ -9,6 +9,7 @@ const logs = require('./commands/logs');
 const remove = require('./commands/remove');
 const endpoint = require('./commands/endpoint');
 const config = require('./commands/config');
+const token = require('./commands/token');
 
 // init program
 yargs
@@ -22,4 +23,5 @@ yargs
   .command(list)
   .command(logs)
   .command(remove)
+  .command(token)
   .command(config).argv;

test/index.js

@@ -5,6 +5,7 @@ const deploy = require('./deploy');
 const logs = require('./logs');
 const list = require('./list');
 const config = require('./config');
+const token = require('./token');
 
 // run tests
 login();
@@ -13,3 +14,4 @@ deploy();
 logs();
 list();
 config();
+token();

test/token.js

@@ -0,0 +1,73 @@
+// npm packages
+const tap = require('tap');
+const nock = require('nock');
+const sinon = require('sinon');
+
+// our packages
+const {handler: token} = require('../src/commands/token');
+const {userConfig, updateConfig} = require('../src/config');
+
+module.exports = () => {
+  // test removal
+  tap.test('Should generate token', t => {
+    // handle correct request
+    const tokenServer = nock('http://localhost:8080').get('/deployToken').reply(200, {token: 'test'});
+    // spy on console
+    const consoleSpy = sinon.spy(console, 'log');
+    // execute login
+    token().then(() => {
+      // make sure log in was successful
+      // check that server was called
+      t.ok(tokenServer.isDone());
+      // first check console output
+      t.deepEqual(
+        consoleSpy.args,
+        [
+          ['Generating new deployment token for:', 'http://localhost:8080'],
+          ['New token generated:'],
+          [''],
+          ['test'],
+          [''],
+          ['WARNING!', 'Make sure to write it down, you will not be able to get it again!'],
+        ],
+        'Correct log output'
+      );
+      // restore console
+      console.log.restore();
+      tokenServer.done();
+      t.end();
+    });
+  });
+
+  // test
+  tap.test('Should deauth on 401', t => {
+    // copy original config for restoration
+    const originalConfig = Object.assign({}, userConfig);
+    // handle correct request
+    const tokenServer = nock('http://localhost:8080').get('/deployToken').reply(401);
+    // spy on console
+    const consoleSpy = sinon.spy(console, 'log');
+    // execute login
+    token().then(() => {
+      // make sure log in was successful
+      // check that server was called
+      t.ok(tokenServer.isDone());
+      // first check console output
+      t.deepEqual(
+        consoleSpy.args,
+        [
+          ['Generating new deployment token for:', 'http://localhost:8080'],
+          ['Error: authorization expired!', 'Please, relogin and try again.'],
+        ],
+        'Correct log output'
+      );
+      // restore console
+      console.log.restore();
+      // tear down nock
+      tokenServer.done();
+      // restore original config
+      updateConfig(originalConfig);
+      t.end();
+    });
+  });
+};