forked from embee/woomy
new music module
This commit is contained in:
parent
d1acf688c9
commit
6d60845546
1 changed files with 246 additions and 0 deletions
246
src/modules/music.js
Normal file
246
src/modules/music.js
Normal file
|
|
@ -0,0 +1,246 @@
|
|||
// 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 < 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 = []
|
||||
|
||||
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 = []
|
||||
// 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(await ytdl(exports.getLinkFromID(v.video.videoId), { highWaterMark: 1024 * 1024 * 32 }), { type: 'opus' })
|
||||
} catch (err) {
|
||||
if (playNext && playNext === true) {
|
||||
guild.queue.splice(1, 1)
|
||||
} else {
|
||||
guild.queue.pop()
|
||||
}
|
||||
|
||||
client.logger.error(err)
|
||||
return message.channel.send(`<:error:466995152976871434> An error has occured! If this issue persists, please contact my developers with this:\n\`${err}\``)
|
||||
}
|
||||
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.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 = []
|
||||
|
||||
connection.disconnect()
|
||||
}
|
||||
})
|
||||
}
|
||||
} else {
|
||||
return message.channel.channelsend('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)
|
||||
}
|
||||
}
|
||||
Loading…
Reference in a new issue