Move Play to voice group

2026-06-26 22:32:52 +02:00 · 2021-03-02 20:33:58 -05:00
parent 1f987befa6
commit 375dbf9602
4 changed files with 3 additions and 7 deletions
@@ -0,0 +1,99 @@
+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: 1,
+				duration: 10
+			},
+			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.say('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.say('I cannot play this video.');
+		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.say('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;
+		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);
+	}
+};