codebee 0.1.14 → 0.1.16

package/app/main.py

@@ -304,6 +304,28 @@ class Handler(BaseHTTPRequestHandler):
                     return self._send(200, "（报告尚未生成）", "text/markdown; charset=utf-8")
                 return self._send(200, p.read_text(encoding="utf-8", errors="replace"),
                                   "text/markdown; charset=utf-8")
+            m = re.match(r"^/api/runs/([^/]+)/cover$", path)
+            if m:
+                # 封面图在运行目录而非任务工作目录，「成果」扫描够不着它，
+                # 前端封面卡内嵌缩略图走这条专用通道（文件名固定无穿越面）
+                run = store.get_run(m.group(1))
+                if not run:
+                    return self._json(404, {"error": "not found"})
+                p = paths.RUNS_DIR / m.group(1) / "cover.png"
+                if not p.is_file():
+                    return self._json(404, {"error": "封面尚未生成"})
+                return self._send(200, p.read_bytes(), "image/png")
+            m = re.match(r"^/api/runs/([^/]+)/share$", path)
+            if m:
+                from core import share_page
+                run = store.get_run(m.group(1))
+                if not run:
+                    return self._json(404, {"error": "not found"})
+                task = store.get_task(run.get("task_id") or "") if run.get("task_id") else None
+                rp = paths.RUNS_DIR / m.group(1) / "report.md"
+                report = rp.read_text(encoding="utf-8", errors="replace") if rp.is_file() else ""
+                html = share_page.render_share_html(run, task, report)
+                return self._send(200, html, "text/html; charset=utf-8")
             m = re.match(r"^/api/runs/([^/]+)/files$", path)
             if m:
                 run = store.get_run(m.group(1))
@@ -437,6 +459,9 @@ class Handler(BaseHTTPRequestHandler):
             if path == "/api/automation":
                 return self._json(200, {"tasks": automation.list_tasks(),
                                         "templates": automation.templates()})
+            if path == "/api/zentao":
+                from core import zentao
+                return self._json(200, zentao.view())
             m = re.match(r"^/api/automation/([^/]+)$", path)
             if m:
                 t = automation.get_task(m.group(1))
@@ -724,6 +749,10 @@ class Handler(BaseHTTPRequestHandler):
         if path == "/api/models/refresh":
             from core import modelhub
             n, err = modelhub.refresh_models(self._body().get("id") or "")
+            if modelhub.is_no_list_note(err):
+                # 网关无 /models 列表接口（zcode-plan 等）：预期情况按成功回报，
+                # 前端给中性提示而非红色「获取失败」
+                return self._json(200, {"ok": True, "count": 0, "note": err})
             return self._json(200, {"ok": bool(n), "count": n, "message": err})
         if path == "/api/models/add":
             # 手工添加模型：厂商列表接口调不通时直接填模型名进列表
@@ -955,6 +984,30 @@ class Handler(BaseHTTPRequestHandler):
                 return self._json(200, {"ok": True, "task": t, "run_id": run_id or ""})
             ok = automation.delete(tid)
             return self._json(200, {"ok": True}) if ok else self._json(404, {"error": "not found"})
+        if path == "/api/zentao/config":
+            from core import zentao
+            try:
+                cfg = zentao.save_config(self._body())
+            except ValueError as e:
+                return self._json(400, {"error": str(e)})
+            return self._json(200, {"ok": True, "config": cfg})
+        if path == "/api/zentao/test":
+            from core import zentao
+            body = self._body()
+            pw = str(body.get("password") or "").strip()
+            ok, msg = zentao.test_connection(
+                base_url=body.get("base_url"), account=body.get("account"),
+                password=pw or None)     # 空 = 用已保存密码
+            return self._json(200, {"ok": ok, "message": msg})
+        if path == "/api/zentao/scan":
+            from core import zentao
+            res = zentao.scan_now()      # 网络操作同步做（ThreadingHTTPServer 不堵别的请求）
+            return self._json(200, dict(res, ok=bool(res.get("ok"))))
+        if path == "/api/zentao/modules":
+            from core import zentao
+            pid = (self._body() or {}).get("product")
+            res = zentao.fetch_modules(pid)
+            return self._json(200, dict(res, ok=bool(res.get("ok"))))
         # 外部目录路由必须在通配的 market/<id>/(install|remove) 之前——
         # 否则 "remote" 会被当成包名吞掉
         if path == "/api/market/remote/refresh":
@@ -1949,6 +2002,13 @@ def main():
     def _step(msg):
         print("[CodeBee] %s (%.1fs)" % (msg, time.time() - _t0), flush=True)
 
+    _ver = ""
+    try:
+        _pj = Path(__file__).resolve().parents[1] / "package.json"
+        _ver = str(json.loads(_pj.read_text(encoding="utf-8")).get("version") or "")
+    except Exception:
+        pass
+    print("[CodeBee] v%s" % (_ver or "?"), flush=True)
     _step("正在准备数据目录…")
     paths.ensure_dirs()
     from core import attachments
@@ -2040,6 +2100,13 @@ def main():
     n_auto = automation.start()   # 自动化：加载定时任务并拉起调度线程（错过的一次性任务不补跑）
     if n_auto:
         print("[CodeBee] 自动化：%d 个定时任务已加载" % n_auto)
+    try:
+        from core import zentao
+        n_zt = zentao.start()   # 禅道集成：加载扫描配置与认领台账（调度挂在 automation tick）
+        if n_zt:
+            print("[CodeBee] 禅道集成：%d 条认领记录已加载" % n_zt)
+    except Exception:
+        pass
     tok = remote.token()
     import atexit
     import os as _os
@@ -2056,7 +2123,8 @@ def main():
     except OSError as e:
         # Windows 的 SO_REUSEADDR 允许两个进程同时 LISTEN 同一端口（请求随机
         # 分发，表现为"时好时坏"）；加独占锁后双起在这里干净失败并指路。
-        import os
+        # （此处不能局部 import os：会让 os 变 main() 的局部名，后面 2127 行
+        #  的 os.name 直接 UnboundLocalError——顶部已有全局导入，直接用）
         hint = ""
         if os.name == "nt":
             hint = ("（Windows 排查：netstat -ano | findstr :%d 找到 PID，"
@@ -2084,14 +2152,17 @@ def main():
               % (remote.PUBLIC_URL, tok))
     elif not args.no_public_tunnel and args.host != "127.0.0.1":
         if remote.has_local_creds():
+            extra = "，或 start-public.bat" if os.name == "nt" else ""
             print("[CodeBee] （检测到已有 Cloudflare 隧道凭据：临时隧道在此类机器上不可用，"
-                  "请用 --public-url 配固定域名，或 start-public.bat）")
+                  "请用 --public-url 配固定域名%s）" % extra)
         elif remote.start_quick_tunnel(args.port, _announce_public):
             print("[CodeBee] 正在建立 Cloudflare 快速隧道（公网地址几秒后打印；"
                   "若打不开说明当前网络不支持，可改用固定域名或 Tailscale）…")
         else:
+            cf = ("winget install Cloudflare.cloudflared" if os.name == "nt"
+                  else "brew install cloudflared")
             print("[CodeBee] （未检测到 cloudflared，跳过公网隧道；"
-                  "winget install Cloudflare.cloudflared 后重启即可获得公网地址）")
+                  "%s 后重启即可获得公网地址）" % cf)
     atexit.register(remote.stop_quick_tunnel)
     if args.host != "127.0.0.1":
         lan = remote.lan_ip()
@@ -2105,9 +2176,21 @@ def main():
         print("[CodeBee] 远程访问受令牌保护；手机打开一次带 token 的地址后会记住。")
     if not args.no_browser:
         threading.Timer(1.0, lambda: webbrowser.open("http://127.0.0.1:%d" % args.port)).start()
+    if os.name != "nt":
+        # POSIX 的 SIGTERM（kill <pid>）/SIGHUP（关终端窗口）默认动作是立即
+        # 终止进程：atexit 不跑，快速隧道的 cloudflared 子进程会变孤儿。转成
+        # SystemExit 走正常退出路径；Ctrl+C 的 KeyboardInterrupt 本就触发
+        # atexit，无需登记。Windows 语义不同，不碰。
+        import signal as _signal
+
+        def _graceful_exit(signum, frame):
+            raise SystemExit(0)
+
+        _signal.signal(_signal.SIGTERM, _graceful_exit)
+        _signal.signal(_signal.SIGHUP, _graceful_exit)
     try:
         httpd.serve_forever()
-    except KeyboardInterrupt:
+    except (KeyboardInterrupt, SystemExit):
         print("\n[CodeBee] 已退出")