Add support for tasks

Author: nt1m

app.js

@@ -18,7 +18,8 @@ const users = require('./lib/users');
 boot.executeInParallel([
   boot.forwardHttp,
   boot.ensureHttpsCertificates,
-  boot.ensureDockerTlsCertificates
+  boot.ensureDockerTlsCertificates,
+  boot.loadTasks,
 ], () => {
   // You can customize these values in './db.json'.
   const hostname = db.get('hostname', 'localhost');

join.js

@@ -27,7 +27,8 @@ boot.executeInParallel([
   boot.forwardHttp,
   boot.ensureHttpsCertificates,
   boot.ensureDockerTlsCertificates,
-  boot.verifyJanitorOAuth2Access
+  boot.verifyJanitorOAuth2Access,
+  boot.loadTasks,
 ], () => {
   boot.registerDockerClient(() => {
     log('[ok] joined cluster as [hostname = ' + hostname + ']');

lib/boot.js

@@ -8,6 +8,7 @@ const certificates = require('./certificates');
 const db = require('./db');
 const log = require('./log');
 const oauth2 = require('./oauth2');
+const tasks = require('./tasks');
 
 const hostname = db.get('hostname', 'localhost');
 
@@ -345,3 +346,10 @@ exports.registerDockerClient = function (next) {
     next();
   });
 };
+
+exports.loadTasks = function (callback) {
+  setInterval(() => {
+    tasks.check();
+  }, 60000);
+  callback();
+};

lib/tasks.js

@@ -0,0 +1,53 @@
+const db = require('./db');
+const log = require('./log');
+let taskCounter = 0; // Used to generate unique task IDs
+
+exports.tasks = db.get('tasks');
+exports.taskTypes = new Map();
+
+exports.check = function () {
+  const now = Date.now();
+  for (const task of Object.values(exports.tasks)) {
+    if (now > Number(task.date)) {
+      exports.execute(task);
+    }
+  }
+};
+
+exports.addType = function (type, task) {
+  if (exports.taskTypes.has(type)) {
+    throw new Error('[fail] task', task, 'already exists');
+  }
+
+  exports.taskTypes.set(type, task);
+};
+
+exports.add = function (date, type, data) {
+  taskCounter++;
+
+  const msSince1970 = date.getTime();
+  const taskId = `${type}-${msSince1970}-${taskCounter}`;
+
+  const task = {
+    id: taskId,
+    date: msSince1970,
+    type,
+    data,
+  };
+
+  exports.tasks[taskId] = task;
+  db.save();
+  return task;
+};
+
+exports.remove = function (id) {
+  delete exports.tasks[id];
+  db.save();
+};
+
+exports.execute = function ({ type, data, id }) {
+  const task = exports.taskTypes.get(type);
+  log('[ok] Task', id, 'executed', data);
+  task(data);
+  exports.remove(id);
+};