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('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);
};