GitHub/dbot
3
0
Fork 1
mirror of https://github.com/reality/dbot.git synced 2024-11-27 06:19:24 +01:00
Code Issues Projects Releases Wiki Activity

xkcd title / relevent text search feature done right.

Browse Source
This commit is contained in:
Scritches 2018-04-26 19:12:20 -04:00
parent 0c8c377b08
commit bbfc4c18a6
1 changed files with 59 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

82
modules/link/link.js
View File

@ -9,15 +9,15 @@ var request = require('request'),
var link = function(dbot) { var link = function(dbot) {
this.urlRegex = /(\b(https?|ftp|file):\/\/[-A-Z0-9+&@#\/%?=~_|!:,.;]*[-A-Z0-9+&@#\/%=~_|])/ig; this.urlRegex = /(\b(https?|ftp|file):\/\/[-A-Z0-9+&@#\/%?=~_|!:,.;]*[-A-Z0-9+&@#\/%=~_|])/ig;
this.links = {}; this.links = {};
this.handlers = []; this.handlers = [];
this.api = { this.api = {
'addHandler': function(name, regex, handler) { 'addHandler': function(name, regex, handler) {
this.handlers.push({ this.handlers.push({
'name': name, 'name': name,
'regex': regex, 'regex': regex,
'callback': handler 'callback': handler
}); });
}, },
@ -43,8 +43,8 @@ var link = function(dbot) {
}, },
'udLookup': function(query, callback) { 'udLookup': function(query, callback) {
var reqUrl = 'http://api.urbandictionary.com/v0/define?term=' + var reqUrl = 'http://api.urbandictionary.com/v0/define?term=' +
encodeURI(query); encodeURI(query);
request(reqUrl, function(error, response, body) { request(reqUrl, function(error, response, body) {
try { try {
@ -62,14 +62,14 @@ var link = function(dbot) {
'parseLink': function(link, callback) { 'parseLink': function(link, callback) {
var handler = false; var handler = false;
for(var i=0;i<this.handlers.length;i++) { for(var i=0;i<this.handlers.length;i++) {
var matches = this.handlers[i].regex.exec(link); var matches = this.handlers[i].regex.exec(link);
if(matches) { if(matches) {
console.log(this.handlers[i].name); console.log(this.handlers[i].name);
handler = this.handlers[i]; handler = this.handlers[i];
break; break;
} }
} }
if(handler) { if(handler) {
this.handlers[i].callback(matches, this.handlers[i].name, function(parsed) { this.handlers[i].callback(matches, this.handlers[i].name, function(parsed) {
callback(parsed); callback(parsed);
@ -81,7 +81,7 @@ var link = function(dbot) {
} }
} }
}; };
var commands = { var commands = {
'~title': function(event) { '~title': function(event) {
var link = this.links[event.channel.name]; var link = this.links[event.channel.name];
@ -95,9 +95,10 @@ var link = function(dbot) {
event.reply(dbot.t('link', { 'link': title} )); event.reply(dbot.t('link', { 'link': title} ));
}); });
}, },
'~xkcd': function(event) { '~xkcd': function(event) {
var comicId = event.params[1] || ""; //var comicId = event.params[1] || "";
var comicId = event.params.slice(1).join(' ');
if(comicId == "*") { if(comicId == "*") {
request("http://xkcd.com/info.0.json", function(error, response, body){ request("http://xkcd.com/info.0.json", function(error, response, body){
@ -108,18 +109,48 @@ var link = function(dbot) {
dbot.commands['~xkcd'](event); dbot.commands['~xkcd'](event);
} }
} catch(err) { }; } catch(err) { };
}); });
} else { } else {
if (isNaN(parseInt(comicId))) { if (isNaN(parseInt(comicId))) {
var relevantUrl = 'https://relevantxkcd.appspot.com/process';
request({ request({
url: relevantUrl, url: 'http://www.explainxkcd.com/wiki/api.php',
qs: { qs: {
action:'xkcd', action: 'query',
query: comicId format: 'json',
} generator: 'search',
gsrwhat: 'text',
gsrsearch: comicId,
prop: 'info|categories',
gsrlimit: 50
},
json: true
}, function(err, res, body) { }, function(err, res, body) {
comicId = body.split(' ').slice(2)[0].trim(); if(!body) {
event.reply(dbot.t("no-hits"));
return;
}
var pages = _.values(body.query.pages);
// page titles must be of the format "####: $$$$$$"
pages = _.filter(pages, p => p.title.indexOf(':') > 0);
if (pages.length > 0) {
// See if any of these matches are exact title matches
var match = false;
_.each(pages, function(p) {
var title = p.title.slice(p.title.indexOf(':')+2).trim();
if(title.toLowerCase() == comicId.toLowerCase()) {
match = p;
}
});
if (match) {
// We got a match! Get the ID and let's get tf out of here.
comicId = match.title.slice(0, match.title.indexOf(':'));
} else {
comicId = pages[0].title.slice(0, pages[0].title.indexOf(':'));
}
var link = "http://xkcd.com/"+comicId+"/info.0.json"; var link = "http://xkcd.com/"+comicId+"/info.0.json";
request(link, function(error, response, body) { request(link, function(error, response, body) {
@ -132,6 +163,11 @@ var link = function(dbot) {
} }
} catch(err) { }; } catch(err) { };
}); });
} else {
event.reply(dbot.t("no-hits"));
}
}); });
} else { } else {
if(comicId !== "") { if(comicId !== "") {
@ -152,7 +188,7 @@ var link = function(dbot) {
} }
} }
}, },
'~ud': function(event) { '~ud': function(event) {
var query = event.input[1]; var query = event.input[1];
@ -175,7 +211,7 @@ var link = function(dbot) {
console.log('DEBUG: got a link'); console.log('DEBUG: got a link');
if(this.config.autoTitle == true) { if(this.config.autoTitle == true) {
this.api.parseLink(urlMatches[0], function(result) { this.api.parseLink(urlMatches[0], function(result) {
event.reply(result); event.reply(result);
}); });
} }
} }