woomy/utils/music.js

211 lines
5.3 KiB
JavaScript

// Copyright 2020 Emily J. / mudkipscience and contributors. Subject to the AGPLv3 license.
const ytdl = require('ytdl-core-discord')
const fetch = require('node-fetch')
const { MessageEmbed } = require('discord.js')
const { utc } = require('moment')
exports.queue = {}
exports.createTimestamp = function (s) {
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.dispatcher = null
guild.playing = false
guild.queue = []
exports.queue[id] = guild
}
return guild
}
exports.getLinkFromID = function (id) {
return 'https://www.youtube.com/watch?v=' + id
}
exports.getVideoByQuery = async function (client, query) {
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()
if (parsed) {
const videos = parsed
if (videos) {
return videos
} else {
return false
}
} else {
return false
}
}
exports.play = async function (client, message, query, ignoreQueue) {
const guild = exports.getGuild(message.guild.id)
if (!message.member.voice.channel && !guild.voiceChannel) {
return message.reply('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)
if (!videos[1]) {
if (!videos[0]) {
video = videos
} 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.playing = false
guild.dispatcher = null
// 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`
}
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.guild))
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('Invalid choice.')
}
break
case 3:
if (videos[2]) {
video = videos[2]
} else {
return message.channel.send('Invalid choice.')
}
break
case 4:
if (videos[3]) {
video = videos[3]
} else {
return message.channel.send('Invalid choice.')
}
break
case 5:
if (videos[4]) {
video = videos[4]
} else {
return message.channel.send('Invalid choice.')
}
break
default:
return message.channel.send('Invalid choice.')
}
}
if (!video && videos[0]) {
video = videos[0]
} else if (!video) {
video = videos
}
// Add video to queue
guild.queue.push({ video: video, requestedBy: message.member.id })
}
// Figure out if the bot should add it to queue or play it right now
if (guild.playing) {
message.reply('added **' + video.title + '** to the queue')
} else {
guild.playing = true
guild.voiceChannel = vc
const connection = await vc.join()
const v = guild.queue[0]
guild.dispatcher = connection.play(await ytdl(exports.getLinkFromID(v.video.videoId), { highWaterMark: 1024 * 1024 * 32 }), { type: 'opus' })
guild.dispatcher.setVolume(0.25)
message.channel.send('Playing **' + v.video.title + '**')
// play next in queue on end
guild.dispatcher.once('finish', () => {
guild.queue.shift()
guild.playing = false
if (guild.queue.length > 0) {
exports.play(message, null, true)
} else {
guild.dispatcher = null
connection.disconnect()
}
})
}
} else {
return message.reply('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)
}
}