Merge pull request #45 from csivitu/dev

feat: CLI to spawn workers

bin/code-executor

@@ -0,0 +1,3 @@
+#!/usr/bin/env node
+
+require('../dist/src/index').cli(process.argv);

package.json

@@ -1,6 +1,6 @@
 {
   "name": "code-executor",
-  "version": "0.2.4",
+  "version": "0.2.5",
   "description": "A CLI/library to execute code against test cases in various languages and obtain relevant results.",
   "main": "dist/src/index.js",
   "keywords": [
@@ -12,6 +12,9 @@
     "parallel-execution",
     "code-runner"
   ],
+  "bin": {
+    "code-executor": "bin/code-executor"
+  },
   "scripts": {
     "build": "npm run clean && tsc && npm run build:dockerfiles",
     "build:dockerfiles": "shx cp -r src/langs dist/src/langs",
@@ -33,6 +36,7 @@
   "homepage": "https://github.com/csivitu/code-executor#readme",
   "devDependencies": {
     "@types/bull": "^3.14.2",
+    "@types/commander": "^2.12.2",
     "@types/dockerode": "^2.5.34",
     "@types/node": "^14.6.3",
     "@types/randomstring": "^1.1.6",
@@ -56,6 +60,7 @@
   },
   "dependencies": {
     "bull": "^3.18.0",
+    "commander": "^6.2.0",
     "del": "^6.0.0",
     "dockerode": "^3.2.1",
     "randomstring": "^1.1.5",

src/WorkerCLI.ts

@@ -0,0 +1,61 @@
+import commander, { Command } from 'commander';
+import logger from './utils/logger';
+import { WorkerCLIOptions } from './models';
+import Worker from './Worker';
+
+class WorkerCLI {
+    readonly program: commander.Command;
+
+    readonly args: string[];
+
+    constructor(args: string[]) {
+        this.program = new Command();
+        this.args = args;
+        this.init();
+    }
+
+    init(): void {
+        this.program
+            .command('spawn-worker')
+            .alias('sw')
+            .description('spawn worker process')
+            .action(() => {
+                const options = this.parseOptions();
+                WorkerCLI.spawnWorker(options);
+            });
+
+        this.program
+            .option('-r, --redis <redis>', 'URL for the redis instance')
+            .option('-q, --queue <queue>', 'name of the redis queue')
+            .option('-l, --langs <langs...>', 'list of languages to build');
+    }
+
+    parseOptions(): WorkerCLIOptions {
+        const opts = this.program.opts();
+
+        const options: WorkerCLIOptions = {
+            redis: opts.redis || 'redis://127.0.0.1:6379',
+            queue: opts.queue || 'myExecutor',
+            langs: opts.langs || [],
+        };
+
+        return options;
+    }
+
+    static async spawnWorker(options: WorkerCLIOptions): Promise<void> {
+        logger.info('Spawning Workers...');
+
+        const worker = new Worker(options.queue, options.redis);
+        await worker.build(options.langs.length ? options.langs : undefined);
+        worker.start();
+    }
+
+    async start(): Promise<void> {
+        await this.program.parseAsync(this.args);
+    }
+}
+
+export default async function cli(args: string[]): Promise<void> {
+    const workerCLI = new WorkerCLI(args);
+    workerCLI.start();
+}

src/index.ts

@@ -1,3 +1,5 @@
 export { default as CodeExecutor, languages } from './CodeExecutor';
 
 export { default as Worker } from './Worker';
+
+export { default as cli } from './WorkerCLI';

src/models/index.ts

@@ -35,3 +35,9 @@ export interface WorkerOptions {
     memory?: number;
     CPUs?: number;
 }
+
+export interface WorkerCLIOptions {
+    queue: string;
+    redis: string;
+    langs: string[];
+}