Add way to check for server and traefik updates, closes #62

Author: yamalight

src/commands/update.js

@@ -2,6 +2,7 @@
 const _ = require('lodash');
 const got = require('got');
 const chalk = require('chalk');
+const ora = require('ora');
 
 // our packages
 const {userConfig, isLoggedIn, logout} = require('../config');
@@ -10,18 +11,58 @@ const {userConfig, isLoggedIn, logout} = require('../config');
 const validTargets = ['traefik', 'server'];
 
 exports.command = ['update [target]'];
-exports.describe = 'update given target';
+exports.describe = 'check for updates or update given target';
 exports.builder = {
   target: {
     alias: 't',
     description: `Target for update (${validTargets.join(', ')})`,
   },
 };
-exports.handler = async ({target} = {target: 'self'}) => {
+exports.handler = async ({target}) => {
   if (!isLoggedIn()) {
     return;
   }
 
+  // construct shared request params
+  const options = {
+    headers: {
+      Authorization: `Bearer ${userConfig.token}`,
+    },
+    json: true,
+  };
+
+  // if no target given - check for update
+  if (!target || !target.length) {
+    // show loader
+    const spinner = ora('Checking for update...').start();
+
+    // services request url
+    const remoteUrl = `${userConfig.endpoint}/version`;
+    // send request
+    const {body, statusCode} = await got.get(remoteUrl, options);
+    if (statusCode !== 200 || body.error) {
+      spinner.fail('Error checking for update');
+      console.log(body.error || 'Oops. Something went wrong! Try again please.');
+      return;
+    }
+
+    if (body.serverUpdate || body.traefikUpdate) {
+      spinner.warn('Updates available!');
+    } else {
+      spinner.succeed('You are up to date!');
+    }
+
+    console.log();
+    console.log(chalk.bold('Exoframe Server:'));
+    console.log(`  current: ${body.server}`);
+    console.log(`  latest: ${body.latestServer}`);
+    console.log();
+    console.log(chalk.bold('Traefik:'));
+    console.log(`  current: ${body.traefik}`);
+    console.log(`  latest: ${body.latestTraefik}`);
+    return;
+  }
+
   if (!validTargets.includes(target)) {
     console.log(
       chalk.red('Error!'),
@@ -35,13 +76,6 @@ exports.handler = async ({target} = {target: 'self'}) => {
 
   // services request url
   const remoteUrl = `${userConfig.endpoint}/update/${target}`;
-  // construct shared request params
-  const options = {
-    headers: {
-      Authorization: `Bearer ${userConfig.token}`,
-    },
-    json: true,
-  };
   // try sending request
   try {
     const {body, statusCode} = await got.post(remoteUrl, options);

test/update.js

@@ -92,6 +92,49 @@ module.exports = () => {
     });
   });
 
+  // test version check
+  tap.test('Should display versions', t => {
+    // handle correct request
+    const response = {
+      server: '0.18.0',
+      latestServer: '0.19.1',
+      serverUpdate: true,
+      traefik: 'v1.3.0',
+      latestTraefik: 'v1.3.2',
+      traefikUpdate: true,
+    };
+    const updateServer = nock('http://localhost:8080')
+      .get('/version')
+      .reply(200, response);
+    // spy on console
+    const consoleSpy = sinon.spy(console, 'log');
+    // execute login
+    update({}).then(() => {
+      // make sure log in was successful
+      // check that server was called
+      t.ok(updateServer.isDone());
+      // first check console output
+      t.deepEqual(
+        consoleSpy.args,
+        [
+          [],
+          ['Exoframe Server:'],
+          ['  current: 0.18.0'],
+          ['  latest: 0.19.1'],
+          [],
+          ['Traefik:'],
+          ['  current: v1.3.0'],
+          ['  latest: v1.3.2'],
+        ],
+        'Correct log output'
+      );
+      // restore console
+      console.log.restore();
+      updateServer.done();
+      t.end();
+    });
+  });
+
   // test deauth
   tap.test('Should deauth on 401', t => {
     // copy original config for restoration