MusareFork/backend/core.js

const async = require("async");

class DeferredPromise {
    constructor() {
        this.promise = new Promise((resolve, reject) => {
            this.reject = reject;
            this.resolve = resolve;
        });
    }
}

class MovingAverageCalculator {
    constructor() {
        this.count = 0;
        this._mean = 0;
    }

    update(newValue) {
        this.count++;
        const differential = (newValue - this._mean) / this.count;
        this._mean += differential;
    }

    get mean() {
        this.validate();
        return this._mean;
    }

    validate() {
        if (this.count === 0) throw new Error("Mean is undefined");
    }
}

class CoreClass {
    constructor(name) {
        this.name = name;
        this.status = "UNINITIALIZED";
        // this.log("Core constructor");
        this.jobQueue = async.priorityQueue(
            (job, callback) => this._runJob(job, callback),
            10 // How many jobs can run concurrently
        );
        this.jobQueue.pause();
        this.runningJobs = [];
        this.priorities = {};
        this.stage = 0;
        this.jobStatistics = {};

        this.registerJobs();
    }

    setStatus(status) {
        this.status = status;
        this.log("INFO", `Status changed to: ${status}`);
        if (this.status === "READY") this.jobQueue.resume();
        else if (this.status === "FAIL" || this.status === "LOCKDOWN")
            this.jobQueue.pause();
    }

    getStatus() {
        return this.status;
    }

    setStage(stage) {
        this.stage = stage;
    }

    getStage() {
        return this.stage;
    }

    _initialize() {
        this.setStatus("INITIALIZING");
        this.initialize()
            .then(() => {
                this.setStatus("READY");
                this.moduleManager.onInitialize(this);
            })
            .catch((err) => {
                console.error(err);
                this.setStatus("FAILED");
                this.moduleManager.onFail(this);
            });
    }

    log() {
        let _arguments = Array.from(arguments);
        const type = _arguments[0];
        _arguments.splice(0, 1);
        const start = `|${this.name.toUpperCase()}|`;
        const numberOfTabsNeeded = 4 - Math.ceil(start.length / 8);
        _arguments.unshift(`${start}${Array(numberOfTabsNeeded).join("\t")}`);

        if (type === "INFO") {
            _arguments[0] = _arguments[0] + "\x1b[36m";
            _arguments.push("\x1b[0m");
            console.log.apply(null, _arguments);
        } else if (type === "ERROR") {
            _arguments[0] = _arguments[0] + "\x1b[31m";
            _arguments.push("\x1b[0m");
            console.error.apply(null, _arguments);
        }
    }

    registerJobs() {
        let props = [];
        let obj = this;
        do {
            props = props.concat(Object.getOwnPropertyNames(obj));
        } while ((obj = Object.getPrototypeOf(obj)));

        const jobNames = props
            .sort()
            .filter(
                (prop) =>
                    typeof this[prop] == "function" &&
                    prop === prop.toUpperCase()
            );

        jobNames.forEach((jobName) => {
            this.jobStatistics[jobName] = {
                successful: 0,
                failed: 0,
                total: 0,
                averageTiming: new MovingAverageCalculator(),
            };
        });
    }

    runJob(name, payload, options = {}) {
        let deferredPromise = new DeferredPromise();
        const job = { name, payload, onFinish: deferredPromise };

        if (options.bypassQueue) {
            this._runJob(job, () => {});
        } else {
            const priority = this.priorities[name] ? this.priorities[name] : 10;
            this.jobQueue.push(job, priority);
        }

        return deferredPromise.promise;
    }

    setModuleManager(moduleManager) {
        this.moduleManager = moduleManager;
    }

    _runJob(job, cb) {
        this.log("INFO", `Running job ${job.name}`);
        const startTime = Date.now();
        this.runningJobs.push(job);
        const newThis = Object.assign(
            Object.create(Object.getPrototypeOf(this)),
            this
        );
        newThis.runJob = (...args) => {
            if (args.length === 2) args.push({});
            args[2].bypassQueue = true;
            return this.runJob.apply(this, args);
        };
        this[job.name]
            .apply(newThis, [job.payload])
            .then((response) => {
                this.log("INFO", `Ran job ${job.name} successfully`);
                this.jobStatistics[job.name].successful++;
                job.onFinish.resolve(response);
            })
            .catch((error) => {
                this.log("INFO", `Running job ${job.name} failed`);
                this.jobStatistics[job.name].failed++;
                job.onFinish.reject(error);
            })
            .finally(() => {
                const endTime = Date.now();
                const executionTime = endTime - startTime;
                this.jobStatistics[job.name].total++;
                this.jobStatistics[job.name].averageTiming.update(
                    executionTime
                );
                this.runningJobs.splice(this.runningJobs.indexOf(job), 1);
                cb();
            });
    }
}

module.exports = CoreClass;