YandexMusicDiscordBot/MusicBot/cogs/voice.py

import logging
from typing import cast

import discord
from discord.ext.commands import Cog

from yandex_music import ClientAsync as YMClient
from yandex_music.exceptions import UnauthorizedError

from MusicBot.database import BaseUsersDatabase
from MusicBot.cogs.utils import VoiceExtension, menu_views
from MusicBot.ui import QueueView, generate_queue_embed

def setup(bot: discord.Bot):
    bot.add_cog(Voice(bot))

users_db = BaseUsersDatabase()

async def get_vibe_stations_suggestions(ctx: discord.AutocompleteContext) -> list[str]:
    if not ctx.interaction.user or not ctx.value or len(ctx.value) < 2:
        return []

    token = await users_db.get_ym_token(ctx.interaction.user.id)
    if not token:
        logging.info(f"[GENERAL] User {ctx.interaction.user.id} has no token")
        return []

    try:
        client = await YMClient(token).init()
    except UnauthorizedError:
        logging.info(f"[GENERAL] User {ctx.interaction.user.id} provided invalid token")
        return []

    stations = await client.rotor_stations_list()
    return [station.station.name for station in stations if station.station and ctx.value in station.station.name][:100]


class Voice(Cog, VoiceExtension):

    voice = discord.SlashCommandGroup("voice", "Команды, связанные с голосовым каналом.")
    queue = discord.SlashCommandGroup("queue", "Команды, связанные с очередью треков.")

    def __init__(self, bot: discord.Bot):
        VoiceExtension.__init__(self, bot)
        self.typed_bot: discord.Bot = bot

    @Cog.listener()
    async def on_voice_state_update(self, member: discord.Member, before: discord.VoiceState, after: discord.VoiceState) -> None:
        gid = member.guild.id
        guild = await self.db.get_guild(gid, projection={'current_menu': 1})

        channel = after.channel or before.channel
        if not channel:
            logging.warning(f"[VOICE] No channel found for member {member.id}")
            return

        vc = cast(discord.VoiceClient | None, discord.utils.get(self.typed_bot.voice_clients, guild=await self.typed_bot.fetch_guild(gid)))

        for member in channel.members:
            if member.id == self.typed_bot.user.id:  # type: ignore  # should be logged in
                logging.info(f"[VOICE] Voice state update for member {member.id} in guild {member.guild.id}")
                break
        else:
            logging.debug(f"[VOICE] Bot is not in the channel {channel.id}")
            return

        if not vc:
            logging.info(f"[VOICE] No voice client found for guild {gid}")
            return

        if len(channel.members) == 1:
            logging.info(f"[VOICE] Clearing history and stopping playback for guild {gid}")

            if member.guild.id in menu_views:
                menu_views[member.guild.id].stop()
                del menu_views[member.guild.id]

            if guild['current_menu']:
                message = self.typed_bot.get_message(guild['current_menu'])
                if message:
                    await message.delete()

            await self.db.update(gid, {
                'previous_tracks': [], 'next_tracks': [], 'votes': {},
                'current_track': None, 'current_menu': None, 'vibing': False,
                'repeat': False, 'shuffle': False, 'is_stopped': True
            })
            vc.stop()

            if member.guild.id in menu_views:
                menu_views[member.guild.id].stop()
                del menu_views[member.guild.id]

    @Cog.listener()
    async def on_raw_reaction_add(self, payload: discord.RawReactionActionEvent) -> None:
        logging.info(f"[VOICE] Reaction added by user {payload.user_id} in channel {payload.channel_id}")
        if not self.typed_bot.user or not payload.member:
            return

        bot_id = self.typed_bot.user.id
        if payload.user_id == bot_id:
            return

        channel = cast(discord.VoiceChannel, self.typed_bot.get_channel(payload.channel_id))
        if not channel:
            return

        try:
            message = await channel.fetch_message(payload.message_id)
        except discord.Forbidden:
            logging.info(f"[VOICE] Bot does not have permissions to read messages in channel {payload.channel_id}")
            return
        except discord.NotFound:
            logging.info(f"[VOICE] Message {payload.message_id} not found in channel {payload.channel_id}")
            return

        if not message or message.author.id != bot_id:
            return

        if not await self.users_db.get_ym_token(payload.user_id):
            await message.remove_reaction(payload.emoji, payload.member)
            await channel.send("Для участия в голосовании необходимо авторизоваться через /account login.", delete_after=15)
            return

        guild_id = payload.guild_id
        if not guild_id:
            return

        guild = await self.db.get_guild(guild_id)
        votes = guild['votes']

        if str(payload.message_id) not in votes:
            logging.info(f"[VOICE] Message {payload.message_id} not found in votes")
            return

        vote_data = votes[str(payload.message_id)]

        if payload.emoji.name == '✅':
            logging.info(f"[VOICE] User {payload.user_id} voted positively for message {payload.message_id}")
            vote_data['positive_votes'].append(payload.user_id)
        elif payload.emoji.name == '❌':
            logging.info(f"[VOICE] User {payload.user_id} voted negatively for message {payload.message_id}")
            vote_data['negative_votes'].append(payload.user_id)

        total_members = len(channel.members)
        required_votes = 2 if total_members <= 5 else 4 if total_members <= 10 else 6 if total_members <= 15 else 9
        if len(vote_data['positive_votes']) >= required_votes:
            logging.info(f"[VOICE] Enough positive votes for message {payload.message_id}")
            await message.delete()
            await self.proccess_vote(payload, guild, channel, vote_data)
            del votes[str(payload.message_id)]

        elif len(vote_data['negative_votes']) >= required_votes:
            logging.info(f"[VOICE] Enough negative votes for message {payload.message_id}")
            await message.clear_reactions()
            await message.edit(content='Запрос был отклонён.', delete_after=15)
            del votes[str(payload.message_id)]

        await self.db.update(guild_id, {'votes': votes})

    @Cog.listener()
    async def on_raw_reaction_remove(self, payload: discord.RawReactionActionEvent) -> None:
        logging.info(f"[VOICE] Reaction removed by user {payload.user_id} in channel {payload.channel_id}")
        if not self.typed_bot.user:
            return

        guild_id = payload.guild_id
        if not guild_id:
            return

        guild = await self.db.get_guild(guild_id, projection={'votes': 1})
        votes = guild['votes']

        if str(payload.message_id) not in votes:
            logging.info(f"[VOICE] Message {payload.message_id} not found in votes")
            return

        channel = cast(discord.VoiceChannel, self.typed_bot.get_channel(payload.channel_id))
        if not channel:
            return

        try:
            message = await channel.fetch_message(payload.message_id)
        except discord.Forbidden:
            logging.info(f"[VOICE] Bot does not have permissions to read messages in channel {payload.channel_id}")
            return
        except discord.NotFound:
            logging.info(f"[VOICE] Message {payload.message_id} not found in channel {payload.channel_id}")
            return

        if not message or message.author.id != self.typed_bot.user.id:
            return

        vote_data = votes[str(payload.message_id)]
        if payload.emoji.name == '✔️':
            logging.info(f"[VOICE] User {payload.user_id} removed positive vote for message {payload.message_id}")
            del vote_data['positive_votes'][payload.user_id]
        elif payload.emoji.name == '❌':
            logging.info(f"[VOICE] User {payload.user_id} removed negative vote for message {payload.message_id}")
            del vote_data['negative_votes'][payload.user_id]

        await self.db.update(guild_id, {'votes': votes})
    
    @voice.command(name="menu", description="Создать или обновить меню проигрывателя.")
    async def menu(self, ctx: discord.ApplicationContext) -> None:
        logging.info(f"[VOICE] Menu command invoked by user {ctx.author.id} in guild {ctx.guild.id}")
        if await self.voice_check(ctx):
            await self.send_menu_message(ctx)

    @voice.command(name="join", description="Подключиться к голосовому каналу, в котором вы сейчас находитесь.")
    async def join(self, ctx: discord.ApplicationContext) -> None:
        logging.info(f"[VOICE] Join command invoked by user {ctx.author.id} in guild {ctx.guild.id}")

        member = cast(discord.Member, ctx.author)
        guild = await self.db.get_guild(ctx.guild.id, projection={'allow_change_connect': 1})
        vc = await self.get_voice_client(ctx)

        if not member.guild_permissions.manage_channels and not guild['allow_change_connect']:
            response_message = "❌ У вас нет прав для выполнения этой команды."
        elif vc and vc.is_connected():
            response_message = "❌ Бот уже находится в голосовом канале. Выключите его с помощью команды /voice leave."
        elif isinstance(ctx.channel, discord.VoiceChannel):
            try:
                await ctx.channel.connect()
            except TimeoutError:
                response_message = "❌ Не удалось подключиться к голосовому каналу."
            else:
                response_message = "✅ Подключение успешно!"
        else:
            response_message = "❌ Вы должны отправить команду в чате голосового канала."

        logging.info(f"[VOICE] Join command response: {response_message}")
        await ctx.respond(response_message, delete_after=15, ephemeral=True)

    @voice.command(description="Заставить бота покинуть голосовой канал.")
    async def leave(self, ctx: discord.ApplicationContext) -> None:
        logging.info(f"[VOICE] Leave command invoked by user {ctx.author.id} in guild {ctx.guild.id}")

        member = cast(discord.Member, ctx.author)
        guild = await self.db.get_guild(ctx.guild.id, projection={'allow_change_connect': 1})

        if not member.guild_permissions.manage_channels and not guild['allow_change_connect']:
            logging.info(f"[VOICE] User {ctx.author.id} does not have permissions to execute leave command in guild {ctx.guild.id}")
            await ctx.respond("❌ У вас нет прав для выполнения этой команды.", delete_after=15, ephemeral=True)
            return

        if (vc := await self.get_voice_client(ctx)) and await self.voice_check(ctx) and vc.is_connected:
            res = await self.stop_playing(ctx, vc=vc, full=True)
            if not res:
                await ctx.respond("❌ Не удалось отключиться.", delete_after=15, ephemeral=True)
                return

            await vc.disconnect(force=True)
            await ctx.respond("✅ Отключение успешно!", delete_after=15, ephemeral=True)
            logging.info(f"[VOICE] Successfully disconnected from voice channel in guild {ctx.guild.id}")
        else:
            await ctx.respond("❌ Бот не подключен к голосовому каналу.", delete_after=15, ephemeral=True)

    @queue.command(description="Очистить очередь треков и историю прослушивания.")
    async def clear(self, ctx: discord.ApplicationContext) -> None:
        logging.info(f"[VOICE] Clear queue command invoked by user {ctx.author.id} in guild {ctx.guild.id}")

        if not await self.voice_check(ctx):
            return

        member = cast(discord.Member, ctx.author)
        channel = cast(discord.VoiceChannel, ctx.channel)

        if len(channel.members) > 2 and not member.guild_permissions.manage_channels:
            logging.info(f"Starting vote for stopping playback in guild {ctx.guild.id}")
            
            response_message = f"{member.mention} хочет очистить историю прослушивания и очередь треков.\n\n Выполнить действие?."
            message = cast(discord.Interaction, await ctx.respond(response_message, delete_after=60))
            response = await message.original_response()

            await response.add_reaction('✅')
            await response.add_reaction('❌')
            
            await self.db.update_vote(
                ctx.guild_id,
                response.id,
                {
                    'positive_votes': list(),
                    'negative_votes': list(),
                    'total_members': len(channel.members),
                    'action': 'clear_queue',
                    'vote_content': None
                }
            )
            return

        await self.db.update(ctx.guild.id, {'previous_tracks': [], 'next_tracks': []})
        await ctx.respond("✅ Очередь и история сброшены.", delete_after=15, ephemeral=True)
        logging.info(f"[VOICE] Queue and history cleared in guild {ctx.guild.id}")

    @queue.command(description="Получить очередь треков.")
    async def get(self, ctx: discord.ApplicationContext) -> None:
        logging.info(f"[VOICE] Get queue command invoked by user {ctx.author.id} in guild {ctx.guild.id}")

        if not await self.voice_check(ctx):
            return
        await self.users_db.update(ctx.user.id, {'queue_page': 0})

        tracks = await self.db.get_tracks_list(ctx.guild.id, 'next')
        embed = generate_queue_embed(0, tracks)
        await ctx.respond(embed=embed, view=await QueueView(ctx).init(), ephemeral=True)

        logging.info(f"[VOICE] Queue embed sent to user {ctx.author.id} in guild {ctx.guild.id}")

    @voice.command(description="Прервать проигрывание, удалить историю, очередь и текущий плеер.")
    async def stop(self, ctx: discord.ApplicationContext) -> None:
        logging.info(f"[VOICE] Stop command invoked by user {ctx.author.id} in guild {ctx.guild.id}")

        if not await self.voice_check(ctx):
            return

        member = cast(discord.Member, ctx.author)
        channel = cast(discord.VoiceChannel, ctx.channel)

        if len(channel.members) > 2 and not member.guild_permissions.manage_channels:
            logging.info(f"Starting vote for stopping playback in guild {ctx.guild.id}")

            response_message = f"{member.mention} хочет полностью остановить проигрывание.\n\n Выполнить действие?."
            message = cast(discord.Interaction, await ctx.respond(response_message, delete_after=60))
            response = await message.original_response()

            await response.add_reaction('✅')
            await response.add_reaction('❌')

            await self.db.update_vote(
                ctx.guild_id,
                response.id,
                {
                    'positive_votes': list(),
                    'negative_votes': list(),
                    'total_members': len(channel.members),
                    'action': 'stop',
                    'vote_content': None
                }
            )
            return

        res = await self.stop_playing(ctx, full=True)
        if res:
            await ctx.respond("✅ Воспроизведение остановлено.", delete_after=15, ephemeral=True)
        else:
            await ctx.respond("❌ Произошла ошибка при остановке воспроизведения.", delete_after=15, ephemeral=True)

    @voice.command(name='vibe', description="Запустить Мою Волну.")
    @discord.option(
        "запрос",
        parameter_name='name',
        description="Название станции.",
        type=discord.SlashCommandOptionType.string,
        autocomplete=discord.utils.basic_autocomplete(get_vibe_stations_suggestions),
        required=False
    )
    async def user_vibe(self, ctx: discord.ApplicationContext, name: str | None = None) -> None:
        logging.info(f"[VOICE] Vibe (user) command invoked by user {ctx.user.id} in guild {ctx.guild_id}")
        if not await self.voice_check(ctx):
            return

        guild = await self.db.get_guild(ctx.guild.id, projection={'current_menu': 1, 'vibing': 1})

        if guild['vibing']:
            logging.info(f"[VOICE] Action declined: vibing is already enabled in guild {ctx.guild.id}")
            await ctx.respond("❌ Моя Волна уже включена. Используйте /voice stop, чтобы остановить воспроизведение.", delete_after=15, ephemeral=True)
            return

        await ctx.defer(invisible=False)
        if name:
            token = await users_db.get_ym_token(ctx.user.id)
            if not token:
                logging.info(f"[GENERAL] User {ctx.user.id} has no token")
                return

            try:
                client = await YMClient(token).init()
            except UnauthorizedError:
                logging.info(f"[GENERAL] User {ctx.user.id} provided invalid token")
                return

            stations = await client.rotor_stations_list()
            for content in stations:
                if content.station and content.station.name == name and content.ad_params:
                    break
            else:
                content = None

            if not content:
                logging.debug(f"[VOICE] Station {name} not found")
                await ctx.respond("❌ Станция не найдена.", delete_after=15, ephemeral=True)
                return

            _type, _id = content.ad_params.other_params.split(':') if content.ad_params else (None, None)

            if not _type or not _id:
                logging.debug(f"[VOICE] Station {name} has no ad params")
                await ctx.respond("❌ Станция не найдена.", delete_after=15, ephemeral=True)
                return
        else:
            _type, _id = 'user', 'onyourwave'
            content = None
        
        member = cast(discord.Member, ctx.author)
        channel = cast(discord.VoiceChannel, ctx.channel)
        
        if len(channel.members) > 2 and not member.guild_permissions.manage_channels:
            logging.info(f"Starting vote for stopping playback in guild {ctx.guild.id}")

            if _type == 'user' and _id == 'onyourwave':
                station = "Моя Волна"
            elif content and content.station:
                station = content.station.name
            else:
                logging.warning(f"[VOICE] Station {name} not found")
                await ctx.respond("❌ Станция не найдена.", delete_after=15, ephemeral=True)
                return

            response_message = f"{member.mention} хочет запустить станцию **{station}**.\n\n Выполнить действие?"
            message = cast(discord.WebhookMessage, await ctx.respond(response_message, delete_after=60))

            await message.add_reaction('✅')
            await message.add_reaction('❌')
            
            await self.db.update_vote(
                ctx.guild_id,
                message.id,
                {
                    'positive_votes': list(),
                    'negative_votes': list(),
                    'total_members': len(channel.members),
                    'action': 'vibe_station',
                    'vote_content': [_type, _id, ctx.user.id]
                }
            )
            return
        
        feedback = await self.update_vibe(ctx, _type, _id)

        if not feedback:
            await ctx.respond("❌ Операция не удалась. Возможно, у вес нет подписки на Яндекс Музыку.", delete_after=15, ephemeral=True)
            return

        if guild['current_menu']:
            await ctx.respond("✅ Моя Волна включена.", delete_after=15, ephemeral=True)
        else:
            await self.send_menu_message(ctx, disable=True)

        next_track = await self.db.get_track(ctx.guild_id, 'next')
        if next_track:
            await self._play_track(ctx, next_track)