const Command = require('../../structures/Command'); const { MessageEmbed } = require('discord.js'); const request = require('node-superfetch'); const ytdl = require('ytdl-core'); const { shorten, verify } = require('../../util/Util'); const { GOOGLE_KEY } = process.env; module.exports = class PlayCommand extends Command { constructor(client) { super(client, { name: 'play', aliases: ['play-music', 'music'], group: 'voice', memberName: 'play', description: 'Plays a YouTube video in your voice channel.', guildOnly: true, throttling: { usages: 2, duration: 60 }, userPermissions: ['CONNECT', 'SPEAK'], credit: [ { name: 'Google', url: 'https://www.google.com/', reason: 'YouTube Data API', reasonURL: 'https://developers.google.com/youtube/v3/' } ], args: [ { key: 'query', prompt: 'What video would you like to play? Either a URL or search query.', type: 'string' } ] }); } async run(msg, { query }) { const connection = this.client.voice.connections.get(msg.guild.id); if (!connection) { const usage = this.client.registry.commands.get('join').usage(); return msg.reply(`I am not in a voice channel. Use ${usage} to fix that!`); } if (this.client.dispatchers.has(msg.guild.id)) return msg.reply('I am already playing audio in this server.'); const result = await this.searchForVideo(query, msg.channel.nsfw || false); if (!result) return msg.reply('Could not find any results for your query.'); const data = await ytdl.getInfo(result); const canPlay = this.canUseVideo(data, msg.channel.nsfw || false); if (!canPlay) return msg.reply('I cannot play this video.'); if (canPlay === 'length') return msg.reply('This video is longer than 15 minutes, so I can\'t play it.'); await msg.reply('Is this the video you want to play? Type **[y]es** or **[n]o**.', { embed: this.generateEmbed(data) }); const verification = await verify(msg.channel, msg.author); if (!verification) return msg.reply('Aborting playback.'); const dispatcher = connection.play(ytdl(result, { filter: 'audioonly' })); this.client.dispatchers.set(msg.guild.id, dispatcher); dispatcher.once('finish', () => this.client.dispatchers.delete(msg.guild.id)); dispatcher.once('error', () => this.client.dispatchers.delete(msg.guild.id)); return msg.reply(`🔉 Now playing **${shorten(data.videoDetails.title, 70)}**!`); } async searchForVideo(query, nsfw) { if (ytdl.validateURL(query)) return ytdl.getURLVideoID(query); if (ytdl.validateID(query)) return query; const { body } = await request .get('https://www.googleapis.com/youtube/v3/search') .query({ part: 'snippet', type: 'video', maxResults: 1, q: query, safeSearch: nsfw ? 'none' : 'strict', key: GOOGLE_KEY }); if (!body.items.length) return null; const data = body.items[0]; return data.id.videoId; } canUseVideo(data, nsfw) { if (data.videoDetails.isPrivate || data.videoDetails.isLiveContent) return false; if (data.videoDetails.age_restricted && nsfw) return false; if (Number.parseInt(data.videoDetails.lengthSeconds, 10) > 900) return 'length'; return true; } generateEmbed(data) { return new MessageEmbed() .setColor(0xDD2825) .setTitle(shorten(data.videoDetails.title, 70)) .setDescription(shorten(data.videoDetails.description, 100)) .setAuthor('YouTube', 'https://i.imgur.com/kKHJg9Q.png', 'https://www.youtube.com/') .setURL(data.videoDetails.video_url) .setThumbnail(data.videoDetails.thumbnails.length ? data.videoDetails.thumbnails[0].url : null) .addField('❯ ID', data.videoDetails.videoId, true) .addField('❯ Publish Date', data.videoDetails.publishDate, true); } };