protobuf.js/cli/util.js

"use strict";
var fs            = require("fs"),
    path          = require("path"),
    child_process = require("child_process"),
    Module        = require("module");

var protobuf = require("..");

function basenameCompare(a, b) {
    var aa = String(a).replace(/\.\w+$/, "").split(/(-?\d*\.?\d+)/g),
        bb = String(b).replace(/\.\w+$/, "").split(/(-?\d*\.?\d+)/g);
    for (var i = 0, k = Math.min(aa.length, bb.length); i < k; i++) {
        var x = parseFloat(aa[i]) || aa[i].toLowerCase(),
            y = parseFloat(bb[i]) || bb[i].toLowerCase();
        if (x < y)
            return -1;
        if (x > y)
            return 1;
    }
    return a.length < b.length ? -1 : 0;
};

exports.requireAll = function requireAll(dirname) {
    dirname   = path.join(__dirname, dirname);
    var files = fs.readdirSync(dirname).sort(basenameCompare),
        all = {};
    files.forEach(function(file) {
        var basename = path.basename(file, ".js"),
            extname  = path.extname(file);
        if (extname === ".js")
            all[basename] = require(path.join(dirname, file));
    });
    return all;
};

exports.inspect = function inspect(object, indent) {
    if (!object)
        return "";
    var chalk = require("chalk");
    var sb = [];
    if (!indent)
        indent = "";
    var ind = indent ? indent.substring(0, indent.length - 2) + "└ " : "";
    sb.push(
        ind + chalk.bold(object.toString()) + (object.visible ? " (visible)" : ""),
        indent + chalk.gray("parent: ") + object.parent
    );
    if (object instanceof protobuf.Field) {
        if (object.extend !== undefined)
            sb.push(indent + chalk.gray("extend: ") + object.extend);
        if (object.partOf)
            sb.push(indent + chalk.gray("oneof : ") + object.oneof);
    }
    sb.push("");
    if (object.fieldsArray)
        object.fieldsArray.forEach(function(field) {
            sb.push(inspect(field, indent + "  "));
        });
    if (object.oneofsArray)
        object.oneofsArray.forEach(function(oneof) {
            sb.push(inspect(oneof, indent + "  "));
        });
    if (object.methodsArray)
        object.methodsArray.forEach(function(service) {
            sb.push(inspect(service, indent + "  "));
        });
    if (object.nestedArray)
        object.nestedArray.forEach(function(nested) {
            sb.push(inspect(nested, indent + "  "));
        });
    return sb.join("\n");
};

exports.setup = function() {
    // this one is important. without it, this folder won't be searched anymore.
    try { fs.mkdirSync(path.join(__dirname, "node_modules")); } catch (e) {}

    // find out which modules are missing
    var pkg = require(path.join(__dirname, "..", "package.json"));
    var install = [];
    var semver;
    pkg.cliDependencies.forEach(function(name) {
        var version = pkg.dependencies[name] || pkg.devDependencies[name];
        try {
            var mPath = require.resolve(name + "/package.json"); // jsdoc has no main file
            var mPkg  = JSON.parse(fs.readFileSync(mPath));
            if (semver && !semver.satisfies(mPkg.version, version))
                throw Error(mPkg.version + " is outdated");
        } catch (e) {
            process.stderr.write("installing " + name + "@" + version + " (" + e.message + ")\n");
            install.push(version ? name + "@" + version : name);
        }
        if (name === "semver")
            semver = require("semver");
    });
    if (!install.length) {
        try { fs.rmdirSync(path.join(__dirname, "node_modules")); } catch (e) {}
        return;
    }

    // if any are missing, install them. this relies on an empty package.json in cli/.
    child_process.execSync("npm --silent install " + install.join(" "), {
        cwd: __dirname,
        stdio: "ignore"
    });
};

exports.wrap = function(OUTPUT, options) {
    var name = options.wrap || "default";
    var wrap;
    try {
        // try built-in wrappers first
        wrap = fs.readFileSync(path.join(__dirname, "wrappers", name + ".js")).toString("utf8");
    } catch (e) {
        // otherwise fetch the custom one
        wrap = fs.readFileSync(path.resolve(process.cwd(), name)).toString("utf8");
    }
    wrap = wrap.replace(/%ROOT%/g, JSON.stringify(options.root || "default"));
    wrap = wrap.replace(/%DEPENDENCY%/g, JSON.stringify(options.dependency || "protobufjs"));
    wrap = wrap.replace(/( *)%OUTPUT%/, function($0, $1) {
        return $1.length ? OUTPUT.replace(/^/mg, $1) : OUTPUT;
    });
    if (options.lint !== "")
        wrap = "/*" + options.lint + "*/\n" + wrap;
    return wrap.replace(/\r?\n/, "\n");
};

exports.pad = function(str, len, l) {
    while (str.length < len)
        str = l ? str + " " : " " + str;
    return str;
};

exports.reserved = function(name) {
    return /^(?:do|if|in|for|let|new|try|var|case|else|enum|eval|false|null|this|true|void|with|break|catch|class|const|super|throw|while|yield|delete|export|import|public|return|static|switch|typeof|default|extends|finally|package|private|continue|debugger|function|arguments|interface|protected|implements|instanceof)$/.test(name);
};