You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 

118 lines
3.3 KiB

#!/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);
});
});
}
}