Newer
Older
vmk-demo-bot / admin / node_modules / browser-sync-ui / lib / plugins / network-throttle / network-throttle.js
var Immutable = require("immutable");

module.exports.init = function (ui) {

    var optPath = ["network-throttle"];
    var serverOptPath = optPath.concat(["servers"]);
    var listenHost = ui.options.get("listen");
    ui.servers  = {};

    ui.setOptionIn(optPath, Immutable.fromJS({
        name: "network-throttle",
        title: "Network Throttle",
        active: false,
        targets: require("./targets")
    }));

    ui.setOptionIn(serverOptPath, Immutable.Map({}));

    /**
     * @param input
     * @returns {number}
     */
    function getPortArg(input) {
        input = input.trim();
        if (input.length && input.match(/\d{3,5}/)) {
            input = parseInt(input, 10);
        } else {
            input = ui.bs.options.get("port") + 1;
        }
        return input;
    }

    /**
     * @returns {string}
     */
    function getTargetUrl() {
        return require("url").parse(ui.bs.options.getIn(["urls", "local"]));
    }

    var methods = {
        /**
         * @param data
         */
        "server:create": function (data) {

            data.port = getPortArg(data.port);
            data.cb   = data.cb || function () { /* noop */};

            /**
             * @param opts
             */
            function saveThrottleInfo (opts) {

                var urls = getUrls(ui.bs.options.set("port", opts.port).toJS());

                ui.setOptionIn(serverOptPath.concat([opts.port]), Immutable.fromJS({
                    urls: urls,
                    speed: opts.speed
                }));

                setTimeout(function () {

                    ui.socket.emit("ui:network-throttle:update", {
                        servers: ui.getOptionIn(serverOptPath).toJS(),
                        event: "server:create"
                    });

                    ui.servers[opts.port] = opts.server;

                    data.cb(null, opts);

                }, 300);

            }

            /**
             * @param err
             * @param port
             */
            function createThrottle (err, port) {

                var target = getTargetUrl();

                var args = {
                    port: port,
                    target: target,
                    speed: data.speed
                };

                if (ui.bs.getOption("scheme") === "https") {
                    var httpsOpts = require("browser-sync/lib/server/utils").getHttpsOptions(ui.bs.options);
                    args.key  = httpsOpts.key;
                    args.cert = httpsOpts.cert;
                }

                args.server = require("./throttle-server")(args, listenHost);
                require('server-destroy')(args.server);
                args.server.listen(port, listenHost);

                saveThrottleInfo(args);
            }

            /**
             * Try for a free port
             */
            ui.bs.utils.portscanner.findAPortNotInUse(data.port, data.port + 100, (listenHost || "127.0.0.1"), function (err, port) {
                if (err) {
                    return createThrottle(err);
                } else {
                    createThrottle(null, port);
                }
            });
        },
        /**
         * @param data
         */
        "server:destroy": function (data) {
            if (ui.servers[data.port]) {
                ui.servers[data.port].destroy();
                ui.setMany(function (item) {
                    item.deleteIn(serverOptPath.concat([parseInt(data.port, 10)]));
                });
                delete ui.servers[data.port];
            }
            ui.socket.emit("ui:network-throttle:update", {
                servers: ui.getOptionIn(serverOptPath).toJS(),
                event: "server:destroy"
            });
        },
        /**
         * @param event
         */
        event: function (event) {
            methods[event.event](event.data);
        }
    };

    return methods;
};

/**
 * Get local + external urls with a different port
 * @param opts
 * @returns {List<T>|List<any>}
 */
function getUrls (opts) {

    var list    = [];

    var bsLocal = require("url").parse(opts.urls.local);

    list.push([bsLocal.protocol + "//", bsLocal.hostname, ":", opts.port].join(""));

    if (opts.urls.external) {
        var external = require("url").parse(opts.urls.external);
        list.push([bsLocal.protocol + "//", external.hostname, ":", opts.port].join(""));
    }

    return Immutable.List(list);
}