mhm/lib/host-entry.js

Multi Host Manager

var fs = require("fs");
var net = require("net");
var path = require("path");

var blankPromiseFunc = function(resolve, reject) {
  resolve();
};

var startHostPromiseFunc = blankPromiseFunc;
var stopHostPromiseFunc = blankPromiseFunc;
var restartHostPromiseFunc = function(resolve, reject) {
  new Promise(stopHostPromiseFunc).then(function() {
    new Promise(startHostPromiseFunc).then(function() {
      resolve();
    }, function() { reject(); });
  }, function() { reject(); });
};
var refreshHostPromiseFunc = blankPromiseFunc;

function startHost() {

}

function stopHost() {

}

function restartHost() {

}

function refreshHost() {

}

if(process.argv.length > 2) {
  var hostDir = process.argv[2];
  var hostEntryPath = process.argv[3];
  var hostRuntimePath = process.argv[4];
  var msgPort = parseInt(process.argv[5]);

  var originalConsoleLog = console.log;

  var runtimeData = {
    ports: {}
  };

  var updateRuntimeData = function() {
    fs.writeFile(hostRuntimePath, JSON.stringify(runtimeData), function(err) {
      if(err) { console.log("Could not update runtime data for host"); }
    });
  };
  updateRuntimeData();

  var listenerPortOverrides = [];

  var messageServer = net.createServer();
  var messageListener = messageServer.listen(0, "localhost", function() {

    messageServer.on("connection", function(socket) {
      var data = "";
      socket.on("data", function(chunk) {
        if(chunk)
        data += chunk;
      });

      socket.on("end", function(chunk) {
        if(chunk)
        data += chunk;

        try {
          data = JSON.parse(data);
        } catch(err) {
          console.log("recieved invalid message");
        }

        if(data.ports) {
          var ports = data.ports;
        }


      });
    });

    runtimeData.messagePort = messageListener.address().port;
    updateRuntimeData();

  });


  process.cwd(hostDir);

  var http = require("http");
  var httpCreateServer = http.createServer;
  http.createServer = function() {
    var server = httpCreateServer.apply(this, arguments);
    var serverListen = server.listen;
    server.listen = function() {

      // If port was explicitly set.
      if(arguments.length > 0 && typeof arguments[0] == "number") {
        var port = arguments[0];
        var args = arguments;
        args[0] = 0;
        var listener = serverListen.apply(this, args);
        if(!runtimeData.ports["http"]) runtimeData.ports["http"] = {};
        runtimeData.ports.http[port] = listener.address().port;
        updateRuntimeData();
        return listener;
      }

      return serverListen.apply(this, arguments);
    };

    return server;
  };

  console.log("requiring '%s'", hostEntryPath);
  var hostModule = require(hostEntryPath);

  if(typeof hostModule.start == "function") {
    startHostPromiseFunc = hostModule.start;
  }
}