kanna-code 0.23.1 → 0.25.0

package/src/server/ws-router.ts

@@ -22,7 +22,7 @@ export interface ClientState {
 
 interface CreateWsRouterArgs {
   store: EventStore
-  diffStore?: Pick<DiffStore, "getSnapshot" | "refreshSnapshot" | "generateCommitMessage" | "commitFiles" | "discardFile" | "ignoreFile">
+  diffStore?: Pick<DiffStore, "getProjectSnapshot" | "refreshSnapshot" | "listBranches" | "previewMergeBranch" | "mergeBranch" | "syncBranch" | "checkoutBranch" | "createBranch" | "generateCommitMessage" | "commitFiles" | "discardFile" | "ignoreFile" | "readPatch">
   agent: AgentCoordinator
   terminals: TerminalManager
   keybindings: KeybindingsManager
@@ -57,12 +57,19 @@ export function createWsRouter({
 }: CreateWsRouterArgs) {
   const sockets = new Set<ServerWebSocket<ClientState>>()
   const resolvedDiffStore = diffStore ?? {
-    getSnapshot: () => ({ status: "unknown", branchName: undefined, hasUpstream: undefined, files: [] as const }),
+    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,
+    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 }),
+    syncBranch: async () => ({ ok: true, action: "fetch" as const, branchName: undefined, snapshotChanged: false }),
+    checkoutBranch: async () => ({ ok: true, branchName: undefined, snapshotChanged: false }),
+    createBranch: async () => ({ ok: true, branchName: "main", snapshotChanged: 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 }),
+    readPatch: async () => ({ patch: "" }),
   }
 
   function createEnvelope(id: string, topic: SubscriptionTopic): ServerEnvelope {
@@ -137,6 +144,20 @@ export function createWsRouter({
       }
     }
 
+    if (topic.type === "project-git") {
+      return {
+        v: PROTOCOL_VERSION,
+        type: "snapshot",
+        id,
+        snapshot: {
+          type: "project-git",
+          data: store.getProject(topic.projectId)
+            ? resolvedDiffStore.getProjectSnapshot(topic.projectId)
+            : null,
+        },
+      }
+    }
+
     return {
       v: PROTOCOL_VERSION,
       type: "snapshot",
@@ -148,8 +169,7 @@ export function createWsRouter({
           agent.getActiveStatuses(),
           agent.getDrainingChatIds(),
           topic.chatId,
-          (chatId) => store.getRecentChatHistory(chatId, topic.recentLimit ?? DEFAULT_CHAT_RECENT_LIMIT),
-          (chatId) => resolvedDiffStore.getSnapshot(chatId)
+          (chatId) => store.getRecentChatHistory(chatId, topic.recentLimit ?? DEFAULT_CHAT_RECENT_LIMIT)
         ),
       },
     }
@@ -250,6 +270,14 @@ export function createWsRouter({
 
   agent.setBackgroundErrorReporter?.(broadcastError)
 
+  function resolveChatProject(chatId: string) {
+    const chat = store.getChat(chatId)
+    if (!chat) throw new Error("Chat not found")
+    const project = store.getProject(chat.projectId)
+    if (!project) throw new Error("Project not found")
+    return { chat, project }
+  }
+
   async function handleCommand(ws: ServerWebSocket<ClientState>, message: Extract<ClientEnvelope, { type: "command" }>) {
     const { command, id } = message
     try {
@@ -322,6 +350,18 @@ export function createWsRouter({
           send(ws, { v: PROTOCOL_VERSION, type: "ack", id })
           break
         }
+        case "project.readDiffPatch": {
+          const project = store.getProject(command.projectId)
+          if (!project) {
+            throw new Error("Project not found")
+          }
+          const result = await resolvedDiffStore.readPatch({
+            projectPath: project.localPath,
+            path: command.path,
+          })
+          send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
+          return
+        }
         case "system.openExternal": {
           await openExternal(command)
           send(ws, { v: PROTOCOL_VERSION, type: "ack", id })
@@ -355,30 +395,87 @@ export function createWsRouter({
           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)
+          const { project } = resolveChatProject(command.chatId)
+          const changed = await resolvedDiffStore.refreshSnapshot(project.id, 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")
+        case "chat.listBranches": {
+          const { project } = resolveChatProject(command.chatId)
+          const result = await resolvedDiffStore.listBranches({
+            projectPath: project.localPath,
+          })
+          send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
+          return
+        }
+        case "chat.previewMergeBranch": {
+          const { project } = resolveChatProject(command.chatId)
+          const result = await resolvedDiffStore.previewMergeBranch({
+            projectPath: project.localPath,
+            branch: command.branch,
+          })
+          send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
+          return
+        }
+        case "chat.mergeBranch": {
+          const { project } = resolveChatProject(command.chatId)
+          const result = await resolvedDiffStore.mergeBranch({
+            projectId: project.id,
+            projectPath: project.localPath,
+            branch: command.branch,
+          })
+          send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
+          if (result.snapshotChanged) {
+            broadcastSnapshots()
           }
-          const project = store.getProject(chat.projectId)
-          if (!project) {
-            throw new Error("Project not found")
+          return
+        }
+        case "chat.checkoutBranch": {
+          const { project } = resolveChatProject(command.chatId)
+          const result = await resolvedDiffStore.checkoutBranch({
+            projectId: project.id,
+            projectPath: project.localPath,
+            branch: command.branch,
+            bringChanges: command.bringChanges,
+          })
+          send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
+          if (result.snapshotChanged) {
+            broadcastSnapshots()
+          }
+          return
+        }
+        case "chat.syncBranch": {
+          const { project } = resolveChatProject(command.chatId)
+          const result = await resolvedDiffStore.syncBranch({
+            projectId: project.id,
+            projectPath: project.localPath,
+            action: command.action,
+          })
+          send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
+          if (result.snapshotChanged) {
+            broadcastSnapshots()
           }
+          return
+        }
+        case "chat.createBranch": {
+          const { project } = resolveChatProject(command.chatId)
+          const result = await resolvedDiffStore.createBranch({
+            projectId: project.id,
+            projectPath: project.localPath,
+            name: command.name,
+            baseBranchName: command.baseBranchName,
+          })
+          send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
+          if (result.snapshotChanged) {
+            broadcastSnapshots()
+          }
+          return
+        }
+        case "chat.generateCommitMessage": {
+          const { project } = resolveChatProject(command.chatId)
           const result = await resolvedDiffStore.generateCommitMessage({
             projectPath: project.localPath,
             paths: command.paths,
@@ -387,16 +484,9 @@ export function createWsRouter({
           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 { project } = resolveChatProject(command.chatId)
           const result = await resolvedDiffStore.commitFiles({
-            chatId: command.chatId,
+            projectId: project.id,
             projectPath: project.localPath,
             paths: command.paths,
             summary: command.summary,
@@ -410,16 +500,9 @@ 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 { project } = resolveChatProject(command.chatId)
           const result = await resolvedDiffStore.discardFile({
-            chatId: command.chatId,
+            projectId: project.id,
             projectPath: project.localPath,
             path: command.path,
           })
@@ -430,16 +513,9 @@ export function createWsRouter({
           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 { project } = resolveChatProject(command.chatId)
           const result = await resolvedDiffStore.ignoreFile({
-            chatId: command.chatId,
+            projectId: project.id,
             projectPath: project.localPath,
             path: command.path,
           })
@@ -461,9 +537,7 @@ export function createWsRouter({
         }
         case "chat.loadHistory": {
           const chat = store.getChat(command.chatId)
-          if (!chat) {
-            throw new Error("Chat not found")
-          }
+          if (!chat) throw new Error("Chat not found")
           const page = store.getMessagesPageBefore(command.chatId, command.beforeCursor, command.limit)
           send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result: page })
           return

package/src/shared/protocol.ts

@@ -1,6 +1,7 @@
 import type {
   AgentProvider,
   ChatAttachment,
+  ChatDiffSnapshot,
   ChatHistoryPage,
   ChatSnapshot,
   DiffCommitMode,
@@ -24,6 +25,7 @@ export type SubscriptionTopic =
   | { type: "update" }
   | { type: "keybindings" }
   | { type: "chat"; chatId: string; recentLimit?: number }
+  | { type: "project-git"; projectId: string }
   | { type: "terminal"; terminalId: string }
 
 export interface TerminalSnapshot {
@@ -48,6 +50,7 @@ export type ClientCommand =
   | { type: "project.open"; localPath: string }
   | { type: "project.create"; localPath: string; title: string }
   | { type: "project.remove"; projectId: string }
+  | { type: "project.readDiffPatch"; projectId: string; path: string }
   | { type: "system.ping" }
   | { type: "update.check"; force?: boolean }
   | { type: "update.install" }
@@ -78,6 +81,58 @@ export type ClientCommand =
       planMode?: boolean
     }
   | { type: "chat.refreshDiffs"; chatId: string }
+  | { type: "chat.listBranches"; chatId: string }
+  | {
+      type: "chat.previewMergeBranch"
+      chatId: string
+      branch:
+      | { kind: "local"; name: string }
+      | { kind: "remote"; name: string; remoteRef: string }
+      | {
+          kind: "pull_request"
+          name: string
+          prNumber: number
+          headRefName: string
+          headRepoCloneUrl?: string
+          isCrossRepository?: boolean
+          remoteRef?: string
+        }
+    }
+  | {
+      type: "chat.mergeBranch"
+      chatId: string
+      branch:
+      | { kind: "local"; name: string }
+      | { kind: "remote"; name: string; remoteRef: string }
+      | {
+          kind: "pull_request"
+          name: string
+          prNumber: number
+          headRefName: string
+          headRepoCloneUrl?: string
+          isCrossRepository?: boolean
+          remoteRef?: string
+        }
+    }
+  | { type: "chat.syncBranch"; chatId: string; action: "fetch" | "pull" | "push" | "publish" }
+  | {
+      type: "chat.checkoutBranch"
+      chatId: string
+      branch:
+      | { kind: "local"; name: string }
+      | { kind: "remote"; name: string; remoteRef: string }
+      | {
+          kind: "pull_request"
+          name: string
+          prNumber: number
+          headRefName: string
+          headRepoCloneUrl?: string
+          isCrossRepository?: boolean
+          remoteRef?: string
+        }
+      bringChanges?: boolean
+    }
+  | { type: "chat.createBranch"; chatId: string; name: string; baseBranchName?: 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 }
@@ -102,6 +157,7 @@ export type ServerSnapshot =
   | { type: "update"; data: UpdateSnapshot }
   | { type: "keybindings"; data: KeybindingsSnapshot }
   | { type: "chat"; data: ChatSnapshot | null }
+  | { type: "project-git"; data: ChatDiffSnapshot | null }
   | { type: "terminal"; data: TerminalSnapshot | null }
 
 export type ServerEnvelope =

package/src/shared/types.ts

@@ -273,6 +273,9 @@ export type KeybindingAction =
   | "openInFinder"
   | "openInEditor"
   | "addSplitTerminal"
+  | "jumpToSidebarChat"
+  | "createChatInCurrentProject"
+  | "openAddProject"
 
 export const DEFAULT_KEYBINDINGS: Record<KeybindingAction, string[]> = {
   toggleEmbeddedTerminal: ["cmd+j", "ctrl+`"],
@@ -280,6 +283,9 @@ export const DEFAULT_KEYBINDINGS: Record<KeybindingAction, string[]> = {
   openInFinder: ["cmd+alt+f", "ctrl+alt+f"],
   openInEditor: ["cmd+shift+o", "ctrl+shift+o"],
   addSplitTerminal: ["cmd+/", "ctrl+/"],
+  jumpToSidebarChat: ["cmd+alt"],
+  createChatInCurrentProject: ["cmd+alt+n"],
+  openAddProject: ["cmd+alt+o"],
 }
 
 export interface KeybindingsSnapshot {
@@ -472,39 +478,140 @@ export interface ChatDiffFile {
   path: string
   changeType: "added" | "deleted" | "modified" | "renamed"
   isUntracked: boolean
-  patch: string
+  additions: number
+  deletions: number
+  patchDigest: string
   mimeType?: string
   size?: number
 }
 
-export interface ChatDiffSnapshot {
-  status: "unknown" | "ready" | "no_repo"
+export interface ChatBranchHistoryEntry {
+  sha: string
+  summary: string
+  description: string
+  authorName?: string
+  authoredAt: string
+  tags: string[]
+  githubUrl?: string
+}
+
+export interface ChatBranchHistorySnapshot {
+  entries: ChatBranchHistoryEntry[]
+}
+
+export type ChatBranchListEntryKind = "local" | "remote" | "pull_request"
+
+export interface ChatBranchListEntry {
+  id: string
+  kind: ChatBranchListEntryKind
+  name: string
+  displayName: string
+  updatedAt?: string
+  description?: string
+  remoteRef?: string
+  prNumber?: number
+  prTitle?: string
+  headRefName?: string
+  headLabel?: string
+  headRepoCloneUrl?: string
+  isCrossRepository?: boolean
+}
+
+export interface ChatBranchListResult {
+  currentBranchName?: string
+  defaultBranchName?: string
+  recent: ChatBranchListEntry[]
+  local: ChatBranchListEntry[]
+  remote: ChatBranchListEntry[]
+  pullRequests: ChatBranchListEntry[]
+  pullRequestsStatus: "available" | "unavailable" | "error"
+  pullRequestsError?: string
+}
+
+export interface BranchMetadata {
   branchName?: string
+  defaultBranchName?: string
+  originRepoSlug?: string
   hasUpstream?: boolean
-  files: ChatDiffFile[]
 }
 
-export type DiffCommitMode = "commit_and_push" | "commit_only"
+export interface UpstreamStatus {
+  aheadCount?: number
+  behindCount?: number
+  lastFetchedAt?: string
+}
+
+export interface ChatDiffSnapshot extends BranchMetadata, UpstreamStatus {
+  status: "unknown" | "ready" | "no_repo"
+  files: ChatDiffFile[]
+  branchHistory?: ChatBranchHistorySnapshot
+}
 
-export interface DiffCommitSuccess {
+export interface BranchActionSuccess {
   ok: true
-  mode: DiffCommitMode
   branchName?: string
-  pushed: boolean
   snapshotChanged: boolean
 }
 
-export interface DiffCommitFailure {
+export interface BranchActionFailure {
   ok: false
-  mode: DiffCommitMode
-  phase: "commit" | "push"
   title: string
   message: string
   detail?: string
-  localCommitCreated?: boolean
+  cancelled?: boolean
   snapshotChanged?: boolean
 }
 
+export type ChatSyncSuccess = BranchActionSuccess & {
+  action: "fetch" | "pull" | "push" | "publish"
+  aheadCount?: number
+  behindCount?: number
+}
+
+export type ChatSyncFailure = BranchActionFailure & {
+  action: "fetch" | "pull" | "push" | "publish"
+}
+
+export type ChatSyncResult = ChatSyncSuccess | ChatSyncFailure
+
+export type DiffCommitMode = "commit_and_push" | "commit_only"
+
+export type ChatCheckoutBranchSuccess = BranchActionSuccess
+export type ChatCheckoutBranchFailure = BranchActionFailure
+export type ChatCheckoutBranchResult = ChatCheckoutBranchSuccess | ChatCheckoutBranchFailure
+
+export type ChatCreateBranchSuccess = BranchActionSuccess & { branchName: string }
+export type ChatCreateBranchFailure = BranchActionFailure
+export type ChatCreateBranchResult = ChatCreateBranchSuccess | ChatCreateBranchFailure
+
+export type ChatMergePreviewStatus = "up_to_date" | "mergeable" | "conflicts" | "error"
+
+export interface ChatMergePreviewResult {
+  currentBranchName?: string
+  targetBranchName: string
+  targetDisplayName: string
+  status: ChatMergePreviewStatus
+  commitCount: number
+  hasConflicts: boolean
+  message: string
+  detail?: string
+}
+
+export type ChatMergeBranchSuccess = BranchActionSuccess
+export type ChatMergeBranchFailure = BranchActionFailure
+export type ChatMergeBranchResult = ChatMergeBranchSuccess | ChatMergeBranchFailure
+
+export type DiffCommitSuccess = BranchActionSuccess & {
+  mode: DiffCommitMode
+  pushed: boolean
+}
+
+export type DiffCommitFailure = BranchActionFailure & {
+  mode: DiffCommitMode
+  phase: "commit" | "push"
+  localCommitCreated?: boolean
+}
+
 export type DiffCommitResult = DiffCommitSuccess | DiffCommitFailure
 
 export interface ContextWindowUpdatedEntry extends TranscriptEntryBase {
@@ -682,7 +789,6 @@ export interface ChatSnapshot {
   runtime: ChatRuntime
   messages: TranscriptEntry[]
   history: ChatHistorySnapshot
-  diffs: ChatDiffSnapshot
   availableProviders: ProviderCatalogEntry[]
 }