|
|
#!/usr/bin/env node
"use strict"; var dope = require("console-dope"), connect = require("connect"), http = require("http"), fs = require("fs"), Model = require("nature").Model, w = require("wodge"), path = require("path"), loadConfig = require("config-master"), morgan = require("morgan"), serveStatic = require("serve-static"), directory = require('serve-index'), compress = require('compression');
var usage = "usage: ws [--directory|-d <directory>] [--port|-p <port>] [--log-format|-f dev|default|short|tiny] [--compress|-c]";
function halt(message){ dope.red.log("Error: %s", message); dope.log(usage); process.exit(1); }
/* parse command-line args */ var argv = new Model() .define({ name: "port", alias: "p", type: "number", defaultOption: true, value: 8000 }) .define({ name: "log-format", alias: "f", type: "string" }) .define({ name: "help", alias: "h", type: "boolean" }) .define({ name: "directory", alias: "d", type: "string", value: process.cwd() }) .define({ name: "compress", alias: "c", type: "boolean" }) .on("error", function(err){ halt(err.message); });
/* Merge together options from "~/.local-web-server.json", "{cwd}/.local-web-server.json" and "{cwd}/package.json", in that order. */ var storedConfig = loadConfig( path.join(w.getHomeDir(), ".local-web-server.json"), path.join(process.cwd(), ".local-web-server.json"), path.join(process.cwd(), "package.json:local-web-server") ); argv.set(storedConfig);
/* Finally, set the options from the command-line, overriding all defaults. */ argv.set(process.argv); /* Die here if invalid args received */ if (!argv.valid) halt(argv.validationMessages);
if (argv.help){ dope.log(usage);
} else { var total = { req: 0, bytes: 0, connections: 0 };
process.on("SIGINT", function(){ dope.showCursor(); dope.log(); process.exit(0); });
/* customised logger :date token, purely to satisfy Logstalgia. */ morgan.token("date", function(){ var a = new Date(); return (a.getDate() + "/" + a.getUTCMonth() + "/" + a.getFullYear() + ":" + a.toTimeString()) .replace("GMT", "").replace(" (BST)", ""); });
var app = connect();
/* log using --log-format (if supplied), else output statics */ if(argv["log-format"]){ app.use(morgan(argv["log-format"])); } else { app.use(function(req, res, next){ dope.column(1).write(++total.req); next(); }); }
/* --compress enables compression */ if (argv.compress) app.use(compress());
/* static file server including directory browsing support */ app.use(serveStatic(path.resolve(argv.directory))) .use(directory(path.resolve(argv.directory), { icons: true }));
/* launch server */ var server = http.createServer(app) .on("error", function(err){ if (err.code === "EADDRINUSE"){ halt("port " + argv.port + " is already is use"); } else { halt(err.message); } }) .listen(argv.port);
/* write status to stderr so stdout can be piped to disk ($ ws > log.txt) */ if (path.resolve(argv.directory) === process.cwd()){ dope.error("serving at %underline{%s}", "http://localhost:" + argv.port); } else { dope.error("serving %underline{%s} at %underline{%s}", argv.directory, "http://localhost:" + argv.port); }
/* in stats mode, monitor connections and bytes transferred */ if (!argv["log-format"]){ dope.hideCursor(); dope.log("%underline{Requests} %underline{Data} %underline{Connections}"); server.on("connection", function(socket){ var oldWrite = socket.write; socket.write = function(data) { if (!Buffer.isBuffer(data)) { data = new Buffer(data); } oldWrite.call(this, data); total.bytes += data.length; dope.column(12).write(w.padRight(w.bytesToSize(total.bytes, 2), 12)); }; dope.column(24).write(++total.connections); socket.on("close", function(){ dope.column(24).write(w.padRight(--total.connections)); }); }); } }
|