// Copyright 2020 Emily J. / mudkipscience and contributors. Subject to the AGPLv3 license. const ytdl = require('ytdl-core') const fetch = require('node-fetch') const { MessageEmbed } = require('discord.js') const { utc } = require('moment') exports.queue = {} exports.createTimestamp = function (s) { if (s < 1) { return 'LIVE' } else if (s >= 3600) { return utc(s * 1000).format('HH:mm:ss') } else { return utc(s * 1000).format('mm:ss') } } exports.getGuild = function (id) { let guild = exports.queue[id] if (!guild) { guild = {} guild.queue = [] guild.playing = false guild.paused = false guild.dispatcher = null guild.skippers = [] guild.fixers = [] exports.queue[id] = guild } return guild } exports.getLinkFromID = function (id) { return 'https://www.youtube.com/watch?v=' + id } exports.getVideoByQuery = async function (client, query, message) { let res try { const id = await ytdl.getURLVideoID(query) res = await fetch(`${client.config.endpoints.invidious}v1/videos/${id}`) } catch (err) { res = await fetch(`${client.config.endpoints.invidious}v1/search?q=${encodeURIComponent(query)}`) } const parsed = await res.json().catch(function (e) { return message.channel.send('<:error:466995152976871434> An error has occured: ' + e) }) if (parsed) { const videos = parsed if (videos) { return videos } else { return false } } else { return false } } exports.play = async function (client, message, query, playNext, ignoreQueue) { const guild = exports.getGuild(message.guild.id) guild.message = message // message.channel.startTyping() if (!message.member.voice.channel && !guild.voiceChannel) { message.channel.stopTyping() return message.channel.send('<:error:466995152976871434> You have to be connected to a voice channel to use this command!') } const vc = message.member.voice.channel let video let videos if (!ignoreQueue) { videos = await exports.getVideoByQuery(client, query, message) if (!videos[1]) { if (!videos[0]) { video = videos message.channel.stopTyping() } else { video = videos[0] } } } if (videos || ignoreQueue) { if (!ignoreQueue) { // Fix the bot if somehow broken // music "playing", nothing in queue if ((guild.playing || guild.dispatcher) && guild.queue.length === 0) { guild.queue = [] guild.playing = false guild.paused = false guild.skippers = [] guild.fixers = [] // music not playing, something is in queue } else if ((!guild.playing || !guild.dispatcher) && guild.queue.length > 0) { guild.queue = [] } if (!video) { let output = '' let i = 0 for (i = 0; i < 5; i++) { if (!videos[i]) break output += `\`${i + 1}:\` **[${videos[i].title}](https://www.youtube.com/watch?v=${videos[i].videoId})** \`[${exports.createTimestamp(videos[i].lengthSeconds)}]\`\n` } message.channel.stopTyping() const embed = new MessageEmbed() embed.setTitle('Please reply with a number `1-' + i + '` to select which song you want to add to the queue.') embed.setColor(client.embedColour(message)) embed.setDescription(output) let selection = await client.awaitReply(message, embed) selection = Number(selection) switch (selection) { case 1: video = videos[0] break case 2: if (videos[1]) { video = videos[1] } else { return message.channel.send('<:error:466995152976871434> Invalid choice.') } break case 3: if (videos[2]) { video = videos[2] } else { return message.channel.send('<:error:466995152976871434> Invalid choice.') } break case 4: if (videos[3]) { video = videos[3] } else { return message.channel.send('<:error:466995152976871434> Invalid choice.') } break case 5: if (videos[4]) { video = videos[4] } else { return message.channel.send('<:error:466995152976871434> Invalid choice.') } break default: return message.channel.send('<:error:466995152976871434> Invalid choice.') } } if (!video && videos[0]) { video = videos[0] } else if (!video) { video = videos } // Add video to queue if (playNext === true) { guild.queue.splice(1, 0, { video: video, requestedBy: message.author }) } else { guild.queue.push({ video: video, requestedBy: message.author }) } } // Figure out if the bot should add it to queue or play it right now if (guild.playing) { message.channel.send('<:success:466995111885144095> Queued **' + video.title + '** `[' + exports.createTimestamp(video.lengthSeconds) + ']`') } else { guild.playing = true guild.voiceChannel = vc if (!guild.channel) { guild.channel = message.channel } const connection = await vc.join() const v = guild.queue[0] try { guild.dispatcher = connection.play(ytdl(v.video.videoId, { type: 'opus', bitrate: 'auto' })); } catch (err) { if (playNext && playNext === true) { guild.queue.splice(1, 1) } else { guild.queue.pop() } client.logger.error(err.stack) return message.channel.send(`<:error:466995152976871434> An error has occured: \n\`${err}\``) // return message.channel.send('<:error:466995152976871434> YouTube have made changes to their site that break Woomy\'s music module. An announcement will be made in the development server when this issue is resolved.') } guild.dispatcher.setVolume(0.25) guild.channel.send('<:player:467216674622537748> Now playing: **' + v.video.title + '** `[' + exports.createTimestamp(v.video.lengthSeconds) + ']`') // play next in queue on end guild.dispatcher.on('error', (err) => { console.error('[MUSIC ERROR] ' + String(err)); }); guild.dispatcher.once('finish', () => { guild.queue.shift() guild.playing = false if (guild.queue.length > 0) { exports.play(client, message, null, false, true) } else { guild.queue = [] guild.playing = false guild.paused = false guild.skippers = [] guild.fixers = [] connection.disconnect() } }) } } else { return message.channel.send('failed to find the video!') } } exports.setVolume = function (guild, target) { const g = exports.getGuild(guild.id) if (g.dispatcher) { g.dispatcher.setVolume(target) } } exports.skip = function (guild, reason) { const g = exports.getGuild(guild.id) if (g.dispatcher) { g.dispatcher.end(reason) } }