thelounge/src/plugins/irc-events/link.js

var _ = require("lodash");
var cheerio = require("cheerio");
var Msg = require("../../models/msg");
var request = require("superagent");

module.exports = function(irc, network) {
	var client = this;
	irc.on("message", function(data) {
		var links = [];
		var split = data.message.split(" ");
		_.each(split, function(w) {
			var match = w.indexOf("http://") === 0 || w.indexOf("https://") === 0;
			if (match) {
				links.push(w);
			}
		});

		if (links.length === 0) {
			return;
		}

		var self = data.to.toLowerCase() == irc.me.toLowerCase();
		var chan = _.findWhere(network.channels, {name: self ? data.from : data.to});
		if (typeof chan === "undefined") {
			return;
		}

		var msg = new Msg({
			type: Msg.Type.TOGGLE,
			time: ""
		});
		chan.messages.push(msg);
		client.emit("msg", {
			chan: chan.id,
			msg: msg
		});

		_.each(links, function(url) {
			fetch(url, function(res) {
				parse(msg.id, url, res, client);
			});
		});
	});
};

function parse(id, url, res, client) {
	var type = "";
	var head = "";
	var body = "";
	var link = url;
	switch (res.type) {
	case "text/html":
		var $ = cheerio.load(res.res.text);
		type = "link";
		head = $("title").text();
		body = "No description found.";
		break;

	case "image/png":
	case "image/gif":
	case "image/jpg":
	case "image/jpeg":
		type = "image";
		break;

	default:
		return;
	}
	client.emit("toggle", {
		id: id,
		type: type,
		head: head,
		body: body,
		link: link,
	});
}

function fetch(url, cb) {
	var req = request.get(url);
	req.end(function(e, res) {
		if (res) {
			cb(res);
		}
	});
}