import logging

from pyrogram import filters
from pyrogram.enums.chat_member_status import ChatMemberStatus
from pyrogram.types import Message

from classes.pyroclient import PyroClient
from classes.pyrogroup import PyroGroup

logger = logging.getLogger(__name__)


@PyroClient.on_message(
    ~filters.scheduled
    & filters.group
    & filters.command(["timeout_join"], prefixes=["/"])  # type: ignore
)
async def command_timeout_join(app: PyroClient, message: Message):
    group = await PyroGroup.create_if_not_exists(message.chat.id, None, True)
    locale = group.select_locale(app, message.from_user)

    if (await app.get_chat_member(group.id, message.from_user.id)).status not in [
        ChatMemberStatus.ADMINISTRATOR,
        ChatMemberStatus.OWNER,
    ]:
        await message.reply_text(
            app._("permission_denied", "messages", locale=locale), quote=True
        )
        return

    if len(message.command) < 2 or not message.command[1].isdigit():
        await message.reply_text(
            app._("timeout_join_invalid", "messages", locale=locale), quote=True
        )
        return

    timeout = int(message.command[1])

    if timeout < 10 or timeout > 600:
        await message.reply_text(
            app._("timeout_invalid_number", "messages", locale=locale), quote=True
        )
        return

    await message.reply_text(
        app._("timeout_join_set", "messages", locale=locale).format(timeout=timeout)
    )

    logger.info(
        "Timeout on join in group %s has been set to %s",
        group.id,
        timeout,
    )

    await group.set_timeout_join(timeout)