feat: Add /account likes and user playlists search.

MusicBot/cogs/utils/misc.py

@@ -1,10 +1,36 @@
 from math import ceil
 from typing import Any
 
+from yandex_music import Track
 from discord.ui import View, Button, Item
 from discord import ButtonStyle, Interaction, ApplicationContext, Embed
 
-from MusicBot.cogs.utils.voice import VoiceExtension
+from MusicBot.cogs.utils.voice_extension import VoiceExtension
+
+def generate_likes_embed(tracks: list[Track]) -> Embed:
+    track_count = len(tracks)
+    cover_url = "https://avatars.yandex.net/get-music-user-playlist/11418140/favorit-playlist-cover.bb48fdb9b9f4/300x300"
+
+    embed = Embed(
+        title="Мне нравится",
+        description="Треки, которые вам понравились.",
+        color=0xce3a26,
+    )
+    embed.set_thumbnail(url=cover_url)
+
+    duration = 0
+    for track in tracks:
+        if track.duration_ms:
+            duration += track.duration_ms
+
+    duration_m = duration // 60000
+    duration_s = ceil(duration / 1000) - duration_m * 60
+    embed.add_field(name="Длительность", value=f"{duration_m}:{duration_s:02}")
+
+    if track_count is not None:
+        embed.add_field(name="Треки", value=str(track_count))
+
+    return embed
 
 def generate_playlist_embed(page: int, playlists: list[tuple[str, int]]) -> Embed:
     count = 15 * page
@@ -35,6 +61,38 @@ def generate_queue_embed(page: int, tracks_list: list[dict[str, Any]]) -> Embed:
         embed.add_field(name=f"{i} - {track['title']} - {duration_m}:{duration_s:02d}", value="", inline=False)
     return embed
 
+class PlayLikesButton(Button, VoiceExtension):
+    def __init__(self, playlist: list[Track], **kwargs):
+        Button.__init__(self, **kwargs)
+        VoiceExtension.__init__(self)
+        self.playlist = playlist
+    
+    async def callback(self, interaction: Interaction):
+        if not interaction.guild or not await self.voice_check(interaction):
+            return
+
+        gid = interaction.guild.id
+        guild = self.db.get_guild(gid)
+
+        if guild['current_track'] is not None:
+            self.db.modify_track(gid, self.playlist, 'next', 'extend')
+            response_message = f"Плейлист **«Мне нравится»** был добавлен в очередь."
+        else:
+            track = self.playlist.pop(0)
+            self.db.modify_track(gid, self.playlist, 'next', 'extend')
+            await self.play_track(interaction, track)
+            response_message = f"Сейчас играет плейлист **«Мне нравится»**!"
+
+        if guild['current_player'] is not None and interaction.message:
+            await interaction.message.delete()
+
+        await interaction.respond(response_message, delete_after=15)
+
+class ListenLikesPlaylist(View):
+    def __init__(self, playlist: list[Track], *items: Item, timeout: float | None = None, disable_on_timeout: bool = False):
+        super().__init__(*items, timeout=timeout, disable_on_timeout=disable_on_timeout)
+        self.add_item(PlayLikesButton(playlist, label="Слушать в голосовом канале", style=ButtonStyle.gray))
+
 class MPNextButton(Button, VoiceExtension):
     def __init__(self, **kwargs):
         Button.__init__(self, **kwargs)
@@ -47,7 +105,7 @@ class MPNextButton(Button, VoiceExtension):
         page = user['playlists_page'] + 1
         self.users_db.update(interaction.user.id, {'playlists_page': page})
         embed = generate_playlist_embed(page, user['playlists'])
-        await interaction.edit(embed=embed, view=MyPlalistsView(interaction))
+        await interaction.edit(embed=embed, view=MyPlalists(interaction))
 
 class MPPrevButton(Button, VoiceExtension):
     def __init__(self, **kwargs):
@@ -61,9 +119,9 @@ class MPPrevButton(Button, VoiceExtension):
         page = user['playlists_page'] - 1
         self.users_db.update(interaction.user.id, {'playlists_page': page})
         embed = generate_playlist_embed(page, user['playlists'])
-        await interaction.edit(embed=embed, view=MyPlalistsView(interaction))
+        await interaction.edit(embed=embed, view=MyPlalists(interaction))
 
-class MyPlalistsView(View, VoiceExtension):
+class MyPlalists(View, VoiceExtension):
     def __init__(self, ctx: ApplicationContext | Interaction, *items: Item, timeout: float | None = 3600, disable_on_timeout: bool = True):
         View.__init__(self, *items, timeout=timeout, disable_on_timeout=disable_on_timeout)
         VoiceExtension.__init__(self)
@@ -83,7 +141,6 @@ class MyPlalistsView(View, VoiceExtension):
         self.add_item(prev_button)
         self.add_item(next_button)
 
-
 class QNextButton(Button, VoiceExtension):
     def __init__(self, **kwargs):
         Button.__init__(self, **kwargs)