2020-04-30 04:06:15 +00:00
// Copyright 2020 Emily J. / mudkipscience and contributors. Subject to the AGPLv3 license.
2021-03-01 00:56:33 +00:00
const ytdl = require ( 'ytdl-core' )
2020-04-30 04:06:15 +00:00
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 = [ ]
2020-05-28 07:52:15 +00:00
guild . fixers = [ ]
2020-04-30 04:06:15 +00:00
exports . queue [ id ] = guild
}
return guild
}
exports . getLinkFromID = function ( id ) {
2020-06-06 09:24:12 +00:00
return 'https://www.youtube.com/watch?v=' + id
2020-04-30 04:06:15 +00:00
}
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
2020-06-06 07:57:46 +00:00
2020-05-09 06:48:25 +00:00
// message.channel.startTyping()
2020-04-30 04:06:15 +00:00
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 = [ ]
2020-05-28 07:52:15 +00:00
guild . fixers = [ ]
2021-03-18 02:44:03 +00:00
guild . channel = null
2020-04-30 04:06:15 +00:00
// music not playing, something is in queue
2020-06-06 07:57:46 +00:00
} else if ( ( ! guild . playing || ! guild . dispatcher ) && guild . queue . length > 0 ) {
2020-04-30 04:06:15 +00:00
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 {
2021-03-18 02:02:59 +00:00
guild . dispatcher = connection . play ( ytdl ( v . video . videoId , { type : 'opus' , bitrate : 'auto' } ) ) ;
2020-04-30 04:06:15 +00:00
} catch ( err ) {
if ( playNext && playNext === true ) {
guild . queue . splice ( 1 , 1 )
} else {
guild . queue . pop ( )
}
2021-03-01 00:56:33 +00:00
client . logger . error ( err . stack )
2020-05-30 03:55:04 +00:00
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.')
2020-04-30 04:06:15 +00:00
}
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
2020-06-06 08:14:56 +00:00
guild . dispatcher . on ( 'error' , ( err ) => {
console . error ( '[MUSIC ERROR] ' + String ( err ) ) ;
} ) ;
2020-06-06 08:07:27 +00:00
2020-04-30 04:06:15 +00:00
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 = [ ]
2020-05-28 07:52:15 +00:00
guild . fixers = [ ]
2021-03-18 02:44:03 +00:00
guild . channel = null
2020-04-30 04:06:15 +00:00
connection . disconnect ( )
}
} )
}
} else {
2020-05-07 10:49:40 +00:00
return message . channel . send ( 'failed to find the video!' )
2020-04-30 04:06:15 +00:00
}
}
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 )
}
2020-05-09 06:48:25 +00:00
}