@hupan56/wlkj 2.7.12 → 3.1.2

package/templates/qoder/scripts/domain/report/status_snapshot.py

@@ -0,0 +1,191 @@
+#!/usr/bin/env python3
+"""
+status_snapshot.py — /wl-status 的一次性数据聚合器 (性能优化)
+
+问题: wl-status skill 让 AI 逐个 cat/ls 收集数据, 实测产生 5 轮迭代、10+ 次工具调用。
+本脚本把 status 需要的全部数据源**一次返回**, 把 5 轮迭代压成 1 轮。
+
+聚合的数据源 (来自 wl-status SKILL.md):
+  1. .developer          — 当前开发者
+  2. .index-meta.json    — 图谱新鲜度
+  3. eval-history.jsonl  — EVA 评估历史 (健康度趋势)
+  4. feedback.jsonl      — 学习反馈记录
+  5. config.yaml         — 项目配置
+  6. task.py list        — 任务列表
+  7. team_sync.py status — 团队同步状态
+  8. tasks 目录          — 任务文件清单
+  9. learning/patterns   — 学习模式
+
+输出: JSON, 缺失的字段标 null 并注明原因。AI 拿到后直接算健康度, 不用再探查。
+
+用法:
+  python status_snapshot.py            # 当前仓库
+  python status_snapshot.py /path/to/repo
+"""
+import json
+import os
+import subprocess
+import sys
+from datetime import datetime
+
+
+def _repo_root():
+    """优先用命令行参数, 否则从环境反推。"""
+    if len(sys.argv) > 1 and os.path.isdir(sys.argv[1]):
+        return os.path.abspath(sys.argv[1])
+    # 同 task.py 的逻辑: 找 .qoder 标记
+    d = os.getcwd()
+    for _ in range(8):
+        if os.path.isdir(os.path.join(d, ".qoder")):
+            return d
+        d = os.path.dirname(d)
+    return os.getcwd()
+
+
+def _read_text(path, limit=20000):
+    """安全读文本文件, 不存在返回 None。"""
+    if not os.path.exists(path):
+        return None
+    try:
+        with open(path, "r", encoding="utf-8") as f:
+            return f.read(limit)
+    except Exception as e:
+        return f"<read_error: {e}>"
+
+
+def _read_json(path):
+    if not os.path.exists(path):
+        return None
+    try:
+        with open(path, "r", encoding="utf-8") as f:
+            return json.load(f)
+    except Exception as e:
+        return {"_error": str(e)}
+
+
+def _read_jsonl_tail(path, n=10):
+    """读 jsonl 最后 n 行 (评估/反馈历史, 只关心最近的)。"""
+    if not os.path.exists(path):
+        return None
+    try:
+        with open(path, "r", encoding="utf-8") as f:
+            lines = [l.strip() for l in f if l.strip()]
+        out = []
+        for l in lines[-n:]:
+            try:
+                out.append(json.loads(l))
+            except json.JSONDecodeError:
+                continue
+        return out
+    except Exception:
+        return None
+
+
+def _run_script(script_rel, repo, args=None, timeout=20):
+    """运行 .qoder/scripts 下的脚本, 返回 (stdout, ok)。"""
+    script = os.path.join(repo, script_rel)
+    if not os.path.exists(script):
+        return None, False
+    try:
+        cmd = [sys.executable, script] + (args or [])
+        r = subprocess.run(
+            cmd, cwd=repo, capture_output=True, text=True, timeout=timeout,
+            encoding="utf-8", errors="replace",
+        )
+        return r.stdout.strip() if r.returncode == 0 else r.stderr.strip(), r.returncode == 0
+    except Exception as e:
+        return f"<run_error: {e}>", False
+
+
+def _list_dir(path):
+    if not os.path.isdir(path):
+        return None
+    try:
+        items = sorted(os.listdir(path))
+        return items[:50]  # 限 50 项防爆
+    except Exception:
+        return None
+
+
+def _index_freshness(meta):
+    """从 .index-meta.json 算新鲜度(距今天数)。"""
+    if not isinstance(meta, dict):
+        return None
+    last = meta.get("last_sync") or meta.get("last_build") or meta.get("updated")
+    if not last:
+        return None
+    try:
+        # 兼容多种时间格式
+        ts = str(last).replace("Z", "+00:00")
+        dt = datetime.fromisoformat(ts)
+        if dt.tzinfo:
+            dt = dt.replace(tzinfo=None)
+        return (datetime.now() - dt).days
+    except Exception:
+        return None
+
+
+def snapshot(repo=None):
+    """收集 wl-status 需要的全部数据, 返回 dict。"""
+    repo = repo or _repo_root()
+    qd = os.path.join(repo, ".qoder")
+
+    data = {"_repo": repo, "_ts": datetime.now().isoformat(timespec="seconds")}
+
+    # 1. 开发者身份
+    data["developer"] = _read_text(os.path.join(qd, ".developer")) or "(未设置)"
+
+    # 2. 图谱索引新鲜度
+    meta = _read_json(os.path.join(repo, "data", "index", ".index-meta.json"))
+    data["index_meta"] = meta
+    data["index_age_days"] = _index_freshness(meta)
+
+    # 3. EVA 评估历史 (健康度趋势)
+    data["eval_history"] = _read_jsonl_tail(
+        os.path.join(qd, "learning", "eval-history.jsonl"), n=10)
+
+    # 4. 学习反馈
+    data["feedback_count"] = None
+    fb = os.path.join(qd, "learning", "feedback.jsonl")
+    if os.path.exists(fb):
+        try:
+            with open(fb, encoding="utf-8") as f:
+                data["feedback_count"] = sum(1 for _ in f)
+        except Exception:
+            pass
+
+    # 5. 项目配置 (只取关键摘要, 防过大)
+    cfg = _read_text(os.path.join(qd, "config.yaml"), limit=4000)
+    data["config_summary"] = cfg[:4000] if cfg else None
+
+    # 6. 任务列表 (调 task.py list)
+    task_out, task_ok = _run_script(".qoder/scripts/domain/task/task.py", repo, args=["list"])
+    data["task_list"] = task_out if task_ok else None
+    data["task_list_error"] = None if task_ok else task_out
+
+    # 7. 团队同步状态 (调 team_sync.py status)
+    sync_out, sync_ok = _run_script(".qoder/scripts/domain/task/team_sync.py", repo,
+                                    args=["status"], timeout=15)
+    data["team_sync"] = sync_out if sync_ok else None
+    data["team_sync_error"] = None if sync_ok else sync_out
+
+    # 8. 任务目录结构
+    data["tasks_dir"] = _list_dir(os.path.join(qd, "workspace", "tasks"))
+
+    # 9. 学习模式
+    data["patterns"] = _list_dir(os.path.join(qd, "learning", "patterns"))
+
+    # 10. 成员目录 (团队规模)
+    data["members"] = _list_dir(os.path.join(qd, "workspace", "members"))
+
+    return data
+
+
+def main():
+    repo = sys.argv[1] if len(sys.argv) > 1 else None
+    data = snapshot(repo)
+    print(json.dumps(data, ensure_ascii=False, indent=2, default=str))
+
+
+if __name__ == "__main__":
+    main()