import logging from typing import Literal, cast import discord from yandex_music import Track, Album, Artist, Playlist from discord.ui import View, Button, Item from discord import ButtonStyle, Interaction, Embed from MusicBot.cogs.utils.voice_extension import VoiceExtension from MusicBot.cogs.utils.misc import generate_track_embed, generate_album_embed, generate_artist_embed, generate_playlist_embed class PlayButton(Button, VoiceExtension): def __init__(self, item: Track | Album | Artist | Playlist | list[Track], **kwargs): Button.__init__(self, **kwargs) VoiceExtension.__init__(self, None) self.item = item async def callback(self, interaction: Interaction) -> None: logging.debug(f"Callback triggered for type: '{type(self.item).__name__}'") if not interaction.guild: logging.warning("No guild found in context.") return if not await self.voice_check(interaction): logging.debug("Voice check failed") return gid = interaction.guild.id guild = self.db.get_guild(gid) channel = cast(discord.VoiceChannel, interaction.channel) member = cast(discord.Member, interaction.user) action: Literal['add_track', 'add_album', 'add_artist', 'add_playlist'] if isinstance(self.item, Track): tracks = [self.item] action = 'add_track' vote_message = f"{member.mention} хочет добавить трек **{self.item.title}** в очередь.\n\n Голосуйте за добавление." response_message = f"Трек **{self.item.title}** был добавлен в очередь." play_message = f"Сейчас играет: **{self.item.title}**!" elif isinstance(self.item, Album): album = await self.item.with_tracks_async() if not album or not album.volumes: logging.debug("Failed to fetch album tracks") await interaction.respond("Не удалось получить треки альбома.", ephemeral=True) return tracks = [track for volume in album.volumes for track in volume] action = 'add_album' vote_message = f"{member.mention} хочет добавить альбом **{self.item.title}** в очередь.\n\n Голосуйте за добавление." response_message = f"Альбом **{self.item.title}** был добавлен в очередь." play_message = f"Сейчас играет: **{self.item.title}**!" elif isinstance(self.item, Artist): artist_tracks = await self.item.get_tracks_async() if not artist_tracks: logging.debug("Failed to fetch artist tracks") await interaction.respond("Не удалось получить треки артиста.", ephemeral=True) return tracks = artist_tracks.tracks.copy() action = 'add_artist' vote_message = f"{member.mention} хочет добавить треки от **{self.item.name}** в очередь.\n\n Голосуйте за добавление." response_message = f"Песни артиста **{self.item.name}** были добавлены в очередь." play_message = f"Сейчас играет: **{self.item.name}**!" elif isinstance(self.item, Playlist): short_tracks = await self.item.fetch_tracks_async() if not short_tracks: logging.debug("Failed to fetch playlist tracks") await interaction.respond("❌ Не удалось получить треки из плейлиста.", delete_after=15) return tracks = [cast(Track, short_track.track) for short_track in short_tracks] action = 'add_playlist' vote_message = f"{member.mention} хочет добавить плейлист **{self.item.title}** в очередь.\n\n Голосуйте за добавление." response_message = f"Плейлист **{self.item.title}** был добавлен в очередь." play_message = f"Сейчас играет: **{self.item.title}**!" elif isinstance(self.item, list): tracks = self.item.copy() if not tracks: logging.debug("Empty tracks list") await interaction.respond("❌ Не удалось получить треки.", delete_after=15) return action = 'add_playlist' vote_message = f"{member.mention} хочет добавить плейлист **** в очередь.\n\n Голосуйте за добавление." response_message = f"Плейлист **«Мне нравится»** был добавлен в очередь." else: raise ValueError(f"Unknown item type: '{type(self.item).__name__}'") if guild.get(f'vote_{action}') and len(channel.members) > 2 and not member.guild_permissions.manage_channels: logging.debug(f"Starting vote for '{action}'") message = cast(discord.Interaction, await interaction.respond(vote_message, delete_after=30)) response = await message.original_response() await response.add_reaction('✅') await response.add_reaction('❌') self.db.update_vote( gid, response.id, { 'positive_votes': list(), 'negative_votes': list(), 'total_members': len(channel.members), 'action': action, 'vote_content': [track.to_dict() for track in tracks] } ) else: logging.debug(f"Skipping vote for '{action}'") if guild['current_track'] is not None: self.db.modify_track(gid, tracks, 'next', 'extend') else: track = tracks.pop(0) self.db.modify_track(gid, tracks, 'next', 'extend') await self.play_track(interaction, track) response_message = f"Сейчас играет: **{tracks[0].title}**!" current_player = None if guild['current_player']: current_player = await self.get_player_message(interaction, guild['current_player']) if current_player and interaction.message: logging.debug(f"Deleting interaction message '{interaction.message.id}': current player '{current_player.id}' found") await interaction.message.delete() else: await interaction.respond(response_message, delete_after=15) class ListenView(View): def __init__(self, item: Track | Album | Artist | 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) logging.debug(f"Creating view for type: '{type(item).__name__}'") if isinstance(item, Track): link_app = f"yandexmusic://album/{item.albums[0].id}/track/{item.id}" link_web = f"https://music.yandex.ru/album/{item.albums[0].id}/track/{item.id}" elif isinstance(item, Album): link_app = f"yandexmusic://album/{item.id}" link_web = f"https://music.yandex.ru/album/{item.id}" elif isinstance(item, Artist): link_app = f"yandexmusic://artist/{item.id}" link_web = f"https://music.yandex.ru/artist/{item.id}" elif isinstance(item, Playlist): link_app = f"yandexmusic://playlist/{item.playlist_uuid}" link_web = f"https://music.yandex.ru/playlist/{item.playlist_uuid}" elif isinstance(item, list): # Can't open other person's likes self.add_item(PlayButton(item, label="Слушать в голосовом канале", style=ButtonStyle.gray)) return self.button1: Button = Button(label="Слушать в приложении", style=ButtonStyle.gray, url=link_app) self.button2: Button = Button(label="Слушать в браузере", style=ButtonStyle.gray, url=link_web) self.button3: PlayButton = PlayButton(item, label="Слушать в голосовом канале", style=ButtonStyle.gray) if item.available: # self.add_item(self.button1) # Discord doesn't allow well formed URLs in buttons for some reason. self.add_item(self.button2) self.add_item(self.button3) async def generate_item_embed(item: Track | Album | Artist | Playlist) -> Embed: """Generate item embed. Args: item (yandex_music.Track | yandex_music.Album | yandex_music.Artist | yandex_music.Playlist): Item to be processed. Returns: discord.Embed: Item embed. """ logging.debug(f"Generating embed for type: '{type(item).__name__}'") if isinstance(item, Track): return await generate_track_embed(item) elif isinstance(item, Album): return await generate_album_embed(item) elif isinstance(item, Artist): return await generate_artist_embed(item) elif isinstance(item, Playlist): return await generate_playlist_embed(item) else: raise ValueError(f"Unknown item type: {type(item).__name__}")