thelounge/lib/server.js

var config = require("../config");
var http = require("connect");
var io = require("socket.io");
var irc = require("slate-irc");
var net = require("net");
var _ = require("lodash");

// Models

var Chan = require("./models/chan");
var Msg = require("./models/msg");
var Network = require("./models/network");
var User = require("./models/user");

var sockets = null;
var networks = [
	new Network({host: "Status"})
];

var events = [
	"join",
	"kick",
	"mode",
	"motd",
	"message",
	"names",
	"nick",
	"notice",
	"part",
	"quit",
	"topic",
	"welcome",
	"whois",
];

module.exports = listen;

function listen() {
	var app = http()
		.use(http.static("client"))
		.listen(9000);
	
	var self = this;
	sockets = io.listen(app, {log: 0}).sockets.on("connection", function(s) {
		s.on("input", input);
		s.emit("networks", {
			networks: networks
		});
	});
	
	config.networks.forEach(function(n) {
		connect(n);
	});
}

function connect(params) {
	params = _.extend(
		config.defaults,
		params
	);
	
	var host = params.host;
	var port = params.port;
	
	var stream = net.connect({
		port: port,
		host: host,
	});
	
	stream.on("error", function(e) {
		console.log(e);
	});
	
	var client = irc(stream);
	var network = new Network({
		host: host,
		client: client,
	});
	
	networks.push(network);
	sockets.emit("networks", {
		networks: networks
	});

	client.nick(params.nick);
	client.user(params.nick, params.realname);
	client.once("welcome", function() {
		(params.channels || []).forEach(function(c) {
			client.join(c);
		});
	});
	
	events.forEach(function(e) {
		client.on(e, function() {
			event.apply(network, [e, arguments]);
		});
	});
}

function input(data) {
	var target = find(data.id);
	if (!target) {
		return;
	}
	
	var network = target.network;
	var chan = target.chan;
	
	var client = network.client;
		
	var id = data.id;
	var text = data.text;
	
	var args = text.replace(/^\//, '').split(" ");
	var cmd = text.charAt(0) == "/" ? args[0].toLowerCase() : "";
	
	switch (cmd) {
	case "":
		args.unshift(
			"msg",
			chan.name
		);
	case "msg":
		var user;
		var text = _.tail(args, 2).join(" ");
		if (cmd == "msg") console.log(args);
		if (client) {
			user = client.me;
			client.send(args[1], text);
		}
		var chan = _.findWhere(network.channels, {name: args[1]});
		if (typeof chan !== "undefined") {
			var msg = new Msg({
				from: user,
				text: text,
			});
			chan.messages.push(msg)
			sockets.emit("msg", {
				id: chan.id,
				msg: msg,
			});
		}
		break;
	
	case "server":
	case "connect":
		if (args[1]) {
			connect({host: args[1]});
		}
		break;
	
	case "join":
	case "nick":
		if (client && args[1]) {
			client[cmd].apply(client, args.slice(1));
		}
		break;
	
	case "leave":
	case "part":
		var id = chan.id;
		if (chan.type == "query" || !chan.users.length) {
			remove(id);
			sockets.emit("part", {
				id: id,
			});
		} else if (client) {
			client.part(chan.name);
		}
		break;
	
	case "topic":
		if (client) {
			var msg = "TOPIC";
			msg += " " + chan.name;
			msg += args[1] ? " :" + args.slice(1).join(" ") : "";
			client.write(msg);
		}
		break;
	
	case "query":
	case "whois":
		if (client && args[1]) {
			client.whois(args[1]);
		}
		break;
	
	case "kick":
		if (client && args[1]) {
			client.kick(chan.name, args[1]);
		}
		break;
	
	case "op":
	case "deop":
	case "voice":
	case "devoice":
	case "mode":
		if (!client || !args[1]) {
			break;
		}
		var mode;
		var user;
		if (cmd != "mode") {
			user = args[1];
			mode = {
			     "op": "+o",
			   "deop": "-o",
			  "voice": "+v",
			"devoice": "-v",
			}[cmd];
		} else if (!args[2]) {
			break;
		} else {
			mode = args[1];
			user = args[2];
		}
		client.mode(
			chan.name,
			mode,
			user
		);
		break;
	
	case "quit":
	case "disconnect":
		if (client) {
			client.quit();
			networks = _.without(networks, network);
			sockets.emit("networks", {
				networks: networks
			});
		}
		break;
	}
}

function event(e, data) {
	var data = _.last(data);
	var channels = this.channels;
	
	switch (e) {
	case "join":
		var chan = _.findWhere(channels, {name: data.channel});
		if (typeof chan === "undefined") {
			chan = new Chan({
				name: data.channel,
			});
			channels.push(chan);
			sockets.emit("join", {
				id: this.id,
				chan: chan,
			});
		}
		var users = chan.users;
		users.push(new User({name: data.nick}));
		sockets.emit("users", {
			id: chan.id,
			users: users,
		});
		var msg = new Msg({
			from: data.nick,
			type: "join",
		});
		chan.messages.push(msg);
		sockets.emit("msg", {
			id: chan.id,
			msg: msg,
		});
		break;
	
	case "kick":
		var chan = _.findWhere(channels, {name: data.channel});
		if (data.client == this.client.me) {
			chan.users = [];
		} else {
			chan.users = _.without(chan.users, _.findWhere(chan.users, {name: data.client}));
		}
		sockets.emit("users", {
			id: chan.id,
			users: chan.users,
		});
		var msg = new Msg({
			type: "kick",
			from: data.nick,
			text: data.client,
		});
		chan.messages.push(msg);
		sockets.emit("msg", {
			id: chan.id,
			msg: msg,
		});
		break;
	
	case "mode":
		var chan = _.findWhere(channels, {name: data.target});
		if (typeof chan !== "undefined") {
			var msg = new Msg({
				type: "mode",
				from: data.nick,
				text: data.mode + " " + data.client,
			});
			chan.messages.push(msg);
			sockets.emit("msg", {
				id: chan.id,
				msg: msg,
			});
		}
		break;
	
	case "motd":
		var chan = channels[0];
		data.motd.forEach(function(m) {
			var msg = new Msg({
				type: "motd",
				from: "-!-",
				text: m,
			});
			chan.messages.push(msg);
			sockets.emit("msg", {
				id: chan.id,
				msg: msg,
			});
		});
		break;
	
	case "message":
		var chan = _.findWhere(channels, {name: data.to});
		if (typeof chan === "undefined") {
			chan = new Chan({
				name: data.from,
				type: "query",
			});
			channels.push(chan);
			sockets.emit("join", {
				id: this.id,
				chan: chan,
			});
		}
		var type = "";
		var text = data.message;
		var network = this;
		text.split(" ").forEach(function(w) {
			if (w == network.client.me) type = "highlight";
		});
		var msg = new Msg({
			type: type,
			from: data.from,
			text: text,
		});
		chan.messages.push(msg);
		sockets.emit("msg", {
			id: chan.id,
			msg: msg,
		});
		break;
	
	case "names":
		var chan = _.findWhere(channels, {name: data.channel});
		chan.users = [];
		data.names.forEach(function(n) {
			chan.users.push(new User({name: n}));
		});
		sockets.emit("users", {
			id: chan.id,
			users: chan.users,
		});
		break;
	
	case "nick":
		if (data["new"] == this.client.me) {
			var chan = channels[0];
			var msg = new Msg({
				from: "-!-",
				text: "You're now known as " + data["new"],
			});
			chan.messages.push(msg);
			sockets.emit("msg", {
				id: chan.id,
				msg: msg,
			});
		}
		channels.forEach(function(chan) {
			var user = _.findWhere(chan.users, {name: data.nick});
			if (!user) {
				return;
			}
			user.name = data["new"];
			sockets.emit("users", {
				id: chan.id,
				users: chan.users,
			});
			var msg = new Msg({
				type: "nick",
				from: data.nick,
				text: data["new"],
			});
			chan.messages.push(msg);
			sockets.emit("msg", {
				id: chan.id,
				msg: msg,
			});
		});
		break;
	
	case "notice":
		var chan = channels[0];
		var msg = new Msg({
			type: "notice",
			from: data.to == "*" ? "-!-" : data.from,
			text: data.message,
		});
		chan.messages.push(msg);
		sockets.emit("msg", {
			id: chan.id,
			msg: msg,
		});
		break;
	
	case "part":
		var chan = _.findWhere(channels, {name: data.channels[0]});
		if (data.nick == this.client.me) {
			remove(chan.id);
			sockets.emit("part", {
				id: chan.id,
			});
		} else {
			chan.users = _.without(chan.users, _.findWhere(chan.users, {name: data.nick}));
			sockets.emit("users", {
				id: chan.id,
				users: chan.users,
			});
			var msg = new Msg({
				type: "part",
				from: data.nick,
			});
			chan.messages.push(msg);
			sockets.emit("msg", {
				id: chan.id,
				msg: msg,
			});
		}
		break;
	
	case "quit":
		channels.forEach(function(chan) {
			var user = _.findWhere(chan.users, {name: data.nick});
			if (!user) {
				return;
			}
			chan.users = _.without(chan.users, user);
			sockets.emit("users", {
				id: chan.id,
				users: chan.users,
			});
			var msg = new Msg({
				type: "quit",
				from: data.nick,
			});
			chan.messages.push(msg);
			sockets.emit("msg", {
				id: chan.id,
				msg: msg,
			});
		});
		break;
	
	case "topic":
		var chan = _.findWhere(channels, {name: data.channel});
		var from = data.nick || chan.name;
		var msg = new Msg({
			type: "topic",
			from: from,
			text: data.topic,
		});
		chan.messages.push(msg);
		sockets.emit("msg", {
			id: chan.id,
			msg: msg,
		});
		break;
	
	case "welcome":
		// Leaving this empty for now.
		break;
	
	case "whois":
		if (!data) {
			var chan = channels[0];
			var msg = new Msg({
				type: "error",
				text: "No such nick/channel.",
			});
			chan.messages.push(msg);
			sockets.emit("msg", {
				id: chan.id,
				msg: msg,
			});
		} else {
			var chan = _.findWhere(channels, {name: data.nickname});
			if (typeof chan === "undefined") {
				chan = new Chan({
					type: "query",
					name: data.nickname,
				});
				channels.push(chan);
				sockets.emit("join", {
					id: this.id,
					chan: chan,
				});
			}
			var i = 0;
			for (var k in data) {
				if (i++ == 5) break;
				var msg = new Msg({
					type: "whois",
					from: "-!-",
					text: k + ": " + data[k],
				});
				chan.messages.push(msg);
				sockets.emit("msg", {
					id: chan.id,
					msg: msg,
				});
			}
		}
		break;
	}
}

// Utils

function find(id) {
	var result = false;
	networks.forEach(function(n) {
		result = {
			network: n,
			chan: _.findWhere(n.channels, {id: id}),
		};
		if (!result.chan) {
			result = false;
		}
	});
	return result;
}

function remove(id) {
	networks.forEach(function(n) {
		n.channels = _.without(n.channels, _.findWhere(n.channels, {id: id}));
	});
}