woomy/modules/music.js

const ytdl = require('ytdl-core-discord')
const youtubeInfo = require('youtube-info')
const getYoutubeId = require('get-youtube-id')
const fetch = require('node-fetch')

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

  client.music.isYoutubeLink = function (input) {
    return input.startsWith('https://www.youtube.com/') || input.startsWith('http://www.youtube.com/') || input.startsWith('https://youtube.com/') || input.startsWith('http://youtube.com/') || input.startsWith('https://youtu.be/') || input.startsWith('http://youtu.be/') || input.startsWith('http://m.youtube.com/') || input.startsWith('https://m.youtube.com/')
  }

  client.music.search = async function (query) {
    return new Promise(function (resolve, reject) {
      try {
        fetch('https://www.googleapis.com/youtube/v3/search?part=id&type=video&q=' + encodeURIComponent(query) + '&key=' + process.env.YTKEY)
          .then(res => res.json())
          .then(json => {
            if (!json.items) { reject() }
            resolve(json.items[0])
          })
      } catch (err) {
        client.logger.error('Music search err: ', err)
        throw err
      }
    })
  }

  client.music.getGuild = function (id) {
    if (client.music.guilds[id]) return client.music.guilds[id]
    return client.music.guilds[id] = {
      queue: [],
      playing: false,
      paused: false,
      dispatcher: null,
      skippers: []
    }
  }
  
  client.music.getMeta = async function(id)
  {
      return new Promise(function(resolve, reject)
      {
          youtubeInfo(id, function(err, videoInfo)
          {
              if(err) throw err
  
              resolve(videoInfo)
          })
      })
  }
  
  client.music.play = async function(message, input, bypassQueue)
  {
      let voiceChannel = message.member.voice.channel
      if(!voiceChannel) return message.channel.send('<:error:466995152976871434> You need to be in a voice channel to use this command!')

      let permissions = voiceChannel.permissionsFor(client.user)
      if (!permissions.has('CONNECT')) {
          return message.channel.send('<:error:466995152976871434> I do not have permission to join your voice channel.')
      }
      if (!permissions.has('SPEAK')) {
          return message.channel.send('<:error:466995152976871434> I do not have permission to join your voice channel.')
      }
      if (voiceChannel.joinable != true) {
          return message.channel.send("<:error:466995152976871434> I do not have permission to join your voice channel.")
      }

      let id = undefined

      if(client.music.isYoutubeLink(input))
      {
          id = await getYoutubeId(input)
      } else {
          let item = await client.music.search(input)
          if(!item) {
            return message.channel.send(`<:error:466995152976871434> No results found.`)
          }
          id = item.id.videoId
      }

      // music "playing", nothing in queue
      if((client.music.getGuild(message.guild.id).playing || client.music.getGuild(message.guild.id).dispatcher) && client.music.getGuild(message.guild.id).queue.length == 0) {
        client.music.getGuild(message.guild.id).playing = false;
        client.music.getGuild(message.guild.id).dispatcher = null;
      // music not playing, something is in queue
      } else if(!client.music.getGuild(message.guild.id).playing && !client.music.getGuild(message.guild.id).dispatcher && client.music.getGuild(message.guild.id).queue.length > 0) {
        client.music.getGuild(message.guild.id).queue = [];
      };

      if(client.music.getGuild(message.guild.id).queue.length == 0 || bypassQueue)
      {
          let meta = await client.music.getMeta(id)

          if(!bypassQueue) client.music.getGuild(message.guild.id).queue.push({input: input, id: id, requestedBy: message.author, title: meta.title, author: meta.owner, thumbnail: meta.thumbnailUrl, duration: meta.duration})

          let connection = await new Promise((resolve, reject) =>
          {
              voiceChannel.join().then((connection) =>
              {
                  resolve(connection)
              })
          })

          function end(silent)
          {
              client.music.getGuild(message.guild.id).queue.shift()
              client.music.getGuild(message.guild.id).dispatcher = null

              if(client.music.getGuild(message.guild.id).queue.length > 0)
              {
                  client.music.play(message, client.music.getGuild(message.guild.id).queue[0].input, true)
              } else {
                  client.music.getGuild(message.guild.id).playing = false

                  if(!silent) {
                    message.channel.send("<:play:467216788187512832> Queue is empty! Disconnecting from the voice channel.")
                  }
                  
                  connection.disconnect()
              }
          }

          client.music.getGuild(message.guild.id).playing = true

          let song = client.music.getGuild(message.guild.id).queue[0]

          try
          {
              let dispatcher = client.music.getGuild(message.guild.id).dispatcher = connection.play(await ytdl("https://www.youtube.com/watch?v=" + id, {highWaterMark: 1024 * 1024 * 32}), {type: 'opus'})

              dispatcher.on('finish', (a, b) =>
              {
                  end(a == "silent")
              })
          } catch(err) {
              message.channel.send('<:error:466995152976871434> Failed to play **' + song.title + '** ' + err)

              end()
          }
          
          client.music.getGuild(message.guild.id).skippers = []
          message.channel.send(`<:play:467216788187512832> Now playing: **${song.title}**`)
      } else {
          let meta = await client.music.getMeta(id)
          let song = {input: input, id: id, requestedBy: message.author, title: meta.title, author: meta.owner, thumbnail: meta.thumbnailUrl, duration: meta.duration}
          
          client.music.getGuild(message.guild.id).queue.push(song)

          message.channel.send(`<:success:466995111885144095> Added to queue: **${song.title}**`)
      }
  }

  // MUSIC - TIMESTAMP
  client.createTimestamp = function (duration) {
    var hrs = ~~(duration / 60 / 60)
    var min = ~~(duration / 60) % 60
    var sec = ~~(duration - min * 60)
  
    if (String(hrs).length < 2) {
      hrs = '0' + String(hrs) + ':'
    }
  
    if (String(min).length < 2) {
      min = '0' + String(min)
    }
  
    if (String(sec).length < 2) {
      sec = '0' + String(sec)
    }

    if (hrs === '00:') {
      hrs = ''
    }

    var time = hrs + min + ':' + sec
    return time
  }
}