From 035737312360279efd5026b225f30ef14aa55abf Mon Sep 17 00:00:00 2001
From: Max Isom <code@maxisom.me>
Date: Wed, 18 Mar 2020 19:17:47 -0500
Subject: Limit playlist adds to 50 songs

---
 src/commands/play.ts | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

(limited to 'src/commands')

diff --git a/src/commands/play.ts b/src/commands/play.ts
index bcafdbf..eceac94 100644
--- a/src/commands/play.ts
+++ b/src/commands/play.ts
@@ -71,6 +71,7 @@ export default class implements Command {
     }
 
     const newSongs: QueuedSong[] = [];
+    let extraMsg = '';
 
     // Test if it's a complete URL
     try {
@@ -95,7 +96,15 @@ export default class implements Command {
           }
         }
       } else if (url.protocol === 'spotify:' || url.host === 'open.spotify.com') {
-        const [convertedSongs] = await this.getSongs.spotifySource(args[0]);
+        const [convertedSongs, nSongsNotFound, totalSongs] = await this.getSongs.spotifySource(args[0]);
+
+        if (totalSongs > 50) {
+          extraMsg = 'a random sample of 50 songs was taken';
+        }
+
+        if (nSongsNotFound !== 0) {
+          extraMsg += `and ${nSongsNotFound.toString()} songs were not found`;
+        }
 
         newSongs.push(...convertedSongs);
       }
@@ -122,10 +131,14 @@ export default class implements Command {
 
     const firstSong = newSongs[0];
 
+    if (extraMsg !== '') {
+      extraMsg = ` (${extraMsg})`;
+    }
+
     if (newSongs.length === 1) {
-      await res.stop(`u betcha, **${firstSong.title}** added to the queue`);
+      await res.stop(`u betcha, **${firstSong.title}** added to the queue${extraMsg}`);
     } else {
-      await res.stop(`u betcha, **${firstSong.title}** and ${newSongs.length - 1} other songs were added to the queue`);
+      await res.stop(`u betcha, **${firstSong.title}** and ${newSongs.length - 1} other songs were added to the queue${extraMsg}`);
     }
 
     if (this.playerManager.get(msg.guild!.id).voiceConnection === null) {
-- 
cgit v1.2.3