项目初始化

bot/router.py

@@ -0,0 +1,109 @@
+"""消息路由：skills 快查 / 网页操作 / 截图预览 / Cursor 通用任务。"""
+
+from __future__ import annotations
+
+import re
+
+import env_config
+from browser_parser import is_browser_intent, parse_browser_request
+from browser_service import format_browser_caption, run_browser_automation
+from cursor_runner import run_cursor_task, strip_mention
+from image_extract import find_image_paths, strip_fake_image_markdown
+from preview_service import capture_preview, is_preview_intent, resolve_preview_request
+from skills_service import handle_command, parse_command
+from bot_types import RouteResult
+
+
+def routing_mode() -> str:
+    return (env_config.env("ROUTING_MODE", "hybrid") or "hybrid").lower()
+
+
+def _normalize(text: str) -> str:
+    return re.sub(r"@\S+\s*", "", text).strip().lower()
+
+
+def is_skills_fast_command(text: str) -> bool:
+    raw = _normalize(text)
+    if not raw:
+        return True
+    if raw in {"help", "帮助", "?", "h"}:
+        return True
+
+    cmd = parse_command(text)
+    if cmd.kind in {"help", "list", "detail"}:
+        return True
+    if cmd.kind == "search" and re.match(r"^(search|搜索|find|查)\s+", raw):
+        return True
+    return False
+
+
+async def _run_browser(text: str, on_progress=None) -> RouteResult:
+    if parse_browser_request(text) is None:
+        raise RuntimeError("无法解析网页操作步骤")
+
+    if on_progress:
+        await on_progress("正在按步骤执行网页操作…")
+
+    result = await run_browser_automation(text)
+    return RouteResult(
+        source="browser",
+        text=format_browser_caption(result),
+        image_path=str(result.screenshot_path),
+    )
+
+
+async def _run_preview(text: str, on_progress=None) -> RouteResult:
+    preview_req = resolve_preview_request(text)
+    if preview_req is None:
+        raise RuntimeError("无法解析截图请求")
+
+    if on_progress:
+        await on_progress(f"正在访问并截图：{preview_req.url or '默认地址'}…")
+
+    result = await capture_preview(preview_req.url, preview_req.port)
+    caption = (
+        f"**页面预览**\n"
+        f"> URL：`{result.final_url or result.url}`\n"
+        f"> 项目：`{env_config.env('CURSOR_CWD', '')}`\n"
+        f"> dev server：{'已自动启动' if result.started_dev_server else '使用已有服务'}"
+    )
+    return RouteResult(
+        source="preview",
+        text=caption,
+        image_path=str(result.screenshot_path),
+    )
+
+
+async def route_message(text: str, on_progress=None) -> RouteResult:
+    task = strip_mention(text)
+    if not task:
+        return RouteResult("skills", handle_command("help"))
+
+    if is_browser_intent(text):
+        return await _run_browser(text, on_progress=on_progress)
+
+    if resolve_preview_request(text) is not None:
+        return await _run_preview(text, on_progress=on_progress)
+
+    mode = routing_mode()
+    if mode == "skills":
+        return RouteResult("skills", handle_command(text))
+
+    if mode == "cursor" or not is_skills_fast_command(text):
+        reply = await run_cursor_task(task, on_progress=on_progress)
+        reply = strip_fake_image_markdown(reply)
+
+        image_path: str | None = None
+        paths = find_image_paths(reply)
+        if paths:
+            image_path = str(paths[0])
+        elif is_preview_intent(text) or is_browser_intent(text):
+            if on_progress:
+                await on_progress("未找到截图文件，改用 Playwright 自动执行…")
+            if is_browser_intent(text):
+                return await _run_browser(text, on_progress=on_progress)
+            return await _run_preview(text, on_progress=on_progress)
+
+        return RouteResult("cursor", reply, image_path=image_path)
+
+    return RouteResult("skills", handle_command(text))