discord-dj/lib/Audio/Playlist/YoutubePlaylist.js

Discord DJ Bot. Let you play music in your server. Inspired by PlugDJ

"use strict";

var https = require('https');

var Playlist = require('../../Logic/Playlist.js');
var YoutubeVideo = require('../YoutubeVideo.js');
var Utils = require('../../Logic/Utils.js');

class YoutubePlaylist extends Playlist {

    constructor(url, key) {
        super();
        this.lastIndex = -1;
        this.key = key;
        this.id = Utils.YTPlaylistRegex.exec(url);
        if(this.id == null) throw new Error("Invalid Playlist URL");
        this.id = this.id[1];
        this.videos = [];
        this.loadPage(null);
    }

    loadPage(token) {
        var url = 'https://www.googleapis.com/youtube/v3/playlistItems?part=contentDetails&maxResults=50';
        url += '&playlistId=' + this.id + '&key=' + this.key;
        if(token != null) url += '&pageToken=' + token;
        https.get(url, function(res) {
            var body = '';
            res.on('data', function(d) {
                body += d;
            });
            res.on('end', function() {
                var parsed = JSON.parse(body);
                if(typeof parsed.error != 'undefined') {
                    this.emit('error', new Error(parsed.error.message));
                    return;
                }
                parsed.items.forEach(function(item) {
                    this.videos.push(item.contentDetails.videoId);
                }.bind(this));
                this.emit('update');
                if(typeof parsed.nextPageToken == 'string') {
                    this.loadPage(parsed.nextPageToken);
                }
            }.bind(this));
        }.bind(this)).on('error', function(error) {
            this.emit('error', error);
        });
    }

    shuffle() {
        this.shuffleArray(this.videos);
    }

    getNextSong() {
        if(this.videos.length == 0) return null;
        this.lastIndex++;
        if(this.lastIndex >= this.videos.length) {
            this.lastIndex = 0;
        }
        return new YoutubeVideo('http://www.youtube.com/watch?v=' + this.videos[this.lastIndex]);
    }

    hasNextSong() {
        return this.lastIndex < this.videos.length;
    }

    getLength() {
        return this.videos.length;
    }

}

module.exports = YoutubePlaylist;