56 lines
3.6 KiB
Python
56 lines
3.6 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import time
|
||
|
|
|
||
|
|
from .approving import INBOX_KINDS
|
||
|
|
from .errors import ToolError
|
||
|
|
from .fields import decode, expect
|
||
|
|
from .names import Directory
|
||
|
|
from .records import digest, iso_epoch
|
||
|
|
from .register import operator_rows, runnable
|
||
|
|
from .signed import signing
|
||
|
|
|
||
|
|
SNOOZE_LIMIT = 8 * 86400
|
||
|
|
|
||
|
|
|
||
|
|
class Inbox:
|
||
|
|
def inbox(self, now: float, tasks: dict[str, dict], directory: Directory) -> list[dict]:
|
||
|
|
base = {"agent": "", "detail": "", "green": False, "gate": "", "score": "", "warnings": [], "options": [], "recommended": "", "command": "", "directory": "", "sha256": "", "runnable": False}
|
||
|
|
items = []
|
||
|
|
for review in self.reviews.pending():
|
||
|
|
created = iso_epoch(review["created"]) or now
|
||
|
|
gate = review.get("gate", "")
|
||
|
|
fields = {"title": review["title"], "detail": review["branch"], "green": gate == "pass", "gate": gate, "score": review.get("score", ""), "directory": review["worktree"]}
|
||
|
|
items.append({**base, "kind": "review", "id": review["id"], "ts": created, **fields})
|
||
|
|
for ask in (ask for ask in self.asks.pending() if ask["expires"] > now):
|
||
|
|
warnings = signing.warnings(ask["command"])
|
||
|
|
fields = {"title": f"{ask['tool']}: {ask['command'][:200]}", "agent": directory.now(ask["session"]), "green": not warnings, "warnings": warnings}
|
||
|
|
items.append({**base, "kind": "ask", "id": ask["id"], "ts": ask["created_ts"], **fields, "command": ask["command"], "directory": ask["cwd"]})
|
||
|
|
for decision in self.decisions.pending():
|
||
|
|
recommended = decision.get("recommended", "")
|
||
|
|
fields = {"title": decision["question"][:300], "agent": decision["agent"], "green": recommended != "", "options": decision["options"], "recommended": recommended}
|
||
|
|
items.append({**base, "kind": "decision", "id": decision["id"], "ts": decision["created_ts"], **fields})
|
||
|
|
held = self.open_jobs()
|
||
|
|
for row in (row for row in operator_rows(tasks) if row["task"] not in held):
|
||
|
|
ok = runnable(row)
|
||
|
|
fields = {"title": row["command"], "detail": row["why"], "command": row["command"], "directory": row["directory"], "sha256": digest(row["command"]) if ok else "", "runnable": ok}
|
||
|
|
items.append({**base, "kind": "operator", "id": row["task"], "ts": self.watch.first(row["task"], now), **fields})
|
||
|
|
for item in items:
|
||
|
|
mark = self.marks.get(f"{item['kind']}:{item['id']}")
|
||
|
|
item.update(pinned=mark["pinned"], until=mark["until"] if mark["until"] > now else 0.0, held=self.batches.holds(item["kind"], item["id"]))
|
||
|
|
return sorted(items, key=lambda item: (not item["pinned"], -item["ts"], item["id"]))
|
||
|
|
|
||
|
|
def mark(self, body: bytes) -> dict:
|
||
|
|
payload = decode(body)
|
||
|
|
expect(payload, ("kind", "id", "until", "pinned"))
|
||
|
|
kind_name, until, pinned = payload["kind"], payload["until"], payload["pinned"]
|
||
|
|
if kind_name not in INBOX_KINDS or not isinstance(payload["id"], str) or not 0 < len(payload["id"]) <= 200:
|
||
|
|
raise ToolError(f"kind must be one of {', '.join(INBOX_KINDS)} and id an item id")
|
||
|
|
now = time.time()
|
||
|
|
if isinstance(until, bool) or not isinstance(until, int | float) or not (until == 0 or now < until <= now + SNOOZE_LIMIT):
|
||
|
|
raise ToolError("until must be 0 or a moment within the next 8 days")
|
||
|
|
if not isinstance(pinned, bool):
|
||
|
|
raise ToolError("pinned must be true or false")
|
||
|
|
mark = {"until": float(until), "pinned": pinned}
|
||
|
|
self.marks.put(f"{kind_name}:{payload['id']}", mark)
|
||
|
|
return {"kind": kind_name, "id": payload["id"], **mark}
|