orca/src/relay/git-handler-branch-cleanup.ts

import {
  branchHasNoUnmergedChangesWithLazyTargetRefresh,
  getBranchCleanupTargetRefs
} from '../shared/git-branch-cleanup'
import type { GitCapabilityCache } from '../shared/git-capability-cache'
import type { GitExec } from './git-handler-ops'
import { parseWorktreeList } from './git-handler-utils'

export async function deleteAlreadyMergedRelayBranchAfterSafeDeleteFailure(
  git: GitExec,
  repoPath: string,
  branchName: string,
  branchHead: string,
  capabilities: GitCapabilityCache
): Promise<boolean> {
  const runGit = (args: string[], options?: { stdin?: string }) =>
    options ? git(args, repoPath, options) : git(args, repoPath)
  const targetRefs = await getBranchCleanupTargetRefs(runGit, branchName)
  // Why: SSH worktrees hit the same squash-merge shape as local worktrees.
  // Git's no-op merge proof lets us clean up only branches whose changes
  // already exist on the saved base ref.
  if (
    !(await branchHasNoUnmergedChangesWithLazyTargetRefresh(
      runGit,
      branchName,
      targetRefs,
      capabilities
    ))
  ) {
    return false
  }
  await deleteRelayBranchAtExpectedHead(git, repoPath, branchName, branchHead)
  return true
}

export async function forceDeletePreservedRelayBranch(
  git: GitExec,
  repoPath: string,
  branchName: string,
  expectedHead: string
): Promise<void> {
  if (!branchName || branchName.includes('\0') || branchName.startsWith('-')) {
    throw new Error('Invalid branch name for preserved branch delete.')
  }
  if (!expectedHead) {
    throw new Error('Expected branch head is required for preserved branch delete.')
  }
  await deleteRelayBranchAtExpectedHead(git, repoPath, branchName, expectedHead, () => {
    return new Error(
      `Local branch "${branchName}" changed after the workspace was deleted. Review it before deleting it.`
    )
  })
}

async function deleteRelayBranchAtExpectedHead(
  git: GitExec,
  repoPath: string,
  branchName: string,
  expectedHead: string,
  mapUpdateRefError?: (error: unknown) => Error
): Promise<void> {
  if (await isRelayBranchCheckedOut(git, repoPath, branchName)) {
    throw new Error(`Local branch "${branchName}" is checked out in another worktree.`)
  }
  try {
    await git(['update-ref', '-d', `refs/heads/${branchName}`, expectedHead], repoPath)
  } catch (error) {
    // Why: only stale ref writes get the force-delete message; checkout guards
    // and removeWorktree cleanup still rely on their distinct/raw failures.
    throw mapUpdateRefError?.(error) ?? error
  }
  if (await isRelayBranchCheckedOut(git, repoPath, branchName)) {
    try {
      await git(['update-ref', `refs/heads/${branchName}`, expectedHead, ''], repoPath)
    } catch (restoreError) {
      console.warn(
        `relay removeWorktree: failed to restore local branch "${branchName}" after concurrent checkout`,
        restoreError
      )
    }
    throw new Error(`Local branch "${branchName}" is checked out in another worktree.`)
  }
  try {
    await git(['config', '--remove-section', `branch.${branchName}`], repoPath)
  } catch {
    // Best-effort parity with `git branch -D`; stale config is harmless after
    // the expected ref was deleted.
  }
}

async function isRelayBranchCheckedOut(
  git: GitExec,
  repoPath: string,
  branchName: string
): Promise<boolean> {
  const { stdout } = await git(['worktree', 'list', '--porcelain'], repoPath)
  return parseWorktreeList(stdout).some(
    (worktree) =>
      typeof worktree.branch === 'string' &&
      worktree.branch.replace(/^refs\/heads\//, '') === branchName
  )
}