orca/config/scripts/reclaim-worktree-space.mjs

#!/usr/bin/env node
import { mkdir } from 'node:fs/promises'
import { resolve, join } from 'node:path'
import { lock } from 'proper-lockfile'
import { runProcess } from '@orca/process-host'

export async function reclaimWorktreeSpace({
  repoRoot = process.cwd(),
  apply = false,
  automatic = false,
  execute = runProcess,
  acquireLock = lock,
  commonDirectory,
  testsStopped = false,
  generatedMinAgeDays = 30,
  excludeWorktree,
  releaseCacheMinAgeDays = 30
} = {}) {
  if (!Number.isFinite(releaseCacheMinAgeDays) || releaseCacheMinAgeDays < 0) {
    throw new Error('--release-cache-min-age-days requires a non-negative number')
  }
  if (!Number.isFinite(generatedMinAgeDays) || generatedMinAgeDays < 0) {
    throw new Error('--generated-min-age-days requires a non-negative number')
  }
  if (!commonDirectory) {
    const result = await runProcess({
      program: 'git',
      args: ['-C', repoRoot, 'rev-parse', '--git-common-dir']
    })
    if (result.code !== 0 || result.timedOut || !result.stdout.trim()) {
      throw new Error('Run this command inside a local Git repository, or supply --repo')
    }
    commonDirectory = resolve(repoRoot, result.stdout.trim())
  }
  const cacheRoot = join(commonDirectory, 'orca-cache')
  const lockPath = join(cacheRoot, 'storage-cleanup')
  await mkdir(cacheRoot, { recursive: true })
  let release
  try {
    release = await acquireLock(lockPath, { realpath: false, stale: 600_000, retries: 0 })
  } catch (error) {
    if (error.code === 'ELOCKED') {
      throw new Error('Another worktree cleanup is already running')
    }
    throw error
  }
  let cleanupFailed = false
  let cleanupError
  try {
    const scripts = [
      'reclaim-electron-dists.mjs',
      'reclaim-dev-electron-bundles.mjs',
      'reclaim-vite-caches.mjs',
      'reclaim-vitest-module-caches.mjs',
      'reclaim-worktree-generated-files.mjs',
      'reclaim-dev-worktree-profiles.mjs'
    ]
    if (!automatic && testsStopped) {
      scripts.unshift('reclaim-release-test-caches.mjs')
    }
    for (const script of scripts) {
      const args = [resolve(import.meta.dirname, script), '--repo', repoRoot]
      if (apply) {
        args.push('--apply')
      }
      if (excludeWorktree && script !== 'reclaim-release-test-caches.mjs') {
        args.push('--exclude-worktree', excludeWorktree)
      }
      if (script === 'reclaim-release-test-caches.mjs') {
        args.unshift('--import', import.meta.resolve('tsx'))
        args.push('--min-age-days', String(releaseCacheMinAgeDays))
        if (testsStopped) {
          args.push('--tests-stopped')
        }
      }
      if (script === 'reclaim-worktree-generated-files.mjs') {
        args.push('--generated-min-age-days', String(generatedMinAgeDays))
      }
      const result = await execute({
        program: process.execPath,
        args,
        env: process.env,
        stdio: 'inherit',
        timeoutMs: 10 * 60 * 1000,
        terminationBarrier: true
      })
      if (result.code !== 0 || result.timedOut) {
        throw new Error(`${script} failed; rerun cleanup after resolving the error`)
      }
    }
  } catch (error) {
    cleanupFailed = true
    cleanupError = error
  }
  try {
    await release()
  } catch (error) {
    if (!cleanupFailed) {
      throw error
    }
    console.error(
      `Failed to release cleanup lock: ${error instanceof Error ? error.message : String(error)}`
    )
  }
  if (cleanupFailed) {
    throw cleanupError
  }
  return { completed: true }
}

if (process.argv[1] && resolve(process.argv[1]) === resolve(import.meta.filename)) {
  import('./reclaim-worktree-command.mjs')
    .then(({ runReclaimCommand }) => runReclaimCommand(process.argv.slice(2)))
    .catch((error) => {
      console.error(error.message)
      process.exitCode = 1
    })
}