codebee 0.1.49 → 0.1.51

package/app/main.py

@@ -20,7 +20,17 @@ from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
 from pathlib import Path
 from urllib.parse import parse_qs, unquote, urlparse
 
-from core import automation, catalog, flows, jobs, manager, market, market_remote, registry, remote, settings, store
+# 测试主目录隔离（TUTTI_ISOLATE_HOME=1）：必须先于 core.* 导入生效——
+# modelhub/manager 在 import 期就绑定 ~ 路径。launch/运行前防线会直写
+# ~/.claude 等真实 CLI 配置（2026-09-22 a.test 毒配置案），测试服务一律带上
+# 本开关，绝不落真实用户目录。
+if os.environ.get("TUTTI_ISOLATE_HOME", "").strip() == "1":
+    import tempfile as _tempfile
+    _fake_home = _tempfile.mkdtemp(prefix="tutti-home-")
+    os.environ["USERPROFILE"] = _fake_home
+    os.environ["HOME"] = _fake_home
+
+from core import automation, catalog, flows, jobs, manager, market, market_remote, preview, registry, remote, settings, store
 from core import paths
 from core import health
 import pick_dialog
@@ -54,6 +64,10 @@ def _utf8_bytes(data):
 PORT = 8765  # main() 启动时更新；/api/connect 组装扫码地址用
 _BOOT_TS = time.time()  # 服务进程启动时间戳：/api/pet_state 下发，蜜蜂据此识别「服务换人了」并让位
 
+# SSE 空闲心跳间隔（秒）：长 CLI 步骤期间状态零变化，注释行心跳保持中间层
+# 不掐空闲流、半死连接尽早显形（EventSource 不派发注释事件，纯链路层探活）
+SSE_PING_S = 15.0
+
 # 写接口统一限制 JSON 请求体，避免误传文件或异常客户端把 worker 线程和
 # 内存拖垮。16 MiB 足够覆盖任务上下文、故事圣经和附件清单（附件本体走
 # 独立上传接口）。
@@ -196,6 +210,10 @@ class Handler(BaseHTTPRequestHandler):
         # 静态页不设防（无敏感信息）：远程裸地址打开时由前端令牌门引导输入
         if path in ("/", "/index.html"):
             return self._static("index.html")
+        # 运行预览：工作目录里网页成品的只读挂载，供 iframe/新窗口直接打开。
+        # 令牌走路径段（子资源继承不了查询串），鉴权复用 remote.request_authed。
+        if path == preview.BASE or path.startswith(preview.BASE + "/"):
+            return self._api_preview(path)
         if path.startswith("/api/"):
             if not self._authed():
                 return self._json(401, {"error": "需要访问令牌（启动 CodeBee 时控制台会显示）"})
@@ -390,6 +408,10 @@ class Handler(BaseHTTPRequestHandler):
                     files = []
                 return self._json(200, {"workdir": wd, "files": files,
                                         "task_id": run.get("task_id") or ""})
+            m = re.match(r"^/api/runs/([^/]+)/preview$", path)
+            if m:
+                # 网页成品预览：入口 HTML + 同目录代码文件（前端渲染「预览」页签）
+                return self._json(200, preview.list_app(m.group(1), remote.token()))
             m = re.match(r"^/api/tasks/([^/]+)/side$", path)
             if m:
                 # 任务检查器（右缘停靠列）专用：轻量聚合、可轮询，不带 diff 文本
@@ -1892,6 +1914,7 @@ class Handler(BaseHTTPRequestHandler):
             body = s.get("output") or ""
             if not body:
                 body = s.get("summary") or ""
+            running = (s.get("status") or "") == "running"
             items.append({
                 "kind": "agent",
                 "at": s.get("ended_at") or s.get("started_at") or "",
@@ -1904,6 +1927,13 @@ class Handler(BaseHTTPRequestHandler):
                 "run": run_id_of_step,
                 "log": s.get("log") or "",
                 "followups": s.get("followups") or [],
+                # 思考过程（2026-09-22 用户诉求）：内置智能体流式抓的思维链随步骤
+                # 落库，收尾保留；运行中另有 stream（正文实时预览）与 activity
+                # （工具活动行）——前端边跑边打印，不再只有三点打字动画。
+                "thinking": s.get("thinking") or "",
+                "stream": (s.get("stream") or "") if running else "",
+                "activity": (s.get("activity") or []) if running else [],
+                "live": s.get("live") or 0,
             })
 
         for r in runs:
@@ -2058,30 +2088,49 @@ class Handler(BaseHTTPRequestHandler):
     def _api_events(self):
         """SSE 事件驱动：等 store 状态版本变化才构建/推送，空闲连接几乎零开销
         （此前每连接每 0.8s 盲构建全量 payload，多端并发会把 detect 的慢 IO
-        放大成服务假死）。每 ~2s 醒一次顺带检查控制权变化。"""
+        放大成服务假死）。每 ~2s 醒一次；空闲超过 SSE_PING_S 发注释行心跳。"""
         self.send_response(200)
         self.send_header("Content-Type", "text/event-stream; charset=utf-8")
         self.send_header("Cache-Control", "no-store")
         self.send_header("X-Accel-Buffering", "no")  # 防 nginx/隧道缓冲 SSE
         self.end_headers()
+        try:
+            # 半死连接（对端消失/网络黑洞）的写会无限挂起钉死线程；给足量
+            # 超时让它必然显形成异常 → 线程退出 → 客户端断流重连自愈
+            self.connection.settimeout(300)
+        except Exception:
+            pass
         cid = self._client_id()
         ver = store.state_version()
         sent_ctrl = None
-        n = 0
+        last_send = 0.0
         try:
+            # 连接即推一次当前全量状态：客户端重连（服务重启/网络闪断）后
+            # 不必等下一次状态变化才知道真实进度。长 CLI 步骤一跑十来分钟
+            # 期间一次 bump 都没有，等推送就是等个寂寞（2026-09-22 详情
+            # 「完成」侧栏「在跑」实案）。此前靠 sent_ctrl=None 首轮必不等
+            # 的巧合工作，这里改成显式行为。
+            payload = json.dumps(_state_payload(cid, ver), ensure_ascii=False)
+            self.wfile.write(("data: " + payload + "\n\n").encode("utf-8"))
+            self.wfile.flush()
+            sent_ctrl = remote.control_view(cid)
+            last_send = time.time()
             while True:
                 ver = store.wait_state_change(ver, 2.0)
                 ctrl = remote.control_view(cid)
                 if ver == store.state_version() and ctrl == sent_ctrl:
+                    # 空闲心跳：真发出去的字节让中间层不掐空闲流，也让半死
+                    # 连接在下一次写时尽早报错（注释行，EventSource 不派发事件）
+                    if time.time() - last_send >= SSE_PING_S:
+                        self.wfile.write(b": ping\n\n")
+                        self.wfile.flush()
+                        last_send = time.time()
                     continue  # 超时醒来且无变化
                 payload = json.dumps(_state_payload(cid, ver), ensure_ascii=False)
                 self.wfile.write(("data: " + payload + "\n\n").encode("utf-8"))
                 self.wfile.flush()
                 sent_ctrl = ctrl
-                n += 1
-                if n % 20 == 0:  # ~40s 一次注释行：探活兼防中间层断开空闲连接
-                    self.wfile.write(b": ping\n\n")
-                    self.wfile.flush()
+                last_send = time.time()
         except Exception:
             pass  # 客户端断开是常态，线程随进程退出
 
@@ -2269,6 +2318,45 @@ class Handler(BaseHTTPRequestHandler):
         suffix = p.suffix.lower()
         self._send(200, p.read_bytes(), MIME.get(suffix, "application/octet-stream"))
 
+    # ------------------------------------------------------------ 运行预览
+    def _api_preview(self, path):
+        """GET /preview/<run_id>/<令牌>/<相对路径>：工作目录网页成品的只读挂载。
+
+        令牌放路径段而不是查询串：iframe 里的 style.css / script.js 是浏览器按
+        文档 URL 相对解析的，查询串不会被子资源继承（令牌会掉）。
+        """
+        rest = path[len(preview.BASE):].lstrip("/")
+        parts = rest.split("/")
+        if len(parts) < 2:
+            return self._json(404, {"error": "not found"})
+        run_id, tok = unquote(parts[0]), unquote(parts[1])
+        rel = unquote("/".join(parts[2:]))
+        # 令牌直接交给 remote.request_authed：与 /api/* 同一把尺子（本机豁免、
+        # 代理下带转发头的 loopback 算远程），不另写一套鉴权
+        ip, fw = self._forwarded_ip()
+        if not remote.request_authed(ip, fw, tok, self.headers.get("X-CodeBee-Token") or ""):
+            return self._json(401, {"error": "需要访问令牌（启动 CodeBee 时控制台会显示）"})
+        if not rel or rel.endswith("/"):
+            # 目录地址（/<令牌>/ 或 /<令牌>/web/）：回落到该目录的入口页
+            app = preview.list_app(run_id, tok)
+            if not app.get("ok"):
+                return self._json(404, {"error": "没有可预览的页面"})
+            rel = app["entry"]
+            if not self._preview_dir_ok(path, rel):
+                return self._json(404, {"error": "not found"})
+        data, ctype, err = preview.serve(run_id, rel, tok)
+        if err:
+            return self._json(404, {"error": err})
+        return self._send(200, data, ctype)
+
+    def _preview_dir_ok(self, path, entry):
+        """目录请求只允许回落到该目录自己的入口（/x/ 只解析 x/index.html），
+        免得 /sub/ 悄悄把根入口当子目录入口发出去。"""
+        want = urlparse(path).path[len(preview.BASE):].strip("/").split("/")[2:]
+        want = "/".join(unquote(p) for p in want if p)
+        edir = entry.rsplit("/", 1)[0] if "/" in entry else ""
+        return want == edir
+
 
 def _state_payload(client_id="", ver=None):
     agents = registry.effective_agents(catalog.load(), manager.detect_all())