import re
from io import BytesIO

from pyrogram import Client, filters, ContinuePropagation
from pyrogram.types import Message

from PIL import Image

from defs.mihoyo_bbs import get_mihoyo_screenshot
from defs.button import gen_button, Button


@Client.on_message(
    filters.incoming
    & filters.text
    & filters.regex(r"(https://)?(m\.)?www.miyoushe.com/.+/article/\d+")
)
async def bili_dynamic(_: Client, message: Message):
    # sourcery skip: use-named-expression
    try:
        p = re.compile(r"(https://)?(m\.)?www.miyoushe.com/.+/article/\d+")
        article = p.search(message.text)
        if article:
            article_url = article.group()
            if not article_url.startswith(("https://", "http://")):
                article_url = f"https://{article_url}"
            image = await get_mihoyo_screenshot(article_url)
            if image:
                # 将bytes结果转化为字节流
                photo = BytesIO(image)
                photo.name = "screenshot.png"
                pillow_photo = Image.open(BytesIO(image))
                width, height = pillow_photo.size
                if abs(height - width) > 1300:
                    await message.reply_document(
                        document=photo,
                        quote=True,
                        reply_markup=gen_button([Button(0, "Link", article_url)]),
                    )
                else:
                    await message.reply_photo(
                        photo,
                        quote=True,
                        reply_markup=gen_button([Button(0, "Link", article_url)]),
                    )
    except Exception as e:
        print(f"截取米哈游帖子时发生错误：{e}")
    raise ContinuePropagation