woomy/utils/music.js

const ytdl = require('ytdl-core-discord')
const fetch = require('node-fetch')
const { MessageEmbed } = require('discord.js')
const { utc } = require('moment')

module.exports = client => {
  client.music = { guilds: {} }

  client.createTimestamp = function (s) {
    if (s >= 3600) {
      return utc(s * 1000).format('HH:mm:ss')
    } else {
      return utc(s * 1000).format('mm:ss')
    }
  }

  client.music.getGuild = function (id) {
    let guild = client.music.guilds[id]

    if (!guild) {
      guild = {}

      guild.dispatcher = null
      guild.playing = false
      guild.queue = []

      client.music.guilds[id] = guild
    }

    return guild
  }

  client.music.getLinkFromID = function (id) {
    return 'https://www.youtube.com/watch?v=' + id
  }

  client.music.getVideoByQuery = async query => {
    let resp

    try {
      const id = await ytdl.getURLVideoID(query)
      resp = await fetch('https://invidious.snopyta.org/api/v1/videos/' + id)
    } catch (err) {
      resp = await fetch('https://invidious.snopyta.org/api/v1/search?q=' + encodeURIComponent(query))
    }

    const parsed = await resp.json()

    if (parsed) {
      const videos = parsed

      if (videos) {
        return videos
      } else {
        return false
      }
    } else {
      return false
    }
  }

  client.music.play = async function (message, query, ignoreQueue) {
    const guild = client.music.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 client.music.getVideoByQuery(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})** \`[${client.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(client.music.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) {
            client.music.play(message, null, true)
          } else {
            guild.dispatcher = null

            connection.disconnect()
          }
        })
      }
    } else {
      return message.reply('failed to find the video!')
    }
  }

  client.music.setVolume = function (guild, target) {
    const g = client.music.getGuild(guild.id)

    if (g.dispatcher) {
      g.dispatcher.setVolume(target)
    }
  }

  client.music.skip = function (guild, reason) {
    const g = client.music.getGuild(guild.id)

    if (g.dispatcher) {
      g.dispatcher.end(reason)
    }
  }
}