enigma-bbs/core/config_loader.js

//  deps
const paths = require('path');
const async = require('async');
const moment = require('moment');

const _ = require('lodash');
const mapValuesDeep = require('deepdash/getMapValuesDeep')(_);

module.exports = class ConfigLoader {
    constructor(
        {
            hotReload = true,
            defaultConfig = {},
            defaultsCustomizer = null,
            onReload = null,
            keepWsc = false,
        } = {
            hotReload: true,
            defaultConfig: {},
            defaultsCustomizer: null,
            onReload: null,
            keepWsc: false,
        }
    ) {
        this.current = {};

        this.hotReload = hotReload;
        this.defaultConfig = defaultConfig;
        this.defaultsCustomizer = defaultsCustomizer;
        this.onReload = onReload;
        this.keepWsc = keepWsc;
    }

    init(baseConfigPath, cb) {
        this.baseConfigPath = baseConfigPath;
        return this._reload(baseConfigPath, cb);
    }

    get() {
        return this.current;
    }

    _reload(baseConfigPath, cb) {
        let defaultConfig;
        if (_.isFunction(this.defaultConfig)) {
            defaultConfig = this.defaultConfig();
        } else if (_.isObject(this.defaultConfig)) {
            defaultConfig = this.defaultConfig;
        } else {
            defaultConfig = {};
        }

        //
        //  1 - Fetch base configuration from |baseConfigPath|
        //  2 - Merge with |defaultConfig|
        //  3 - Resolve any includes
        //  4 - Resolve @reference and @environment
        //  5 - Perform any validation
        //
        async.waterfall(
            [
                callback => {
                    return this._loadConfigFile(baseConfigPath, callback);
                },
                (config, callback) => {
                    if (_.isFunction(this.defaultsCustomizer)) {
                        const stack = [];
                        const mergedConfig = _.mergeWith(
                            defaultConfig,
                            config,
                            (defaultVal, configVal, key, target, source) => {
                                let path;
                                while (true) {
                                    //  eslint-disable-line no-constant-condition
                                    if (!stack.length) {
                                        stack.push({ source, path: [] });
                                    }

                                    const prev = stack[stack.length - 1];

                                    if (source === prev.source) {
                                        path = prev.path.concat(key);
                                        stack.push({ source: configVal, path });
                                        break;
                                    }

                                    stack.pop();
                                }

                                path = path.join('.');
                                return this.defaultsCustomizer(
                                    defaultVal,
                                    configVal,
                                    key,
                                    path
                                );
                            }
                        );

                        return callback(null, mergedConfig);
                    }

                    return callback(null, _.merge(defaultConfig, config));
                },
                (config, callback) => {
                    const configRoot = paths.dirname(baseConfigPath);
                    return this._resolveIncludes(configRoot, config, callback);
                },
                (config, callback) => {
                    config = this._resolveAtSpecs(config);
                    return callback(null, config);
                },
            ],
            (err, config) => {
                if (!err) {
                    this.current = config;
                }
                return cb(err);
            }
        );
    }

    _convertTo(value, type) {
        switch (type) {
            case 'bool':
            case 'boolean':
                value = '1' === value || 'true' === value.toLowerCase();
                break;

            case 'number':
                {
                    const num = parseInt(value);
                    if (!isNaN(num)) {
                        value = num;
                    }
                }
                break;

            case 'object':
                try {
                    value = JSON.parse(value);
                } catch (e) {
                    //  ignored
                }
                break;

            case 'timestamp':
                {
                    const m = moment(value);
                    if (m.isValid()) {
                        value = m;
                    }
                }
                break;
        }

        return value;
    }

    _resolveEnvironmentVariable(spec) {
        const [, varName, type, array] = spec.split(':');
        if (!varName) {
            return;
        }

        let value = process.env[varName];
        if (!value) {
            //  console is about as good as we can do here
            return console.info(
                `WARNING: environment variable "${varName}" from spec "${spec}" not found!`
            );
        }

        if ('array' === array) {
            value = value.split(',').map(v => this._convertTo(v, type));
        } else {
            value = this._convertTo(value, type);
        }

        return value;
    }

    _loadConfigFile(filePath, cb) {
        const ConfigCache = require('./config_cache');

        const options = {
            filePath,
            hotReload: this.hotReload,
            keepWsc: this.keepWsc,
            callback: this._configFileChanged.bind(this),
        };

        ConfigCache.getConfigWithOptions(options, (err, config) => {
            if (err) {
                err.configPath = options.filePath;
            }
            return cb(err, config);
        });
    }

    _configFileChanged({ fileName, fileRoot }) {
        const reCachedPath = paths.join(fileRoot, fileName);
        if (this.configPaths.includes(reCachedPath)) {
            this._reload(this.baseConfigPath, err => {
                if (_.isFunction(this.onReload)) {
                    this.onReload(err, reCachedPath);
                }
            });
        }
    }

    _resolveIncludes(configRoot, config, cb) {
        if (!Array.isArray(config.includes)) {
            this.configPaths = [this.baseConfigPath];
            return cb(null, config);
        }

        //  If a included file is changed, we need to re-cache, so this
        //  must be tracked...
        const includePaths = config.includes.map(inc => paths.join(configRoot, inc));
        async.eachSeries(
            includePaths,
            (includePath, nextIncludePath) => {
                this._loadConfigFile(includePath, (err, includedConfig) => {
                    if (err) {
                        return nextIncludePath(err);
                    }

                    _.defaultsDeep(config, includedConfig);
                    return nextIncludePath(null);
                });
            },
            err => {
                this.configPaths = [this.baseConfigPath, ...includePaths];
                return cb(err, config);
            }
        );
    }

    _resolveAtSpecs(config) {
        return mapValuesDeep(config, value => {
            if (_.isString(value) && '@' === value.charAt(0)) {
                if (value.startsWith('@reference:')) {
                    const refPath = value.slice(11);
                    value = _.get(config, refPath, value);
                } else if (value.startsWith('@environment:')) {
                    value = this._resolveEnvironmentVariable(value) || value;
                }
            }

            return value;
        });
    }
};