kanna-code 0.22.0 → 0.23.1

package/src/server/ws-router.test.ts

@@ -38,7 +38,7 @@ const DEFAULT_UPDATE_SNAPSHOT: UpdateSnapshot = {
 }
 
 describe("ws-router", () => {
-  test("acks system.ping without broadcasting snapshots", () => {
+  test("acks system.ping without broadcasting snapshots", async () => {
     const router = createWsRouter({
       store: { state: createEmptyState() } as never,
       agent: { getActiveStatuses: () => new Map(), getDrainingChatIds: () => new Set() } as never,
@@ -58,7 +58,7 @@ describe("ws-router", () => {
     const ws = new FakeWebSocket()
 
     ws.data.subscriptions.set("sub-1", { type: "sidebar" })
-    router.handleMessage(
+    await router.handleMessage(
       ws as never,
       JSON.stringify({
         v: 1,
@@ -77,7 +77,7 @@ describe("ws-router", () => {
     ])
   })
 
-  test("acks terminal.input without rebroadcasting terminal snapshots", () => {
+  test("acks terminal.input without rebroadcasting terminal snapshots", async () => {
     const router = createWsRouter({
       store: { state: createEmptyState() } as never,
       agent: { getActiveStatuses: () => new Map(), getDrainingChatIds: () => new Set() } as never,
@@ -98,7 +98,7 @@ describe("ws-router", () => {
     const ws = new FakeWebSocket()
 
     ws.data.subscriptions.set("sub-terminal", { type: "terminal", terminalId: "terminal-1" })
-    router.handleMessage(
+    await router.handleMessage(
       ws as never,
       JSON.stringify({
         v: 1,
@@ -121,7 +121,7 @@ describe("ws-router", () => {
     ])
   })
 
-  test("subscribes and unsubscribes chat topics", () => {
+  test("subscribes and unsubscribes chat topics", async () => {
     const router = createWsRouter({
       store: { state: createEmptyState() } as never,
       agent: { getActiveStatuses: () => new Map(), getDrainingChatIds: () => new Set() } as never,
@@ -140,7 +140,7 @@ describe("ws-router", () => {
     })
     const ws = new FakeWebSocket()
 
-    router.handleMessage(
+    await router.handleMessage(
       ws as never,
       JSON.stringify({
         v: 1,
@@ -160,7 +160,7 @@ describe("ws-router", () => {
       },
     })
 
-    router.handleMessage(
+    await router.handleMessage(
       ws as never,
       JSON.stringify({
         v: 1,
@@ -176,7 +176,7 @@ describe("ws-router", () => {
     })
   })
 
-  test("loads older chat history pages", () => {
+  test("loads older chat history pages", async () => {
     const state = createEmptyState()
     state.projectsById.set("project-1", {
       id: "project-1",
@@ -230,7 +230,7 @@ describe("ws-router", () => {
     })
     const ws = new FakeWebSocket()
 
-    router.handleMessage(
+    await router.handleMessage(
       ws as never,
       JSON.stringify({
         v: 1,
@@ -472,7 +472,7 @@ describe("ws-router", () => {
     })
     const ws = new FakeWebSocket()
 
-    router.handleMessage(
+    await router.handleMessage(
       ws as never,
       JSON.stringify({
         v: 1,
@@ -492,7 +492,7 @@ describe("ws-router", () => {
       },
     })
 
-    router.handleMessage(
+    await router.handleMessage(
       ws as never,
       JSON.stringify({
         v: 1,
@@ -576,7 +576,7 @@ describe("ws-router", () => {
     })
     const ws = new FakeWebSocket()
 
-    router.handleMessage(
+    await router.handleMessage(
       ws as never,
       JSON.stringify({
         v: 1,
@@ -596,7 +596,7 @@ describe("ws-router", () => {
       },
     })
 
-    router.handleMessage(
+    await router.handleMessage(
       ws as never,
       JSON.stringify({
         v: 1,
@@ -625,7 +625,7 @@ describe("ws-router", () => {
       },
     })
 
-    router.handleMessage(
+    await router.handleMessage(
       ws as never,
       JSON.stringify({
         v: 1,
@@ -651,4 +651,186 @@ describe("ws-router", () => {
       },
     })
   })
+
+  test("routes discard diff file commands through the diff store and rebroadcasts chat snapshots", async () => {
+    const state = createEmptyState()
+    state.projectsById.set("project-1", {
+      id: "project-1",
+      localPath: "/tmp/project",
+      title: "Project",
+      createdAt: 1,
+      updatedAt: 1,
+    })
+    state.projectIdsByPath.set("/tmp/project", "project-1")
+    state.chatsById.set("chat-1", {
+      id: "chat-1",
+      projectId: "project-1",
+      title: "Chat",
+      createdAt: 1,
+      updatedAt: 1,
+      unread: false,
+      provider: null,
+      planMode: false,
+      sessionToken: null,
+      lastTurnOutcome: null,
+    })
+
+    const discardCalls: Array<{ chatId: string; projectPath: string; path: string }> = []
+    const diffStore = {
+      getSnapshot: () => ({ status: "ready" as const, files: [] }),
+      refreshSnapshot: async () => false,
+      generateCommitMessage: async () => ({ subject: "", body: "" }),
+      commitFiles: async () => ({ ok: true as const, mode: "commit_only" as const, pushed: false, snapshotChanged: false }),
+      discardFile: async (args: { chatId: string; projectPath: string; path: string }) => {
+        discardCalls.push(args)
+        return { snapshotChanged: true }
+      },
+      ignoreFile: async () => ({ snapshotChanged: false }),
+    }
+
+    const router = createWsRouter({
+      store: {
+        state,
+        getChat: (chatId: string) => state.chatsById.get(chatId) ?? null,
+        getProject: (projectId: string) => state.projectsById.get(projectId) ?? null,
+        getRecentChatHistory: () => ({ entries: [], hasOlder: false, olderCursor: null }),
+      } as never,
+      diffStore: diffStore as never,
+      agent: { getActiveStatuses: () => new Map(), getDrainingChatIds: () => new Set() } as never,
+      terminals: {
+        getSnapshot: () => null,
+        onEvent: () => () => {},
+      } as never,
+      keybindings: {
+        getSnapshot: () => DEFAULT_KEYBINDINGS_SNAPSHOT,
+        onChange: () => () => {},
+      } as never,
+      refreshDiscovery: async () => [],
+      getDiscoveredProjects: () => [],
+      machineDisplayName: "Local Machine",
+      updateManager: null,
+    })
+    const ws = new FakeWebSocket()
+
+    router.handleOpen(ws as never)
+    router.handleMessage(
+      ws as never,
+      JSON.stringify({
+        v: 1,
+        type: "subscribe",
+        id: "chat-sub",
+        topic: { type: "chat", chatId: "chat-1" },
+      })
+    )
+
+    await router.handleMessage(
+      ws as never,
+      JSON.stringify({
+        v: 1,
+        type: "command",
+        id: "discard-1",
+        command: {
+          type: "chat.discardDiffFile",
+          chatId: "chat-1",
+          path: "app.txt",
+        },
+      })
+    )
+
+    expect(discardCalls).toEqual([{
+      chatId: "chat-1",
+      projectPath: "/tmp/project",
+      path: "app.txt",
+    }])
+    expect(ws.sent).toContainEqual({
+      v: PROTOCOL_VERSION,
+      type: "ack",
+      id: "discard-1",
+      result: { snapshotChanged: true },
+    })
+  })
+
+  test("routes ignore diff file commands through the diff store", async () => {
+    const state = createEmptyState()
+    state.projectsById.set("project-1", {
+      id: "project-1",
+      localPath: "/tmp/project",
+      title: "Project",
+      createdAt: 1,
+      updatedAt: 1,
+    })
+    state.projectIdsByPath.set("/tmp/project", "project-1")
+    state.chatsById.set("chat-1", {
+      id: "chat-1",
+      projectId: "project-1",
+      title: "Chat",
+      createdAt: 1,
+      updatedAt: 1,
+      unread: false,
+      provider: null,
+      planMode: false,
+      sessionToken: null,
+      lastTurnOutcome: null,
+    })
+
+    const ignoreCalls: Array<{ chatId: string; projectPath: string; path: string }> = []
+    const router = createWsRouter({
+      store: {
+        state,
+        getChat: (chatId: string) => state.chatsById.get(chatId) ?? null,
+        getProject: (projectId: string) => state.projectsById.get(projectId) ?? null,
+      } as never,
+      diffStore: {
+        getSnapshot: () => ({ status: "ready" as const, files: [] }),
+        refreshSnapshot: async () => false,
+        generateCommitMessage: async () => ({ subject: "", body: "" }),
+        commitFiles: async () => ({ ok: true as const, mode: "commit_only" as const, pushed: false, snapshotChanged: false }),
+        discardFile: async () => ({ snapshotChanged: false }),
+        ignoreFile: async (args: { chatId: string; projectPath: string; path: string }) => {
+          ignoreCalls.push(args)
+          return { snapshotChanged: false }
+        },
+      } as never,
+      agent: { getActiveStatuses: () => new Map(), getDrainingChatIds: () => new Set() } as never,
+      terminals: {
+        getSnapshot: () => null,
+        onEvent: () => () => {},
+      } as never,
+      keybindings: {
+        getSnapshot: () => DEFAULT_KEYBINDINGS_SNAPSHOT,
+        onChange: () => () => {},
+      } as never,
+      refreshDiscovery: async () => [],
+      getDiscoveredProjects: () => [],
+      machineDisplayName: "Local Machine",
+      updateManager: null,
+    })
+    const ws = new FakeWebSocket()
+
+    await router.handleMessage(
+      ws as never,
+      JSON.stringify({
+        v: 1,
+        type: "command",
+        id: "ignore-1",
+        command: {
+          type: "chat.ignoreDiffFile",
+          chatId: "chat-1",
+          path: "scratch.log",
+        },
+      })
+    )
+
+    expect(ignoreCalls).toEqual([{
+      chatId: "chat-1",
+      projectPath: "/tmp/project",
+      path: "scratch.log",
+    }])
+    expect(ws.sent).toContainEqual({
+      v: PROTOCOL_VERSION,
+      type: "ack",
+      id: "ignore-1",
+      result: { snapshotChanged: false },
+    })
+  })
 })

package/src/server/ws-router.ts

@@ -22,7 +22,7 @@ export interface ClientState {
 
 interface CreateWsRouterArgs {
   store: EventStore
-  diffStore?: Pick<DiffStore, "getSnapshot" | "refreshSnapshot" | "generateCommitMessage" | "commitFiles">
+  diffStore?: Pick<DiffStore, "getSnapshot" | "refreshSnapshot" | "generateCommitMessage" | "commitFiles" | "discardFile" | "ignoreFile">
   agent: AgentCoordinator
   terminals: TerminalManager
   keybindings: KeybindingsManager
@@ -61,6 +61,8 @@ export function createWsRouter({
     refreshSnapshot: async () => false,
     generateCommitMessage: async () => ({ subject: "Update selected files", body: "", usedFallback: true, failureMessage: null }),
     commitFiles: async () => ({ ok: true, mode: "commit_only" as const, branchName: undefined, pushed: false, snapshotChanged: false }),
+    discardFile: async () => ({ snapshotChanged: false }),
+    ignoreFile: async () => ({ snapshotChanged: false }),
   }
 
   function createEnvelope(id: string, topic: SubscriptionTopic): ServerEnvelope {
@@ -407,6 +409,46 @@ export function createWsRouter({
           }
           return
         }
+        case "chat.discardDiffFile": {
+          const chat = store.getChat(command.chatId)
+          if (!chat) {
+            throw new Error("Chat not found")
+          }
+          const project = store.getProject(chat.projectId)
+          if (!project) {
+            throw new Error("Project not found")
+          }
+          const result = await resolvedDiffStore.discardFile({
+            chatId: command.chatId,
+            projectPath: project.localPath,
+            path: command.path,
+          })
+          send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
+          if (result.snapshotChanged) {
+            broadcastSnapshots()
+          }
+          return
+        }
+        case "chat.ignoreDiffFile": {
+          const chat = store.getChat(command.chatId)
+          if (!chat) {
+            throw new Error("Chat not found")
+          }
+          const project = store.getProject(chat.projectId)
+          if (!project) {
+            throw new Error("Project not found")
+          }
+          const result = await resolvedDiffStore.ignoreFile({
+            chatId: command.chatId,
+            projectPath: project.localPath,
+            path: command.path,
+          })
+          send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
+          if (result.snapshotChanged) {
+            broadcastSnapshots()
+          }
+          return
+        }
         case "chat.cancel": {
           await agent.cancel(command.chatId)
           send(ws, { v: PROTOCOL_VERSION, type: "ack", id })

package/src/shared/protocol.ts

@@ -80,6 +80,8 @@ export type ClientCommand =
   | { type: "chat.refreshDiffs"; chatId: string }
   | { type: "chat.generateCommitMessage"; chatId: string; paths: string[] }
   | { type: "chat.commitDiffs"; chatId: string; paths: string[]; summary: string; description?: string; mode: DiffCommitMode }
+  | { type: "chat.discardDiffFile"; chatId: string; path: string }
+  | { type: "chat.ignoreDiffFile"; chatId: string; path: string }
   | { type: "chat.cancel"; chatId: string }
   | { type: "chat.stopDraining"; chatId: string }
   | { type: "chat.loadHistory"; chatId: string; beforeCursor: string; limit: number }

package/src/shared/types.ts

@@ -471,6 +471,7 @@ export interface ContextWindowUsageSnapshot {
 export interface ChatDiffFile {
   path: string
   changeType: "added" | "deleted" | "modified" | "renamed"
+  isUntracked: boolean
   patch: string
   mimeType?: string
   size?: number