from core import command
from loguru import logger
from pyrogram import Client
from pyrogram.errors import RPCError
from pyrogram.types import Message
from tools.helpers import Parameters, delete_this


async def check_replied_msg(msg: Message, cmd: str) -> bool:
    replied_msg = msg.reply_to_message
    if not replied_msg:
        await msg.edit_text(f"❗️ Please use `{cmd}` to reply to a message.")
        return False
    elif replied_msg.has_protected_content or replied_msg.chat.has_protected_content:
        await msg.edit_text("😮‍💨 Please don't foward protected messages")
        return False
    else:
        return True


@Client.on_message(command('f'))
async def forward(_: Client, msg: Message):
    """转发目标消息"""
    cmd, num = Parameters.get_int(msg)
    replied_msg = msg.reply_to_message
    if not await check_replied_msg(msg, cmd):
        return

    await delete_this(msg)
    for _ in range(num):
        try:
            await replied_msg.forward(msg.chat.id, disable_notification=True)
        except RPCError as e:
            logger.error(e)


@Client.on_message(command('cp'))
async def copy_forward(cli: Client, msg: Message):
    """无引用转发"""
    cmd, num = Parameters.get_int(msg)
    if not await check_replied_msg(msg, cmd):
        return

    await delete_this(msg)
    for _ in range(num):
        try:
            await cli.copy_message(
                chat_id=msg.chat.id,
                from_chat_id=msg.chat.id,
                message_id=msg.reply_to_message.message_id,
                disable_notification=True
            )
        except RPCError as e:
            logger.error(e)