feat: add typescript support to plugin

README.md

@@ -10,8 +10,14 @@
 
 > Typescript workers plugin for Bree!
 
+
 ## Table of Contents
 
+* [Install](#install)
+* [Usage](#usage)
+* [Contributors](#contributors)
+* [License](#license)
+
 
 ## Install
 
@@ -42,9 +48,15 @@ console.log(@breejs/tsWorker.renderName());
 
 ## Contributors
 
+| Name              | Website                           |
+| ----------------- | --------------------------------- |
+| **Taylor Schley** | <https://github.com/shadowgate15> |
+
 
 ## License
 
+[MIT](LICENSE) © [Taylor Schley](https://github.com/shadowgate15)
+
 
 ##
 

package.json

@@ -7,16 +7,28 @@
     "url": "https://github.com/breejs/ts-worker/issues",
     "email": "taylorschley@me.com"
   },
+  "ava": {
+    "files": [
+      "test/*.js",
+      "test/**/*.js",
+      "!test/jobs"
+    ],
+    "verbose": true
+  },
   "contributors": [
     "Taylor Schley <taylorschley@me.com> (https://github.com/shadowgate15)"
   ],
-  "dependencies": {},
+  "dependencies": {
+    "ts-node": "^10.4.0"
+  },
   "devDependencies": {
     "@commitlint/cli": "latest",
     "@commitlint/config-conventional": "latest",
     "ava": "latest",
+    "bree": "^6.4.0",
     "codecov": "latest",
     "cross-env": "latest",
+    "delay": "^5.0.0",
     "eslint": "latest",
     "eslint-config-xo-lass": "latest",
     "fixpack": "latest",
@@ -28,7 +40,7 @@
     "xo": "latest"
   },
   "engines": {
-    "node": ">= 10"
+    "node": ">= 12.11.0"
   },
   "homepage": "https://github.com/breejs/ts-worker",
   "keywords": [

src/index.js

@@ -1,5 +1,30 @@
+const path = require('path');
+const { Worker } = require('worker_threads');
+
 function plugin(opts, Bree) {
-
+  opts = opts || {};
+
+  const oldInit = Bree.prototype.init;
+
+  // define accepted extensions
+  Bree.prototype.init = function () {
+    if (!this.config.acceptedExtensions.includes('.ts'))
+      this.config.acceptedExtensions.push('.ts');
+
+    oldInit.bind(this)();
+  };
+
+  const oldCreateWorker = Bree.prototype.createWorker;
+
+  Bree.prototype.createWorker = function (filename, options) {
+    if (filename.endsWith('.ts')) {
+      options.workerData.path = filename;
+
+      return new Worker(path.resolve(__dirname, 'worker.js'), options);
+    }
+
+    return oldCreateWorker(filename, options);
+  };
 }
 
 module.exports = plugin;

src/worker.js

@@ -0,0 +1,5 @@
+const { workerData } = require('worker_threads');
+
+require('ts-node').register();
+
+require(workerData.path);

test/jobs/short.js

@@ -0,0 +1,3 @@
+setTimeout(() => {
+  console.log('hello');
+}, 100);

test/jobs/short.ts

@@ -0,0 +1,3 @@
+setTimeout(() => {
+  console.log('hello');
+}, 100);

test/test.js

@@ -1,27 +1,94 @@
+const path = require('path');
 const test = require('ava');
+const delay = require('delay');
+const Bree = require('../../bree/src');
 
-const Script = require('..');
+const plugin = require('../src');
 
-test.beforeEach((t) => {
-  const script = new Script({});
-  Object.assign(t.context, { script });
-});
+Bree.extend(plugin);
+
+const root = path.join(__dirname, 'jobs');
+const baseConfig = {
+  root,
+  timeout: 0,
+  interval: 0,
+  hasSeconds: false,
+  defaultExtension: 'js'
+};
 
-test('returns itself', (t) => {
-  t.true(t.context.script instanceof Script);
+test('will validate job when given ts file string', (t) => {
+  t.notThrows(() => {
+    // eslint-disable-next-line no-unused-vars
+    const bree = new Bree({
+      jobs: ['short.ts'],
+      ...baseConfig
+    });
+  });
 });
 
-test('sets a config object', (t) => {
-  const script = new Script(false);
-  t.true(script instanceof Script);
+test('will validate job when ".ts" is defined in config', (t) => {
+  t.notThrows(() => {
+    // eslint-disable-next-line no-unused-vars
+    const bree = new Bree({
+      ...baseConfig,
+      jobs: ['short.ts'],
+      acceptedExtensions: ['.ts', '.js', '.mjs']
+    });
+  });
 });
 
-test('renders name', (t) => {
-  const { script } = t.context;
-  t.is(script.renderName(), 'script');
+test('will run job defined in ts', async (t) => {
+  t.plan(2);
+
+  const logger = {
+    info: () => {}
+  };
+
+  const bree = new Bree({
+    jobs: ['short.ts'],
+    ...baseConfig,
+    logger
+  });
+
+  bree.start();
+
+  bree.on('worker created', (name) => {
+    t.true(typeof bree.workers[name] === 'object');
+  });
+
+  bree.on('worker deleted', (name) => {
+    t.true(typeof bree.workers[name] === 'undefined');
+  });
+
+  await delay(100);
+
+  await bree.stop();
 });
 
-test('sets a default name', (t) => {
-  const { script } = t.context;
-  t.is(script._name, 'script');
+test('will run job defined in js', async (t) => {
+  t.plan(2);
+
+  const logger = {
+    info: () => {}
+  };
+
+  const bree = new Bree({
+    jobs: ['short.js'],
+    ...baseConfig,
+    logger
+  });
+
+  bree.start();
+
+  bree.on('worker created', (name) => {
+    t.true(typeof bree.workers[name] === 'object');
+  });
+
+  bree.on('worker deleted', (name) => {
+    t.true(typeof bree.workers[name] === 'undefined');
+  });
+
+  await delay(100);
+
+  await bree.stop();
 });