from pyrogram.raw.functions.messages import GetStickerSet
from pyrogram.raw.functions.stickers import CreateStickerSet
from pyrogram.raw.types import InputStickerSetShortName, InputStickerSetItem, InputDocument
from pyrogram.raw.types.messages import StickerSet

from pagermaid.listener import listener
from pagermaid.services import bot
from pagermaid.enums import Message


class NoStickerSetNameError(Exception):
    """
        Occurs when no username is provided
    """

    def __init__(self, string: str = "贴纸包不存在"):
        super().__init__(
            string
        )


async def get_pack(name: str):
    try:
        return await bot.invoke(GetStickerSet(
            stickerset=InputStickerSetShortName(short_name=name),
            hash=0
        ))
    except Exception as e:  # noqa
        raise NoStickerSetNameError() from e


async def create_sticker_set(sticker_set: str, title: str, is_animated: bool, is_video: bool, stickers):
    try:
        await bot.invoke(
            CreateStickerSet(
                user_id=await bot.resolve_peer((await bot.get_me()).id),
                title=title,
                short_name=sticker_set,
                stickers=stickers,
                animated=is_animated,
                videos=is_video,
                software="pagermaid-pyro",
            )
        )
    except Exception as e:
        raise NoStickerSetNameError("贴纸包名称或者链接非法或者已被占用，请换一个") from e


async def process_old_sticker_set(sticker_set: str):
    pack: StickerSet = await get_pack(sticker_set)
    is_animated = pack.set.animated
    is_video = pack.set.videos
    hash_map = {}
    for i in pack.packs:
        for j in i.documents:
            hash_map[j] = i.emoticon
    stickers = [
        InputStickerSetItem(
            document=InputDocument(
                id=i.id,
                access_hash=i.access_hash,
                file_reference=i.file_reference,
            ),
            emoji=hash_map.get(i.id, "👀")
        ) for i in pack.documents
    ]
    return stickers, is_animated, is_video


@listener(
    command="copy_sticker_set",
    parameters="贴纸包链接 贴纸包名称",
    description="复制某个贴纸包",
)
async def copy_sticker_set(message: Message):
    if (not message.reply_to_message) or (not message.reply_to_message.sticker) or (
            not message.reply_to_message.sticker.set_name):
        return await message.edit("请先回复一张需要复制的贴纸包的贴纸")
    sticker_set = message.reply_to_message.sticker.set_name
    if len(message.parameter) < 2:
        return await message.edit("请指定贴纸包链接和贴纸包名称，例如 <code>xxxx_sticker xxxx 的贴纸包</code>")
    set_name = message.parameter[0]
    name = " ".join(message.parameter[1:])
    try:
        stickers, is_animated, is_video = await process_old_sticker_set(sticker_set)
        await create_sticker_set(set_name, name, is_animated, is_video, stickers)
    except Exception as e:
        return await message.edit(f"复制贴纸包失败：{e}")
    await message.edit(f"复制贴纸包成功 <a href=\"https://t.me/addstickers/{set_name}\">{name}</a>")