Release 1.0.9: Timeline aligned with Grok Build (session events + edit diffs)

Author: 3xian

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "pinkcode",
-  "version": "1.0.8",
+  "version": "1.0.9",
   "description": "Desktop mission control for Grok agents.",
   "type": "module",
   "keywords": [

package.json

@@ -1,6 +1,6 @@
 [package]
 name = "pinkcode"
-version = "1.0.8"
+version = "1.0.9"
 description = "Desktop control plane for Grok Build multi-task observability"
 authors = ["PinkCode"]
 license = "Apache-2.0"

src-tauri/Cargo.lock

@@ -2,7 +2,7 @@
   "$schema": "https://schema.tauri.app/config/2",
   "productName": "PinkCode",
   "mainBinaryName": "PinkCode",
-  "version": "1.0.8",
+  "version": "1.0.9",
   "identifier": "com.pinkcode.app",
   "build": {
     "beforeDevCommand": "npm run dev",

src-tauri/Cargo.toml

@@ -11,6 +11,7 @@ import {
   reduceShellUpdate,
   settleLifecycleItems,
   settleStreamingItems,
+  type UpdateDescription,
 } from "./liveTimeline";
 import type { TimelineItem } from "../types";
 
@@ -267,6 +268,43 @@ describe("live timeline reducer", () => {
     expect(terminal?.detail).toBe("1.2k tok");
   });
 
+  it("emits the goal-complete milestone once per goal id", () => {
+    const shells = createTimelineReducerState();
+    let map = new Map<string, TimelineItem[]>();
+    const complete: UpdateDescription = {
+      kind: "event",
+      title: "Goal complete — 5m end-to-end.",
+      goalId: "g1",
+    };
+    map = reduceAgentUpdate(
+      map,
+      { handleId: "h1", sessionId: "sess-1", description: complete, now: 1, nextId: () => "g1" },
+      shells,
+    );
+    expect(map.get("sess-1")?.length).toBe(1);
+    // A repeated complete broadcast (replay / leader re-emit) must not stack.
+    const same = reduceAgentUpdate(
+      map,
+      { handleId: "h1", sessionId: "sess-1", description: complete, now: 2, nextId: () => "g1b" },
+      shells,
+    );
+    expect(same).toBe(map);
+    expect(same.get("sess-1")?.length).toBe(1);
+    // A different goal still renders.
+    const other = reduceAgentUpdate(
+      map,
+      {
+        handleId: "h1",
+        sessionId: "sess-1",
+        description: { ...complete, goalId: "g2" },
+        now: 3,
+        nextId: () => "g2",
+      },
+      shells,
+    );
+    expect(other.get("sess-1")?.length).toBe(2);
+  });
+
   it("timeline elapsed always owns turn_completed title", () => {
     const shells = createTimelineReducerState();
     let map = new Map<string, TimelineItem[]>();

src-tauri/tauri.conf.json

@@ -26,6 +26,8 @@ export type ShellIndexes = Map<string, Map<string, number>>;
 export interface TimelineReducerState {
   shellIndexes: ShellIndexes;
   suppressedToolIds: Map<string, Set<string>>;
+  /** Goal ids that already emitted their "Goal complete" milestone (Grok just_completed). */
+  completedGoalIds: Set<string>;
 }
 export type TimelineRetention = "live" | "history";
 
@@ -39,6 +41,7 @@ export function createTimelineReducerState(): TimelineReducerState {
   return {
     shellIndexes: new Map(),
     suppressedToolIds: new Map(),
+    completedGoalIds: new Set(),
   };
 }
 
@@ -288,6 +291,15 @@ export function reduceAgentUpdate(
     input.streaming !== undefined ? input.streaming : textUpdate ? true : undefined;
   let listChanged = false;
 
+  // Emit the "Goal complete" milestone once per goal (Grok Build gates on the
+  // just-completed transition; a repeated complete broadcast must not stack).
+  if (description.kind === "event" && description.goalId) {
+    if (reducerState.completedGoalIds.has(description.goalId)) {
+      return previous;
+    }
+    reducerState.completedGoalIds.add(description.goalId);
+  }
+
   if (!textUpdate) {
     for (let index = 0; index < list.length; index++) {
       if (list[index].streaming && list[index].handleId === handleId) {

src/hooks/liveTimeline.test.ts

@@ -477,8 +477,9 @@ export function useAgentEvents(
           return;
         }
 
-        // Non-lifecycle notifications (e.g. FileWritten for modified files) are parsed centrally in describeUpdate.
-        // This removes duplicated handling and lets the parser own all notification shapes.
+        // Non-lifecycle notifications (x.ai session events, subagent/task
+        // lifecycle) are parsed centrally in describeUpdate — one parser owns
+        // all notification shapes.
         const desc = describeUpdate({
           method,
           params: params as any,

src/hooks/liveTimeline.ts

@@ -0,0 +1,134 @@
+import { describe, expect, it } from "vitest";
+import { extractEditDiff, isEditToolUpdate } from "./editDiff";
+
+describe("isEditToolUpdate", () => {
+  it("detects Edit kind and edit-class tool names", () => {
+    // ACP ToolKind serializes lowercase on the wire ("kind":"edit").
+    expect(isEditToolUpdate({ kind: "edit", title: "Edit `src/a.ts`" })).toBe(true);
+    expect(isEditToolUpdate({ kind: "Edit", title: "Edit `src/a.ts`" })).toBe(true);
+    expect(isEditToolUpdate({ kind: "read", title: "Read `src/a.ts`" })).toBe(false);
+    expect(
+      isEditToolUpdate({
+        kind: "Other",
+        _meta: { "x.ai/tool": { name: "search_replace" } },
+      }),
+    ).toBe(true);
+    expect(
+      isEditToolUpdate({
+        kind: "Other",
+        _meta: { "x.ai/tool": { name: "apply_patch" } },
+      }),
+    ).toBe(true);
+    expect(
+      isEditToolUpdate({
+        kind: "Other",
+        _meta: { "x.ai/tool": { name: "run_terminal_command" } },
+      }),
+    ).toBe(false);
+  });
+});
+
+describe("extractEditDiff", () => {
+  it("builds a unified diff from SearchReplace rawOutput details", () => {
+    const diff = extractEditDiff({
+      kind: "Edit",
+      title: "Edit `src/main.ts`",
+      rawOutput: {
+        type: "SearchReplace",
+        EditsApplied: {
+          old_string: "let x = 1;",
+          new_string: "let x = 2;",
+          absolute_path: "/tmp/main.ts",
+          edits: {
+            details: [
+              {
+                old_string: "let x = 1;",
+                new_string: "let x = 2;",
+                old_line: 5,
+                new_line: 5,
+                context_before: "fn main() {\n",
+                context_after: "}",
+                line_prefix: "",
+              },
+            ],
+          },
+        },
+      },
+    });
+    // Hunk starts at the first shown old line — the context line at 4.
+    expect(diff).toContain("@@ -4,3 +4,3 @@");
+    expect(diff).toContain("-let x = 1;");
+    expect(diff).toContain("+let x = 2;");
+    expect(diff).toContain(" fn main() {");
+  });
+
+  it("accepts snake_case raw_output too (disk hydrate tolerance)", () => {
+    const diff = extractEditDiff({
+      kind: "Edit",
+      raw_output: {
+        type: "SearchReplace",
+        EditsApplied: {
+          edits: {
+            details: [
+              {
+                old_string: "a",
+                new_string: "b",
+                old_line: 1,
+                new_line: 1,
+              },
+            ],
+          },
+        },
+      },
+    });
+    expect(diff).toContain("-a");
+    expect(diff).toContain("+b");
+  });
+
+  it("falls back to content diff blocks (write tool full content)", () => {
+    const diff = extractEditDiff({
+      kind: "Edit",
+      title: "Write `src/out.txt`",
+      content: [
+        {
+          type: "diff",
+          path: "src/out.txt",
+          new_text: "hello\nworld\n",
+        },
+      ],
+    });
+    // Pure insertion: zero old lines ("-1,0"), two new lines.
+    expect(diff).toContain("@@ -1,0 +1,2 @@");
+    expect(diff).toContain("+hello");
+    expect(diff).toContain("+world");
+  });
+
+  it("renders replacements against the old text via LCS alignment", () => {
+    const diff = extractEditDiff({
+      kind: "Edit",
+      content: [
+        {
+          type: "diff",
+          path: "x.txt",
+          old_text: "keep\nremove\nkeep2\n",
+          new_text: "keep\nadded\nkeep2\n",
+        },
+      ],
+    });
+    expect(diff).toContain("-remove");
+    expect(diff).toContain("+added");
+    expect(diff).toContain(" keep");
+  });
+
+  it("returns undefined for non-edit output", () => {
+    expect(
+      extractEditDiff({ kind: "Read", rawOutput: { type: "ReadFile" } }),
+    ).toBeUndefined();
+    expect(
+      extractEditDiff({
+        kind: "Edit",
+        rawOutput: { type: "Bash", output: [], exit_code: 0 },
+      }),
+    ).toBeUndefined();
+  });
+});