claude-coding-flow 1.0.0 → 1.1.0

package/dashboard/main.py

@@ -0,0 +1,253 @@
+import argparse
+import os
+from pathlib import Path
+
+from fastapi import FastAPI, HTTPException
+from fastapi.staticfiles import StaticFiles
+from fastapi.responses import FileResponse
+
+try:
+    import db
+    import logger
+    from models import ModuleCreate, TaskCreate, PhaseUpdate, TaskStatusUpdate
+except ImportError:
+    from dashboard import db, logger
+    from dashboard.models import ModuleCreate, TaskCreate, PhaseUpdate, TaskStatusUpdate
+
+app = FastAPI()
+
+STATIC_DIR = Path(__file__).parent / "static"
+app.mount("/static", StaticFiles(directory=str(STATIC_DIR)), name="static")
+
+
+@app.get("/")
+def index():
+    return FileResponse(STATIC_DIR / "index.html")
+
+
+# ── Module API ──
+
+@app.get("/api/modules")
+def list_modules():
+    modules = db.get_modules()
+    for m in modules:
+        tasks = db.get_module_tasks(m["id"])
+        m["task_count"] = len(tasks)
+        m["latest_task"] = tasks[-1]["title"] if tasks else None
+        m.pop("_path", None)
+    return modules
+
+
+@app.get("/api/modules/{module_id}")
+def get_module(module_id: str):
+    module = db.get_module(module_id)
+    if not module:
+        raise HTTPException(status_code=404, detail="Module not found")
+    module.pop("_path", None)
+    module["tasks"] = db.get_module_tasks(module_id)
+    return module
+
+
+@app.post("/api/modules")
+def create_module(body: ModuleCreate):
+    return {"message": "模块由 code-gen 创建，Dashboard 仅读取"}
+
+
+# ── Task API ──
+
+@app.get("/api/tasks")
+def list_tasks():
+    return db.get_tasks()
+
+
+@app.get("/api/tasks/{task_id}")
+def get_task(task_id: str):
+    task = db.get_task(task_id)
+    if not task:
+        raise HTTPException(status_code=404, detail="Task not found")
+    return task
+
+
+@app.post("/api/tasks")
+def create_task(body: TaskCreate):
+    return {"message": "任务由 code-gen 创建，Dashboard 仅读取"}
+
+
+@app.delete("/api/tasks/{task_id}")
+def delete_task(task_id: str):
+    import shutil
+    task_dir = db.get_task_dir(task_id)
+    if not task_dir:
+        raise HTTPException(status_code=404, detail="Task not found")
+    shutil.rmtree(task_dir, ignore_errors=True)
+    return {"ok": True}
+
+
+# ── Log & Snapshot API ──
+
+@app.get("/api/tasks/{task_id}/logs")
+def get_logs(task_id: str):
+    task_dir = db.get_task_dir(task_id)
+    if not task_dir:
+        raise HTTPException(status_code=404, detail="Task not found")
+    return {
+        "logs": logger.get_logs(task_dir),
+        "token_summary": logger.get_token_summary(task_dir),
+        "text_log": logger.get_text_log(task_dir),
+    }
+
+
+@app.get("/api/tasks/{task_id}/snapshots")
+def get_snapshots(task_id: str):
+    task_dir = db.get_task_dir(task_id)
+    if not task_dir:
+        raise HTTPException(status_code=404, detail="Task not found")
+    return logger.list_snapshots(task_dir)
+
+
+@app.get("/api/tasks/{task_id}/snapshots/{snapshot_name}/{filepath:path}")
+def get_snapshot_file(task_id: str, snapshot_name: str, filepath: str):
+    task_dir = db.get_task_dir(task_id)
+    if not task_dir:
+        raise HTTPException(status_code=404, detail="Task not found")
+    content = logger.get_snapshot_file(task_dir, snapshot_name, filepath)
+    if content is None:
+        raise HTTPException(status_code=404, detail="Snapshot file not found")
+    return {"file": filepath, "content": content}
+
+
+@app.get("/api/tasks/{task_id}/plan")
+def get_plan(task_id: str):
+    task_dir = db.get_task_dir(task_id)
+    if not task_dir:
+        raise HTTPException(status_code=404, detail="Task not found")
+    plan = logger.get_plan(task_dir)
+    if plan is None:
+        raise HTTPException(status_code=404, detail="Plan not found")
+    return {"plan": plan}
+
+
+# ── 按类型过滤模块 ──
+
+@app.get("/api/modules/type/{module_type}")
+def list_modules_by_type(module_type: str):
+    if module_type not in ("code-gen", "doc-gen", "bug-fix"):
+        raise HTTPException(status_code=400, detail="Invalid module type")
+    modules = db.get_modules_by_type(module_type)
+    for m in modules:
+        tasks = db.get_module_tasks(m["id"])
+        m["task_count"] = len(tasks)
+        m["latest_task"] = tasks[-1]["title"] if tasks else None
+        m.pop("_path", None)
+    return modules
+
+
+# ── doc-gen 产物 API ──
+
+@app.get("/api/tasks/{task_id}/doc-artifacts")
+def get_doc_artifacts(task_id: str):
+    task_dir = db.get_task_dir(task_id)
+    if not task_dir:
+        raise HTTPException(status_code=404, detail="Task not found")
+    return db.get_doc_gen_artifacts(task_dir)
+
+
+@app.get("/api/tasks/{task_id}/doc-artifacts/{artifact_type}/{filepath:path}")
+def get_doc_artifact_file(task_id: str, artifact_type: str, filepath: str):
+    task_dir = db.get_task_dir(task_id)
+    if not task_dir:
+        raise HTTPException(status_code=404, detail="Task not found")
+
+    if artifact_type == "develop":
+        path = os.path.join(task_dir, filepath)
+    elif artifact_type in ("requirement", "images"):
+        path = os.path.join(task_dir, artifact_type, filepath)
+    else:
+        raise HTTPException(status_code=400, detail="Invalid artifact type")
+
+    if not os.path.exists(path):
+        raise HTTPException(status_code=404, detail="File not found")
+
+    if filepath.lower().endswith(('.png', '.jpg', '.jpeg', '.gif', '.webp', '.bmp')):
+        return FileResponse(path)
+
+    try:
+        with open(path, "r", encoding="utf-8") as f:
+            return {"content": f.read()}
+    except UnicodeDecodeError:
+        return FileResponse(path)
+
+
+# ── bug-fix 信息 API ──
+
+@app.get("/api/tasks/{task_id}/bug-info")
+def get_bug_info(task_id: str):
+    task_dir = db.get_task_dir(task_id)
+    if not task_dir:
+        raise HTTPException(status_code=404, detail="Task not found")
+    return db.get_bug_fix_info(task_dir)
+
+
+# ── 任务链 API ──
+
+@app.get("/api/modules/{module_id}/task-chain")
+def get_task_chain(module_id: str):
+    module = db.get_module(module_id)
+    if not module:
+        raise HTTPException(status_code=404, detail="Module not found")
+    return db.get_task_chain(module_id)
+
+
+@app.get("/api/tasks/{task_id}/cross-relations")
+def get_cross_relations(task_id: str):
+    task = db.get_task(task_id)
+    if not task:
+        raise HTTPException(status_code=404, detail="Task not found")
+    return db.get_cross_relations(task_id)
+
+
+# ── CLI ──
+
+def cli():
+    parser = argparse.ArgumentParser(description="Code-Gen Dashboard CLI")
+    sub = parser.add_subparsers(dest="command")
+
+    # serve
+    sub.add_parser("serve", help="启动 Dashboard 服务")
+
+    # list modules
+    sub.add_parser("modules", help="列出所有模块")
+
+    # scan
+    sub.add_parser("scan", help="扫描 worktree 目录并显示信息")
+
+    args = parser.parse_args()
+
+    if args.command == "modules":
+        for m in db.get_modules():
+            tasks = db.get_module_tasks(m["id"])
+            print(f"{m['id']}  {m['name']}  ({len(tasks)} tasks)")
+
+    elif args.command == "scan":
+        modules = db.get_modules()
+        if not modules:
+            print("worktree 为空，暂无模块")
+        for m in modules:
+            tasks = db.get_module_tasks(m["id"])
+            print(f"\n模块: {m['name']} ({m['id']})")
+            for t in tasks:
+                phase_status = " ".join(
+                    f"P{p['phase']}:{p['status']}" for p in t.get("phases", [])
+                )
+                print(f"  任务: {t['title']} ({t['id']}) [{t['status']}] {phase_status}")
+
+    elif args.command == "serve":
+        import uvicorn
+        uvicorn.run(app, host="0.0.0.0", port=8000)
+
+    else:
+        parser.print_help()
+
+
+if __name__ == "__main__":
+    cli()

package/dashboard/models.py

@@ -0,0 +1,26 @@
+from typing import Optional, List
+from pydantic import BaseModel
+
+
+class ModuleCreate(BaseModel):
+    name: str
+    type: str = "code-gen"
+    id: Optional[str] = None
+
+
+class TaskCreate(BaseModel):
+    id: str
+    module_id: str
+    title: str
+    requirements_doc: str = ""
+
+
+class PhaseUpdate(BaseModel):
+    phase: int
+    status: str  # running / completed / failed
+    detail: Optional[str] = None
+    error: Optional[str] = None
+
+
+class TaskStatusUpdate(BaseModel):
+    status: str  # running / completed / failed

package/dashboard/requirements.txt

@@ -0,0 +1,2 @@
+fastapi
+uvicorn