enigma-bbs/core/event_scheduler.js

/* jslint node: true */
'use strict';

//  ENiGMA½
const PluginModule = require('./plugin_module.js').PluginModule;
const Config = require('./config.js').get;
const Log = require('./logger.js').log;
const { Errors } = require('./enig_error.js');

const _ = require('lodash');
const later = require('@breejs/later');
const path = require('path');
const pty = require('node-pty');
const sane = require('sane');
const moment = require('moment');
const paths = require('path');
const fse = require('fs-extra');

exports.getModule = EventSchedulerModule;
exports.EventSchedulerModule = EventSchedulerModule; //  allow for loadAndStart

exports.moduleInfo = {
    name: 'Event Scheduler',
    desc: 'Support for scheduling arbritary events',
    author: 'NuSkooler',
};

const SCHEDULE_REGEXP = /(?:^|or )?(@watch:)([^\0]+)?$/;
const ACTION_REGEXP = /@(method|execute):([^\0]+)?$/;

class ScheduledEvent {
    constructor(events, name) {
        this.name = name;
        this.schedule = this.parseScheduleString(events[name].schedule);
        this.action = this.parseActionSpec(events[name].action);
        if (this.action) {
            this.action.args = events[name].args || [];
        }
    }

    get isValid() {
        if (
            !this.schedule ||
            (!this.schedule.sched && !this.schedule.watchFile) ||
            !this.action
        ) {
            return false;
        }

        if ('method' === this.action.type && !this.action.location) {
            return false;
        }

        return true;
    }

    parseScheduleString(schedStr) {
        if (!schedStr) {
            return false;
        }

        let schedule = {};

        const m = SCHEDULE_REGEXP.exec(schedStr);
        if (m) {
            schedStr = schedStr.substr(0, m.index).trim();

            if ('@watch:' === m[1]) {
                schedule.watchFile = m[2];
            }
        }

        if (schedStr.length > 0) {
            const sched = later.parse.text(schedStr);
            if (-1 === sched.error) {
                schedule.sched = sched;
            }
        }

        //  return undefined if we couldn't parse out anything useful
        if (!_.isEmpty(schedule)) {
            return schedule;
        }
    }

    parseActionSpec(actionSpec) {
        if (actionSpec) {
            if ('@' === actionSpec[0]) {
                const m = ACTION_REGEXP.exec(actionSpec);
                if (m) {
                    if (m[2].indexOf(':') > -1) {
                        const parts = m[2].split(':');
                        return {
                            type: m[1],
                            location: parts[0],
                            what: parts[1],
                        };
                    } else {
                        return {
                            type: m[1],
                            what: m[2],
                        };
                    }
                }
            } else {
                return {
                    type: 'execute',
                    what: actionSpec,
                };
            }
        }
    }

    executeAction(reason, cb) {
        Log.info(
            { eventName: this.name, action: this.action, reason: reason },
            'Executing scheduled event action...'
        );

        if ('method' === this.action.type) {
            const modulePath = path.join(__dirname, '../', this.action.location); //  enigma-bbs base + supplied location (path/file.js')
            try {
                const methodModule = require(modulePath);
                methodModule[this.action.what](this.action.args, err => {
                    if (err) {
                        Log.debug(
                            {
                                error: err.message,
                                eventName: this.name,
                                action: this.action,
                            },
                            'Error performing scheduled event action'
                        );
                    }

                    return cb(err);
                });
            } catch (e) {
                Log.warn(
                    { error: e.message, eventName: this.name, action: this.action },
                    'Failed to perform scheduled event action'
                );

                return cb(e);
            }
        } else if ('execute' === this.action.type) {
            const opts = {
                //  :TODO: cwd
                name: this.name,
                cols: 80,
                rows: 24,
                env: process.env,
            };

            let proc;
            try {
                proc = pty.spawn(this.action.what, this.action.args, opts);
            } catch (e) {
                Log.warn({
                    error: 'Failed to spawn @execute process',
                    reason: e.message,
                    eventName: this.name,
                    action: this.action,
                    what: this.action.what,
                    args: this.action.args,
                });
                return cb(e);
            }

            proc.once('exit', exitCode => {
                if (exitCode) {
                    Log.warn(
                        { eventName: this.name, action: this.action, exitCode: exitCode },
                        'Bad exit code while performing scheduled event action'
                    );
                }
                return cb(
                    exitCode
                        ? Errors.ExternalProcess(
                              `Bad exit code while performing scheduled event action: ${exitCode}`
                          )
                        : null
                );
            });
        }
    }
}

function EventSchedulerModule(options) {
    PluginModule.call(this, options);

    const config = Config();
    if (_.has(config, 'eventScheduler')) {
        this.moduleConfig = config.eventScheduler;
    }

    const self = this;
    this.runningActions = new Set();

    this.performAction = function (schedEvent, reason) {
        if (self.runningActions.has(schedEvent.name)) {
            return; //  already running
        }

        self.runningActions.add(schedEvent.name);

        schedEvent.executeAction(reason, () => {
            self.runningActions.delete(schedEvent.name);
        });
    };
}

//  convienence static method for direct load + start
EventSchedulerModule.loadAndStart = function (cb) {
    const loadModuleEx = require('./module_util.js').loadModuleEx;

    const loadOpts = {
        name: path.basename(__filename, '.js'),
        path: __dirname,
    };

    loadModuleEx(loadOpts, (err, mod) => {
        if (err) {
            return cb(err);
        }

        const modInst = new mod.getModule();
        modInst.startup(err => {
            return cb(err, modInst);
        });
    });
};

EventSchedulerModule.prototype.startup = function (cb) {
    this.eventTimers = [];
    const self = this;

    if (this.moduleConfig && _.has(this.moduleConfig, 'events')) {
        const events = Object.keys(this.moduleConfig.events).map(name => {
            return new ScheduledEvent(this.moduleConfig.events, name);
        });

        events.forEach(schedEvent => {
            if (!schedEvent.isValid) {
                Log.warn({ eventName: schedEvent.name }, 'Invalid scheduled event entry');
                return;
            }

            Log.debug(
                {
                    eventName: schedEvent.name,
                    schedule: this.moduleConfig.events[schedEvent.name].schedule,
                    action: schedEvent.action,
                    next: schedEvent.schedule.sched
                        ? moment(
                              later.schedule(schedEvent.schedule.sched).next(1)
                          ).format('ddd, MMM Do, YYYY @ h:m:ss a')
                        : 'N/A',
                },
                'Scheduled event loaded'
            );

            if (schedEvent.schedule.sched) {
                this.eventTimers.push(
                    later.setInterval(() => {
                        self.performAction(schedEvent, 'Schedule');
                    }, schedEvent.schedule.sched)
                );
            }

            if (schedEvent.schedule.watchFile) {
                const watcher = sane(paths.dirname(schedEvent.schedule.watchFile), {
                    glob: `**/${paths.basename(schedEvent.schedule.watchFile)}`,
                });

                //  :TODO: should track watched files & stop watching @ shutdown?

                ['change', 'add', 'delete'].forEach(event => {
                    watcher.on(event, (fileName, fileRoot) => {
                        const eventPath = paths.join(fileRoot, fileName);
                        if (schedEvent.schedule.watchFile === eventPath) {
                            self.performAction(schedEvent, `Watch file: ${eventPath}`);
                        }
                    });
                });

                fse.exists(schedEvent.schedule.watchFile, exists => {
                    if (exists) {
                        self.performAction(
                            schedEvent,
                            `Watch file: ${schedEvent.schedule.watchFile}`
                        );
                    }
                });
            }
        });
    }

    cb(null);
};

EventSchedulerModule.prototype.shutdown = function (cb) {
    if (this.eventTimers) {
        this.eventTimers.forEach(et => et.clear());
    }

    cb(null);
};