from nonebot import get_bot
from nonebot import require
from nonebot.adapters.onebot.v11 import Bot, Message, MessageEvent, MessageSegment
from nonebot.log import logger

require("nonebot_plugin_htmlrender")
from nonebot_plugin_htmlrender import md_to_pic


async def send_markdown(
    bot: Bot,
    event: MessageEvent,
    markdown: str,
    fallback_text: str | None = None,
) -> None:
    """发送 Markdown

    优先尝试 OneBot 后端的原生 markdown 消息段;后端不支持/失败时
    自动降级为 htmlrender 把 Markdown 渲染成图片发送,保证任何后端都能看到。
    """
    # 1. 尝试原生 markdown 消息段(不同后端字段可能不同,失败自动降级)
    try:
        seg = MessageSegment("markdown", {"content": markdown})
        await bot.send(event=event, message=Message(seg))
        return
    except Exception as e:
        logger.warning(f"原生 markdown 发送失败,尝试降级为图片: {e}")

    # 2. 降级:htmlrender 渲染成图片
    try:
        img = await md_to_pic(md=markdown, width=540, device_scale_factor=4)
        await bot.send(event=event, message=MessageSegment.image(img))
    except Exception as e:
        logger.error(f"Markdown 渲染图片失败,发送纯文本: {e}")
        await bot.send(event=event, message=fallback_text or markdown)


async def send_poke(group_id: int | None = None, user_id: int = 0):
    """发送戳一戳(OneBot V11 send_poke API)

    group_id 为空时向私聊发送戳一戳;失败返回 None
    """
    bot = get_bot()
    params = {"user_id": int(user_id)}
    if group_id:
        params["group_id"] = int(group_id)
    logger.debug(f"send_poke: {params}")
    try:
        return await bot.call_api("send_poke", **params)
    except Exception as e:
        logger.error(f"send_poke 失败: {e}")
        return None


def get_reply_message(event: MessageEvent) -> Message:
    """获取消息内容:若为回复消息则返回被回复的消息,否则返回当前消息"""
    if event.reply is not None:
        logger.info(f"此条消息为回复消息: {event.reply.message}")
        return event.reply.message
    logger.info(f"此条消息为普通消息: {event.message}")
    return event.message


def build_forward_nodes(
    self_id: int,
    entries: list[tuple[str, str]],
    sender_name: str | None = None,
) -> list[dict]:
    """构造合并转发节点列表

    entries: [(发送者名, 消息内容), ...]
    sender_name: 非空时所有节点统一使用该昵称(用于把插件名等放进 content,
    而不是作为转发节点的发送者昵称)
    """
    return [
        {
            "type": "node",
            "data": {
                "name": sender_name if sender_name is not None else name,
                "uin": str(self_id),
                "content": content,
            },
        }
        for name, content in entries
    ]


async def send_forward_msg(
    bot: Bot,
    event: MessageEvent,
    entries: list[tuple[str, str]],
    fallback: str,
    sender_name: str | None = None,
):
    """群聊发送合并转发消息,私聊或发送失败时回退纯文本

    entries: [(发送者名, 消息内容), ...]
    sender_name: 非空时统一用于节点发送者昵称(插件名等应写入 content)
    """
    group_id = getattr(event, "group_id", None)
    if group_id:
        try:
            await bot.send_group_forward_msg(
                group_id=group_id,
                messages=build_forward_nodes(event.self_id, entries, sender_name),
            )
            return
        except Exception as e:
            logger.warning(f"合并转发发送失败,回退文本: {e}")
    await bot.send(event, fallback)