agent-trace 0.2.8 → 0.2.10

package/agent-trace.cjs

@@ -24122,6 +24122,10 @@ th{color:var(--text-dim);font-size:10px;text-transform:uppercase;letter-spacing:
 .outcome-prs{border-top:1px solid var(--panel-border);padding-top:6px;margin-top:4px}
 .outcome-pr{display:flex;align-items:center;gap:8px;padding:3px 0;font-size:12px}
 .pr-badge{font-size:10px;padding:1px 5px;border-radius:3px;font-weight:600;text-transform:uppercase;border:1px solid rgba(74,222,128,.3);color:var(--green)}
+.pr-badge.open{border-color:rgba(34,211,238,.3);color:var(--cyan)}
+.pr-badge.merged{border-color:rgba(192,132,252,.3);color:var(--purple)}
+.pr-badge.closed{border-color:rgba(248,113,113,.3);color:var(--red)}
+.pr-badge.draft{border-color:rgba(102,102,102,.3);color:var(--text-muted)}
 .pr-label{color:var(--cyan);font-weight:600}
 .pr-repo{color:var(--text-muted)}
 .pr-link{color:var(--text-dim);text-decoration:none;font-size:11px}
@@ -24470,7 +24474,7 @@ function renderReplay() {
     }
     if (replay.pullRequests.length > 0) {
       h += '<div class="outcome-prs">';
-      replay.pullRequests.forEach(function(pr){ h += '<div class="outcome-pr"><span class="pr-badge">' + esc(pr.state) + '</span><span class="pr-label">PR #' + pr.prNumber + '</span><span class="pr-repo">' + esc(pr.repo) + '</span>' + (pr.url ? '<a class="pr-link" href="' + esc(pr.url) + '" target="_blank" rel="noopener noreferrer">' + esc(pr.url) + '</a>' : '') + '</div>'; });
+      replay.pullRequests.forEach(function(pr){ h += '<div class="outcome-pr"><span class="pr-badge ' + esc(pr.state) + '">' + esc(pr.state) + '</span><span class="pr-label">PR #' + pr.prNumber + '</span><span class="pr-repo">' + esc(pr.repo) + '</span>' + (pr.url ? '<a class="pr-link" href="' + esc(pr.url) + '" target="_blank" rel="noopener noreferrer">' + esc(pr.url) + '</a>' : '') + '</div>'; });
       h += '</div>';
     }
     h += '</div>';
@@ -25107,9 +25111,9 @@ var SqliteClient = class {
     this.db = new import_better_sqlite3.default(dbPath);
     this.db.pragma("journal_mode = WAL");
     this.db.pragma("synchronous = NORMAL");
+    this.migrateCacheTokenColumns();
     this.migrateDeduplicateEvents();
     this.db.exec(SCHEMA_SQL);
-    this.migrateCacheTokenColumns();
     this.migrateRebuildBrokenTraces();
   }
   async insertJsonEachRow(request) {
@@ -28545,15 +28549,35 @@ function toUpdatedTrace(existing, envelope) {
   const prUrl = readString4(payload, ["pr_url", "prUrl"]);
   const prRepo = readString4(payload, ["pr_repo", "prRepo"]);
   const prNumberRaw = readNumber3(payload, ["pr_number", "prNumber"]);
+  const prState = readString4(payload, ["pr_state", "prState"]) ?? "open";
+  const prMergedAt = readString4(payload, ["pr_merged_at", "prMergedAt"]);
   if (prUrl !== void 0 && prRepo !== void 0 && prNumberRaw !== void 0) {
-    const alreadyTracked = existingPullRequests.some((pr) => pr.prNumber === prNumberRaw && pr.repo === prRepo);
-    if (!alreadyTracked) {
+    const trackedIndex = existingPullRequests.findIndex((pr) => pr.prNumber === prNumberRaw && pr.repo === prRepo);
+    if (trackedIndex === -1) {
       existingPullRequests.push({
         repo: prRepo,
         prNumber: prNumberRaw,
-        state: "open",
-        url: prUrl
+        state: prState,
+        url: prUrl,
+        ...prState === "merged" && prMergedAt !== void 0 ? { mergedAt: prMergedAt } : {}
       });
+    } else if (prState !== "open") {
+      const tracked = existingPullRequests[trackedIndex];
+      existingPullRequests[trackedIndex] = {
+        ...tracked,
+        state: prState,
+        ...prState === "merged" && prMergedAt !== void 0 ? { mergedAt: prMergedAt } : {}
+      };
+    }
+  } else if (prState !== "open" && prRepo !== void 0 && prNumberRaw !== void 0) {
+    const trackedIndex = existingPullRequests.findIndex((pr) => pr.prNumber === prNumberRaw && pr.repo === prRepo);
+    if (trackedIndex !== -1) {
+      const tracked = existingPullRequests[trackedIndex];
+      existingPullRequests[trackedIndex] = {
+        ...tracked,
+        state: prState,
+        ...prState === "merged" && prMergedAt !== void 0 ? { mergedAt: prMergedAt } : {}
+      };
     }
   }
   return {
@@ -28767,12 +28791,15 @@ var SqlitePersistence = class {
   }
   async persistAcceptedEvent(event, trace) {
     const eventWrite = this.eventWriter.writeEvent(event).catch((error) => {
+      console.error(`[agent-trace] sqlite_events write failed: ${String(error)}`);
       this.writeFailures.push(`sqlite_events: ${String(error)}`);
     });
     const traceWrite = this.sessionTraceWriter.writeTrace(trace).catch((error) => {
+      console.error(`[agent-trace] sqlite_traces write failed: ${String(error)}`);
       this.writeFailures.push(`sqlite_traces: ${String(error)}`);
     });
     const sessionWrite = this.postgresWriter.writeTrace(trace).catch((error) => {
+      console.error(`[agent-trace] sqlite_sessions write failed: ${String(error)}`);
       this.writeFailures.push(`sqlite_sessions: ${String(error)}`);
     });
     await Promise.all([eventWrite, traceWrite, sessionWrite]);
@@ -29423,6 +29450,40 @@ function extractPrUrl(payload) {
   if (prUrlMatch !== null && prUrlMatch[0] !== void 0) return prUrlMatch[0];
   return void 0;
 }
+function extractPrState(payload) {
+  const record = payload;
+  const explicit = readString5(record, "pr_state") ?? readString5(record, "prState");
+  if (explicit !== void 0) return explicit;
+  const output = readString5(record, "tool_response") ?? readString5(record, "toolResponse") ?? readString5(record, "stdout") ?? readString5(record, "output");
+  const command = pickCommand2(payload);
+  const combined = [command, output].filter((s) => s !== void 0).join("\n");
+  if (combined.length === 0) return void 0;
+  const isGhPr = /\bgh\s+pr\b/.test(combined);
+  if (!isGhPr) return void 0;
+  const jsonState = combined.match(/"state"\s*:\s*"(MERGED|CLOSED|OPEN|DRAFT)"/i);
+  if (jsonState?.[1] !== void 0) {
+    const s = jsonState[1].toLowerCase();
+    return s === "open" ? "open" : s;
+  }
+  const isDraft = /"isDraft"\s*:\s*true/i.test(combined);
+  if (isDraft) return "draft";
+  if (/\bgh\s+pr\s+merge\b/.test(combined) && output !== void 0 && !/error|failed|not merged/i.test(output)) {
+    return "merged";
+  }
+  if (/\bMerged\b/.test(combined)) return "merged";
+  if (/\bClosed\b/.test(combined)) return "closed";
+  if (/\bDraft\b/.test(combined)) return "draft";
+  return void 0;
+}
+function extractPrMergedAt(payload) {
+  const record = payload;
+  const explicit = readString5(record, "pr_merged_at") ?? readString5(record, "prMergedAt");
+  if (explicit !== void 0) return explicit;
+  const output = readString5(record, "tool_response") ?? readString5(record, "toolResponse") ?? readString5(record, "stdout") ?? readString5(record, "output");
+  if (output === void 0) return void 0;
+  const match = output.match(/"mergedAt"\s*:\s*"([^"]+)"/);
+  return match?.[1] ?? void 0;
+}
 function parsePrFromUrl(url) {
   const match = url.match(/https:\/\/github\.com\/([^/]+\/[^/]+)\/pull\/(\d+)/);
   if (match === null || match[1] === void 0 || match[2] === void 0) return void 0;
@@ -29828,6 +29889,14 @@ function enrichHookPayloadWithGitContext(payload, provider, baselineStore, now)
       }
     }
   }
+  const prState = extractPrState(payload);
+  if (prState !== void 0) {
+    patch["pr_state"] = prState;
+  }
+  const prMergedAt = extractPrMergedAt(payload);
+  if (prMergedAt !== void 0) {
+    patch["pr_merged_at"] = prMergedAt;
+  }
   if (Object.keys(patch).length === 0) {
     return {
       payload,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-trace",
-  "version": "0.2.8",
+  "version": "0.2.10",
   "description": "Self-hosted observability for AI coding agents. One command, zero config.",
   "license": "Apache-2.0",
   "bin": {