@agenthippo-ai/agenthippo 0.1.0

package/out/scripts/analytics/fleet-workflow.sh

@@ -0,0 +1,263 @@
+#!/usr/bin/env bash
+# Workflows view writer: emit human-friendly multi-agent workflow artifacts.
+# Seeded as a sidecar of the built-in Workflows view; writes to the workspace
+# (AGENTIDE_WORKSPACE_ROOT or pwd), independent of where this script is seeded.
+# Active:  <workspace>/.agent-hippo/fleet-workflows/<workflowId>.json
+# History: <workspace>/.agent-hippo/fleet-workflows/history/<workflowId>-<runId>.json
+set -euo pipefail
+
+WORKSPACE="${AGENTIDE_WORKSPACE_ROOT:-$(pwd)}"
+OUT_DIR="$WORKSPACE/.agent-hippo/fleet-workflows"
+HISTORY_DIR="$OUT_DIR/history"
+mkdir -p "$OUT_DIR" "$HISTORY_DIR"
+
+CMD="${1:-}"
+shift || true
+
+workflow="" title="" subtitle="" orchestrator="" orch_status="" orch_summary=""
+step="" agent="" label="" status="" summary="" artifact=""
+
+while [[ $# -gt 0 ]]; do
+	case "$1" in
+		--workflow) workflow="$2"; shift 2 ;;
+		--title) title="$2"; shift 2 ;;
+		--subtitle) subtitle="$2"; shift 2 ;;
+		--orchestrator) orchestrator="$2"; shift 2 ;;
+		--status) status="$2"; shift 2 ;;
+		--summary) summary="$2"; shift 2 ;;
+		--step) step="$2"; shift 2 ;;
+		--agent) agent="$2"; shift 2 ;;
+		--label) label="$2"; shift 2 ;;
+		--artifact) artifact="$2"; shift 2 ;;
+		*) echo "Unknown arg: $1" >&2; exit 1 ;;
+	esac
+done
+
+if [[ -z "$workflow" ]]; then
+	echo "Usage: fleet-workflow.sh {init|orchestrator|step|finish} --workflow ID ..." >&2
+	exit 1
+fi
+
+OUT_FILE="$OUT_DIR/${workflow}.json"
+NOW="$(date -u +"%Y-%m-%dT%H:%M:%SZ")"
+
+export _FW_WORKFLOW="$workflow" _FW_TITLE="$title" _FW_SUBTITLE="$subtitle" \
+	_FW_ORCH="$orchestrator" _FW_ORCH_STATUS="$orch_status" _FW_ORCH_SUMMARY="$orch_summary" \
+	_FW_STEP="$step" _FW_AGENT="$agent" _FW_LABEL="$label" _FW_STATUS="$status" _FW_SUMMARY="$summary" \
+	_FW_ARTIFACT="$artifact" _FW_HISTORY_DIR="$HISTORY_DIR"
+
+python3 - "$CMD" "$OUT_FILE" "$NOW" <<'PY'
+import json, os, re, sys, shutil
+
+cmd, out_file, now = sys.argv[1:4]
+history_dir = os.environ.get("_FW_HISTORY_DIR", "")
+env = {
+    "workflow": os.environ.get("_FW_WORKFLOW", ""),
+    "title": os.environ.get("_FW_TITLE", ""),
+    "subtitle": os.environ.get("_FW_SUBTITLE", ""),
+    "orchestrator": os.environ.get("_FW_ORCH", ""),
+    "orch_status": os.environ.get("_FW_ORCH_STATUS", ""),
+    "orch_summary": os.environ.get("_FW_ORCH_SUMMARY", ""),
+    "step": os.environ.get("_FW_STEP", ""),
+    "agent": os.environ.get("_FW_AGENT", ""),
+    "label": os.environ.get("_FW_LABEL", ""),
+    "status": os.environ.get("_FW_STATUS", ""),
+    "summary": os.environ.get("_FW_SUMMARY", ""),
+    "artifact": os.environ.get("_FW_ARTIFACT", ""),
+}
+
+STEP_COLORS = {1: "#fbbf24", 2: "#38bdf8", 3: "#a78bfa", 4: "#34d399", 5: "#f472b6"}
+
+
+def run_id_from_started(started_at: str) -> str:
+    raw = (started_at or now).replace(":", "").replace("-", "").replace("T", "-").replace("Z", "")
+    return re.sub(r"[^0-9A-Za-z-]", "", raw)[:24] or "run"
+
+
+def load_doc(path=out_file):
+    if os.path.isfile(path):
+        try:
+            with open(path, encoding="utf-8") as f:
+                return json.load(f)
+        except (json.JSONDecodeError, OSError):
+            pass
+    return None
+
+
+def fresh_doc():
+    run_id = run_id_from_started(now)
+    return {
+        "workflowId": env["workflow"],
+        "runId": run_id,
+        "title": env["title"] or env["workflow"],
+        "subtitle": "",
+        "status": "in_progress",
+        "startedAt": now,
+        "updatedAt": now,
+        "finishedAt": "",
+        "orchestrator": {
+            "agent": env["orchestrator"] or "coordinator",
+            "label": "Orchestrator",
+            "status": "pending",
+            "summary": "",
+        },
+        "steps": [],
+    }
+
+
+def save_doc(doc, path=out_file):
+    doc["updatedAt"] = now
+    os.makedirs(os.path.dirname(path), exist_ok=True)
+    tmp = path + ".tmp"
+    with open(tmp, "w", encoding="utf-8") as f:
+        json.dump(doc, f, indent=2)
+        f.write("\n")
+    os.replace(tmp, path)
+
+
+def archive_doc(doc):
+    if not history_dir or not doc:
+        return None
+    wf_id = doc.get("workflowId") or env["workflow"]
+    run_id = doc.get("runId") or run_id_from_started(doc.get("startedAt", now))
+    doc = dict(doc)
+    doc["archivedAt"] = now
+    doc["kind"] = "history"
+    dest = os.path.join(history_dir, f"{wf_id}-{run_id}.json")
+    doc["artifact"] = dest
+    save_doc(doc, dest)
+    return dest
+
+
+def upsert_step(doc, step_num, agent, label, status, summary, artifact=""):
+    steps = doc.setdefault("steps", [])
+    found = next((s for s in steps if int(s.get("step", 0)) == step_num), None)
+    if not found:
+        found = {
+            "step": step_num,
+            "role": "worker",
+            "agent": agent,
+            "label": label or agent,
+            "status": "pending",
+            "summary": "",
+            "startedAt": "",
+            "finishedAt": "",
+            "artifact": "",
+            "color": STEP_COLORS.get(step_num, "#94a3b8"),
+        }
+        steps.append(found)
+    if agent:
+        found["agent"] = agent
+    if label:
+        found["label"] = label
+    if status:
+        found["status"] = status
+        if status == "in_progress" and not found.get("startedAt"):
+            found["startedAt"] = now
+        if status == "done":
+            found["finishedAt"] = now
+    if summary:
+        found["summary"] = summary
+    if artifact:
+        found["artifact"] = artifact
+    found["color"] = STEP_COLORS.get(step_num, found.get("color", "#94a3b8"))
+    steps.sort(key=lambda s: int(s.get("step", 0)))
+
+
+def derive_workflow_status(doc):
+    steps = doc.get("steps") or []
+    orch = doc.get("orchestrator") or {}
+    if orch.get("status") == "done" and steps and all(s.get("status") == "done" for s in steps):
+        return "done"
+    if any(s.get("status") == "in_progress" for s in steps):
+        return "in_progress"
+    if orch.get("status") == "in_progress":
+        return "in_progress"
+    if steps and all(s.get("status") == "done" for s in steps):
+        return "done"
+    return doc.get("status") or "in_progress"
+
+
+def sync_orch_summary(doc):
+    steps = doc.get("steps") or []
+    done_count = sum(1 for s in steps if s.get("status") == "done")
+    total = len(steps)
+    active = next((s for s in steps if s.get("status") == "in_progress"), None)
+    orch = doc.setdefault("orchestrator", {})
+    if active:
+        orch["summary"] = f"Running: {active.get('label', active.get('agent', ''))}"
+    elif done_count == total and total > 0:
+        orch["summary"] = f"All {total} steps complete"
+    elif total:
+        orch["summary"] = f"{done_count} of {total} steps done"
+
+
+existing = load_doc()
+doc = existing if existing else fresh_doc()
+doc["workflowId"] = env["workflow"] or doc.get("workflowId", "")
+doc.setdefault("kind", "active")
+
+if cmd == "init":
+    if existing and existing.get("status") == "done":
+        archive_doc(existing)
+        doc = fresh_doc()
+    elif not existing:
+        doc = fresh_doc()
+    if env["title"]:
+        doc["title"] = env["title"]
+    if env["subtitle"]:
+        doc["subtitle"] = env["subtitle"]
+    if env["orchestrator"]:
+        doc["orchestrator"]["agent"] = env["orchestrator"]
+        doc["orchestrator"]["label"] = env["orchestrator"].replace("-", " ").title()
+    doc["status"] = "in_progress"
+    doc["orchestrator"]["status"] = "in_progress"
+    doc["finishedAt"] = ""
+    doc["startedAt"] = now
+    doc["runId"] = run_id_from_started(now)
+    doc["kind"] = "active"
+
+elif cmd == "orchestrator":
+    orch = doc.setdefault("orchestrator", {})
+    if env["orch_status"]:
+        orch["status"] = env["orch_status"]
+    if env["orch_summary"]:
+        orch["summary"] = env["orch_summary"]
+    doc["status"] = derive_workflow_status(doc)
+
+elif cmd == "step":
+    step_num = int(env["step"] or "0")
+    if step_num < 1:
+        raise SystemExit("step requires --step >= 1")
+    upsert_step(doc, step_num, env["agent"], env["label"], env["status"] or "in_progress", env["summary"], env["artifact"])
+    doc["status"] = derive_workflow_status(doc)
+    sync_orch_summary(doc)
+
+elif cmd == "finish":
+    doc["status"] = "done"
+    doc["orchestrator"]["status"] = "done"
+    doc["finishedAt"] = now
+    if env["summary"]:
+        doc["orchestrator"]["summary"] = env["summary"]
+    else:
+        doc["orchestrator"]["summary"] = "Pipeline complete"
+    for s in doc.get("steps", []):
+        if s.get("status") not in ("done", "failed"):
+            s["status"] = "done"
+    archived = archive_doc(doc)
+    try:
+        os.remove(out_file)
+    except FileNotFoundError:
+        pass
+    if archived:
+        print(f"Fleet workflow → archived {os.path.basename(archived)} (done)")
+    else:
+        print(f"Fleet workflow → removed active slot (done)")
+    sys.exit(0)
+
+else:
+    raise SystemExit(f"Unknown command: {cmd}")
+
+save_doc(doc)
+print(f"Fleet workflow → {os.path.basename(out_file)} ({doc['status']})")
+PY