forked from GitHub/dbot
97 lines
3.1 KiB
JavaScript
97 lines
3.1 KiB
JavaScript
/**
|
|
* Module Name: youtube
|
|
* Description: Various Youtube functionality
|
|
*/
|
|
var _ = require('underscore')._,
|
|
request = require('request');
|
|
|
|
var youtube = function(dbot) {
|
|
this.ApiRoot = 'https://gdata.youtube.com/feeds/api';
|
|
this.params = {
|
|
'alt': 'json',
|
|
'v': 2
|
|
};
|
|
this.LinkRegex = /^.*(youtu.be\/|v\/|u\/\w\/|embed\/|watch\?v=|\&v=)([^#\&\?]*).*/;
|
|
|
|
this.api = {
|
|
'search': function(term, callback) {
|
|
var qs = _.clone(this.params);
|
|
request.get(this.ApiRoot + '/videos', {
|
|
'qs': _.extend(qs, {
|
|
'q': term,
|
|
'max-results': 1
|
|
}),
|
|
'json': true
|
|
}, function(error, response, body) {
|
|
callback(body);
|
|
}.bind(this));
|
|
}
|
|
};
|
|
|
|
this.internalAPI = {
|
|
'formatLink': function(v) {
|
|
var seconds = v['media$group']['yt$duration'].seconds,
|
|
minutes = Math.floor(seconds / 60),
|
|
seconds = ((seconds%60 < 10) ? "0"+seconds%60 : seconds%60);
|
|
|
|
if(!_.has(v, 'yt$rating')) {
|
|
v['yt$rating'] = {
|
|
'numLikes': 0,
|
|
'numDislikes': 0
|
|
};
|
|
}
|
|
if(!_.has(v, 'yt$statistics')) {
|
|
v['yt$statistics'] = { 'viewCount': 0 };
|
|
}
|
|
|
|
var res = dbot.t('yt_video', {
|
|
'title': v.title['$t'],
|
|
'plays': v['yt$statistics'].viewCount.toString().replace(/(\d)(?=(\d\d\d)+(?!\d))/g, "$1,"),
|
|
'author': v.author[0].name['$t'],
|
|
'likes': v['yt$rating'].numLikes,
|
|
'dislikes': v['yt$rating'].numDislikes,
|
|
'minutes': minutes,
|
|
'seconds': seconds
|
|
});
|
|
|
|
var link = v.link[0].href.match(this.LinkRegex);
|
|
if(link) {
|
|
res += ' - http://youtu.be/' + link[2];
|
|
}
|
|
|
|
return res;
|
|
}.bind(this)
|
|
};
|
|
|
|
this.commands = {
|
|
'~youtube': function(event) {
|
|
this.api.search(event.input[1], function(body) {
|
|
if(_.isObject(body) && _.has(body, 'feed') && _.has(body.feed, 'entry')) {
|
|
event.reply(this.internalAPI.formatLink(body.feed.entry[0]));
|
|
} else {
|
|
event.reply(dbot.t('yt_noresults'));
|
|
}
|
|
}.bind(this));
|
|
}
|
|
};
|
|
this.commands['~youtube'].regex = [/^youtube (.+)$/, 2];
|
|
|
|
this.onLoad = function() {
|
|
dbot.api.link.addHandler(this.name, this.LinkRegex,
|
|
function(match, name, callback) {
|
|
request.get(this.ApiRoot + '/videos/' + match[2], {
|
|
'qs': this.params,
|
|
'json': true
|
|
}, function(error, response, body) {
|
|
if(_.isObject(body) && _.has(body, 'entry')) {
|
|
callback(this.internalAPI.formatLink(body.entry));
|
|
}
|
|
}.bind(this));
|
|
}.bind(this));
|
|
}.bind(this);
|
|
};
|
|
|
|
exports.fetch = function(dbot) {
|
|
return new youtube(dbot);
|
|
};
|