@saadnvd1/agent-os 0.3.513 → 0.3.516

package/lib/status/pr-stages.ts

@@ -1,7 +1,6 @@
 // A task's PR stage for the sidebar, from state AgentOS already holds: the
 // PR as the last poll saw it, the stored review and CI checks, escalations,
 // asks and the armed flag. Nothing here asks GitHub.
-import { db, type Session } from "../db";
 import { cachedPR, storedPR } from "../tasks/session";
 import { derivePrStage } from "../tasks/pr-stage";
 import { getCheck } from "../orchestrator/checks";
@@ -9,30 +8,23 @@ import { holdingEscalations } from "../orchestrator/gates";
 import { openBySubject, workSubject } from "../orchestrator/asks";
 import { ciSettleLeft } from "../orchestrator/task-state";
 import { knownCommitTime } from "../orchestrator/facts";
+import { prMemo, type Pass } from "./incremental";
 import type { SessionStatusResponse } from "./collect";
 
-type TaskRow = Session & { workspace_id: string };
-
 export function addPrStages(
-  statuses: Record<string, SessionStatusResponse>
+  statuses: Record<string, SessionStatusResponse>,
+  pass: Pass
 ): void {
-  const tasks = db
-    .prepare(
-      `SELECT s.*, p.workspace_id FROM sessions s
-         JOIN projects p ON p.id = s.project_id
-        WHERE s.task_status = 'running' AND s.pr_number IS NOT NULL
-          AND s.archived_at IS NULL`
-    )
-    .all() as TaskRow[];
-  for (const t of tasks) {
+  for (const t of pass.rows.values()) {
+    if (t.task_status !== "running" || !t.pr_number || !t.project_id) continue;
     const status = statuses[t.id];
-    if (!status) continue;
-    try {
-      const pr = cachedPR(t.id) ?? storedPR(t);
-      const head = pr?.head;
-      const stage = derivePrStage({
-        pr,
-        settleIn:
+    if (!status || t.project_workspace_id === null) continue;
+    const workspaceId = t.project_workspace_id;
+    const stage = prMemo(pass, t.id, () => {
+      try {
+        const pr = cachedPR(t.id) ?? storedPR(t);
+        const head = pr?.head;
+        const settleIn =
           pr?.checks === "pass" && head
             ? ciSettleLeft({
                 taskId: t.id,
@@ -40,16 +32,27 @@ export function addPrStages(
                 checkCount: pr.checkCount ?? 0,
                 committedAt: knownCommitTime(head),
               })
-            : 0,
-        review: head ? (getCheck(t.id, head, "review")?.status ?? null) : null,
-        held:
-          holdingEscalations(t.id).length > 0 ||
-          !!openBySubject(t.workspace_id, workSubject(t.id)),
-        armed: !!t.merge_armed_at,
-      });
-      if (stage) status.prStage = stage;
-    } catch (error) {
-      console.error(`[status] PR stage of ${t.id.slice(0, 8)}:`, error);
-    }
+            : 0;
+        return {
+          value: derivePrStage({
+            pr,
+            settleIn,
+            review: head
+              ? (getCheck(t.id, head, "review")?.status ?? null)
+              : null,
+            held:
+              holdingEscalations(t.id).length > 0 ||
+              !!openBySubject(workspaceId, workSubject(t.id)),
+            armed: !!t.merge_armed_at,
+          }),
+          // Counting down to the end of the settle wait: time moves it.
+          keep: settleIn <= 0,
+        };
+      } catch (error) {
+        console.error(`[status] PR stage of ${t.id.slice(0, 8)}:`, error);
+        return { value: null, keep: false };
+      }
+    });
+    if (stage) status.prStage = stage;
   }
 }

package/lib/tasks/session.ts

@@ -18,6 +18,7 @@ import { lookupPR } from "./pr-poll";
 import { captureMergedTask } from "../session-logs";
 import { notifyStatusChanged } from "../status/hub";
 import type { TaskPR } from "./state";
+import { markStatusDirty } from "../status/dirty";
 
 export const expandHome = (p: string) => p.replace(/^~/, os.homedir());
 
@@ -190,6 +191,7 @@ export async function prFor(
   if (renamed && !refused) moveBranch(session, renamed);
   const before = prCache.get(session.id)?.pr;
   prCache.set(session.id, { at: Date.now(), pr });
+  markStatusDirty(session.id);
   // A PR whose stage inputs moved is news for the sidebar's badge.
   if (prSignature(before) !== prSignature(pr)) notifyStatusChanged();
   if (pr) {
@@ -279,4 +281,5 @@ export function storedPR(s: Session): TaskPR | null {
 
 export function forgetPR(sessionId: string): void {
   prCache.delete(sessionId);
+  markStatusDirty(sessionId);
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@saadnvd1/agent-os",
-  "version": "0.3.513",
+  "version": "0.3.516",
   "description": "Self-hosted web UI for managing Claude Code sessions",
   "bin": {
     "agent-os": "scripts/agent-os"

package/server.ts

@@ -318,6 +318,7 @@ app.prepare().then(async () => {
   const connect = demo
     ? { hostname: () => null }
     : startConnect({ onRequest, onUpgrade }, port);
+  policy.machineHost = () => connect.hostname();
   // HTTPS on the tailnet (its own port) for passkeys and other secure-origin APIs.
   if (
     !demo &&