opencode/packages/opencode/test/project/worktree-remove.test.ts

import { $ } from "bun"
import { describe, expect } from "bun:test"
import * as fs from "fs/promises"
import path from "path"
import { Effect, Layer } from "effect"
import { CrossSpawnSpawner } from "@opencode-ai/core/cross-spawn-spawner"
import { Worktree } from "../../src/worktree"
import { provideTmpdirInstance } from "../fixture/fixture"
import { testEffect } from "../lib/effect"

const it = testEffect(Layer.mergeAll(Worktree.defaultLayer, CrossSpawnSpawner.defaultLayer))
const wintest = process.platform === "win32" ? it.live : it.live.skip

describe("Worktree.remove", () => {
  it.live("continues when git remove exits non-zero after detaching", () =>
    provideTmpdirInstance(
      (root) =>
        Effect.gen(function* () {
          const svc = yield* Worktree.Service
          const name = `remove-regression-${Date.now().toString(36)}`
          const branch = `opencode/${name}`
          const dir = path.join(root, "..", name)

          yield* Effect.promise(() => $`git worktree add --no-checkout -b ${branch} ${dir}`.cwd(root).quiet())
          yield* Effect.promise(() => $`git reset --hard`.cwd(dir).quiet())

          const real = (yield* Effect.promise(() => $`which git`.quiet().text())).trim()
          expect(real).toBeTruthy()

          const bin = path.join(root, "bin")
          const shim = path.join(bin, "git")
          yield* Effect.promise(() => fs.mkdir(bin, { recursive: true }))
          yield* Effect.promise(() =>
            Bun.write(
              shim,
              [
                "#!/bin/bash",
                `REAL_GIT=${JSON.stringify(real)}`,
                'if [ "$1" = "worktree" ] && [ "$2" = "remove" ]; then',
                '  "$REAL_GIT" "$@" >/dev/null 2>&1',
                '  echo "fatal: failed to remove worktree: Directory not empty" >&2',
                "  exit 1",
                "fi",
                'exec "$REAL_GIT" "$@"',
              ].join("\n"),
            ),
          )
          yield* Effect.promise(() => fs.chmod(shim, 0o755))

          const prev = yield* Effect.acquireRelease(
            Effect.sync(() => {
              const prev = process.env.PATH ?? ""
              process.env.PATH = `${bin}${path.delimiter}${prev}`
              return prev
            }),
            (prev) =>
              Effect.sync(() => {
                process.env.PATH = prev
              }),
          )
          void prev

          const ok = yield* svc.remove({ directory: dir })

          expect(ok).toBe(true)
          expect(
            yield* Effect.promise(() =>
              fs
                .stat(dir)
                .then(() => true)
                .catch(() => false),
            ),
          ).toBe(false)

          const list = yield* Effect.promise(() => $`git worktree list --porcelain`.cwd(root).quiet().text())
          expect(list).not.toContain(`worktree ${dir}`)

          const ref = yield* Effect.promise(() =>
            $`git show-ref --verify --quiet refs/heads/${branch}`.cwd(root).quiet().nothrow(),
          )
          expect(ref.exitCode).not.toBe(0)
        }),
      { git: true },
    ),
  )

  wintest("stops fsmonitor before removing a worktree", () =>
    provideTmpdirInstance(
      (root) =>
        Effect.gen(function* () {
          const svc = yield* Worktree.Service
          const name = `remove-fsmonitor-${Date.now().toString(36)}`
          const branch = `opencode/${name}`
          const dir = path.join(root, "..", name)

          yield* Effect.promise(() => $`git worktree add --no-checkout -b ${branch} ${dir}`.cwd(root).quiet())
          yield* Effect.promise(() => $`git reset --hard`.cwd(dir).quiet())
          yield* Effect.promise(() => $`git config core.fsmonitor true`.cwd(dir).quiet())
          yield* Effect.promise(() => $`git fsmonitor--daemon stop`.cwd(dir).quiet().nothrow())
          yield* Effect.promise(() => Bun.write(path.join(dir, "tracked.txt"), "next\n"))
          yield* Effect.promise(() => $`git diff`.cwd(dir).quiet())

          const before = yield* Effect.promise(() => $`git fsmonitor--daemon status`.cwd(dir).quiet().nothrow())
          expect(before.exitCode).toBe(0)

          const ok = yield* svc.remove({ directory: dir })

          expect(ok).toBe(true)
          expect(
            yield* Effect.promise(() =>
              fs
                .stat(dir)
                .then(() => true)
                .catch(() => false),
            ),
          ).toBe(false)

          const ref = yield* Effect.promise(() =>
            $`git show-ref --verify --quiet refs/heads/${branch}`.cwd(root).quiet().nothrow(),
          )
          expect(ref.exitCode).not.toBe(0)
        }),
      { git: true },
    ),
  )
})