LGBot/src/modules/unmute.py

from telebot.async_telebot import AsyncTeleBot
from telebot.types import Message, User, ChatPermissions
import asyncio
import logging
from database import db

from action_reporter import action_reporter
from utils import (
    delete_messages,
    check_admin_status,
    check_target_status,
)

from config import COMMAND_MESSAGES

# Получаем логгер для текущего модуля
logger = logging.getLogger(__name__)

# Регистрирует все обработчики команд
def register_handlers(bot: AsyncTeleBot):

    # Обработчик команды /unmute
    @bot.message_handler(commands=['unmute'])
    async def unmute_command(message: Message):

        # Определяем целевого пользователя
        target_user = None

        # Разбиваем текст сообщения на части
        parts_msg = message.text.split()

        # Команда /unmute help
        if len(parts_msg) == 2 and parts_msg[1].strip() in ('help', 'помощь'):
            
            # Отправляем инструкцию
            await bot.send_message(
                chat_id=message.chat.id, 
                text=COMMAND_MESSAGES['manual_unmute'], 
                message_thread_id=message.message_thread_id,
            )

            # Удаляем сообщения через 30 секунд
            await delete_messages(bot, message, time_sleep=30, number_message=2)
            return

        try:

            # Проверяем, является ли отправитель администратором
            if await check_admin_status(bot, message) == 1:
                return

            # Команда через ответ на сообщение, если одно слово (/unmute)
            if len(parts_msg) == 1:

                # Если это топик
                if message.is_topic_message:

                    # Если без ответа на сообщение
                    if message.message_thread_id == message.reply_to_message.message_id:
                        
                        # Удаляем сообщение через 3 секунды
                        await delete_messages(bot, message, time_sleep=3, number_message=1)
                        return

                    # Если с ответом на сообщение
                    else:

                        # Собираем данные
                        target_user = message.reply_to_message.from_user

                # Если это General (обычный чат)
                elif message.reply_to_message and not message.is_topic_message:

                    # Собираем данные
                    target_user = message.reply_to_message.from_user
                    message.message_thread_id = None
                
                # Удаляем сообщение, если команда неправильная
                else:
                    
                    # Удаляем сообщение через 3 секунды
                    await delete_messages(bot, message, time_sleep=3, number_message=1)
                    return
            
            # Команда через тег или ID, если два слова
            elif len(parts_msg) == 2:
                
                # Собираем данные
                identifier = parts_msg[1].strip()
                
                # Делаем поиск по ID
                if identifier.isdigit():

                    # Ищем пользователя в базе данных
                    user_info = db.get_user(int(identifier))

                    # Если нашли пользователя
                    if user_info:

                        # Создаем объект пользователя
                        target_user = User(
                            id=user_info[0],
                            first_name=user_info[1],
                            username=user_info[2],
                            is_bot=False
                        )
                
                # Делаем поиск по тегу
                elif identifier.startswith('@'):

                    # Ищем пользователя в базе данных
                    user_info = db.get_user_by_username(identifier[1:])

                    # Если нашли пользователя
                    if user_info:

                        # Создаем объект пользователя
                        target_user = User(
                            id=user_info[0],
                            first_name=user_info[1],
                            username=user_info[2],
                            is_bot=False
                        )

            # Если пользователь не найден в базе данных
            if not target_user:

                # Отправляем предупреждение
                await bot.send_message(
                    chat_id=message.chat.id, 
                    text=COMMAND_MESSAGES['user_not_found'], 
                    message_thread_id=message.message_thread_id,
                )

                # Удаляем сообщения через 5 секунд
                await delete_messages(bot, message, time_sleep=5, number_message=2)
                return

            # Проверяем статус целевого пользователя
            if await check_target_status(bot, message, target_user) == 1:
                return

            try:

                # Убираем ограничения (можно писать в чат)
                permissions = ChatPermissions(
                    can_send_messages=True,
                    can_send_media_messages=True,
                    can_send_polls=True,
                    can_send_other_messages=True,
                    can_add_web_page_previews=True,
                    can_change_info=False,
                    can_invite_users=True,
                    can_pin_messages=False
                )

                # Выполняем размут
                await bot.restrict_chat_member(
                    chat_id=message.chat.id,
                    user_id=target_user.id,
                    permissions=permissions,
                )

                # Отправляем лог в админ-чат
                await action_reporter.log_action(
                    action="РАЗМУТ",
                    user_id=target_user.id,
                    admin_id=message.from_user.id,
                    reason=None,
                    duration=None,
                    photo_path=None,
                )
                
                # Отправляем сообщения, что пользователь получил размут
                await bot.send_message(
                    chat_id=message.chat.id, 
                    text=COMMAND_MESSAGES['unmuted'], 
                    message_thread_id=message.message_thread_id,
                )

                # Записываем действие в логи
                logger.info(f"Пользователь {target_user.id} получил размут от администратора {message.from_user.id}.")
                
                # Удаляем сообщения через 5 секунд
                await delete_messages(bot, message, time_sleep=5, number_message=2)
                
            except Exception as e:

                # Отправляем ошибку
                await bot.send_message(
                    chat_id=message.chat.id, 
                    text=COMMAND_MESSAGES['error'].format(e=str(e)), 
                    message_thread_id=message.message_thread_id,
                )

                # Записываем ошибку в логи
                logger.error(f"Ошибка размута: {str(e)}")

                # Удаляем сообщения через 5 секунд
                await delete_messages(bot, message, time_sleep=5, number_message=2)
                
        except Exception as e:

            # Отправляем ошибку
            await bot.send_message(
                chat_id=message.chat.id, 
                text=COMMAND_MESSAGES['general_error'], 
                message_thread_id=message.message_thread_id,
            )

            # Записываем ошибку в логи
            logger.error(f"Общая ошибка в unmute_command: {str(e)}")

            # Удаляем сообщения через 5 секунд
            await delete_messages(bot, message, time_sleep=5, number_message=2)