cloudron-box/src/taskworker.js

#!/usr/bin/env node

'use strict';

const apptask = require('./apptask.js'),
    async = require('async'),
    backups = require('./backups.js'),
    cloudron = require('./cloudron.js'),
    database = require('./database.js'),
    domains = require('./domains.js'),
    externalLdap = require('./externalldap.js'),
    fs = require('fs'),
    mail = require('./mail.js'),
    reverseProxy = require('./reverseproxy.js'),
    safe = require('safetydance'),
    settings = require('./settings.js'),
    tasks = require('./tasks.js'),
    updater = require('./updater.js');

const TASKS = { // indexed by task type
    app: apptask.run,
    backup: backups.backupBoxAndApps,
    update: updater.update,
    checkCerts: reverseProxy.checkCerts,
    setupDnsAndCert: cloudron.setupDnsAndCert,
    cleanBackups: backups.cleanup,
    syncExternalLdap: externalLdap.sync,
    changeMailLocation: mail.changeLocation,
    syncDnsRecords: domains.syncDnsRecords,

    _identity: (arg, progressCallback, callback) => callback(null, arg),
    _error: (arg, progressCallback, callback) => callback(new Error(`Failed for arg: ${arg}`)),
    _crash: (arg) => { throw new Error(`Crashing for arg: ${arg}`); }, // the test looks for this debug string in the log file
    _sleep: (arg) => setTimeout(process.exit, arg)
};

if (process.argv.length !== 4) {
    console.error('Pass the taskid and logfile as argument');
    process.exit(1);
}

const taskId = process.argv[2];
const logFile = process.argv[3];

function setupLogging(callback) {
    const logfileStream = fs.createWriteStream(logFile, { flags:'a' });
    process.stdout.write = process.stderr.write = logfileStream.write.bind(logfileStream);

    callback();
}

// Main process starts here
const startTime = new Date();

async.series([
    setupLogging,
    database.initialize,
    settings.initCache
], async function (error) {
    if (error) {
        console.error(error);
        return process.exit(50);
    }

    const debug = require('debug')('box:taskworker'); // require this here so that logging handler is already setup

    process.on('SIGTERM', () => process.exit(0)); // sent as timeout notification

    // ensure we log task crashes with the task logs
    process.on('uncaughtException', function (e) { debug(e); process.exit(1); });

    debug(`Starting task ${taskId}. Logs are at ${logFile}`);

    const [getError, task] = await safe(tasks.get(taskId));
    if (getError || !task) {
        debug(getError ? `Error getting task: ${getError.message}` : 'Task not found');
        return process.exit(50);
    }

    const [updateError] = await safe(tasks.update(taskId, { percent: 2, error: null }));
    if (updateError) {
        debug(updateError);
        return process.exit(50);
    }

    const progressCallback = async function (progress, callback) {
        await safe(tasks.update(taskId, progress));
        if (callback) callback();
    };

    const resultCallback = async (error, result) => {
        // Error object has properties with enumerable: false (https://mattcbaker.com/posts/stringify-javascript-error/)
        const progress = {
            result: result || null,
            error: error ? JSON.parse(JSON.stringify(error, Object.getOwnPropertyNames(error))) : null
        };

        debug(`Task took ${(new Date() - startTime)/1000} seconds`);

        await safe(tasks.setCompleted(taskId, progress));
        process.exit(error ? 50 : 0);
    };

    try {
        TASKS[task.type].apply(null, task.args.concat(progressCallback).concat(resultCallback));
    } catch (error) {
        debug('Uncaught exception in task', error);
        process.exit(1); // do not call setCompleted() intentionally. the task code must be resilient enough to handle it
    }
});