Reload workers when updating

package.json

@@ -26,7 +26,9 @@
     "find-up": "^2.1.0",
     "lowdb": "^1.0.0",
     "opentmi-jsclient": "git+https://github.com/opentmi/opentmi-jsclient.git#v0.3.0",
-    "yargs": "^11.0.0"
+    "yargs": "^11.0.0",
+    "lodash": "^4.17.5",
+    "table": "^4.0.2"
   },
   "devDependencies": {
     "eslint": "^4.17.0",

src/arguments.js

@@ -5,7 +5,10 @@ const yargs = require('yargs');
 
 // app modules
 const {authenticateUser, logoutUser} = require('./cmds/login');
-const {showVersion, updateVersion} = require('./cmds/update');
+const {showVersion} = require('./cmds/show');
+const {updateVersion} = require('./cmds/update');
+const {reloadWorkers} = require('./cmds/reload');
+const {listenResults} = require('./cmds/listen');
 const config = require('./config');
 
 
@@ -18,6 +21,12 @@ yargs.usage('Usage: $0 <command> [options]')
   .command('update', 'Update something from server', argv =>
     argv
       .command('version <revision>', 'Update server version', () => {}, updateVersion))
+  .command('reload', 'Reload something from server', argv =>
+    argv
+      .command('workers', 'Reload workers', () => {}, reloadWorkers))
+  .command('listen', 'Listen events', argv =>
+    argv
+      .command('results', 'Listen new results', () => {}, listenResults))
   .config(config.getState())
   .demandCommand()
   .alias('h', 'help')

src/cmds/listen.js

@@ -0,0 +1,60 @@
+const _ = require('lodash');
+const {Transport} = require('opentmi-jsclient');
+const {createStream} = require('table');
+
+
+const untilSignal = (signal='SIGINT') => {
+  console.log(`listening until ${signal}..`);
+  return new Promise((resolve) => {
+    process.on(signal, resolve);
+  });
+};
+
+const colored = (value) => {
+  const reset ='\x1B[49m';
+  const red = '\x1B[41m';
+  const green = '\x1B[42m';
+  const yellow = '\x1B[43m';
+
+  const map = {
+    pass: green,
+    fail: red,
+    inconclusive: yellow
+  };
+  const color = _.get(map, value, yellow);
+  return `${color}${value}${reset}`
+};
+
+
+const listenResults = (argv) => {
+  const transport = new Transport(argv.host);
+  transport.token = argv.token;
+
+  console.log(`Start listening new results`);
+  const config = {
+      columnDefault: {
+          width: 50
+      },
+      columnCount: 2
+  };
+  const stream = createStream(config);
+  transport
+    .connect('/results')
+    .then(() => transport.sio('/results'))
+    .then((sio) => {
+      sio.on('new', (result) => {
+        const tc = result.tcid;
+        const verdict = colored(_.get(result, 'exec.verdict'));
+        stream.write([tc, verdict]);
+      });
+      return untilSignal();
+    })
+    .then(() => {
+      transport.disconnect();
+    });
+};
+
+
+module.exports = {
+  listenResults
+};

src/cmds/reload.js

@@ -0,0 +1,17 @@
+const {Transport, Admin} = require('opentmi-jsclient');
+
+const reloadWorkers = (argv) => {
+  const transport = new Transport(argv.host);
+  transport.token = argv.token;
+  const admin = new Admin(transport);
+  console.log('Start reloading workers..');
+  return admin.reloadWorkers()
+    .then(() => { console.log(`Reload success`); })
+    .catch((error) => {
+      console.error(`failed: ${error.message}`);
+    });
+};
+
+module.exports = {
+  reloadWorkers
+};

src/cmds/show.js

@@ -0,0 +1,16 @@
+const {Transport, Admin} = require('opentmi-jsclient');
+
+const showVersion = (argv) => {
+  const transport = new Transport(argv.host);
+  transport.token = argv.token;
+  const admin = new Admin(transport);
+  admin.version()
+    .then(ver => console.log(ver))
+    .catch((error) => {
+      console.error(`failed: ${error.message}`);
+    });
+};
+
+module.exports = {
+  showVersion
+};

src/cmds/update.js

@@ -1,30 +1,31 @@
 const {Transport, Admin} = require('opentmi-jsclient');
 
-const showVersion = (argv) => {
-  const transport = new Transport(argv.host);
-  transport.token = argv.token;
-  const admin = new Admin(transport);
-  admin.version()
-    .then(ver => console.log(ver))
-    .catch((error) => {
-      console.error(`failed: ${error.message}`);
-    });
-};
 
 const updateVersion = (argv) => {
   const version = argv.revision;
   const transport = new Transport(argv.host);
   transport.token = argv.token;
   const admin = new Admin(transport);
   console.log(`Start updating to ${version}`);
-  admin.upgrade(version)
-    .then(ver => console.log(`Backend updated to ${ver}`))
-    .catch((error) => {
-      console.error(`failed: ${error.message}`);
+  transport
+    .connect()
+    .then(() => transport.sio())
+    .then((sio) => {
+      sio.on('status', (status) => {
+        console.log(status);
+      });
+      return admin.upgrade(version)
+        .then(ver => console.log(`Backend updated to ${JSON.stringify(ver)}`))
+        .catch((error) => {
+          console.error(`failed: ${error.message}`);
+        });
+    })
+    .then(() => {
+      transport.disconnect();
     });
 };
 
+
 module.exports = {
-  showVersion,
   updateVersion
 };