dbot/modules/youtube/youtube.js

/**
 * 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 += ' - https://youtu.be/' + link[2];
            }

            return res;
        }.bind(this)
    };

    this.commands = {
        '~yt': 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['~yt'].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);
};