kanna-code 0.19.0 → 0.21.0

package/src/server/terminal-manager.test.ts

@@ -1,5 +1,5 @@
-import { afterEach, beforeAll, describe, expect, test } from "bun:test"
-import { mkdtemp, rm } from "node:fs/promises"
+import { afterAll, afterEach, beforeAll, describe, expect, test } from "bun:test"
+import { mkdtemp, mkdir, rm } from "node:fs/promises"
 import os from "node:os"
 import path from "node:path"
 import { TerminalManager } from "./terminal-manager"
@@ -13,10 +13,19 @@ const isSupportedPlatform = process.platform !== "win32" && typeof Bun.Terminal
 const describeIfSupported = isSupportedPlatform ? describe : describe.skip
 
 let tempProjectPath = ""
+let tempHomePath = ""
+const originalHome = process.env.HOME
+const originalZdotdir = process.env.ZDOTDIR
+const originalHistfile = process.env.HISTFILE
 
 beforeAll(async () => {
   if (!isSupportedPlatform) return
   tempProjectPath = await mkdtemp(path.join(os.tmpdir(), "kanna-terminal-manager-"))
+  tempHomePath = await mkdtemp(path.join(os.tmpdir(), "kanna-terminal-home-"))
+  await mkdir(path.join(tempHomePath, ".config"), { recursive: true })
+  process.env.HOME = tempHomePath
+  process.env.ZDOTDIR = tempHomePath
+  process.env.HISTFILE = path.join(tempHomePath, ".zsh_history")
 })
 
 afterEach(async () => {
@@ -25,6 +34,30 @@ afterEach(async () => {
   tempProjectPath = await mkdtemp(path.join(os.tmpdir(), "kanna-terminal-manager-"))
 })
 
+afterAll(async () => {
+  if (originalHome === undefined) {
+    delete process.env.HOME
+  } else {
+    process.env.HOME = originalHome
+  }
+
+  if (originalZdotdir === undefined) {
+    delete process.env.ZDOTDIR
+  } else {
+    process.env.ZDOTDIR = originalZdotdir
+  }
+
+  if (originalHistfile === undefined) {
+    delete process.env.HISTFILE
+  } else {
+    process.env.HISTFILE = originalHistfile
+  }
+
+  if (tempHomePath) {
+    await rm(tempHomePath, { recursive: true, force: true })
+  }
+})
+
 async function waitFor(check: () => boolean, timeoutMs: number, intervalMs = 25) {
   const startedAt = Date.now()
   while (Date.now() - startedAt < timeoutMs) {
@@ -70,8 +103,8 @@ describeIfSupported("TerminalManager", () => {
     const { manager, getOutput } = await createSession(terminalId)
 
     try {
-      manager.write(terminalId, 'python3 -c "import time; time.sleep(30)"\r')
-      await waitFor(() => getOutput().includes("time.sleep(30)"), COMMAND_TIMEOUT_MS)
+      manager.write(terminalId, `python3 -c "import time; print('__KANNA_SLEEP__', flush=True); time.sleep(30)"\r`)
+      await waitFor(() => getOutput().includes("__KANNA_SLEEP__"), COMMAND_TIMEOUT_MS)
 
       manager.write(terminalId, "\x03")
       manager.write(terminalId, "printf '__KANNA_AFTER_INT__\\n'\r")

package/src/server/uploads.ts

@@ -108,6 +108,10 @@ export function inferAttachmentContentType(fileName: string, fallbackType?: stri
   return fallbackType || DEFAULT_BINARY_MIME_TYPE
 }
 
+export function inferProjectFileContentType(fileName: string, fallbackType?: string): string {
+  return inferAttachmentContentType(fileName, fallbackType)
+}
+
 export async function deleteProjectUpload(args: {
   localPath: string
   storedName: string

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

@@ -41,7 +41,7 @@ describe("ws-router", () => {
   test("acks system.ping without broadcasting snapshots", () => {
     const router = createWsRouter({
       store: { state: createEmptyState() } as never,
-      agent: { getActiveStatuses: () => new Map() } as never,
+      agent: { getActiveStatuses: () => new Map(), getDrainingChatIds: () => new Set() } as never,
       terminals: {
         getSnapshot: () => null,
         onEvent: () => () => {},
@@ -80,7 +80,7 @@ describe("ws-router", () => {
   test("acks terminal.input without rebroadcasting terminal snapshots", () => {
     const router = createWsRouter({
       store: { state: createEmptyState() } as never,
-      agent: { getActiveStatuses: () => new Map() } as never,
+      agent: { getActiveStatuses: () => new Map(), getDrainingChatIds: () => new Set() } as never,
       terminals: {
         getSnapshot: () => null,
         onEvent: () => () => {},
@@ -124,7 +124,7 @@ describe("ws-router", () => {
   test("subscribes and unsubscribes chat topics", () => {
     const router = createWsRouter({
       store: { state: createEmptyState() } as never,
-      agent: { getActiveStatuses: () => new Map() } as never,
+      agent: { getActiveStatuses: () => new Map(), getDrainingChatIds: () => new Set() } as never,
       terminals: {
         getSnapshot: () => null,
         onEvent: () => () => {},
@@ -373,7 +373,7 @@ describe("ws-router", () => {
 
     const router = createWsRouter({
       store: { state: createEmptyState() } as never,
-      agent: { getActiveStatuses: () => new Map() } as never,
+      agent: { getActiveStatuses: () => new Map(), getDrainingChatIds: () => new Set() } as never,
       terminals: {
         getSnapshot: () => null,
         onEvent: () => () => {},
@@ -474,7 +474,7 @@ describe("ws-router", () => {
 
     const router = createWsRouter({
       store: { state: createEmptyState() } as never,
-      agent: { getActiveStatuses: () => new Map() } as never,
+      agent: { getActiveStatuses: () => new Map(), getDrainingChatIds: () => new Set() } as never,
       terminals: {
         getSnapshot: () => null,
         onEvent: () => () => {},

package/src/server/ws-router.ts

@@ -4,6 +4,7 @@ import type { ClientEnvelope, ServerEnvelope, SubscriptionTopic } from "../share
 import { isClientEnvelope } from "../shared/protocol"
 import type { AgentCoordinator } from "./agent"
 import type { DiscoveredProject } from "./discovery"
+import { DiffStore } from "./diff-store"
 import { EventStore } from "./event-store"
 import { openExternal } from "./external-open"
 import { KeybindingsManager } from "./keybindings"
@@ -14,10 +15,12 @@ import { deriveChatSnapshot, deriveLocalProjectsSnapshot, deriveSidebarData } fr
 
 export interface ClientState {
   subscriptions: Map<string, SubscriptionTopic>
+  snapshotSignatures: Map<string, string>
 }
 
 interface CreateWsRouterArgs {
   store: EventStore
+  diffStore?: Pick<DiffStore, "getSnapshot" | "refreshSnapshot" | "generateCommitMessage" | "commitFiles">
   agent: AgentCoordinator
   terminals: TerminalManager
   keybindings: KeybindingsManager
@@ -31,8 +34,17 @@ function send(ws: ServerWebSocket<ClientState>, message: ServerEnvelope) {
   ws.send(JSON.stringify(message))
 }
 
+function ensureSnapshotSignatures(ws: ServerWebSocket<ClientState>) {
+  if (!ws.data.snapshotSignatures) {
+    ws.data.snapshotSignatures = new Map()
+  }
+
+  return ws.data.snapshotSignatures
+}
+
 export function createWsRouter({
   store,
+  diffStore,
   agent,
   terminals,
   keybindings,
@@ -42,6 +54,12 @@ export function createWsRouter({
   updateManager,
 }: CreateWsRouterArgs) {
   const sockets = new Set<ServerWebSocket<ClientState>>()
+  const resolvedDiffStore = diffStore ?? {
+    getSnapshot: () => ({ status: "unknown", branchName: undefined, files: [] as const }),
+    refreshSnapshot: async () => false,
+    generateCommitMessage: async () => ({ subject: "Update selected files", body: "", usedFallback: true, failureMessage: null }),
+    commitFiles: async () => false,
+  }
 
   function createEnvelope(id: string, topic: SubscriptionTopic): ServerEnvelope {
     if (topic.type === "sidebar") {
@@ -121,14 +139,28 @@ export function createWsRouter({
       id,
       snapshot: {
         type: "chat",
-        data: deriveChatSnapshot(store.state, agent.getActiveStatuses(), topic.chatId, (chatId) => store.getMessages(chatId)),
+        data: deriveChatSnapshot(
+          store.state,
+          agent.getActiveStatuses(),
+          agent.getDrainingChatIds(),
+          topic.chatId,
+          (chatId) => store.getMessages(chatId),
+          (chatId) => resolvedDiffStore.getSnapshot(chatId)
+        ),
       },
     }
   }
 
   function pushSnapshots(ws: ServerWebSocket<ClientState>) {
+    const snapshotSignatures = ensureSnapshotSignatures(ws)
     for (const [id, topic] of ws.data.subscriptions.entries()) {
-      send(ws, createEnvelope(id, topic))
+      const envelope = createEnvelope(id, topic)
+      const signature = JSON.stringify(envelope.snapshot)
+      if (snapshotSignatures.get(id) === signature) {
+        continue
+      }
+      snapshotSignatures.set(id, signature)
+      send(ws, envelope)
     }
   }
 
@@ -150,9 +182,14 @@ export function createWsRouter({
 
   function pushTerminalSnapshot(terminalId: string) {
     for (const ws of sockets) {
+      const snapshotSignatures = ensureSnapshotSignatures(ws)
       for (const [id, topic] of ws.data.subscriptions.entries()) {
         if (topic.type !== "terminal" || topic.terminalId !== terminalId) continue
-        send(ws, createEnvelope(id, topic))
+        const envelope = createEnvelope(id, topic)
+        const signature = JSON.stringify(envelope.snapshot)
+        if (snapshotSignatures.get(id) === signature) continue
+        snapshotSignatures.set(id, signature)
+        send(ws, envelope)
       }
     }
   }
@@ -177,18 +214,28 @@ export function createWsRouter({
 
   const disposeKeybindingEvents = keybindings.onChange(() => {
     for (const ws of sockets) {
+      const snapshotSignatures = ensureSnapshotSignatures(ws)
       for (const [id, topic] of ws.data.subscriptions.entries()) {
         if (topic.type !== "keybindings") continue
-        send(ws, createEnvelope(id, topic))
+        const envelope = createEnvelope(id, topic)
+        const signature = JSON.stringify(envelope.snapshot)
+        if (snapshotSignatures.get(id) === signature) continue
+        snapshotSignatures.set(id, signature)
+        send(ws, envelope)
       }
     }
   })
 
   const disposeUpdateEvents = updateManager?.onChange(() => {
     for (const ws of sockets) {
+      const snapshotSignatures = ensureSnapshotSignatures(ws)
       for (const [id, topic] of ws.data.subscriptions.entries()) {
         if (topic.type !== "update") continue
-        send(ws, createEnvelope(id, topic))
+        const envelope = createEnvelope(id, topic)
+        const signature = JSON.stringify(envelope.snapshot)
+        if (snapshotSignatures.get(id) === signature) continue
+        snapshotSignatures.set(id, signature)
+        send(ws, envelope)
       }
     }
   }) ?? (() => {})
@@ -258,6 +305,7 @@ export function createWsRouter({
           const project = store.getProject(command.projectId)
           for (const chat of store.listChatsByProject(command.projectId)) {
             await agent.cancel(chat.id)
+            await agent.closeChat(chat.id)
           }
           if (project) {
             terminals.closeByCwd(project.localPath)
@@ -283,6 +331,7 @@ export function createWsRouter({
         }
         case "chat.delete": {
           await agent.cancel(command.chatId)
+          await agent.closeChat(command.chatId)
           await store.deleteChat(command.chatId)
           send(ws, { v: PROTOCOL_VERSION, type: "ack", id })
           break
@@ -297,11 +346,70 @@ export function createWsRouter({
           send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
           break
         }
+        case "chat.refreshDiffs": {
+          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 changed = await resolvedDiffStore.refreshSnapshot(command.chatId, project.localPath)
+          send(ws, { v: PROTOCOL_VERSION, type: "ack", id })
+          if (changed) {
+            broadcastSnapshots()
+          }
+          return
+        }
+        case "chat.generateCommitMessage": {
+          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.generateCommitMessage({
+            projectPath: project.localPath,
+            paths: command.paths,
+          })
+          send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
+          return
+        }
+        case "chat.commitDiffs": {
+          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 changed = await resolvedDiffStore.commitFiles({
+            chatId: command.chatId,
+            projectPath: project.localPath,
+            paths: command.paths,
+            summary: command.summary,
+            description: command.description,
+          })
+          send(ws, { v: PROTOCOL_VERSION, type: "ack", id })
+          if (changed) {
+            broadcastSnapshots()
+          }
+          return
+        }
         case "chat.cancel": {
           await agent.cancel(command.chatId)
           send(ws, { v: PROTOCOL_VERSION, type: "ack", id })
           break
         }
+        case "chat.stopDraining": {
+          await agent.stopDraining(command.chatId)
+          send(ws, { v: PROTOCOL_VERSION, type: "ack", id })
+          break
+        }
         case "chat.respondTool": {
           await agent.respondTool(command)
           send(ws, { v: PROTOCOL_VERSION, type: "ack", id })
@@ -343,6 +451,11 @@ export function createWsRouter({
       broadcastSnapshots()
     } catch (error) {
       const messageText = error instanceof Error ? error.message : String(error)
+      console.error("[ws-router] command failed", {
+        id,
+        type: command.type,
+        message: messageText,
+      })
       send(ws, { v: PROTOCOL_VERSION, type: "error", id, message: messageText })
     }
   }
@@ -370,20 +483,25 @@ export function createWsRouter({
       }
 
       if (parsed.type === "subscribe") {
+        const snapshotSignatures = ensureSnapshotSignatures(ws)
         ws.data.subscriptions.set(parsed.id, parsed.topic)
+        snapshotSignatures.delete(parsed.id)
         if (parsed.topic.type === "local-projects") {
           void refreshDiscovery().then(() => {
             if (ws.data.subscriptions.has(parsed.id)) {
-              send(ws, createEnvelope(parsed.id, parsed.topic))
+              pushSnapshots(ws)
             }
           })
+          return
         }
-        send(ws, createEnvelope(parsed.id, parsed.topic))
+        pushSnapshots(ws)
         return
       }
 
       if (parsed.type === "unsubscribe") {
+        const snapshotSignatures = ensureSnapshotSignatures(ws)
         ws.data.subscriptions.delete(parsed.id)
+        snapshotSignatures.delete(parsed.id)
         send(ws, { v: PROTOCOL_VERSION, type: "ack", id: parsed.id })
         return
       }

package/src/shared/protocol.ts

@@ -75,7 +75,11 @@ export type ClientCommand =
       effort?: string
       planMode?: boolean
     }
+  | { type: "chat.refreshDiffs"; chatId: string }
+  | { type: "chat.generateCommitMessage"; chatId: string; paths: string[] }
+  | { type: "chat.commitDiffs"; chatId: string; paths: string[]; summary: string; description?: string }
   | { type: "chat.cancel"; chatId: string }
+  | { type: "chat.stopDraining"; chatId: string }
   | { type: "chat.respondTool"; chatId: string; toolUseId: string; result: unknown }
   | { type: "terminal.create"; projectId: string; terminalId: string; cols: number; rows: number; scrollback: number }
   | { type: "terminal.input"; terminalId: string; data: string }

package/src/shared/types.ts

@@ -171,6 +171,16 @@ export function resolveClaudeApiModelId(modelId: string, contextWindow?: ClaudeC
   return contextWindow === "1m" ? `${modelId}[1m]` : modelId
 }
 
+export function resolveClaudeContextWindowTokens(contextWindow: ClaudeContextWindow): number {
+  switch (contextWindow) {
+    case "1m":
+      return 1_000_000
+    case "200k":
+    default:
+      return 200_000
+  }
+}
+
 export type KannaStatus =
   | "idle"
   | "starting"
@@ -440,6 +450,43 @@ export interface StatusEntry extends TranscriptEntryBase {
   status: string
 }
 
+export interface ContextWindowUsageSnapshot {
+  usedTokens: number
+  totalProcessedTokens?: number
+  maxTokens?: number
+  inputTokens?: number
+  cachedInputTokens?: number
+  outputTokens?: number
+  reasoningOutputTokens?: number
+  lastUsedTokens?: number
+  lastInputTokens?: number
+  lastCachedInputTokens?: number
+  lastOutputTokens?: number
+  lastReasoningOutputTokens?: number
+  toolUses?: number
+  durationMs?: number
+  compactsAutomatically: boolean
+}
+
+export interface ChatDiffFile {
+  path: string
+  changeType: "added" | "deleted" | "modified" | "renamed"
+  patch: string
+  mimeType?: string
+  size?: number
+}
+
+export interface ChatDiffSnapshot {
+  status: "unknown" | "ready" | "no_repo"
+  branchName?: string
+  files: ChatDiffFile[]
+}
+
+export interface ContextWindowUpdatedEntry extends TranscriptEntryBase {
+  kind: "context_window_updated"
+  usage: ContextWindowUsageSnapshot
+}
+
 export interface CompactBoundaryEntry extends TranscriptEntryBase {
   kind: "compact_boundary"
 }
@@ -466,6 +513,7 @@ export type TranscriptEntry =
   | ToolResultEntry
   | ResultEntry
   | StatusEntry
+  | ContextWindowUpdatedEntry
   | CompactBoundaryEntry
   | CompactSummaryEntry
   | ContextClearedEntry
@@ -579,6 +627,7 @@ export type HydratedTranscriptMessage =
   | ({ kind: "assistant_text"; text: string; id: string; messageId?: string; timestamp: string; hidden?: boolean })
   | ({ kind: "result"; success: boolean; cancelled?: boolean; result: string; durationMs: number; costUsd?: number; id: string; messageId?: string; timestamp: string; hidden?: boolean })
   | ({ kind: "status"; status: string; id: string; messageId?: string; timestamp: string; hidden?: boolean })
+  | ({ kind: "context_window_updated"; usage: ContextWindowUsageSnapshot; id: string; messageId?: string; timestamp: string; hidden?: boolean })
   | ({ kind: "compact_boundary"; id: string; messageId?: string; timestamp: string; hidden?: boolean })
   | ({ kind: "compact_summary"; summary: string; id: string; messageId?: string; timestamp: string; hidden?: boolean })
   | ({ kind: "context_cleared"; id: string; messageId?: string; timestamp: string; hidden?: boolean })
@@ -592,6 +641,7 @@ export interface ChatRuntime {
   localPath: string
   title: string
   status: KannaStatus
+  isDraining: boolean
   provider: AgentProvider | null
   planMode: boolean
   sessionToken: string | null
@@ -600,6 +650,7 @@ export interface ChatRuntime {
 export interface ChatSnapshot {
   runtime: ChatRuntime
   messages: TranscriptEntry[]
+  diffs: ChatDiffSnapshot
   availableProviders: ProviderCatalogEntry[]
 }