xiao/commands/games-sp/guess-song.js

const Command = require('../../framework/Command');
const request = require('node-superfetch');
const cheerio = require('cheerio');
const { Readable } = require('stream');
const { reactIfAble, base64, list } = require('../../util/Util');
const playlists = require('../../assets/json/guess-song');
const demaster = /(\s(\(|-\s+))((199\d|20[0-2]\d)\s+)?(Remast|Live|Mono|From|Feat|Original|Motion|Deluxe).*/i;
const { SPOTIFY_KEY, SPOTIFY_SECRET } = process.env;

module.exports = class GuessSongCommand extends Command {
	constructor(client) {
		super(client, {
			name: 'guess-song',
			aliases: ['song-guess', 'song-game', 'music-guess', 'guess-music', 'music-game'],
			group: 'games-sp',
			memberName: 'guess-song',
			description: 'Guess what song is playing.',
			throttling: {
				usages: 2,
				duration: 15
			},
			guildOnly: true,
			userPermissions: ['CONNECT', 'SPEAK'],
			clientPermissions: ['ATTACH_FILES'],
			game: true,
			credit: [
				{
					name: 'Spotify',
					url: 'https://www.spotify.com/us/',
					reason: 'API',
					reasonURL: 'https://developer.spotify.com/'
				}
			],
			args: [
				{
					key: 'chart',
					type: 'string',
					oneOf: Object.keys(playlists),
					parse: chart => chart.toLowerCase()
				}
			]
		});

		this.token = null;
		this.charts = new Map();
		this.cache = new Map();
	}

	async run(msg, { chart }) {
		const connection = this.client.dispatchers.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 (!connection.canPlay) return msg.reply('I am already playing audio in this server.');
		if (!this.token) await this.fetchToken();
		const data = await this.fetchRandomSong(chart);
		const { body: previewBody } = await request.get(data.preview);
		connection.play(Readable.from([previewBody]));
		await reactIfAble(msg, this.client.user, '🔉');
		await msg.reply('**You have 30 seconds, what song is this?**');
		const msgs = await msg.channel.awaitMessages({
			filter: res => res.author.id === msg.author.id,
			max: 1,
			time: 30000
		});
		connection.stop();
		if (!msgs.size) return msg.reply(`Time! It's **${data.name}** by **${data.artist}**!`);
		const guess = msgs.first().content.toLowerCase();
		if (!guess.includes(data.name.toLowerCase()) && !guess.includes(data.shortName.toLowerCase())) {
			return msg.reply(`Nope! It's **${data.name}** by **${data.artist}**!`);
		}
		return msg.reply(`Nice! It's **${data.name}** by **${data.artist}**!`);
	}

	async fetchCharts(playlist) {
		if (this.charts.has(playlist)) return this.charts.get(playlist);
		const { body } = await request
			.get(`https://api.spotify.com/v1/playlists/${playlists[playlist]}/tracks`)
			.set({ Authorization: `Bearer ${this.token}` })
			.query({
				market: 'US',
				fields: 'items(track(id))',
				limit: 100
			});
		const songList = body.items.map(item => item.track.id);
		this.charts.set(playlist, songList);
		setTimeout(() => this.charts.delete(playlist), 4.32e+7);
		return songList;
	}

	async fetchRandomSong(playlist) {
		const songs = await this.fetchCharts(playlist);
		const choice = songs[Math.floor(Math.random() * songs.length)];
		return this.fetchSongPreview(choice);
	}

	async fetchSongPreview(id) {
		if (this.cache.has(id)) return this.cache.get(id);
		const { body } = await request
			.get(`https://api.spotify.com/v1/tracks/${id}`)
			.set({ Authorization: `Bearer ${this.token}` })
			.query({ market: 'US' });
		let previewURL = body.preview_url;
		if (!body.preview_url) previewURL = await this.fetchAlternativePreview(id);
		const result = {
			id,
			name: body.name,
			shortName: body.name.replace(demaster, ''),
			artist: list(body.artists.map(artist => artist.name)),
			preview: previewURL
		};
		this.cache.set(id, result);
		return result;
	}

	async fetchAlternativePreview(id) {
		const { text } = await request.get(`https://open.spotify.com/embed/track/${id}`);
		const $ = cheerio.load(text);
		const body = JSON.parse(decodeURIComponent($('script[id="__NEXT_DATA__"]')[0].children[0].data));
		return body.props.pageProps.state.data.entity.audioPreview.url;
	}

	async fetchToken() {
		const { body } = await request
			.post('https://accounts.spotify.com/api/token')
			.set({
				Authorization: `Basic ${base64(`${SPOTIFY_KEY}:${SPOTIFY_SECRET}`)}`,
				'Content-Type': 'application/x-www-form-urlencoded'
			})
			.send('grant_type=client_credentials');
		this.token = body.access_token;
		setTimeout(() => { this.token = null; }, body.expires_in * 1000);
		return body;
	}
};