Fix #2

Author: joseconstela

.gitignore

@@ -0,0 +1 @@
+node_modules

README.md

@@ -16,10 +16,11 @@ $ npm i -g @tideflowio/tideflow-agent
     Usage: index [options]
 
     Options:
-      --noupdate           Opt-out of update version check
-      -v, --version        output the version number
-      -t, --token [token]  Authentication token
-      -h, --help           output usage information
+      -v, --version                          output the version number
+      -c, --concurrency [concurrency]        Max number of jobs the agent should process concurrently
+      -t, --token [token]                    Authentication token
+      --noupdate                             Opt-out of update version check
+      -h, --help                             output usage information
 
     Examples:
       $ tideflow-agent --help

agent.js

@@ -3,35 +3,42 @@ const io = require('socket.io-client')
 const colors = require('colors')
 const pretty = require('./helpers/pretty')
 const runner = require('./runner')
+const os = require('os')
 
 /**
  * @param {Object} cli's parameters parsed via commander's npm package
  */
 module.exports.exec = (program) => {
 
+  const cpusCounts = os.cpus().length
+  const programConcurrency = program.concurrency || cpusCounts
+  let concurrency = programConcurrency >= cpusCounts ? cpusCounts : programConcurrency
+
+  const q = require('queue')({
+    concurrency: concurrency
+  })
+
   let agent = {}
 
   // Shows welcome messages
   console.log(pretty.logo())
   console.log(` || Tideflow.io - agent ${pjson.version}`.blue)
+  console.log(` || Using ${concurrency} as concurrency`.yellow)
 
   const URL = process.env.TF_AGENT_URL || 'http://localhost:1337'
 
   const socket = io(`${URL}?token=${program.token}`)
 
-  // socket.on('connect', function () {})
-
-  // socket.on('reconnect', function () {})
-
   // Execute command
   socket.on('tf.command', function (req) {
     if (!agent.authenticated) return
-    runner.cmd(socket, 'tf.command', req)
+    q.push(runner.cmd(socket, 'tf.command', req))
   })
 
   // Execute code
   socket.on('tf.code', function () {
-    if (!agent.token) return
+    if (!agent.authenticated) return
+    q.push(runner.code(socket, 'tf.code', req))
   })
 
   // Authorize agent

index.js

@@ -1,10 +1,17 @@
 #!/usr/bin/env node
 'use strict'
 const program = require('commander')
+const os = require('os')
 const pkg = require('./package.json')
 
 program
   .version(pkg.version, '-v, --version')
+  
+  .option('-c, --concurrency [concurrency]', 'Max number of jobs the agent should process concurrently', (v) => {
+    const concurrency = parseInt(v)
+    if (!concurrency) { throw new Error('Concurrency parameter must be a valid integer') }
+    return parseInt(concurrency)
+  })
   .option('-t, --token [token]', 'Authentication token', (v) => {
     return v || process.env.TIDEFLOWIO_AGENT_TOKEN
   })
@@ -16,6 +23,8 @@ program
 program.on('--help', function(){
   console.log('')
   console.log('Examples:')
+  console.log('  $ tideflow-agent -t agent-auth-token')
+  console.log('  $ tideflow-agent -c 16 -t agent-auth-token')
   console.log('  $ tideflow-agent --help')
   console.log('  $ tideflow-agent -h')
 })

package-lock.json

@@ -34,6 +34,8 @@
   "dependencies": {
     "colors": "^1.3.3",
     "commander": "^2.19.0",
+    "cross-spawn": "^6.0.5",
+    "queue": "^6.0.0",
     "socket.io-client": "^2.2.0",
     "update-notifier": "^2.5.0"
   },

package.json

@@ -1,4 +1,4 @@
-const { spawn } = require('child_process')
+const spawn = require('cross-spawn')
 const report = require('./helpers/report')
 
 /**
@@ -49,4 +49,19 @@ const cmd = (socket, topic, req) => {
   })
 }
 
-module.exports.cmd = cmd
+module.exports.cmd = cmd
+
+/**
+ * Handles the execution of nodejs code sent from the platform
+ * 
+ * @param {Object} socket Socket's io socket that requested the execution
+ * @param {String} topic Original message's topic
+ * @param {Object} req Original request that came fromthe platform
+ */
+const code = (socket, topic, req) => {
+  return new Promise((resolve, reject) => {
+    return reject('Not supported yet')
+  })
+}
+
+module.exports.code = code