Add /loop-queue command (#989)

1 files changed, 12 insertions, 0 deletions

diff --git a/src/services/player.ts b/src/services/player.ts
index 9754a42..c17a536 100644
--- a/src/services/player.ts
+++ b/src/services/player.ts
@@ -63,6 +63,7 @@ export default class {
   public status = STATUS.PAUSED;
   public guildId: string;
   public loopCurrentSong = false;
+  public loopCurrentQueue = false;
 
   private queue: QueuedSong[] = [];
   private queuePosition = 0;
@@ -545,6 +546,17 @@ export default class {
       return;
     }
 
+    // Automatically re-add current song to queue
+    if (this.loopCurrentQueue && newState.status === AudioPlayerStatus.Idle && this.status === STATUS.PLAYING) {
+      const currentSong = this.getCurrent();
+
+      if (currentSong) {
+        this.add(currentSong);
+      } else {
+        throw new Error('No song currently playing.');
+      }
+    }
+
     if (newState.status === AudioPlayerStatus.Idle && this.status === STATUS.PLAYING) {
       await this.forward(1);
     }