Initial comit; working server.

Author: rpetrano

.gitignore

@@ -0,0 +1 @@
+config.json

command.js

@@ -0,0 +1,10 @@
+var Command = function(name, func) {
+	this.name = name;
+	this.func = func;
+}
+
+Command.prototype.call = function(self, args, stanza) {
+	this.func.call(self, args, stanza);
+}
+
+module.exports = Command;

commands/register.js

@@ -0,0 +1,21 @@
+var loglevel = require('../loglevel.js');
+var User = require('../user.js');
+var Command = require('../command.js');
+
+module.exports = Object.keys(loglevel).map(function(key) {
+	var value = loglevel[key];
+	return new Command('!' + key.toLowerCase(), function(args, stanza) {
+		var jid = stanza.attrs.from;
+		var projectName = args[1];
+
+		this.users = this.users.filter(function(user) {
+			return user.jid !== jid;
+		});
+		this.users.push(new User(jid, projectName, value));
+
+		if (value == loglevel.OFF)
+			this.sendMessage(jid, 'Won\'t bother you anymore');
+		else
+			this.sendMessage(jid, 'Sending logs ' + key + ' for project "' + projectName + '"');
+	});
+});

commands/status.js

@@ -0,0 +1,19 @@
+var loglevel = require('../loglevel.js');
+var Command = require('../command.js');
+
+var XmppWho = function(args, stanza) {
+	var jid = stanza.attrs.from;
+
+	var user = this.users.filter(function(user) { return user.jid === jid; })[0];
+	if (user === undefined || user.logLevel === loglevel.OFF) {
+		this.sendMessage(jid, 'No logs are sent to you.');
+		return;
+	}
+
+	var levelName = loglevel(user.logLevel);
+	this.sendMessage(jid, 'You are receiving logs ' + levelName + ' for project "' + user.projectName + '"');
+}
+
+module.exports = [ '!st', '!status' ].map(function(name) {
+	return new Command(name, XmppWho);
+});

commands/who.js

@@ -0,0 +1,20 @@
+var loglevel = require('../loglevel.js');
+var Command = require('../command.js');
+
+var XmppWho = function(args, stanza) {
+	var msg = this.users.filter(function(user) {
+		return user.logLevel != loglevel.OFF;
+	}).reduce(function(str, user) {
+		var level = loglevel(user.logLevel);
+		return str + user.jid + ' ' + level + ' ' + user.projectName + '\n';
+	}, '');
+
+	if (msg.length === 0)
+		this.sendMessage(stanza.attrs.from, 'No one is listening.');
+	else
+		this.sendMessage(stanza.attrs.from, msg);
+}
+
+module.exports = [ '!w', '!who' ].map(function(name) {
+	return new Command(name, XmppWho);
+});

config.json.example

@@ -0,0 +1,6 @@
+{
+	"jid": "[email protected]",
+	"password": "",
+	"port":     9989,
+	"host":     "localhost"
+}

loglevel.js

@@ -0,0 +1,11 @@
+var loglevel = function(value) {
+	return Object.keys(loglevel).filter(function(key) {return loglevel[key] === value})[0];
+}
+loglevel.TRACE = 0;
+loglevel.DEBUG = 1;
+loglevel.INFO  = 2;
+loglevel.WARN  = 3;
+loglevel.ERROR = 4;
+loglevel.OFF   = 5;
+
+module.exports = loglevel;

main.js

@@ -0,0 +1,17 @@
+var config = require('./config.json');
+
+var Xmpp = require('./xmpp.js');
+var commands = Array.prototype.concat(
+	require('./commands/register.js'),
+	require('./commands/status.js'),
+	require('./commands/who.js')
+);
+
+var xmpp = new Xmpp();
+xmpp.connect(config.jid, config.password);
+xmpp.setAvailable();
+xmpp.registerCommand(commands);
+
+var Server = require('./server.js');
+var server = new Server(xmpp);
+server.listen(config.port, config.host);

message.js

@@ -0,0 +1,12 @@
+var qs = require('querystring');
+var url = require('url');
+
+var Message = function(request, body) {
+	var query = url.parse(request.url, true).query;
+
+	this.body = body;
+	this.projectName = query.projectName;
+	this.logLevel = query.level;
+}
+
+module.exports = Message;

server.js

@@ -0,0 +1,38 @@
+var http = require('http');
+var Message = require('./message.js');
+var credentials = 'superusr1245!:passpasspass123';
+
+var HttpServer = function(xmpp) {
+	var server = http.createServer();
+
+	server.on('request', function(request, response) {
+		var auth = request.headers.authorization;
+		if (auth === undefined || new Buffer(auth.replace(/Basic /, ''), 'base64').toString('utf-8') !== credentials) {
+			response.writeHead(401);
+			response.end('Not authorized.');
+			return;
+		}
+
+		if (request.method === 'GET') {
+			var projectName = require('url').parse(request.url).path.substring(1);
+			var level = xmpp.getLevel(projectName).toString();
+			response.writeHead(200);
+			response.end(level, 'utf-8');
+		} else {
+			var body = '';
+			request.on('data', function(chunk) {
+				body += chunk;
+			});
+
+			request.on('end', function() {
+				xmpp.log(new Message(request, body));
+				response.writeHead(200);
+				response.end();
+			});
+		}
+	});
+
+	return server;
+}
+
+module.exports = HttpServer;

user.js

@@ -0,0 +1,7 @@
+var loglevel = require('./loglevel.js');
+var User = function(jid, projectName, logLevel) {
+	this.jid = jid;
+	this.projectName = projectName;
+	this.logLevel = logLevel;
+}
+module.exports = User;

xmpp.js

@@ -0,0 +1,96 @@
+var xmpp = require('node-xmpp');
+var loglevel = require('./loglevel.js');
+
+var Xmpp = function() {
+	this.client = null;
+	this.users = [ ];
+	this.jid = '';
+}
+
+Xmpp.prototype = {
+	respondToPings: function() {
+		var self = this;
+		self.client.on('stanza', function(stanza) {
+			if (stanza.is('iq') && stanza.children.length == 1 && stanza.children[0].is('ping')) {
+				var pong = new xmpp.Element('iq', {
+					from: stanza.attrs.to,
+					to: stanza.attrs.from,
+					id: stanza.attrs.id,
+					type: 'result'
+				});
+				self.client.send(pong);
+			}
+		});
+	},
+	connect: function(jid, password) {
+		var self = this;
+
+		self.jid = jid;
+		self.client = new xmpp.Client({
+			jid: jid,
+			password: password
+		});
+		self.client.setMaxListeners(100);
+		self.client.connection.socket.setTimeout(0);
+		self.client.connection.socket.setKeepAlive(true, 10000);
+		self.respondToPings();
+
+		self.client.on('error', function(e) {
+			console.error(e);
+			process.exit(1);
+		});
+	},
+	setAvailable: function() {
+		var self = this;
+		self.client.on('online', function() {
+			self.client.send(
+				new xmpp.Element('presence',{
+					type: 'chat'
+				}).c('show').t('chat').up()
+			);
+		});
+	},
+	registerCommand: function(command) {
+		var self = this;
+		if (Array.isArray(command)) {
+			for (var i in command)
+				self.registerCommand(command[i]);
+			return;
+		}
+
+		self.client.on('stanza', function(stanza) {
+			var body = stanza.getChild('body');
+			if (!stanza.is('message') || stanza.attrs.type !== 'chat' || body == undefined)
+				return;
+
+			var msg = body.getText();
+			var args = msg.split(/\s+/);
+			if (args[0] === command.name)
+				command.call(self, args, stanza);
+		});
+	},
+	sendMessage: function(to, msg) {
+		this.client.send(new xmpp.Element('message', {
+			to: to,
+			type: 'chat'
+		}).c('body').t(msg));
+	},
+	log: function(msg) {
+		for (var i in this.users) {
+			var user = this.users[i];
+			if (user.projectName == msg.projectName && user.logLevel <= msg.logLevel)
+				this.sendMessage(user.jid, msg.body);
+		}
+	},
+	getLevel: function(projectName) {
+		return this.users.filter(function(user) {
+			return user.projectName === projectName;
+		}).map(function(user) {
+			return user.logLevel;
+		}).reduce(function(minLevel, level) {
+			return Math.min(minLevel, level);
+		}, loglevel.OFF);
+	}
+}
+
+module.exports = Xmpp;