kanna-code 0.23.1 → 0.24.0

package/src/server/diff-store.ts

@@ -1,23 +1,47 @@
 import { mkdtemp, readFile, rm, stat, writeFile } from "node:fs/promises"
 import { tmpdir } from "node:os"
 import path from "node:path"
-import type { ChatDiffFile, ChatDiffSnapshot, DiffCommitMode, DiffCommitResult } from "../shared/types"
+import type {
+  ChatBranchHistoryEntry,
+  ChatBranchHistorySnapshot,
+  ChatBranchListEntry,
+  ChatBranchListResult,
+  ChatCheckoutBranchResult,
+  ChatCreateBranchResult,
+  ChatDiffFile,
+  ChatDiffSnapshot,
+  ChatSyncResult,
+  DiffCommitMode,
+  DiffCommitResult,
+} from "../shared/types"
 import { generateCommitMessageDetailed } from "./generate-commit-message"
 import { inferProjectFileContentType } from "./uploads"
 
 interface StoredChatDiffState {
   status: ChatDiffSnapshot["status"]
   branchName?: string
+  defaultBranchName?: string
+  originRepoSlug?: string
   hasUpstream?: boolean
+  aheadCount?: number
+  behindCount?: number
+  lastFetchedAt?: string
   files: ChatDiffFile[]
+  branchHistory: ChatBranchHistorySnapshot
 }
 
 function createEmptyState(): StoredChatDiffState {
   return {
     status: "unknown",
     branchName: undefined,
+    defaultBranchName: undefined,
+    originRepoSlug: undefined,
     hasUpstream: undefined,
+    aheadCount: undefined,
+    behindCount: undefined,
+    lastFetchedAt: undefined,
     files: [],
+    branchHistory: { entries: [] },
   }
 }
 
@@ -27,8 +51,27 @@ function snapshotsEqual(left: StoredChatDiffState | undefined, right: StoredChat
   }
   if (left.status !== right.status) return false
   if (left.branchName !== right.branchName) return false
+  if (left.defaultBranchName !== right.defaultBranchName) return false
+  if (left.originRepoSlug !== right.originRepoSlug) return false
   if (left.hasUpstream !== right.hasUpstream) return false
+  if (left.aheadCount !== right.aheadCount) return false
+  if (left.behindCount !== right.behindCount) return false
+  if (left.lastFetchedAt !== right.lastFetchedAt) return false
   if (left.files.length !== right.files.length) return false
+  if (left.branchHistory.entries.length !== right.branchHistory.entries.length) return false
+  const sameHistory = left.branchHistory.entries.every((entry, index) => {
+    const other = right.branchHistory.entries[index]
+    return Boolean(other)
+      && entry.sha === other.sha
+      && entry.summary === other.summary
+      && entry.description === other.description
+      && entry.authorName === other.authorName
+      && entry.authoredAt === other.authoredAt
+      && entry.githubUrl === other.githubUrl
+      && entry.tags.length === other.tags.length
+      && entry.tags.every((tag, tagIndex) => tag === other.tags[tagIndex])
+  })
+  if (!sameHistory) return false
   return left.files.every((file, index) => {
     const other = right.files[index]
     return Boolean(other)
@@ -72,6 +115,24 @@ async function runGit(args: string[], cwd: string) {
   }
 }
 
+async function runCommand(args: string[]) {
+  const process = Bun.spawn(args, {
+    stdout: "pipe",
+    stderr: "pipe",
+  })
+  const [stdout, stderr, exitCode] = await Promise.all([
+    new Response(process.stdout).text(),
+    new Response(process.stderr).text(),
+    process.exited,
+  ])
+
+  return {
+    stdout,
+    stderr,
+    exitCode,
+  }
+}
+
 function formatGitFailure(result: Awaited<ReturnType<typeof runGit>>) {
   return [result.stderr.trim(), result.stdout.trim()].filter(Boolean).join("\n")
 }
@@ -160,6 +221,312 @@ async function hasUpstreamBranch(repoRoot: string) {
   return upstream.exitCode === 0 && upstream.stdout.trim().length > 0
 }
 
+async function getLastFetchedAt(repoRoot: string) {
+  const gitDirResult = await runGit(["rev-parse", "--git-dir"], repoRoot)
+  if (gitDirResult.exitCode !== 0) {
+    return undefined
+  }
+
+  const gitDir = gitDirResult.stdout.trim()
+  const fetchHeadPath = path.resolve(repoRoot, gitDir, "FETCH_HEAD")
+  try {
+    const fetchHeadStat = await stat(fetchHeadPath)
+    return fetchHeadStat.mtime.toISOString()
+  } catch {
+    return undefined
+  }
+}
+
+async function getUpstreamStatusCounts(repoRoot: string) {
+  const result = await runGit(["rev-list", "--left-right", "--count", "HEAD...@{upstream}"], repoRoot)
+  if (result.exitCode !== 0) {
+    return { aheadCount: undefined, behindCount: undefined }
+  }
+
+  const [aheadRaw, behindRaw] = result.stdout.trim().split(/\s+/u)
+  const aheadCount = Number.parseInt(aheadRaw ?? "", 10)
+  const behindCount = Number.parseInt(behindRaw ?? "", 10)
+  return {
+    aheadCount: Number.isFinite(aheadCount) ? aheadCount : undefined,
+    behindCount: Number.isFinite(behindCount) ? behindCount : undefined,
+  }
+}
+
+async function getOriginRemoteUrl(repoRoot: string) {
+  const result = await runGit(["remote", "get-url", "origin"], repoRoot)
+  if (result.exitCode !== 0) {
+    return null
+  }
+  const remoteUrl = result.stdout.trim()
+  return remoteUrl.length > 0 ? remoteUrl : null
+}
+
+async function getGitHubRemoteSlugs(repoRoot: string) {
+  const remotesResult = await runGit(["remote"], repoRoot)
+  if (remotesResult.exitCode !== 0) {
+    return new Map<string, string>()
+  }
+
+  const remoteNames = remotesResult.stdout
+    .split(/\r?\n/u)
+    .map((line) => line.trim())
+    .filter(Boolean)
+
+  const remoteSlugEntries = await Promise.all(remoteNames.map(async (remoteName) => {
+    const remoteUrlResult = await runGit(["remote", "get-url", remoteName], repoRoot)
+    if (remoteUrlResult.exitCode !== 0) {
+      return null
+    }
+    const repoSlug = extractGitHubRepoSlug(remoteUrlResult.stdout.trim())
+    return repoSlug ? [remoteName, repoSlug.toLowerCase()] as const : null
+  }))
+
+  return new Map(remoteSlugEntries.filter((entry): entry is readonly [string, string] => Boolean(entry)))
+}
+
+async function getLocalBranchNames(repoRoot: string) {
+  const result = await runGit(["for-each-ref", "--format=%(refname:short)", "refs/heads"], repoRoot)
+  if (result.exitCode !== 0) {
+    throw new Error(result.stderr.trim() || "Failed to list local branches")
+  }
+  return result.stdout
+    .split(/\r?\n/u)
+    .map((line) => line.trim())
+    .filter(Boolean)
+    .sort((left, right) => left.localeCompare(right))
+}
+
+async function getRemoteBranchNames(repoRoot: string) {
+  const result = await runGit(["for-each-ref", "--format=%(refname:short)", "refs/remotes"], repoRoot)
+  if (result.exitCode !== 0) {
+    throw new Error(result.stderr.trim() || "Failed to list remote branches")
+  }
+  return result.stdout
+    .split(/\r?\n/u)
+    .map((line) => line.trim())
+    .filter((line) => line.length > 0 && !line.endsWith("/HEAD"))
+    .sort((left, right) => left.localeCompare(right))
+}
+
+async function getBranchUpdatedAtMap(repoRoot: string, refPrefix: "refs/heads" | "refs/remotes") {
+  const result = await runGit(
+    ["for-each-ref", "--format=%(refname:short)\t%(committerdate:iso-strict)", refPrefix],
+    repoRoot
+  )
+  if (result.exitCode !== 0) {
+    throw new Error(result.stderr.trim() || "Failed to read branch update times")
+  }
+
+  const entries = new Map<string, string>()
+  for (const line of result.stdout.split(/\r?\n/u)) {
+    const trimmed = line.trim()
+    if (!trimmed) continue
+    const [name, updatedAt] = trimmed.split("\t")
+    if (!name || !updatedAt || (refPrefix === "refs/remotes" && name.endsWith("/HEAD"))) {
+      continue
+    }
+    entries.set(name, updatedAt)
+  }
+  return entries
+}
+
+async function resolveDefaultBranchName(repoRoot: string) {
+  const originHead = await runGit(["symbolic-ref", "--quiet", "--short", "refs/remotes/origin/HEAD"], repoRoot)
+  if (originHead.exitCode === 0) {
+    const ref = originHead.stdout.trim()
+    if (ref.startsWith("origin/")) {
+      return ref.slice("origin/".length)
+    }
+  }
+
+  const localBranches = await getLocalBranchNames(repoRoot)
+  if (localBranches.includes("main")) return "main"
+  if (localBranches.includes("master")) return "master"
+  return (await getBranchName(repoRoot)) ?? localBranches[0] ?? undefined
+}
+
+async function getRecentBranchNames(repoRoot: string) {
+  const result = await runGit(["reflog", "--format=%gs", "--max-count=100", "HEAD"], repoRoot)
+  if (result.exitCode !== 0) {
+    return []
+  }
+
+  const recent: string[] = []
+  const seen = new Set<string>()
+  for (const line of result.stdout.split(/\r?\n/u)) {
+    const match = /checkout: moving from .* to (?<branch>.+)$/u.exec(line.trim())
+    const branch = match?.groups?.branch?.trim()
+    if (!branch || branch === "HEAD" || branch.startsWith("refs/")) {
+      continue
+    }
+    if (seen.has(branch)) continue
+    seen.add(branch)
+    recent.push(branch)
+  }
+  return recent
+}
+
+export function extractGitHubRepoSlug(remoteUrl: string | null | undefined) {
+  if (!remoteUrl) return null
+
+  const sshMatch = /^git@github\.com:(?<owner>[^/]+)\/(?<repo>[^/]+?)(?:\.git)?$/u.exec(remoteUrl)
+  if (sshMatch?.groups?.owner && sshMatch.groups.repo) {
+    return `${sshMatch.groups.owner}/${sshMatch.groups.repo}`
+  }
+
+  const sshProtocolMatch = /^ssh:\/\/git@github\.com\/(?<owner>[^/]+)\/(?<repo>[^/]+?)(?:\.git)?$/u.exec(remoteUrl)
+  if (sshProtocolMatch?.groups?.owner && sshProtocolMatch.groups.repo) {
+    return `${sshProtocolMatch.groups.owner}/${sshProtocolMatch.groups.repo}`
+  }
+
+  const httpsMatch = /^https?:\/\/github\.com\/(?<owner>[^/]+)\/(?<repo>[^/]+?)(?:\.git)?$/u.exec(remoteUrl)
+  if (httpsMatch?.groups?.owner && httpsMatch.groups.repo) {
+    return `${httpsMatch.groups.owner}/${httpsMatch.groups.repo}`
+  }
+
+  return null
+}
+
+interface GitHubPullRequestResponseItem {
+  number: number
+  title: string
+  head?: {
+    ref?: string
+    label?: string
+    repo?: {
+      clone_url?: string
+      full_name?: string
+    } | null
+  }
+  base?: {
+    ref?: string
+  }
+}
+
+type FetchLike = (input: string, init?: RequestInit) => Promise<Response>
+
+type GitHubCliApiLike = (path: string) => Promise<GitHubPullRequestResponseItem[] | null>
+
+interface FetchGitHubPullRequestsDeps {
+  fetchImpl?: FetchLike
+  ghApiImpl?: GitHubCliApiLike
+}
+
+async function fetchGitHubPullRequestsViaGh(path: string): Promise<GitHubPullRequestResponseItem[] | null> {
+  const result = await runCommand([
+    "gh",
+    "api",
+    "-H",
+    "Accept: application/vnd.github+json",
+    path,
+  ])
+  if (result.exitCode !== 0) {
+    return null
+  }
+
+  const json = JSON.parse(result.stdout)
+  return Array.isArray(json) ? json as GitHubPullRequestResponseItem[] : []
+}
+
+export async function fetchGitHubPullRequests(
+  repoSlug: string,
+  deps: FetchLike | FetchGitHubPullRequestsDeps = fetch
+): Promise<GitHubPullRequestResponseItem[]> {
+  const fetchImpl = typeof deps === "function" ? deps : (deps.fetchImpl ?? fetch)
+  const ghApiImpl = typeof deps === "function" ? fetchGitHubPullRequestsViaGh : (deps.ghApiImpl ?? fetchGitHubPullRequestsViaGh)
+  const ghPath = `repos/${repoSlug}/pulls?state=open&per_page=50`
+
+  try {
+    const ghPulls = await ghApiImpl(ghPath)
+    if (ghPulls) {
+      return ghPulls
+    }
+  } catch {
+    // Fall back to an unauthenticated HTTP request when `gh` is unavailable.
+  }
+
+  const response = await fetchImpl(`https://api.github.com/repos/${repoSlug}/pulls?state=open&per_page=50`, {
+    headers: {
+      Accept: "application/vnd.github+json",
+    },
+  })
+
+  if (!response.ok) {
+    throw new Error(`GitHub pull requests request failed with status ${response.status}`)
+  }
+
+  const json = await response.json()
+  return Array.isArray(json) ? json as GitHubPullRequestResponseItem[] : []
+}
+
+function buildGitHubCommitUrl(remoteUrl: string | null, sha: string) {
+  const slug = extractGitHubRepoSlug(remoteUrl)
+  return slug ? `https://github.com/${slug}/commit/${sha}` : undefined
+}
+
+async function listCommitTags(repoRoot: string, sha: string) {
+  const result = await runGit(["tag", "--points-at", sha], repoRoot)
+  if (result.exitCode !== 0) {
+    return []
+  }
+  return result.stdout
+    .split(/\r?\n/u)
+    .map((line) => line.trim())
+    .filter(Boolean)
+    .sort((left, right) => left.localeCompare(right))
+}
+
+async function getBranchHistory(args: {
+  repoRoot: string
+  ref: string
+  limit: number
+}): Promise<ChatBranchHistorySnapshot> {
+  const logResult = await runGit(
+    [
+      "log",
+      "--max-count",
+      String(args.limit),
+      "--pretty=format:%H%x1f%s%x1f%b%x1f%an%x1f%aI%x1e",
+      args.ref,
+    ],
+    args.repoRoot
+  )
+
+  if (logResult.exitCode !== 0) {
+    throw new Error(logResult.stderr.trim() || "Failed to read git log")
+  }
+
+  const remoteUrl = await getOriginRemoteUrl(args.repoRoot)
+  const entries: ChatBranchHistoryEntry[] = []
+
+  for (const record of logResult.stdout.split("\u001e")) {
+    const trimmed = record.trim()
+    if (!trimmed) continue
+    const [sha, summary, description, authorName, authoredAt] = trimmed.split("\u001f")
+    if (!sha || !summary || !authoredAt) continue
+    entries.push({
+      sha,
+      summary,
+      description: (description ?? "").trim(),
+      authorName: authorName?.trim() || undefined,
+      authoredAt,
+      tags: await listCommitTags(args.repoRoot, sha),
+      githubUrl: buildGitHubCommitUrl(remoteUrl, sha),
+    })
+  }
+
+  return { entries }
+}
+
+function createBranchActionFailure(title: string, detail: string, fallback: string) {
+  return {
+    ok: false,
+    title,
+    message: summarizeGitFailure(detail, fallback),
+    detail,
+  } as const
+}
+
 function parseStatusPaths(output: string): DirtyPathEntry[] {
   const entries: DirtyPathEntry[] = []
   for (const rawLine of output.split(/\r?\n/u)) {
@@ -381,8 +748,19 @@ export class DiffStore {
     return {
       status: state.status,
       branchName: state.branchName,
+      defaultBranchName: state.defaultBranchName,
+      originRepoSlug: state.originRepoSlug,
       hasUpstream: state.hasUpstream,
+      aheadCount: state.aheadCount,
+      behindCount: state.behindCount,
+      lastFetchedAt: state.lastFetchedAt,
       files: [...state.files],
+      branchHistory: {
+        entries: state.branchHistory.entries.map((entry) => ({
+          ...entry,
+          tags: [...entry.tags],
+        })),
+      },
     }
   }
 
@@ -392,8 +770,14 @@ export class DiffStore {
       const nextState = {
         status: "no_repo",
         branchName: undefined,
+        defaultBranchName: undefined,
+        originRepoSlug: undefined,
         hasUpstream: undefined,
+        aheadCount: undefined,
+        behindCount: undefined,
+        lastFetchedAt: undefined,
         files: [],
+        branchHistory: { entries: [] },
       } satisfies StoredChatDiffState
       const changed = !snapshotsEqual(this.states.get(chatId), nextState)
       this.states.set(chatId, nextState)
@@ -402,18 +786,432 @@ export class DiffStore {
 
     const files = await computeCurrentFiles(repo.repoRoot, repo.baseCommit)
     const branchName = await getBranchName(repo.repoRoot)
+    const defaultBranchName = await resolveDefaultBranchName(repo.repoRoot)
+    const originRemoteUrl = await getOriginRemoteUrl(repo.repoRoot)
+    const originRepoSlug = extractGitHubRepoSlug(originRemoteUrl) ?? undefined
     const hasUpstream = await hasUpstreamBranch(repo.repoRoot)
+    const { aheadCount, behindCount } = hasUpstream
+      ? await getUpstreamStatusCounts(repo.repoRoot)
+      : { aheadCount: undefined, behindCount: undefined }
+    const lastFetchedAt = await getLastFetchedAt(repo.repoRoot)
+    const branchHistory = repo.baseCommit
+      ? await getBranchHistory({
+          repoRoot: repo.repoRoot,
+          ref: branchName ?? "HEAD",
+          limit: 20,
+        })
+      : { entries: [] }
     const nextState = {
       status: "ready",
       branchName,
+      defaultBranchName,
+      originRepoSlug,
       hasUpstream,
+      aheadCount,
+      behindCount,
+      lastFetchedAt,
       files,
+      branchHistory,
     } satisfies StoredChatDiffState
     const changed = !snapshotsEqual(this.states.get(chatId), nextState)
     this.states.set(chatId, nextState)
     return changed
   }
 
+  async listBranches(args: {
+    projectPath: string
+  }): Promise<ChatBranchListResult> {
+    const repo = await resolveRepo(args.projectPath)
+    if (!repo) {
+      throw new Error("Project is not in a git repository")
+    }
+
+    const [currentBranchName, defaultBranchName, localBranchNames, remoteBranchNames, recentBranchNames, localUpdatedAtMap, remoteUpdatedAtMap] = await Promise.all([
+      getBranchName(repo.repoRoot),
+      resolveDefaultBranchName(repo.repoRoot),
+      getLocalBranchNames(repo.repoRoot),
+      getRemoteBranchNames(repo.repoRoot),
+      getRecentBranchNames(repo.repoRoot),
+      getBranchUpdatedAtMap(repo.repoRoot, "refs/heads"),
+      getBranchUpdatedAtMap(repo.repoRoot, "refs/remotes"),
+    ])
+
+    const local = localBranchNames.map((name) => ({
+      id: `local:${name}`,
+      kind: "local",
+      name,
+      displayName: name,
+      updatedAt: localUpdatedAtMap.get(name),
+    } satisfies ChatBranchListEntry))
+
+    const remote = remoteBranchNames.map((remoteRef) => ({
+      id: `remote:${remoteRef}`,
+      kind: "remote",
+      name: remoteRef.replace(/^[^/]+\//u, ""),
+      displayName: remoteRef,
+      updatedAt: remoteUpdatedAtMap.get(remoteRef),
+      remoteRef,
+    } satisfies ChatBranchListEntry))
+
+    const localBranchSet = new Set(localBranchNames)
+    const remoteByName = new Map(remote.map((entry) => [entry.name, entry]))
+    const remoteEntriesByName = new Map<string, ChatBranchListEntry[]>()
+    for (const entry of remote) {
+      const entries = remoteEntriesByName.get(entry.name) ?? []
+      entries.push(entry)
+      remoteEntriesByName.set(entry.name, entries)
+    }
+    const recent: ChatBranchListEntry[] = recentBranchNames.flatMap<ChatBranchListEntry>((branchName) => {
+      if (localBranchSet.has(branchName)) {
+        return {
+          id: `recent:local:${branchName}`,
+          kind: "local",
+          name: branchName,
+          displayName: branchName,
+          updatedAt: localUpdatedAtMap.get(branchName),
+        } satisfies ChatBranchListEntry
+      }
+      const remoteEntry = remoteByName.get(branchName)
+      return remoteEntry
+        ? {
+            ...remoteEntry,
+            id: `recent:${remoteEntry.id}`,
+          } satisfies ChatBranchListEntry
+        : []
+    })
+
+    const [remoteUrl, githubRemoteSlugs] = await Promise.all([
+      getOriginRemoteUrl(repo.repoRoot),
+      getGitHubRemoteSlugs(repo.repoRoot),
+    ])
+    const repoSlug = extractGitHubRepoSlug(remoteUrl)
+    let pullRequests: ChatBranchListEntry[] = []
+    const pullRequestRemoteRefs = new Set<string>()
+    const pullRequestHeadNames = new Set<string>()
+    let pullRequestsStatus: ChatBranchListResult["pullRequestsStatus"] = "unavailable"
+    let pullRequestsError: string | undefined
+
+    if (repoSlug) {
+      try {
+        const prs = await fetchGitHubPullRequests(repoSlug)
+        pullRequests = prs.flatMap<ChatBranchListEntry>((pr) => {
+          const headRefName = pr.head?.ref?.trim()
+          if (!headRefName) return []
+          pullRequestHeadNames.add(headRefName)
+          const cloneUrl = pr.head?.repo?.clone_url?.trim() || undefined
+          const fullName = pr.head?.repo?.full_name?.trim() || undefined
+          const headRepoSlug = fullName?.toLowerCase()
+          const matchingRemoteEntries = (remoteEntriesByName.get(headRefName) ?? []).filter((entry) => {
+            const remoteName = entry.remoteRef?.split("/")[0]
+            if (!remoteName) return false
+            const remoteSlug = githubRemoteSlugs.get(remoteName)
+            if (!remoteSlug) return false
+            if (headRepoSlug) {
+              return remoteSlug === headRepoSlug
+            }
+            return remoteName === "origin"
+          })
+          for (const entry of matchingRemoteEntries) {
+            if (entry.remoteRef) {
+              pullRequestRemoteRefs.add(entry.remoteRef)
+            }
+          }
+          const preferredRemoteEntry = matchingRemoteEntries[0] ?? remoteByName.get(headRefName)
+          const remoteRef = preferredRemoteEntry?.remoteRef ?? `origin/${headRefName}`
+          return {
+            id: `pr:${pr.number}`,
+            kind: "pull_request",
+            name: headRefName,
+            displayName: `PR #${pr.number}`,
+            updatedAt: (remoteRef ? remoteUpdatedAtMap.get(remoteRef) : undefined) ?? localUpdatedAtMap.get(headRefName),
+            description: pr.title,
+            remoteRef,
+            prNumber: pr.number,
+            prTitle: pr.title,
+            headRefName,
+            headLabel: pr.head?.label?.trim() || fullName,
+            headRepoCloneUrl: cloneUrl,
+            isCrossRepository: Boolean(fullName && fullName.toLowerCase() !== repoSlug.toLowerCase()),
+          } satisfies ChatBranchListEntry
+        })
+        pullRequestsStatus = "available"
+      } catch (error) {
+        pullRequestsStatus = "error"
+        pullRequestsError = error instanceof Error ? error.message : String(error)
+      }
+    }
+
+    const visibleRemote = remote.filter((entry) => {
+      if (pullRequestHeadNames.has(entry.name)) {
+        return false
+      }
+      return !entry.remoteRef || !pullRequestRemoteRefs.has(entry.remoteRef)
+    })
+    const visibleRemoteByName = new Map(visibleRemote.map((entry) => [entry.name, entry]))
+    const visibleRecent = recent.filter((entry) => entry.kind !== "remote" || !entry.remoteRef || visibleRemoteByName.has(entry.name))
+
+    return {
+      currentBranchName,
+      defaultBranchName,
+      recent: visibleRecent,
+      local,
+      remote: visibleRemote,
+      pullRequests,
+      pullRequestsStatus,
+      pullRequestsError,
+    }
+  }
+
+  async checkoutBranch(args: {
+    chatId: string
+    projectPath: 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
+  }): Promise<ChatCheckoutBranchResult> {
+    const repo = await resolveRepo(args.projectPath)
+    if (!repo) {
+      throw new Error("Project is not in a git repository")
+    }
+
+    const currentDirtyPaths = await listDirtyPaths(repo.repoRoot)
+    if (currentDirtyPaths.length > 0 && !args.bringChanges) {
+      return {
+        ok: false,
+        cancelled: true,
+        title: "Branch switch cancelled",
+        message: "Your current changes were kept on the current branch.",
+        snapshotChanged: false,
+      }
+    }
+
+    let switchResult: Awaited<ReturnType<typeof runGit>>
+    if (args.branch.kind === "local") {
+      switchResult = await runGit(["switch", args.branch.name], repo.repoRoot)
+    } else if (args.branch.kind === "remote") {
+      const localBranchNames = await getLocalBranchNames(repo.repoRoot)
+      if (localBranchNames.includes(args.branch.name)) {
+        switchResult = await runGit(["switch", args.branch.name], repo.repoRoot)
+      } else {
+        switchResult = await runGit(["switch", "--track", "--no-guess", args.branch.remoteRef], repo.repoRoot)
+      }
+    } else {
+      const localBranchNames = await getLocalBranchNames(repo.repoRoot)
+      let localBranchName = args.branch.name
+
+      if (localBranchNames.includes(localBranchName) && args.branch.isCrossRepository) {
+        localBranchName = `${args.branch.name}-pr-${args.branch.prNumber}`
+      }
+
+      if (localBranchNames.includes(localBranchName)) {
+        switchResult = await runGit(["switch", localBranchName], repo.repoRoot)
+      } else if (args.branch.isCrossRepository && args.branch.headRepoCloneUrl) {
+        const fetchResult = await runGit(
+          [
+            "fetch",
+            "--no-tags",
+            args.branch.headRepoCloneUrl,
+            `refs/heads/${args.branch.headRefName}:refs/heads/${localBranchName}`,
+          ],
+          repo.repoRoot
+        )
+        if (fetchResult.exitCode !== 0) {
+          return createBranchActionFailure("Checkout failed", formatGitFailure(fetchResult), "Git could not fetch the pull request branch.")
+        }
+        switchResult = await runGit(["switch", localBranchName], repo.repoRoot)
+      } else {
+        const remoteRef = args.branch.remoteRef ?? `origin/${args.branch.headRefName}`
+        const remoteBranchNames = await getRemoteBranchNames(repo.repoRoot)
+        if (!remoteBranchNames.includes(remoteRef)) {
+          const fetchResult = await runGit(
+            ["fetch", "--no-tags", "origin", `refs/heads/${args.branch.headRefName}:refs/remotes/${remoteRef}`],
+            repo.repoRoot
+          )
+          if (fetchResult.exitCode !== 0) {
+            return createBranchActionFailure("Checkout failed", formatGitFailure(fetchResult), "Git could not fetch the pull request branch.")
+          }
+        }
+        switchResult = await runGit(["switch", "--track", "--no-guess", remoteRef], repo.repoRoot)
+      }
+    }
+
+    if (switchResult.exitCode !== 0) {
+      return createBranchActionFailure("Checkout failed", formatGitFailure(switchResult), "Git could not switch branches.")
+    }
+
+    const snapshotChanged = await this.refreshSnapshot(args.chatId, args.projectPath)
+    return {
+      ok: true,
+      branchName: await getBranchName(repo.repoRoot),
+      snapshotChanged,
+    }
+  }
+
+  async createBranch(args: {
+    chatId: string
+    projectPath: string
+    name: string
+    baseBranchName?: string
+  }): Promise<ChatCreateBranchResult> {
+    const repo = await resolveRepo(args.projectPath)
+    if (!repo) {
+      throw new Error("Project is not in a git repository")
+    }
+
+    const branchName = args.name.trim()
+    if (!branchName) {
+      throw new Error("Branch name is required")
+    }
+
+    const refValidation = await runGit(["check-ref-format", "--branch", branchName], repo.repoRoot)
+    if (refValidation.exitCode !== 0) {
+      return createBranchActionFailure("Create branch failed", formatGitFailure(refValidation), "Branch name is not valid.")
+    }
+
+    const localBranchNames = await getLocalBranchNames(repo.repoRoot)
+    if (localBranchNames.includes(branchName)) {
+      return {
+        ok: false,
+        title: "Create branch failed",
+        message: `A local branch named "${branchName}" already exists.`,
+        snapshotChanged: false,
+      }
+    }
+
+    const baseBranchName = args.baseBranchName?.trim() || await resolveDefaultBranchName(repo.repoRoot) || await getBranchName(repo.repoRoot)
+    if (!baseBranchName) {
+      throw new Error("Could not determine a base branch")
+    }
+
+    const switchResult = await runGit(["switch", "-c", branchName, baseBranchName], repo.repoRoot)
+    if (switchResult.exitCode !== 0) {
+      return createBranchActionFailure("Create branch failed", formatGitFailure(switchResult), "Git could not create the branch.")
+    }
+
+    const snapshotChanged = await this.refreshSnapshot(args.chatId, args.projectPath)
+    return {
+      ok: true,
+      branchName,
+      snapshotChanged,
+    }
+  }
+
+  async syncBranch(args: {
+    chatId: string
+    projectPath: string
+    action: "fetch" | "pull" | "publish"
+  }): Promise<ChatSyncResult> {
+    const repo = await resolveRepo(args.projectPath)
+    if (!repo) {
+      throw new Error("Project is not in a git repository")
+    }
+
+    const hasUpstream = await hasUpstreamBranch(repo.repoRoot)
+    if (args.action === "publish") {
+      const publishResult = await runGit(["push", "-u", "origin", "HEAD"], repo.repoRoot)
+      if (publishResult.exitCode !== 0) {
+        const detail = formatGitFailure(publishResult)
+        const normalized = detail.toLowerCase()
+        let title = "Publish branch failed"
+        let message = summarizeGitFailure(detail, "Git could not publish this branch.")
+
+        if (normalized.includes("could not read from remote repository") || normalized.includes("authentication failed") || normalized.includes("permission denied")) {
+          title = "Remote authentication failed"
+          message = "Git could not authenticate with the remote repository."
+        }
+
+        return {
+          ok: false,
+          action: args.action,
+          title,
+          message,
+          detail,
+          snapshotChanged: false,
+        }
+      }
+
+      const snapshotChanged = await this.refreshSnapshot(args.chatId, args.projectPath)
+      const branchName = await getBranchName(repo.repoRoot)
+      const nextHasUpstream = await hasUpstreamBranch(repo.repoRoot)
+      const { aheadCount, behindCount } = nextHasUpstream
+        ? await getUpstreamStatusCounts(repo.repoRoot)
+        : { aheadCount: undefined, behindCount: undefined }
+
+      return {
+        ok: true,
+        action: args.action,
+        branchName,
+        aheadCount,
+        behindCount,
+        snapshotChanged,
+      }
+    }
+
+    if (args.action === "pull" && !hasUpstream) {
+      return {
+        ok: false,
+        action: args.action,
+        title: "Pull failed",
+        message: "This branch does not have an upstream remote branch configured yet.",
+        snapshotChanged: false,
+      }
+    }
+
+    const syncResult = args.action === "pull"
+      ? await runGit(["pull", "--ff-only"], repo.repoRoot)
+      : await runGit(["fetch", "--all", "--prune"], repo.repoRoot)
+
+    if (syncResult.exitCode !== 0) {
+      const detail = formatGitFailure(syncResult)
+      const normalized = detail.toLowerCase()
+      let title = args.action === "pull" ? "Pull failed" : "Fetch failed"
+      let message = summarizeGitFailure(detail, args.action === "pull" ? "Git could not pull the latest changes." : "Git could not fetch the latest changes.")
+
+      if (args.action === "pull" && normalized.includes("not possible to fast-forward")) {
+        title = "Pull requires merge or rebase"
+        message = "Your branch cannot be fast-forwarded. Rebase or merge manually, then try again."
+      } else if (normalized.includes("could not read from remote repository") || normalized.includes("authentication failed") || normalized.includes("permission denied")) {
+        title = "Remote authentication failed"
+        message = "Git could not authenticate with the remote repository."
+      }
+
+      return {
+        ok: false,
+        action: args.action,
+        title,
+        message,
+        detail,
+        snapshotChanged: false,
+      }
+    }
+
+    const snapshotChanged = await this.refreshSnapshot(args.chatId, args.projectPath)
+    const branchName = await getBranchName(repo.repoRoot)
+    const nextHasUpstream = await hasUpstreamBranch(repo.repoRoot)
+    const { aheadCount, behindCount } = nextHasUpstream
+      ? await getUpstreamStatusCounts(repo.repoRoot)
+      : { aheadCount: undefined, behindCount: undefined }
+
+    return {
+      ok: true,
+      action: args.action,
+      branchName,
+      aheadCount,
+      behindCount,
+      snapshotChanged,
+    }
+  }
+
   async generateCommitMessage(args: {
     projectPath: string
     paths: string[]
@@ -509,12 +1307,14 @@ export class DiffStore {
       return createPushFailure(args.mode, formatGitFailure(pushResult), snapshotChanged)
     }
 
+    const postPushSnapshotChanged = await this.refreshSnapshot(args.chatId, args.projectPath)
+
     return {
       ok: true,
       mode: args.mode,
       branchName,
       pushed: true,
-      snapshotChanged,
+      snapshotChanged: snapshotChanged || postPushSnapshotChanged,
     } satisfies DiffCommitResult
   }