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; } };