forked from GitHub/dbot
114 lines
4.3 KiB
JavaScript
114 lines
4.3 KiB
JavaScript
/**
|
|
* Module Name: imgur
|
|
* Description: Various imgur functionality
|
|
*/
|
|
|
|
var _ = require('underscore')._,
|
|
request = require('request');
|
|
|
|
var imgur = function(dbot) {
|
|
this.internalAPI = {
|
|
'infoString': function(imgData) {
|
|
info = '';
|
|
if(imgData && _.has(imgData, 'data')) {
|
|
imgData = imgData.data;
|
|
if(imgData.title) {
|
|
info += imgData.title + ' - ';
|
|
}
|
|
if(imgData.type) {
|
|
if(imgData.animated) {
|
|
info += 'an animated ' + imgData.type.split('/')[1] + ' with ';
|
|
} else {
|
|
info += 'a ' + imgData.type.split('/')[1] + ' with ';
|
|
}
|
|
} else {
|
|
info += 'an image with ';
|
|
}
|
|
info += imgData.views + ' views (';
|
|
info += imgData.width + 'x' + imgData.height + ')';
|
|
}
|
|
|
|
return info;
|
|
}.bind(this)
|
|
};
|
|
|
|
this.api = {
|
|
'getRandomImage': function(callback) {
|
|
var random = function(len) {
|
|
var chars = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXTZabcdefghiklmnopqrstuvwxyz";
|
|
return len ? chars.charAt(~~(Math.random()*chars.length)) + random(len-1) : "";
|
|
};
|
|
|
|
var ext = [ 'gif', 'png', 'jpg' ];
|
|
var testSlug = random(5);
|
|
var testUrl = 'http://i.imgur.com/' +
|
|
testSlug +
|
|
'.' + ext[_.random(0, ext.length - 1)];
|
|
dbot.db.imgur.totalHttpRequests += 1;
|
|
var image = request(testUrl, function(error, response, body) {
|
|
// 492 is body.length of a removed image
|
|
if(!error && response.statusCode == 200 && body.length != 492) {
|
|
dbot.db.imgur.totalImages += 1;
|
|
callback(testUrl, testSlug);
|
|
} else {
|
|
this.api.getRandomImage(callback);
|
|
}
|
|
}.bind(this));
|
|
},
|
|
|
|
'getImageInfoString': function(slug, callback) {
|
|
this.api.getImageInfo(slug, function(imgData) {
|
|
callback(this.internalAPI.infoString(imgData));
|
|
}.bind(this));
|
|
},
|
|
|
|
'getImageInfo': function(slug, callback) {
|
|
request.get({
|
|
'url': 'https://api.imgur.com/3/image/' + slug + '.json',
|
|
'json': true,
|
|
'headers': {
|
|
'Authorization': 'Client-ID ' + dbot.config.imgur.apikey
|
|
}
|
|
}, function(err, response, body) {
|
|
dbot.db.imgur.totalApiRequests += 1;
|
|
callback(body);
|
|
}.bind(this));
|
|
}
|
|
};
|
|
this.api['getRandomImage'].external = true;
|
|
this.api['getRandomImage'].extMap = [ 'callback' ];
|
|
this.api['getImageInfoString'].external = true;
|
|
this.api['getImageInfoString'].extMap = [ 'slug', 'callback' ];
|
|
|
|
this.commands = {
|
|
'~ri': function(event) {
|
|
this.api.getRandomImage(function(link, slug) {
|
|
this.api.getImageInfo(slug, function(imgData) {
|
|
var info = this.internalAPI.infoString(imgData);
|
|
event.reply(event.user + ': ' + link + ' [' + info + ']');
|
|
}.bind(this));
|
|
}.bind(this));
|
|
}
|
|
}
|
|
|
|
this.onLoad = function() {
|
|
var imgurHandler = function(event, matches, name) {
|
|
if(matches[1]) {
|
|
this.api.getImageInfo(matches[1], function(imgData) {
|
|
var info = this.internalAPI.infoString(imgData);
|
|
if(info) event.reply('[' + info + ']');
|
|
}.bind(this));
|
|
}
|
|
}.bind(this);
|
|
dbot.api.link.addHandler(this.name, /https?:\/\/i\.imgur\.com\/([a-zA-Z0-9]+)\.([jpg|png|gif])/, imgurHandler);
|
|
dbot.api.link.addHandler(this.name, /https?:\/\/imgur\.com\/([a-zA-Z0-9]+)/, imgurHandler);
|
|
if(!_.has(dbot.db.imgur, 'totalHttpRequests')) dbot.db.imgur.totalHttpRequests = 0;
|
|
if(!_.has(dbot.db.imgur, 'totalApiRequests')) dbot.db.imgur.totalApiRequests = 0;
|
|
if(!_.has(dbot.db.imgur, 'totalImages')) dbot.db.imgur.totalImages = 0;
|
|
}.bind(this);
|
|
};
|
|
|
|
exports.fetch = function(dbot) {
|
|
return new imgur(dbot);
|
|
}
|