@usestrict/wsproxy/src/server.js

This is a websocket to tcp proxy, written in node.js. It is dynamic and will proxy to various tcp servers.

/**
 * Dependencies
 */
var http    = require('http');
var https   = require('https');
var fs      = require('fs');
var ws      = require('ws');
var modules = require('./modules');
var mes     = require('./message');


/**
 * Proxy constructor
 */
var Proxy = require('./proxy');


/**
 * Initiate a server
 */
var Server = function Init(config) {
	var opts = {
		clientTracking: false,
		verifyClient:   onRequestConnect
	}

	if(config.ssl) {
		opts.server = https.createServer({
			key: fs.readFileSync( config.key ),
			cert: fs.readFileSync( config.cert ),
		}, function(req, res) {
			res.writeHead(200);
        	res.end("Secure wsProxy running...\n");
		});

		opts.server.listen(config.port)

		mes.status("Starting a secure wsProxy on port %s...", config.port)
	}
	else {
		opts.server = http.createServer(function(req, res) {
			res.writeHead(200);
			res.end("wsProxy running...\n");
		});

		opts.server.listen(config.port)

		mes.status("Starting wsProxy on port %s...", config.port)
	}

	var WebSocketServer = new ws.Server(opts)

	WebSocketServer.on('connection', onConnection);

	return this;
}


/**
 * Before estabilishing a connection
 */
function onRequestConnect(info, callback) {

	// Once we get a response from our modules, pass it through
	modules.method.verify(info, function(res) {
		callback(res);
	})

}


/**
 * Connection passed through verify, lets initiate a proxy
 */
function onConnection(ws) {

	modules.method.connect(ws, function(res) {
		//All modules have processed the connection, lets start the proxy
		new Proxy(ws);
	})

}


/**
 * Exports
 */
module.exports = Server;