@andespindola/crow-skills 0.8.0 → 0.10.0

package/crow_skills/progress.py

@@ -83,7 +83,7 @@ def validate(data):
            "status": data.get("status") if data.get("status") in ("running",) + FINISHED else "running",
            "startedAt": data.get("startedAt"), "updatedAt": data.get("updatedAt"), "endedAt": data.get("endedAt"),
            "session": data.get("session") if isinstance(data.get("session"), dict) else {},
-           "phases": [], "units": []}
+           "phases": [], "units": [], "resources": resources_decision(data.get("resources"))}
     for phase in phases:
         if isinstance(phase, dict) and isinstance(phase.get("id"), str):
             run["phases"].append({"id": phase["id"], "title": safe(phase.get("title") or phase["id"]),
@@ -107,6 +107,42 @@ def validate(data):
     return run
 
 
+RESOURCE_MODES = ("parallel", "reduced", "sequential")
+RESOURCE_LEVELS = ("normal", "warn", "critical", "unknown")
+RESOURCE_SOURCES = ("probe", "os", "assumed")
+
+
+def resources_decision(value):
+    """The optional run-level ``resources`` field (0.9.0): the parallelism decision recorded
+    before a fan-out. Ledgers without it, or with an invalid one, read as None."""
+    if not isinstance(value, dict) or value.get("mode") not in RESOURCE_MODES:
+        return None
+
+    def count(key):
+        number = value.get(key)
+        return number if isinstance(number, int) and not isinstance(number, bool) and 0 <= number <= 256 else None
+
+    return {"mode": value["mode"], "pressure": value.get("pressure") if value.get("pressure") in RESOURCE_LEVELS else None,
+            "source": value.get("source") if value.get("source") in RESOURCE_SOURCES else None,
+            "maxParallelAgents": count("maxParallelAgents"), "maxParallelHeavyJobs": count("maxParallelHeavyJobs"),
+            "maxWorkersPerJob": count("maxWorkersPerJob"),
+            "checkedAt": value.get("checkedAt") if isinstance(value.get("checkedAt"), str) else None,
+            "note": safe(value.get("note") or "", 160)}
+
+
+def resources_line(decision, unicode=True):
+    if not decision:
+        return ""
+    parts = [decision["mode"]]
+    if decision["maxParallelAgents"] is not None:
+        parts.append(("≤" if unicode else "<=") + str(decision["maxParallelAgents"]) + " agente(s)")
+    if decision["maxParallelHeavyJobs"] is not None:
+        parts.append(("≤" if unicode else "<=") + str(decision["maxParallelHeavyJobs"]) + " job(s) pesado(s)")
+    if decision["pressure"]:
+        parts.append("pressão " + decision["pressure"])
+    return "recursos: " + (" · " if unicode else " - ").join(parts) + (" (" + decision["note"] + ")" if decision["note"] else "")
+
+
 def load(path):
     path = Path(path)
     if path.is_symlink() or not path.is_file() or path.stat().st_size > MAX_LEDGER:
@@ -515,6 +551,8 @@ def render(run, usage=None, now=None, color=False, unicode=True, width=20):
              bar(info["progress"], width, unicode) + " " + str(int(info["progress"] * 100)).rjust(3) + "%  " +
              str(info["done"]) + "/" + str(info["total"]) + " unidades" +
              (" - " + str(info["failed"]) + " falha(s)" if info["failed"] else "") + "  " + token_text(usage, unicode)]
+    if run.get("resources"):
+        lines.append(paint("dim", resources_line(run["resources"], unicode)))
     lines += ["", *agent_rows(run, unicode, counts=False), ""]
     phase_ids = [p["id"] for p in run["phases"]]
     groups = [(p, [u for u in run["units"] if u["phase"] == p["id"]]) for p in run["phases"]]
@@ -649,6 +687,18 @@ def mode_text(home, unicode=True):
         return ""
 
 
+def memory_text(home, unicode=True):
+    """'mem ▲ 85% ×2' when the machine is under pressure or parallelism is reduced; else ''."""
+    hook = hook_module()
+    module = hook.resources_module() if hook is not None and hasattr(hook, "resources_module") else None
+    if module is None:
+        return ""
+    try:
+        return module.segment(module.probe(home, processes=False, use_cache=True), unicode)
+    except Exception:  # noqa: BLE001 - the probe never breaks a status line
+        return ""
+
+
 def statusline(home, stdin_text, chain=None, host="claude"):
     """Status line for Claude Code and Antigravity (agy).
 
@@ -684,7 +734,7 @@ def statusline(home, stdin_text, chain=None, host="claude"):
         except (OSError, subprocess.SubprocessError):
             previous = ""
     run = select(discover(cwd, home), cwd, strict=True, session=session)
-    parts = [p for p in (mode_text(home),) if p]
+    parts = [p for p in (mode_text(home), memory_text(home)) if p]
     rows = []
     if active(run):
         if run["host"] == "claude" and host == "claude" and not run["session"].get("id") and session:
@@ -711,12 +761,12 @@ def statusline(home, stdin_text, chain=None, host="claude"):
 
 
 def tmux_segment(home, cwd):
-    mode = mode_text(home)
+    head = " · ".join(p for p in (mode_text(home), memory_text(home)) if p)
     run = select(discover(cwd, home), cwd, strict=True)
     if not active(run):
-        return mode
+        return head.replace("#", "##")
     text = compact(run, tokens(run, home), unicode=True, width=8)
-    return ((mode + " · " if mode else "") + text).replace("#", "##")
+    return ((head + " · " if head else "") + text).replace("#", "##")
 
 
 def main(argv=None):