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

137 lines
2.6 KiB
JavaScript
Raw Normal View History

2014-09-27 21:17:05 +02:00
var _ = require("lodash");
2014-09-27 22:39:14 +02:00
var cheerio = require("cheerio");
2014-09-27 21:17:05 +02:00
var Msg = require("../../models/msg");
2014-12-23 02:06:11 +01:00
var request = require("request");
2014-10-10 23:12:01 +02:00
var Helper = require("../../helper");
2015-10-01 00:39:57 +02:00
var es = require("event-stream");
2014-09-27 21:17:05 +02:00
2015-04-29 21:55:13 +02:00
process.setMaxListeners(0);
2014-09-27 21:17:05 +02:00
module.exports = function(irc, network) {
var client = this;
irc.on("message", function(data) {
2014-10-10 23:12:01 +02:00
var config = Helper.getConfig();
if (!config.prefetch) {
return;
}
2014-12-12 00:48:43 +01:00
2014-09-27 21:17:05 +02:00
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;
}
2015-10-01 00:39:57 +02:00
var self = data.to.toLowerCase() === irc.me.toLowerCase();
2014-09-27 21:17:05 +02:00
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
});
2014-12-12 00:48:43 +01:00
var link = links[0];
fetch(link, function(res) {
parse(msg, link, res, client);
2014-09-27 21:17:05 +02:00
});
});
};
2014-09-28 01:47:04 +02:00
function parse(msg, url, res, client) {
var toggle = msg.toggle = {
id: msg.id,
type: "",
head: "",
body: "",
thumb: "",
2014-09-28 01:47:04 +02:00
link: url
};
2014-09-27 21:17:05 +02:00
switch (res.type) {
case "text/html":
var $ = cheerio.load(res.text);
2014-09-28 01:47:04 +02:00
toggle.type = "link";
toggle.head = $("title").text();
2014-10-14 20:51:27 +02:00
toggle.body =
2015-10-01 00:39:57 +02:00
$("meta[name=description]").attr("content")
|| $("meta[property=\"og:description\"]").attr("content")
2014-10-14 20:51:27 +02:00
|| "No description found.";
2014-12-12 00:48:43 +01:00
toggle.thumb =
2015-10-01 00:39:57 +02:00
$("meta[property=\"og:image\"]").attr("content")
|| $("meta[name=\"twitter:image:src\"]").attr("content")
2014-10-14 20:51:27 +02:00
|| "";
2014-09-27 21:17:05 +02:00
break;
case "image/png":
case "image/gif":
case "image/jpg":
case "image/jpeg":
2014-09-28 01:47:04 +02:00
toggle.type = "image";
2014-09-27 21:17:05 +02:00
break;
default:
return;
}
2014-09-28 01:47:04 +02:00
client.emit("toggle", toggle);
2014-09-27 21:17:05 +02:00
}
function fetch(url, cb) {
2015-01-04 03:58:12 +01:00
try {
var req = request.get(url);
2015-10-01 00:39:57 +02:00
} catch (e) {
2015-01-04 03:58:12 +01:00
return;
}
var length = 0;
2014-12-23 02:06:11 +01:00
var limit = 1024 * 10;
req
2015-10-01 00:39:57 +02:00
.on("response", function(res) {
if (!(/(text\/html|application\/json)/.test(res.headers["content-type"]))) {
res.req.abort();
2014-12-23 02:06:11 +01:00
}
})
2015-10-01 00:39:57 +02:00
.on("error", function() {})
2014-12-23 02:06:11 +01:00
.pipe(es.map(function(data, next) {
length += data.length;
if (length > limit) {
req.response.req.abort();
}
next(null, data);
}))
.pipe(es.wait(function(err, data) {
if (err) return;
var body;
var type;
2014-12-23 02:06:11 +01:00
try {
body = JSON.parse(data);
2015-10-01 00:39:57 +02:00
} catch (e) {
2014-12-23 02:06:11 +01:00
body = {};
}
try {
2015-10-01 00:39:57 +02:00
type = req.response.headers["content-type"].split(/ *; */).shift();
} catch (e) {
type = {};
}
2014-12-23 02:06:11 +01:00
data = {
text: data,
body: body,
type: type
2014-12-23 02:06:11 +01:00
};
cb(data);
}));
2014-09-27 21:17:05 +02:00
}