oh-my-opencode/src/hooks/_auc-mocks-workspace/workspace-resolution.test.ts

import type { PluginInput } from "@opencode-ai/plugin"
import { afterEach, beforeEach, describe, expect, it, mock, spyOn } from "bun:test"
import { existsSync, mkdirSync, rmSync, writeFileSync } from "node:fs"
import { join } from "node:path"
import * as shared from "../../shared"

type PluginEntry = {
  entry: string
  isPinned: boolean
  pinnedVersion: string | null
  configPath: string
}

type ToastMessageGetter = (isUpdate: boolean, version?: string) => string

function createPluginEntry(overrides?: Partial<PluginEntry>): PluginEntry {
  return {
    entry: "oh-my-opencode@3.4.0",
    isPinned: false,
    pinnedVersion: null,
    configPath: "/test/opencode.json",
    ...overrides,
  }
}

const TEST_DIR = join(import.meta.dir, "__test-workspace-resolution__")
const TEST_CACHE_DIR = join(TEST_DIR, "cache")
const TEST_CONFIG_DIR = join(TEST_DIR, "config")

const mockFindPluginEntry = mock((_directory: string): PluginEntry | null => createPluginEntry())
const mockGetCachedVersion = mock((): string | null => "3.4.0")
const mockGetLatestVersion = mock(async (): Promise<string | null> => "3.5.0")
const mockExtractChannel = mock(() => "latest")
const mockInvalidatePackage = mock(() => {})
const mockShowUpdateAvailableToast = mock(
  async (_ctx: PluginInput, _latestVersion: string, _getToastMessage: ToastMessageGetter): Promise<void> => {}
)
const mockShowAutoUpdatedToast = mock(
  async (_ctx: PluginInput, _fromVersion: string, _toVersion: string): Promise<void> => {}
)
const mockSyncCachePackageJsonToIntent = mock(() => ({ synced: true, error: null }))

const mockRunBunInstallWithDetails = mock(
  async (opts?: { outputMode?: string; workspaceDir?: string }) => {
    return { success: true }
  }
)

let importCounter = 0
let getOpenCodeCacheDirSpy: { mockRestore: () => void } | undefined
let getOpenCodeConfigPathsSpy: { mockRestore: () => void } | undefined

async function importFreshBackgroundUpdateCheck(): Promise<typeof import("../auto-update-checker/hook/background-update-check")> {
  mock.module("../auto-update-checker/checker", () => ({
    findPluginEntry: mockFindPluginEntry,
    getCachedVersion: mockGetCachedVersion,
    getLatestVersion: mockGetLatestVersion,
    revertPinnedVersion: mock(() => false),
    syncCachePackageJsonToIntent: mockSyncCachePackageJsonToIntent,
  }))
  mock.module("../auto-update-checker/version-channel", () => ({ extractChannel: mockExtractChannel }))
  mock.module("../auto-update-checker/cache", () => ({ invalidatePackage: mockInvalidatePackage }))
  mock.module("../../cli/config-manager", () => ({
    runBunInstallWithDetails: mockRunBunInstallWithDetails,
  }))
  mock.module("../auto-update-checker/hook/update-toasts", () => ({
    showUpdateAvailableToast: mockShowUpdateAvailableToast,
    showAutoUpdatedToast: mockShowAutoUpdatedToast,
  }))
  mock.module("../../shared/logger", () => ({ log: () => {} }))
  getOpenCodeCacheDirSpy = spyOn(shared, "getOpenCodeCacheDir").mockReturnValue(TEST_CACHE_DIR)
  getOpenCodeConfigPathsSpy = spyOn(shared, "getOpenCodeConfigPaths").mockReturnValue({
    configDir: TEST_CONFIG_DIR,
    configJson: join(TEST_CONFIG_DIR, "opencode.json"),
    configJsonc: join(TEST_CONFIG_DIR, "opencode.jsonc"),
    packageJson: join(TEST_CONFIG_DIR, "package.json"),
    omoConfig: join(TEST_CONFIG_DIR, "oh-my-opencode.json"),
  })

  mock.module("../auto-update-checker/constants", () => ({
    PACKAGE_NAME: "oh-my-opencode",
    CACHE_DIR: TEST_CACHE_DIR,
    USER_CONFIG_DIR: TEST_CONFIG_DIR,
    NPM_REGISTRY_URL: "https://registry.npmjs.org/-/package/oh-my-opencode/dist-tags",
    NPM_FETCH_TIMEOUT: 5000,
    VERSION_FILE: join(TEST_CACHE_DIR, "version"),
    USER_OPENCODE_CONFIG: join(TEST_CONFIG_DIR, "opencode.json"),
    USER_OPENCODE_CONFIG_JSONC: join(TEST_CONFIG_DIR, "opencode.jsonc"),
    INSTALLED_PACKAGE_JSON: join(TEST_CACHE_DIR, "node_modules", "oh-my-opencode", "package.json"),
    getWindowsAppdataDir: () => null,
  }))

  mock.module("../../shared/data-path", () => ({
    getDataDir: () => join(TEST_DIR, "data"),
    getOpenCodeStorageDir: () => join(TEST_DIR, "data", "opencode", "storage"),
    getCacheDir: () => TEST_DIR,
    getOmoOpenCodeCacheDir: () => join(TEST_DIR, "oh-my-opencode"),
    getOpenCodeCacheDir: () => TEST_CACHE_DIR,
  }))
  mock.module("../../shared/opencode-config-dir", () => ({
    getOpenCodeConfigDir: () => TEST_CONFIG_DIR,
    getOpenCodeConfigPaths: () => ({
      configDir: TEST_CONFIG_DIR,
      configJson: join(TEST_CONFIG_DIR, "opencode.json"),
      configJsonc: join(TEST_CONFIG_DIR, "opencode.jsonc"),
      packageJson: join(TEST_CONFIG_DIR, "package.json"),
      omoConfig: join(TEST_CONFIG_DIR, "oh-my-opencode.json"),
    }),
  }))

  const backgroundUpdateCheckModule = await import(`../auto-update-checker/hook/background-update-check?test=${importCounter++}`)
  return backgroundUpdateCheckModule
}

describe("workspace resolution", () => {
  const mockCtx = { directory: "/test" } as PluginInput
  const getToastMessage: ToastMessageGetter = (isUpdate, version) =>
    isUpdate ? `Update to ${version}` : "Up to date"

  beforeEach(() => {
    // Setup test directories
    if (existsSync(TEST_DIR)) {
      rmSync(TEST_DIR, { recursive: true, force: true })
    }
    mkdirSync(TEST_DIR, { recursive: true })

    mockFindPluginEntry.mockReset()
    mockGetCachedVersion.mockReset()
    mockGetLatestVersion.mockReset()
    mockExtractChannel.mockReset()
    mockInvalidatePackage.mockReset()
    mockRunBunInstallWithDetails.mockReset()
    mockShowUpdateAvailableToast.mockReset()
    mockShowAutoUpdatedToast.mockReset()

    mockFindPluginEntry.mockReturnValue(createPluginEntry())
    mockGetCachedVersion.mockReturnValue("3.4.0")
    mockGetLatestVersion.mockResolvedValue("3.5.0")
    mockExtractChannel.mockReturnValue("latest")
    // Note: Don't use mockResolvedValue here - it overrides the function that captures args
    mockSyncCachePackageJsonToIntent.mockReturnValue({ synced: true, error: null })
  })

  afterEach(() => {
    getOpenCodeCacheDirSpy?.mockRestore()
    getOpenCodeConfigPathsSpy?.mockRestore()
    getOpenCodeCacheDirSpy = undefined
    getOpenCodeConfigPathsSpy = undefined
    mock.restore()
    if (existsSync(TEST_DIR)) {
      rmSync(TEST_DIR, { recursive: true, force: true })
    }
  })

  describe("#given config-dir install exists but cache-dir does not", () => {
    it("installs to config-dir, not cache-dir", async () => {
      //#given - config-dir has installation, cache-dir does not
      const { runBackgroundUpdateCheck } = await importFreshBackgroundUpdateCheck()
      mkdirSync(join(TEST_CONFIG_DIR, "node_modules", "oh-my-opencode"), { recursive: true })
      writeFileSync(
        join(TEST_CONFIG_DIR, "package.json"),
        JSON.stringify({ dependencies: { "oh-my-opencode": "3.4.0" } }, null, 2)
      )
      writeFileSync(
        join(TEST_CONFIG_DIR, "node_modules", "oh-my-opencode", "package.json"),
        JSON.stringify({ name: "oh-my-opencode", version: "3.4.0" }, null, 2)
      )

      // cache-dir should NOT exist
      expect(existsSync(TEST_CACHE_DIR)).toBe(false)

      //#when
      await runBackgroundUpdateCheck(mockCtx, true, getToastMessage)

      //#then - install should be called with config-dir
      const mockCalls = mockRunBunInstallWithDetails.mock.calls
      expect(mockCalls[0][0]?.workspaceDir).toBe(TEST_CONFIG_DIR)
    })
  })

  describe("#given both config-dir and cache-dir exist", () => {
    it("prefers config-dir over cache-dir", async () => {
      //#given - both directories have installations
      const { runBackgroundUpdateCheck } = await importFreshBackgroundUpdateCheck()
      mkdirSync(join(TEST_CONFIG_DIR, "node_modules", "oh-my-opencode"), { recursive: true })
      writeFileSync(
        join(TEST_CONFIG_DIR, "package.json"),
        JSON.stringify({ dependencies: { "oh-my-opencode": "3.4.0" } }, null, 2)
      )
      writeFileSync(
        join(TEST_CONFIG_DIR, "node_modules", "oh-my-opencode", "package.json"),
        JSON.stringify({ name: "oh-my-opencode", version: "3.4.0" }, null, 2)
      )

      mkdirSync(join(TEST_CACHE_DIR, "node_modules", "oh-my-opencode"), { recursive: true })
      writeFileSync(
        join(TEST_CACHE_DIR, "package.json"),
        JSON.stringify({ dependencies: { "oh-my-opencode": "3.4.0" } }, null, 2)
      )
      writeFileSync(
        join(TEST_CACHE_DIR, "node_modules", "oh-my-opencode", "package.json"),
        JSON.stringify({ name: "oh-my-opencode", version: "3.4.0" }, null, 2)
      )

      //#when
      await runBackgroundUpdateCheck(mockCtx, true, getToastMessage)

      //#then - install should prefer config-dir
      const mockCalls2 = mockRunBunInstallWithDetails.mock.calls
      expect(mockCalls2[0][0]?.workspaceDir).toBe(TEST_CONFIG_DIR)
    })
  })

  describe("#given only cache-dir install exists", () => {
    it("falls back to cache-dir", async () => {
      //#given - only cache-dir has installation
      const { runBackgroundUpdateCheck } = await importFreshBackgroundUpdateCheck()
      mkdirSync(join(TEST_CACHE_DIR, "node_modules", "oh-my-opencode"), { recursive: true })
      writeFileSync(
        join(TEST_CACHE_DIR, "package.json"),
        JSON.stringify({ dependencies: { "oh-my-opencode": "3.4.0" } }, null, 2)
      )
      writeFileSync(
        join(TEST_CACHE_DIR, "node_modules", "oh-my-opencode", "package.json"),
        JSON.stringify({ name: "oh-my-opencode", version: "3.4.0" }, null, 2)
      )

      // config-dir should NOT exist
      expect(existsSync(TEST_CONFIG_DIR)).toBe(false)

      //#when
      await runBackgroundUpdateCheck(mockCtx, true, getToastMessage)

      //#then - install should fall back to cache-dir
      const mockCalls3 = mockRunBunInstallWithDetails.mock.calls
      expect(mockCalls3[0][0]?.workspaceDir).toBe(TEST_CACHE_DIR)
    })
  })
})