run.js

// a little node webserver designed to run the unit tests herein

var   sys = require("sys"),
     http = require("http"),
      url = require("url"),
     path = require("path"),
       fs = require("fs"),
  connect = require("connect");

var PRIMARY_HOST = "127.0.0.1";

// all bound webservers stored in this lil' object
var boundServers = [ ];

function subHostNames(data) {
  for (var i = 0; i < boundServers.length; i++) {
    var o = boundServers[i]
    var a = o.server.address();
    var from = o.name;
    var to = "http://" + a.address + ":" + a.port;
    data = data.toString().replace(new RegExp(from, 'g'), to);

    // now do another replacement to catch bare hostnames sans http(s)
    // and explicit cases where port is appended
    var fromWithPort;
    if (from.substr(0,5) === 'https') {
        from = from.substr(8);
        fromWithPort = from + ":443";
    } else {
        from = from.substr(7);
        fromWithPort = from + ":80";
    }
    to = to.substr(7);

    if (o.subPath) to += o.subPath;

    data = data.replace(new RegExp(fromWithPort, 'g'), to);
    data = data.replace(new RegExp(from, 'g'), to);
  }

  return data;
}

function createServer(obj) {
    var server = connect.createServer().use(connect.favicon()).use(connect.logger());

    // this file is a *test* harness, to make it go, we'll insert a little handler that
    // substitutes output, changing production URLs to developement URLs.
    server.use(function(req, resp, next) {
        // cache the *real* functions
        var realWrite = resp.write;
        var realEnd = resp.end;
        var realWriteHead = resp.writeHead;

        var buf = undefined;
        var enc = undefined;
        var contentType = undefined;

        resp.writeHead = function (sc, reason, hdrs) {
            var h = undefined;
            if (typeof hdrs === 'object') h = hdrs;
            else if (typeof reason === 'object') h = reason; 
            for (var k in h) {
                if (k.toLowerCase() === 'content-type') {
                    contentType = h[k];
                    break;
                }
            }
            if (!contentType) contentType = resp.getHeader('content-type');
            if (!contentType) contentType = "application/unknown";
            realWriteHead(sc, reason, hdrs);
        };

        resp.write = function (chunk, encoding) {
            if (buf) buf += chunk;
            else buf = chunk;
            enc = encoding;
        };

        resp.end = function() {
            if (!contentType) contentType = resp.getHeader('content-type');
            if (contentType && (contentType === "application/javascript" ||
                                contentType.substr(0,4) === 'text'))
            {
                if (buf) {
                    var l = buf.length;
                    buf = subHostNames(buf);
                    if (l != buf.length) resp.setHeader('Content-Length', buf.length);
                }
            }
            if (buf && buf.length) realWrite.call(resp, buf, enc);
            realEnd.call(resp);
        }

        next();
    });

    // let the specific server interact directly with the connect server to register their middleware 
    if (obj.setup) obj.setup(server);

    // if this site has a handler, we'll run that, otherwise serve statically
    if (obj.handler) server.use(obj.handler);

    // now set up the static resource servin'
    var p = obj.path, ps = path.join(p, "static");
    try { if (fs.statSync(ps).isDirectory()) p = ps; } catch(e) { }
    server.use(connect.static(p));

    // and listen!
    server.listen(obj.port, PRIMARY_HOST);
    return server;
};

// start up webservers on ephemeral ports for each subdirectory here.
var dirs = [
    // the reference verification server.  A version is hosted at
    // browserid.org and may be used, or the RP may perform their
    // own verification.
    {
        name: "https://browserid.org/verify",
        subPath: "/",
        path: path.join(__dirname, "verifier")
    },
    // An example relying party.
    {
        name: "http://rp.eyedee.me",
        path: path.join(__dirname, "rp")
    },
    // A reference primary identity provider.
    {
        name: "https://eyedee.me",
        path: path.join(__dirname, "primary")
    },
    // BrowserID: the secondary + ip + more.
    {
        name: "https://browserid.org",
        path: path.join(__dirname, "authority")
    }
];

function formatLink(server, extraPath) {
  var addr = server.address();
  var url = 'http://' + addr.address + ':' + addr.port;
  if (extraPath) {
    url += extraPath;
  }
  return url;
}

console.log("Running test servers:");
dirs.forEach(function(dirObj) {
  if (!fs.statSync(dirObj.path).isDirectory()) return;
  // does this server have a js handler for custom request handling?
  var handlerPath = path.join(dirObj.path, "server", "run.js");
  var runJS = {};
  try {
    var runJSExists = false;
    try { runJSExists = fs.statSync(handlerPath).isFile() } catch(e) {};
    if (runJSExists) {
      var runJS = require(handlerPath);
    }
  } catch(e) {
    console.log("Error loading " + handlerPath + ": " + e);
    process.exit(1);
  }

  var so = {
    path: dirObj.path,
    server: undefined,
    port: "0",
    name: dirObj.name,
    handler: runJS.handler,
    setup: runJS.setup,
    subPath: dirObj.subPath
  };
  so.server = createServer(so)
  boundServers.push(so);
  console.log("  " + dirObj.name + ": " + formatLink(so.server));
});