fixed 0000013

[JIRC.git] / JIRC.js

#!/usr/bin/env node

//growl('5 new emails', { title: 'Email Client' });
//process.exit(0);
var argv = require('optimist').argv, bindhost="127.0.0.1", bindport="10080";
if(argv.h || argv.help){
    console.log('usage: '+argv.$0+' [options] [bind-host]');
    console.log('');
    console.log('Options:');
    console.log('-q        be quiet');    
    console.log('-v        verbose logging');
    process.exit(0);    
}
if(argv._[0]){
    if(argv._[0].match(/^\d+\.\d+\.\d+\.\d+\:[0-9]{1,5}$/)){
        bindhost=argv._[0].split(":")[0];
        bindport=argv._[0].split(":")[1];        
    } else {
        console.error('Error: bind-host must be in the format [IP]:[PORT]');
        process.exit(0);
    }
}

require('./lib/IRC.js');
global.argv=argv;
global.irc.loadScripts();
global.settings = '';
global.resume = '';

var http = require('http'),
    WebSocketRequest = require('websocket').request,
    ws = require('websocket-server'),
    url = require('url'),
    fs = require('fs'),        
    static = require('node-static'),
    file = new(static.Server)('./public', { cache: 7200, headers: {'X-Hello':'World!'} }),
    httpServer = http.createServer(function(request, response) {
            var queryData = "";        
        request.on('data', function(data) {
            var path = url.parse(request.url).pathname; 
            if(path=="/resume"){        
                queryData += data;
                if(queryData.length > 1e6) {
                    queryData = "";
                    response.writeHead(413, {'Content-Type': 'text/plain'});
                    request.connection.destroy();
                }
            } else if(path=="/settings"){        
                queryData += data;
                if(queryData.length > 1e6) {
                    queryData = "";
                    response.writeHead(413, {'Content-Type': 'text/plain'});
                    request.connection.destroy();
                }            
            }
        });
        
        request.addListener('end', function () {    
            var path = url.parse(request.url).pathname;
            if(path=="/emoticons"){
                var emoticons = fs.readFileSync(
                    __dirname + '/public/images/emoticons/Emoticons.plist', 'utf8'
                );
                response.writeHead(200, {"Content-Type": "application/xml"});
                response.end(emoticons);
            } else if(path=="/resume"){
                response.writeHead(200, {"Content-Type": "application/json"});               
                response.end(JSON.stringify(global.resume));
                if(queryData) {
                    global.resume=JSON.parse(queryData);
                }
            } else if(path=="/settings"){
                response.writeHead(200, {"Content-Type": "application/json"});               
                response.end(JSON.stringify(global.settings));
                if(queryData) {
                    global.settings=JSON.parse(queryData);
                }
            } else{
                file.serve(request, response, function (err, res) {
                    if (err) {
                        console.error((new Date()) + " > Error serving " + request.url + " - " + err.message);
                        response.writeHead(err.status, err.headers);
                        response.end();
                    } else {
                        if(argv.v) console.log((new Date()) + " > " + request.url + " - " + res.message);
                    }
                });
            }
        });
    });
httpServer.listen(bindport, bindhost, function() {
    if(!argv.q) console.log((new Date()) + " Server is listening on " + bindhost + ":" + bindport);
});

// node-websocket-server
var miksagoConnection = require('./node_modules/websocket-server/lib/ws/connection');
var miksagoServer = ws.createServer();
miksagoServer.server = httpServer;
miksagoServer.addListener('connection', function(connection) {
    connection.remoteAddress = connection._socket.remoteAddress;
    connection.sendUTF = connection.send;
    handleConnection(connection);
});


// WebSocket-Node config

var wsServerConfig =  {
    // All options *except* 'httpServer' are required when bypassing
    // WebSocketServer.
    maxReceivedFrameSize: 0x10000,
    maxReceivedMessageSize: 0x100000,
    fragmentOutgoingMessages: true,
    fragmentationThreshold: 0x4000,
    keepalive: true,
    keepaliveInterval: 20000,
    assembleFragments: true,
    // autoAcceptConnections is not applicable when bypassing WebSocketServer
    // autoAcceptConnections: false,
    disableNagleAlgorithm: true,
    closeTimeout: 5000
};

httpServer.on('upgrade', function(req, socket, head) {
    if (typeof req.headers['sec-websocket-version'] !== 'undefined') {
        // WebSocket hybi-08/-09/-10 connection (WebSocket-Node)
        var wsRequest = new WebSocketRequest(socket, req, wsServerConfig);
        try {
            wsRequest.readHandshake();
            var wsConnection = wsRequest.accept(wsRequest.requestedProtocols[0], wsRequest.origin);
            handleConnection(wsConnection);
        }
        catch(e) {
            console.log("WebSocket Request unsupported by WebSocket-Node: " + e.toString());
            return;
        }
    } else {
        // WebSocket hixie-75/-76/hybi-00 connection (node-websocket-server)
        if (req.method === 'GET' &&
            (req.headers.upgrade && req.headers.connection) &&
            req.headers.upgrade.toLowerCase() === 'websocket' &&
            req.headers.connection.toLowerCase() === 'upgrade') {
            new miksagoConnection(miksagoServer.manager, miksagoServer.options, req, socket, head);
        }
    }
});

function handleConnection(connection) {
    if(!argv.q) console.log((new Date()) + " Connection accepted.");    
            global.websocket = connection;
    connection.addListener('message', function(wsMessage) {
        var message = wsMessage;
        // WebSocket-Node adds a "type", node-websocket-server does not
        if (typeof wsMessage.type !== 'undefined') {
            if (wsMessage.type !== 'utf8') {
                return;
            }
            message = wsMessage.utf8Data;
        }
//        console.log("Received Message: " + message);
        if (message.match(/^CONNECT (.*)$/i)) {
            var arrCmd = message.split(" ");
            global.irc.connect(arrCmd[1], arrCmd[2], arrCmd[3], arrCmd[4], arrCmd[5], arrCmd[6]);        
        } else{
            global.irc.send(message);        
        }       
    });
    
    connection.addListener('close', function() {
        if(!argv.q) console.log((new Date()) + " Peer " + connection.remoteAddress + " disconnected.");
        if(!global.resume || global.settings.hold != "on") {
            global.irc.quit('BYE BYE');
            if(!argv.q) console.log((new Date()) + " Disconnected from IRC");            
        }
    });
       
}
if(!argv.n) require("openurl").open("http://" + bindhost + ":" + bindport);