kanna-code 0.25.1 → 0.26.1

package/src/server/event-store.ts

@@ -1,4 +1,4 @@
-import { appendFile, mkdir, rename, writeFile } from "node:fs/promises"
+import { appendFile, mkdir, rename, rm, writeFile } from "node:fs/promises"
 import { existsSync, readFileSync as readFileSyncImmediate } from "node:fs"
 import { homedir } from "node:os"
 import path from "node:path"
@@ -19,6 +19,7 @@ import {
 import { resolveLocalPath } from "./paths"
 
 const COMPACTION_THRESHOLD_BYTES = 2 * 1024 * 1024
+const STALE_EMPTY_CHAT_MAX_AGE_MS = 5 * 60 * 1000
 interface LegacyTranscriptStats {
   hasLegacyData: boolean
   sources: Array<"snapshot" | "messages_log">
@@ -454,6 +455,41 @@ export class EventStore {
     await this.append(this.chatsLogPath, event)
   }
 
+  async pruneStaleEmptyChats(args?: {
+    now?: number
+    maxAgeMs?: number
+    activeChatIds?: Iterable<string>
+  }) {
+    const now = args?.now ?? Date.now()
+    const maxAgeMs = args?.maxAgeMs ?? STALE_EMPTY_CHAT_MAX_AGE_MS
+    const activeChatIds = new Set(args?.activeChatIds ?? [])
+    const prunedChatIds: string[] = []
+
+    for (const chat of this.state.chatsById.values()) {
+      if (chat.deletedAt || activeChatIds.has(chat.id)) continue
+      if (now - chat.createdAt < maxAgeMs) continue
+      if (this.getMessages(chat.id).length > 0) continue
+
+      const event: ChatEvent = {
+        v: STORE_VERSION,
+        type: "chat_deleted",
+        timestamp: now,
+        chatId: chat.id,
+      }
+      await this.append(this.chatsLogPath, event)
+
+      const transcriptPath = this.transcriptPath(chat.id)
+      await rm(transcriptPath, { force: true })
+      if (this.cachedTranscript?.chatId === chat.id) {
+        this.cachedTranscript = null
+      }
+
+      prunedChatIds.push(chat.id)
+    }
+
+    return prunedChatIds
+  }
+
   async setChatProvider(chatId: string, provider: AgentProvider) {
     const chat = this.requireChat(chatId)
     if (chat.provider === provider) return

package/src/server/server.ts

@@ -17,6 +17,7 @@ import { getProjectUploadDir } from "./paths"
 
 const MAX_UPLOAD_FILES = 10
 const MAX_UPLOAD_SIZE_BYTES = 25 * 1024 * 1024
+const STALE_EMPTY_CHAT_PRUNE_INTERVAL_MS = 60 * 1000
 
 export async function persistUploadedFiles(args: {
   projectId: string
@@ -99,7 +100,7 @@ export async function startKannaServer(options: StartKannaServerOptions = {}) {
   const agent = new AgentCoordinator({
     store,
     onStateChange: () => {
-      router.broadcastSnapshots()
+      void router.broadcastSnapshots()
     },
   })
   router = createWsRouter({
@@ -113,6 +114,9 @@ export async function startKannaServer(options: StartKannaServerOptions = {}) {
     machineDisplayName,
     updateManager,
   })
+  const staleEmptyChatPruneInterval = setInterval(() => {
+    void router.broadcastSnapshots()
+  }, STALE_EMPTY_CHAT_PRUNE_INTERVAL_MS)
 
   const distDir = path.join(import.meta.dir, "..", "..", "dist", "client")
 
@@ -188,6 +192,7 @@ export async function startKannaServer(options: StartKannaServerOptions = {}) {
   }
 
   const shutdown = async () => {
+    clearInterval(staleEmptyChatPruneInterval)
     for (const chatId of [...agent.activeTurns.keys()]) {
       await agent.cancel(chatId)
     }

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

@@ -577,7 +577,7 @@ describe("ws-router", () => {
     router.handleOpen(wsA as never)
     router.handleOpen(wsB as never)
 
-    router.handleMessage(
+    await router.handleMessage(
       wsA as never,
       JSON.stringify({
         v: 1,
@@ -586,7 +586,7 @@ describe("ws-router", () => {
         topic: { type: "sidebar" },
       })
     )
-    router.handleMessage(
+    await router.handleMessage(
       wsB as never,
       JSON.stringify({
         v: 1,
@@ -596,7 +596,7 @@ describe("ws-router", () => {
       })
     )
 
-    router.handleMessage(
+    await router.handleMessage(
       wsA as never,
       JSON.stringify({
         v: 1,
@@ -606,8 +606,6 @@ describe("ws-router", () => {
       })
     )
 
-    await Promise.resolve()
-
     expect(wsA.sent.at(-2)).toEqual({
       v: PROTOCOL_VERSION,
       type: "ack",
@@ -667,6 +665,83 @@ describe("ws-router", () => {
     })
   })
 
+  test("prunes stale empty chats before sending sidebar 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-stale", {
+      id: "chat-stale",
+      projectId: "project-1",
+      title: "New Chat",
+      createdAt: 1,
+      updatedAt: 1,
+      unread: false,
+      provider: null,
+      planMode: false,
+      sessionToken: null,
+      lastTurnOutcome: null,
+    })
+
+    let pruneCalls = 0
+    const router = createWsRouter({
+      store: {
+        state,
+        async pruneStaleEmptyChats() {
+          pruneCalls += 1
+          state.chatsById.delete("chat-stale")
+          return ["chat-stale"]
+        },
+      } 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: "subscribe",
+        id: "sidebar-sub-1",
+        topic: { type: "sidebar" },
+      })
+    )
+
+    expect(pruneCalls).toBe(1)
+    expect(ws.sent[0]).toEqual({
+      v: PROTOCOL_VERSION,
+      type: "snapshot",
+      id: "sidebar-sub-1",
+      snapshot: {
+        type: "sidebar",
+        data: {
+          projectGroups: [{
+            groupKey: "project-1",
+            localPath: "/tmp/project",
+            chats: [],
+          }],
+        },
+      },
+    })
+  })
+
   test("broadcasts background title-generation errors to connected clients", () => {
     let reportBackgroundError: ((message: string) => void) | null | undefined
     const router = createWsRouter({

package/src/server/ws-router.ts

@@ -22,7 +22,7 @@ export interface ClientState {
 
 interface CreateWsRouterArgs {
   store: EventStore
-  diffStore?: Pick<DiffStore, "getProjectSnapshot" | "refreshSnapshot" | "listBranches" | "previewMergeBranch" | "mergeBranch" | "syncBranch" | "checkoutBranch" | "createBranch" | "generateCommitMessage" | "commitFiles" | "discardFile" | "ignoreFile" | "readPatch">
+  diffStore?: Pick<DiffStore, "getProjectSnapshot" | "refreshSnapshot" | "initializeGit" | "getGitHubPublishInfo" | "checkGitHubRepoAvailability" | "publishToGitHub" | "listBranches" | "previewMergeBranch" | "mergeBranch" | "syncBranch" | "checkoutBranch" | "createBranch" | "generateCommitMessage" | "commitFiles" | "discardFile" | "ignoreFile" | "readPatch">
   agent: AgentCoordinator
   terminals: TerminalManager
   keybindings: KeybindingsManager
@@ -59,6 +59,10 @@ export function createWsRouter({
   const resolvedDiffStore = diffStore ?? {
     getProjectSnapshot: () => ({ status: "unknown", branchName: undefined, defaultBranchName: undefined, originRepoSlug: undefined, hasUpstream: undefined, aheadCount: undefined, behindCount: undefined, lastFetchedAt: undefined, files: [] as const, branchHistory: { entries: [] as const } }),
     refreshSnapshot: async () => false,
+    initializeGit: async () => ({ ok: true, branchName: undefined, snapshotChanged: false }),
+    getGitHubPublishInfo: async () => ({ ghInstalled: false, authenticated: false, activeAccountLogin: undefined, owners: [], suggestedRepoName: "my-repo" }),
+    checkGitHubRepoAvailability: async () => ({ available: false, message: "Unavailable" }),
+    publishToGitHub: async () => ({ ok: false, title: "Publish failed", message: "Unavailable", snapshotChanged: false }),
     listBranches: async () => ({ recent: [], local: [], remote: [], pullRequests: [], pullRequestsStatus: "unavailable" as const }),
     previewMergeBranch: async () => ({ currentBranchName: undefined, targetBranchName: "", targetDisplayName: "", status: "error" as const, commitCount: 0, hasConflicts: false, message: "Merge preview unavailable." }),
     mergeBranch: async () => ({ ok: false as const, title: "Merge failed", message: "Merge unavailable.", snapshotChanged: false }),
@@ -72,6 +76,19 @@ export function createWsRouter({
     readPatch: async () => ({ patch: "" }),
   }
 
+  function getProtectedChatIds() {
+    return new Set([
+      ...agent.getActiveStatuses().keys(),
+      ...agent.getDrainingChatIds().values(),
+    ])
+  }
+
+  async function maybePruneStaleEmptyChats() {
+    await store.pruneStaleEmptyChats?.({
+      activeChatIds: getProtectedChatIds(),
+    })
+  }
+
   function createEnvelope(id: string, topic: SubscriptionTopic): ServerEnvelope {
     if (topic.type === "sidebar") {
       return {
@@ -175,7 +192,10 @@ export function createWsRouter({
     }
   }
 
-  function pushSnapshots(ws: ServerWebSocket<ClientState>) {
+  async function pushSnapshots(ws: ServerWebSocket<ClientState>, options?: { skipPrune?: boolean }) {
+    if (!options?.skipPrune) {
+      await maybePruneStaleEmptyChats()
+    }
     const snapshotSignatures = ensureSnapshotSignatures(ws)
     for (const [id, topic] of ws.data.subscriptions.entries()) {
       const envelope = createEnvelope(id, topic)
@@ -189,9 +209,10 @@ export function createWsRouter({
     }
   }
 
-  function broadcastSnapshots() {
+  async function broadcastSnapshots() {
+    await maybePruneStaleEmptyChats()
     for (const ws of sockets) {
-      pushSnapshots(ws)
+      await pushSnapshots(ws, { skipPrune: true })
     }
   }
 
@@ -399,7 +420,51 @@ export function createWsRouter({
           const changed = await resolvedDiffStore.refreshSnapshot(project.id, project.localPath)
           send(ws, { v: PROTOCOL_VERSION, type: "ack", id })
           if (changed) {
-            broadcastSnapshots()
+            void broadcastSnapshots()
+          }
+          return
+        }
+        case "chat.initGit": {
+          const { project } = resolveChatProject(command.chatId)
+          const result = await resolvedDiffStore.initializeGit({
+            projectId: project.id,
+            projectPath: project.localPath,
+          })
+          send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
+          if (result.snapshotChanged) {
+            void broadcastSnapshots()
+          }
+          return
+        }
+        case "chat.getGitHubPublishInfo": {
+          const { project } = resolveChatProject(command.chatId)
+          const result = await resolvedDiffStore.getGitHubPublishInfo({
+            projectPath: project.localPath,
+          })
+          send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
+          return
+        }
+        case "chat.checkGitHubRepoAvailability": {
+          const result = await resolvedDiffStore.checkGitHubRepoAvailability({
+            owner: command.owner,
+            name: command.name,
+          })
+          send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
+          return
+        }
+        case "chat.publishToGitHub": {
+          const { project } = resolveChatProject(command.chatId)
+          const result = await resolvedDiffStore.publishToGitHub({
+            projectId: project.id,
+            projectPath: project.localPath,
+            owner: command.owner,
+            name: command.name,
+            visibility: command.visibility,
+            description: command.description,
+          })
+          send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
+          if (result.snapshotChanged) {
+            void broadcastSnapshots()
           }
           return
         }
@@ -429,7 +494,7 @@ export function createWsRouter({
           })
           send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
           if (result.snapshotChanged) {
-            broadcastSnapshots()
+            void broadcastSnapshots()
           }
           return
         }
@@ -443,7 +508,7 @@ export function createWsRouter({
           })
           send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
           if (result.snapshotChanged) {
-            broadcastSnapshots()
+            void broadcastSnapshots()
           }
           return
         }
@@ -456,7 +521,7 @@ export function createWsRouter({
           })
           send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
           if (result.snapshotChanged) {
-            broadcastSnapshots()
+            void broadcastSnapshots()
           }
           return
         }
@@ -470,7 +535,7 @@ export function createWsRouter({
           })
           send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
           if (result.snapshotChanged) {
-            broadcastSnapshots()
+            void broadcastSnapshots()
           }
           return
         }
@@ -495,7 +560,7 @@ export function createWsRouter({
           })
           send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
           if (result.snapshotChanged) {
-            broadcastSnapshots()
+            void broadcastSnapshots()
           }
           return
         }
@@ -508,7 +573,7 @@ export function createWsRouter({
           })
           send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
           if (result.snapshotChanged) {
-            broadcastSnapshots()
+            void broadcastSnapshots()
           }
           return
         }
@@ -521,7 +586,7 @@ export function createWsRouter({
           })
           send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
           if (result.snapshotChanged) {
-            broadcastSnapshots()
+            void broadcastSnapshots()
           }
           return
         }
@@ -580,7 +645,7 @@ export function createWsRouter({
         }
       }
 
-      broadcastSnapshots()
+      await broadcastSnapshots()
     } catch (error) {
       const messageText = error instanceof Error ? error.message : String(error)
       console.error("[ws-router] command failed", {
@@ -600,7 +665,7 @@ export function createWsRouter({
       sockets.delete(ws)
     },
     broadcastSnapshots,
-    handleMessage(ws: ServerWebSocket<ClientState>, raw: string | Buffer | ArrayBuffer | Uint8Array) {
+    async handleMessage(ws: ServerWebSocket<ClientState>, raw: string | Buffer | ArrayBuffer | Uint8Array) {
       let parsed: unknown
       try {
         parsed = JSON.parse(String(raw))
@@ -621,12 +686,12 @@ export function createWsRouter({
         if (parsed.topic.type === "local-projects") {
           void refreshDiscovery().then(() => {
             if (ws.data.subscriptions.has(parsed.id)) {
-              pushSnapshots(ws)
+              void pushSnapshots(ws)
             }
           })
           return
         }
-        pushSnapshots(ws)
+        await pushSnapshots(ws)
         return
       }
 
@@ -638,7 +703,7 @@ export function createWsRouter({
         return
       }
 
-      void handleCommand(ws, parsed)
+      await handleCommand(ws, parsed)
     },
     dispose() {
       agent.setBackgroundErrorReporter?.(null)

package/src/shared/protocol.ts

@@ -81,6 +81,17 @@ export type ClientCommand =
       planMode?: boolean
     }
   | { type: "chat.refreshDiffs"; chatId: string }
+  | { type: "chat.initGit"; chatId: string }
+  | { type: "chat.getGitHubPublishInfo"; chatId: string }
+  | { type: "chat.checkGitHubRepoAvailability"; chatId: string; owner: string; name: string }
+  | {
+      type: "chat.publishToGitHub"
+      chatId: string
+      owner: string
+      name: string
+      visibility: "public" | "private"
+      description?: string
+    }
   | { type: "chat.listBranches"; chatId: string }
   | {
       type: "chat.previewMergeBranch"

package/src/shared/types.ts

@@ -528,6 +528,19 @@ export interface ChatBranchListResult {
   pullRequestsError?: string
 }
 
+export interface GitHubPublishInfo {
+  ghInstalled: boolean
+  authenticated: boolean
+  activeAccountLogin?: string
+  owners: string[]
+  suggestedRepoName: string
+}
+
+export interface GitHubRepoAvailabilityResult {
+  available: boolean
+  message: string
+}
+
 export interface BranchMetadata {
   branchName?: string
   defaultBranchName?: string