Add Vibe Mode

SwagLyricsBot/general_commands.py

@@ -2,7 +2,7 @@
 
 import discord
 import re
-from discord.ext import commands
+from discord.ext import commands, tasks
 
 from SwagLyricsBot import SpotifyClosed, LyricsError, ConsoleColors, NoActivityAccess, NotEnoughArguments
 from SwagLyricsBot.logs import Log
@@ -11,9 +11,12 @@
 
 class GeneralCommands(commands.Cog, name="General"):
 
-    def __init__(self, bot, session):
+    def __init__(self, bot, session, current=None):
+        if current is None:
+            current = {}
         self.bot = bot
         self.session = session
+        self.current = current
 
     @commands.command(name="help")
     async def help_message(self, ctx):
@@ -40,7 +43,7 @@ async def help_message(self, ctx):
     def get_spotify_data(user):
         from SwagLyricsBot.swaglyrics_bot import find_mutual_guild
         """
-        Reads data from discord spotify activity. 
+        Reads data from discord spotify activity.
         """
         if user.dm_channel:
             print("    - Command was raised in DM, finding mutual guild with user...")
@@ -59,6 +62,20 @@ def get_spotify_data(user):
             raise SpotifyClosed()
         return spotify_activity[0].title, spotify_activity[0].artists
 
+    async def send_lyrics(self, ctx, song, artists):
+
+        lyrics = await get_lyrics(song, artists[0], self.session)
+        print('got lyrics')
+        # await log.add_sub_log("Lyrics fetched successfully, splitting it into fields...")
+        split_lyrics = self.chop_string_into_chunks(lyrics, 1024)
+        print('split lyrics')
+        # await log.add_sub_log("Split successfully. Packing into messages...")
+
+        artists_string = self.artists_to_string(artists)
+        await self.send_chunks(ctx, split_lyrics, song, artists_string)
+        # await log.add_sub_log("Lyrics sent successfully.", ConsoleColors.OKGREEN)
+        # log.change_log_success_status(True)
+
     @commands.command(name="swaglyrics", aliases=["sl", "lyrics"])
     async def get_lyrics_command(self, ctx, song=None, artists=None):
         """
@@ -67,37 +84,22 @@ async def get_lyrics_command(self, ctx, song=None, artists=None):
         """
         log = Log(self.session)
 
-        async def send_lyrics():
-            lyrics = await get_lyrics(song, artists[0], self.session)
-            await log.add_sub_log("Lyrics fetched successfully, splitting it into fields...")
-            split_lyrics = self.chop_string_into_chunks(lyrics, 1024)
-            await log.add_sub_log("Split successfully. Packing into messages...")
-
-            await self.send_chunks(ctx, split_lyrics, song, artists_string)
-            await log.add_sub_log(f"Lyrics sent successfully.", ConsoleColors.OKGREEN)
-            log.change_log_success_status(True)
-
         try:
-
-            await log.add_log(
-                f"User {ctx.author} from {ctx.guild or ctx.channel} guild requested lyrics"
-            )
+            await log.add_log(f"User {ctx.author} from {ctx.guild or ctx.channel} guild requested lyrics")
 
             if not (song or artists):
                 await log.add_sub_log("Song data not provided, trying to fetch it automatically...")
                 song, artists = self.get_spotify_data(ctx.author)
             elif artists is None:
                 raise NotEnoughArguments("Not enough arguments! For usage, check `$help`")
             else:
-                tmp = artists
-                artists = list()
-                artists.append(tmp)
+                artists = list(artists)
             artists_string = self.artists_to_string(artists)
             debug_string = f"Getting lyrics for {song} by {artists_string}"
             await log.add_sub_log(debug_string)
             await ctx.send(debug_string)
 
-            await send_lyrics()
+            await self.send_lyrics(ctx, song, artists, log)
         except LyricsError as ex:
             await log.add_sub_log(f"Error raised: {ex}", ConsoleColors.FAIL)
             log.change_log_success_status(None)
@@ -106,10 +108,61 @@ async def send_lyrics():
             await log.add_sub_log(f"Error: {ex}", ConsoleColors.FAIL, True)
             print(traceback.print_exception(type(ex), ex, ex.__traceback__))
             log.change_log_success_status(False)
-            await ctx.send(f"There was an error while processing your request. Please try again in a few seconds.")
+            await ctx.send("There was an error while processing your request. Please try again in a few seconds.")
         finally:
             await log.send_webhook()
 
+    @tasks.loop(seconds=5)
+    async def vibe_mode(self):
+        print('this is inside the loop')
+        for user, info in self.current.items():
+            print(user.id, info)
+            channel = self.bot.get_channel(info['channel'])
+            await channel.send('ope')
+            try:
+                song, artists = self.get_spotify_data(user)
+                print(song, artists)
+                if (song, artists) == info['playing']:
+                    print("continuing")
+                    continue
+                # song has changed
+                self.current[user]['playing'] = (song, artists)
+                artists_string = self.artists_to_string(artists)
+                debug_string = f"Getting lyrics for {song} by {artists_string}"
+                # await log.add_sub_log(debug_string)
+                print('sending lyrics bb')
+                await channel.send(debug_string)
+
+                await self.send_lyrics(channel, song, artists)
+            except LyricsError as e:
+                print('No activity detected, stopping loop.')
+                await channel.send("No activity detected, killing the vibe.")
+                del self.current[user]
+                if not self.current:
+                    print('stopping loop')
+                    self.vibe_mode.cancel()
+
+    @commands.command()
+    async def vibe(self, ctx):
+        prev = self.current.copy()
+        # add current user to dict
+        self.current[ctx.author] = {'playing': (None, None),
+                                    'channel': ctx.channel.id}
+        await ctx.send('one vibe mode coming right up.')
+        print(prev)
+        if not prev:  # compare using previous current
+            print("starting loop")
+            self.vibe_mode.start()
+
+    @commands.command()
+    async def kill(self, ctx):
+        del self.current[ctx.author]
+        print('killed vibe')
+        await ctx.send('https://www.youtube.com/watch?v=GF8aaTu2kg0')
+        if not self.current:
+            print('stopping loop')
+            self.vibe_mode.cancel()
+
     async def send_chunks(self, ctx, chunks, song, artists):
         messages = self.pack_into_messages(chunks)
         i = 0

SwagLyricsBot/lyrics.py

@@ -11,7 +11,8 @@ async def fetch(session, url, **kwargs):
     """
     Uses aiohttp to make http GET requests
     """
-    headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/84.0.4147.89 Safari/537.36'}
+    headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) '
+                             'Chrome/84.0.4147.89 Safari/537.36'}
     async with session.get(url, headers=headers, **kwargs) as resp:
         return await resp.text()