|
|
#!/usr/bin/env node
"use strict"; require("more-console"); var connect = require("connect"), http = require("http"), util = require("util"), Thing = require("nature").Thing, w = require("wodge");
var usage = "usage: ws [--directory|-d <directory>] [--port|-p <port>] [--log-format|-f dev|default|short|tiny]";
function halt(message){ console.red.log("Error: %s", message); console.log(usage); process.exit(1); }
/** parse command-line args */ var argv = new Thing() .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() }) .on("error", function(err){ halt(err.message); }) .set(process.argv);
function handleServerError(err){ if (err.code === "EADDRINUSE"){ halt("port " + argv.port + " is already is use"); } else { halt(err.message); } }
/** Die here if invalid args received */ if (!argv.valid) halt(argv.validationMessages);
/** $ ws --help */ if (argv.help){ console.log(usage);
} else { var total = { req: 0, bytes: 0, connections: 0 };
process.on("SIGINT", function(){ console.show(); console.log(); process.exit(0); });
/** customised connect.logger :date token, purely to satisfy Logstalgia. */ connect.logger.token("date", function(){ var a = new Date(); return (a.getDate() + "/" + a.getUTCMonth() + "/" + a.getFullYear() + ":" + a.toTimeString()) .replace("GMT", "").replace(" (BST)", ""); });
var app = connect();
if(argv["log-format"]){ app.use(connect.logger(argv["log-format"])); } else { app.use(function(req, res, next){ console.column(1).write(++total.req); next(); }); }
app.use(connect.static(argv.directory)) .use(connect.directory(argv.directory, { icons: true }));
var server = http.createServer(app) .on("error", handleServerError) .listen(argv.port);
/* write status to stderr so stdout can be piped to disk ($ ws > log.txt) */ if (argv.directory === process.cwd()){ console.error("serving at %u{%s}", "http://localhost:" + argv.port); } else { console.error("serving %u{%s} at %u{%s}", argv.directory, "http://localhost:" + argv.port); }
if (!argv["log-format"]){ console.hide(); console.log("%u{Requests} %u{Data} %u{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; console.column(12).write((w.bytesToSize(total.bytes, 2) + " ").slice(0,12)); }; console.column(24).write(++total.connections); socket.on("close", function(){ console.column(24).write(--total.connections); }); }); } }
|