From 280cbaed924c4cedf24ac26d4cd6ddfa08e3ba0e Mon Sep 17 00:00:00 2001
From: Carsten Burgard <carsten.burgard@gmail.com>
Date: Fri, 15 Apr 2022 15:41:50 +0200
Subject: [PATCH] added basic functionality for looping queues

---
 PlexBot/bot.py | 54 +++++++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 51 insertions(+), 3 deletions(-)

diff --git a/PlexBot/bot.py b/PlexBot/bot.py
index 3401909..f4cc765 100644
--- a/PlexBot/bot.py
+++ b/PlexBot/bot.py
@@ -37,7 +37,9 @@ Plex:
     q - Print the current queue (This can take very long!)
     stop - Halt playback and leave vc.
     loop - Loop the current song.
-    unloop - Disable looping.
+    loopq - Loop the current queue.
+    unloop - Disable looping the current track.
+    unloopq - Disable looping the current queue.
     pause - Pause playback.
     resume - Resume playback.
     skip - Skip the current song. Give a number as argument to skip more than 1.
@@ -206,6 +208,7 @@ class Plex(commands.Cog):
         self.voice_channel = None
         self.current_track = None
         self.is_looping = False
+        self.loop_queue = None
         self.np_message_id = None
         self.show_queue_message_ids = []        
         self.ctx = None
@@ -322,6 +325,14 @@ class Plex(commands.Cog):
                 self.current_track = await self.play_queue.get()
             except asyncio.exceptions.CancelledError:
                 bot_log.debug("failed to pop queue")
+                
+            if not self.current_track and self.loop_queue:
+                try:                
+                    self.current_track = await self.loop_queue.get()
+                except asyncio.exceptions.CancelledError:
+                    bot_log.debug("failed to pop queue")                    
+                self.play_queue = self.loop_queue
+                self.loop_queue = asyncio.Queue()
 
             
     async def _audio_player_task(self):
@@ -377,6 +388,8 @@ class Plex(commands.Cog):
         Raises:
             None
         """
+        if self.loop_queue:
+            self.bot.loop.call_soon_threadsafe(self.loop_queue.put,self.current_track)
         self.current_track = None
         self.bot.loop.call_soon_threadsafe(self.play_next_event.set)
 
@@ -776,6 +789,23 @@ class Plex(commands.Cog):
         bot_log.debug("Looping "+str(self.current_track))        
         self.is_looping = self.current_track
 
+    @command()
+    async def loopq(self, ctx):
+        """
+        User command to activate looping the current queue
+
+        Args:
+            ctx: discord.ext.commands.Context message context from command
+
+        Returns:
+            None
+
+        Raises:
+            None
+        """
+        bot_log.debug("Looping current queue")
+        self.loop_queue = asyncio.Queue()
+
     @command()
     async def unloop(self, ctx):
         """
@@ -790,8 +820,25 @@ class Plex(commands.Cog):
         Raises:
             None
         """
-        bot_log.debug("Unlooping")                
-        self.is_looping = False        
+        bot_log.debug("Looping current queue")        
+        self.is_looping = False
+
+    @command()
+    async def unloopq(self, ctx):
+        """
+        User command to deactivate looping the current queue
+
+        Args:
+            ctx: discord.ext.commands.Context message context from command
+
+        Returns:
+            None
+
+        Raises:
+            None
+        """
+        bot_log.debug("Unlooping")
+        self.loop_queue = None
             
 
     @command()
@@ -942,6 +989,7 @@ class Plex(commands.Cog):
             None
         """
         self.play_queue = asyncio.Queue()
+        self.loop_queue = None
         bot_log.debug("Cleared queue")
         await ctx.send(":boom: Queue cleared.")