"""
腾讯元宝网页版驱动引擎（yuanbao.tencent.com）。
元宝暂无公开 API，仅支持网页模式。

选择器说明（如页面改版需更新）：
- 输入框：[class*="input-area"] textarea 或 [contenteditable="true"]
- 发送按钮：[class*="send-btn"] 或 [aria-label="发送"]
- 停止生成：[class*="stop"] 相关按钮
- 复制按钮：最后一条回复下的复制按钮
"""
import asyncio
from .base import BaseEngine


class YuanbaoEngine(BaseEngine):
    async def generate(self, prompt: str) -> str:
        await self.page.goto("https://yuanbao.tencent.com/chat")
        await self.page.wait_for_load_state("networkidle")

        # 登录检测
        input_selectors = [
            "textarea[class*='input']",
            "div[class*='input-area'] textarea",
            "[contenteditable='true']",
            "textarea",
        ]
        editor = None
        for sel in input_selectors:
            try:
                loc = self.page.locator(sel).first
                await loc.wait_for(state="visible", timeout=4000)
                editor = loc
                break
            except Exception:
                continue
        if editor is None:
            return "ERROR:REQUIRE_LOGIN"

        # 输入提示词
        await editor.click()
        await self.page.keyboard.insert_text(prompt)
        await asyncio.sleep(0.5)

        # 发送
        sent = False
        for sel in (
            '[class*="send-btn"]',
            '[aria-label="发送"]',
            'button[class*="send"]',
            'button[type="submit"]',
        ):
            try:
                btn = self.page.locator(sel).first
                if await btn.is_visible(timeout=1000):
                    await btn.click()
                    sent = True
                    break
            except Exception:
                continue
        if not sent:
            await self.page.keyboard.press("Enter")

        print("💡 [llm-manager/yuanbao] 已发送提示词，等待腾讯元宝生成响应...")
        await asyncio.sleep(2)

        # 等待生成完毕
        stop_selectors = ['[class*="stop"]', '[aria-label="停止"]', '[aria-label="停止生成"]']
        deadline = asyncio.get_event_loop().time() + 150
        while asyncio.get_event_loop().time() < deadline:
            stop_visible = False
            for sel in stop_selectors:
                try:
                    if await self.page.locator(sel).first.is_visible(timeout=300):
                        stop_visible = True
                        break
                except Exception:
                    pass
            if not stop_visible:
                break
            await asyncio.sleep(2)

        await asyncio.sleep(2)

        # 取结果
        try:
            copy_btn = self.page.locator(
                '[aria-label="复制"], [class*="copy"], button:has(svg[class*="copy"])'
            ).last
            await copy_btn.click()
            await asyncio.sleep(0.5)
            result = await self.page.evaluate("navigator.clipboard.readText()")
            return result.strip()
        except Exception as e:
            return f"ERROR:抓取腾讯元宝内容时异常：{e}"