opencode-onboard 0.4.3 → 0.4.4

package/src/steps/optimization/optimization.test.js

@@ -1,93 +1,101 @@
-import { beforeEach, describe, expect, it, vi } from 'vitest'
-
-vi.mock('@inquirer/prompts', () => ({
-  checkbox: vi.fn(),
-  confirm: vi.fn(),
-}))
-
-vi.mock('../../utils/exec.js', () => ({
-  code: vi.fn(),
-  commandExists: vi.fn(),
-  header: vi.fn(),
-  info: vi.fn(),
-  loading: vi.fn(),
-  success: vi.fn(),
-  warn: vi.fn(),
-}))
-
-vi.mock('./quota.js', () => ({ installQuota: vi.fn() }))
-vi.mock('./caveman.js', () => ({ installCaveman: vi.fn() }))
-vi.mock('./caveman-guidance.js', () => ({ enableCavemanGuidance: vi.fn() }))
-vi.mock('./global.js', () => ({ configureObGlobal: vi.fn() }))
-
-vi.mock('fs-extra', () => ({
-  default: {
-    readJson: vi.fn().mockResolvedValue({
-      info: 'Token optimization info',
-      message: 'Select tools',
-      timeoutMs: 5000,
-      choices: [
-        { value: 'rtk', checked: false },
-        { value: 'quota', checked: false },
-        { value: 'caveman', checked: false },
-      ],
-    }),
-  },
-}))
-
-import { checkbox } from '@inquirer/prompts'
-import { commandExists, warn } from '../../utils/exec.js'
-import { installQuota } from './quota.js'
-import { installCaveman } from './caveman.js'
-import { enableCavemanGuidance } from './caveman-guidance.js'
-import { configureObGlobal } from './global.js'
-import { tokenOptimizationStep } from './index.js'
-
-describe('tokenOptimizationStep()', () => {
-  beforeEach(() => {
-    vi.clearAllMocks()
-  })
-
-  it('runs all optimizations by default selection', async () => {
-    const originalIsTTY = process.stdin.isTTY
-    Object.defineProperty(process.stdin, 'isTTY', { value: true, configurable: true })
-
-    checkbox.mockResolvedValue(['rtk', 'quota', 'caveman'])
-    commandExists.mockResolvedValue(true)
-    installQuota.mockResolvedValue({ optedIn: true, installed: true })
-    installCaveman.mockResolvedValue({ optedIn: true, installed: true })
-    enableCavemanGuidance.mockResolvedValue({ enabled: true })
-    configureObGlobal.mockResolvedValue({ configured: true })
-
-    const result = await tokenOptimizationStep()
-
-    Object.defineProperty(process.stdin, 'isTTY', { value: originalIsTTY, configurable: true })
-
-    expect(commandExists).toHaveBeenCalledWith('rtk')
-    expect(installQuota).toHaveBeenCalledWith({ skipHeader: true, skipPrompt: true })
-    expect(installCaveman).toHaveBeenCalledWith({ skipHeader: true, skipPrompt: true })
-    expect(enableCavemanGuidance).toHaveBeenCalledWith({ optedIn: true, installed: true })
-    expect(configureObGlobal).toHaveBeenCalled()
-    expect(result.rtk.available).toBe(true)
-    expect(result.quota.installed).toBe(true)
-    expect(result.caveman.installed).toBe(true)
-    expect(result.cavemanGuidance.enabled).toBe(true)
-  })
-
-  it('skips all tools when nothing is selected', async () => {
-    checkbox.mockResolvedValue([])
-
-    const result = await tokenOptimizationStep()
-
-    expect(commandExists).not.toHaveBeenCalled()
-    expect(installQuota).not.toHaveBeenCalled()
-    expect(installCaveman).not.toHaveBeenCalled()
-    expect(enableCavemanGuidance).not.toHaveBeenCalled()
-    expect(configureObGlobal).toHaveBeenCalled()
-    expect(warn).toHaveBeenCalledWith('No token optimization tools selected')
-    expect(result.rtk.optedIn).toBe(false)
-    expect(result.quota.optedIn).toBe(false)
-    expect(result.caveman.optedIn).toBe(false)
-    expect(result.cavemanGuidance.enabled).toBe(false)
-  })
-})
+import { beforeEach, describe, expect, it, vi } from 'vitest'
+
+vi.mock('@inquirer/prompts', () => ({
+  checkbox: vi.fn(),
+  confirm: vi.fn(),
+}))
+
+vi.mock('../../utils/exec.js', () => ({
+  code: vi.fn(),
+  commandExists: vi.fn(),
+  header: vi.fn(),
+  info: vi.fn(),
+  loading: vi.fn(),
+  success: vi.fn(),
+  warn: vi.fn(),
+}))
+
+vi.mock('./quota.js', () => ({ installQuota: vi.fn() }))
+vi.mock('./caveman.js', () => ({ installCaveman: vi.fn() }))
+vi.mock('./codegraph.js', () => ({ installCodegraph: vi.fn() }))
+vi.mock('./caveman-guidance.js', () => ({ enableCavemanGuidance: vi.fn() }))
+vi.mock('./global.js', () => ({ configureObGlobal: vi.fn() }))
+
+vi.mock('fs-extra', () => ({
+  default: {
+    readJson: vi.fn().mockResolvedValue({
+      info: 'Token optimization info',
+      message: 'Select tools',
+      timeoutMs: 5000,
+      choices: [
+        { value: 'rtk', checked: false },
+        { value: 'quota', checked: false },
+        { value: 'caveman', checked: false },
+        { value: 'codegraph', checked: false },
+      ],
+    }),
+  },
+}))
+
+import { checkbox } from '@inquirer/prompts'
+import { commandExists, warn } from '../../utils/exec.js'
+import { installQuota } from './quota.js'
+import { installCaveman } from './caveman.js'
+import { installCodegraph } from './codegraph.js'
+import { enableCavemanGuidance } from './caveman-guidance.js'
+import { configureObGlobal } from './global.js'
+import { tokenOptimizationStep } from './index.js'
+
+describe('tokenOptimizationStep()', () => {
+  beforeEach(() => {
+    vi.clearAllMocks()
+  })
+
+  it('runs all optimizations by default selection', async () => {
+    const originalIsTTY = process.stdin.isTTY
+    Object.defineProperty(process.stdin, 'isTTY', { value: true, configurable: true })
+
+    checkbox.mockResolvedValue(['rtk', 'quota', 'caveman', 'codegraph'])
+    commandExists.mockResolvedValue(true)
+    installQuota.mockResolvedValue({ optedIn: true, installed: true })
+    installCaveman.mockResolvedValue({ optedIn: true, installed: true })
+    installCodegraph.mockResolvedValue({ optedIn: true, installed: true })
+    enableCavemanGuidance.mockResolvedValue({ enabled: true })
+    configureObGlobal.mockResolvedValue({ configured: true })
+
+    const result = await tokenOptimizationStep()
+
+    Object.defineProperty(process.stdin, 'isTTY', { value: originalIsTTY, configurable: true })
+
+    expect(commandExists).toHaveBeenCalledWith('rtk')
+    expect(installQuota).toHaveBeenCalledWith({ skipHeader: true, skipPrompt: true })
+    expect(installCaveman).toHaveBeenCalledWith(expect.objectContaining({ skipHeader: true, skipPrompt: true }))
+    expect(installCodegraph).toHaveBeenCalledWith(expect.objectContaining({ skipHeader: true }))
+    expect(enableCavemanGuidance).toHaveBeenCalledWith({ optedIn: true, installed: true })
+    expect(configureObGlobal).toHaveBeenCalled()
+    expect(result.rtk.available).toBe(true)
+    expect(result.quota.installed).toBe(true)
+    expect(result.caveman.installed).toBe(true)
+    expect(result.cavemanGuidance.enabled).toBe(true)
+    expect(result.codegraph.installed).toBe(true)
+  })
+
+  it('skips all tools when nothing is selected', async () => {
+    checkbox.mockResolvedValue([])
+
+    const result = await tokenOptimizationStep()
+
+    expect(commandExists).not.toHaveBeenCalled()
+    expect(installQuota).not.toHaveBeenCalled()
+    expect(installCaveman).not.toHaveBeenCalled()
+    expect(installCodegraph).not.toHaveBeenCalled()
+    expect(enableCavemanGuidance).not.toHaveBeenCalled()
+    expect(configureObGlobal).toHaveBeenCalled()
+    expect(warn).toHaveBeenCalledWith('No token optimization tools selected')
+    expect(result.rtk.optedIn).toBe(false)
+    expect(result.quota.optedIn).toBe(false)
+    expect(result.caveman.optedIn).toBe(false)
+    expect(result.cavemanGuidance.enabled).toBe(false)
+    expect(result.codegraph.optedIn).toBe(false)
+  })
+})

package/src/steps/optimization/quota.js

@@ -1,84 +1,84 @@
-import { confirm } from '@inquirer/prompts'
-import fse from 'fs-extra'
-import path from 'node:path'
-import { fileURLToPath } from 'url'
-import { error, header, info, loading, success, warn } from '../../utils/exec.js'
-
-const __dirname = path.dirname(fileURLToPath(import.meta.url))
-const QUOTA_PRESET_PATH = path.resolve(__dirname, '../../presets/quota.json')
-const quotaPreset = await fse.readJson(QUOTA_PRESET_PATH)
-const PLUGIN = quotaPreset.plugin
-
-function ensurePlugin(config) {
-  if (!Array.isArray(config.plugin)) config.plugin = []
-  if (!config.plugin.includes(PLUGIN)) config.plugin.push(PLUGIN)
-}
-
-function addIfMissing(target, key, value) {
-  if (!(key in target)) target[key] = value
-}
-
-export async function installQuota(options = {}) {
-  if (!options.skipHeader) header('Installing opencode-quota')
-
-  let shouldInstall = true
-  if (!options.skipPrompt && process.stdin.isTTY) {
-    const timeoutMs = quotaPreset.prompt.timeoutMs
-    const choice = await Promise.race([
-      confirm({
-        message: quotaPreset.prompt.message,
-        default: quotaPreset.prompt.default,
-      }),
-      new Promise(resolve => setTimeout(() => resolve(true), timeoutMs)),
-    ])
-    shouldInstall = choice !== false
-  }
-
-  if (!shouldInstall) {
-    warn('Skipped opencode-quota installation')
-    return { optedIn: false, installed: false }
-  }
-
-  loading('configuring opencode-quota...')
-
-  try {
-    const opencodeDir = path.join(process.cwd(), '.opencode')
-    const opencodePath = path.join(opencodeDir, 'opencode.json')
-    const tuiPath = path.join(opencodeDir, 'tui.json')
-    const quotaDir = path.join(opencodeDir, 'opencode-quota')
-    const quotaPath = path.join(quotaDir, 'quota-toast.json')
-
-    const opencode = await fse.pathExists(opencodePath)
-      ? await fse.readJson(opencodePath)
-      : { $schema: 'https://opencode.ai/config.json' }
-
-    const tui = await fse.pathExists(tuiPath)
-      ? await fse.readJson(tuiPath)
-      : { $schema: 'https://opencode.ai/tui.json' }
-
-    ensurePlugin(opencode)
-    ensurePlugin(tui)
-
-    await fse.ensureDir(opencodeDir)
-    await fse.writeJson(opencodePath, opencode, { spaces: 2 })
-    await fse.writeJson(tuiPath, tui, { spaces: 2 })
-
-    const quotaConfig = await fse.pathExists(quotaPath)
-      ? await fse.readJson(quotaPath)
-      : {}
-
-    for (const [key, value] of Object.entries(quotaPreset.defaults)) {
-      addIfMissing(quotaConfig, key, value)
-    }
-
-    await fse.ensureDir(quotaDir)
-    await fse.writeJson(quotaPath, quotaConfig, { spaces: 2 })
-
-    success('opencode-quota configured (manual setup)')
-    info('Restart OpenCode and run /quota to verify')
-    return { optedIn: true, installed: true }
-  } catch (err) {
-    error(`Failed to configure opencode-quota: ${err.message}`)
-    return { optedIn: true, installed: false }
-  }
-}
+import { confirm } from '@inquirer/prompts'
+import fse from 'fs-extra'
+import path from 'node:path'
+import { fileURLToPath } from 'url'
+import { error, header, info, loading, success, warn } from '../../utils/exec.js'
+
+const __dirname = path.dirname(fileURLToPath(import.meta.url))
+const QUOTA_PRESET_PATH = path.resolve(__dirname, '../../presets/quota.json')
+const quotaPreset = await fse.readJson(QUOTA_PRESET_PATH)
+const PLUGIN = quotaPreset.plugin
+
+function ensurePlugin(config) {
+  if (!Array.isArray(config.plugin)) config.plugin = []
+  if (!config.plugin.includes(PLUGIN)) config.plugin.push(PLUGIN)
+}
+
+function addIfMissing(target, key, value) {
+  if (!(key in target)) target[key] = value
+}
+
+export async function installQuota(options = {}) {
+  if (!options.skipHeader) header('Installing opencode-quota')
+
+  let shouldInstall = true
+  if (!options.skipPrompt && process.stdin.isTTY) {
+    const timeoutMs = quotaPreset.prompt.timeoutMs
+    const choice = await Promise.race([
+      confirm({
+        message: quotaPreset.prompt.message,
+        default: quotaPreset.prompt.default,
+      }),
+      new Promise(resolve => { setTimeout(() => resolve(true), timeoutMs) }),
+    ])
+    shouldInstall = choice !== false
+  }
+
+  if (!shouldInstall) {
+    warn('Skipped opencode-quota installation')
+    return { optedIn: false, installed: false }
+  }
+
+  loading('configuring opencode-quota...')
+
+  try {
+    const opencodeDir = path.join(process.cwd(), '.opencode')
+    const opencodePath = path.join(opencodeDir, 'opencode.json')
+    const tuiPath = path.join(opencodeDir, 'tui.json')
+    const quotaDir = path.join(opencodeDir, 'opencode-quota')
+    const quotaPath = path.join(quotaDir, 'quota-toast.json')
+
+    const opencode = await fse.pathExists(opencodePath)
+      ? await fse.readJson(opencodePath)
+      : { $schema: 'https://opencode.ai/config.json' }
+
+    const tui = await fse.pathExists(tuiPath)
+      ? await fse.readJson(tuiPath)
+      : { $schema: 'https://opencode.ai/tui.json' }
+
+    ensurePlugin(opencode)
+    ensurePlugin(tui)
+
+    await fse.ensureDir(opencodeDir)
+    await fse.writeJson(opencodePath, opencode, { spaces: 2 })
+    await fse.writeJson(tuiPath, tui, { spaces: 2 })
+
+    const quotaConfig = await fse.pathExists(quotaPath)
+      ? await fse.readJson(quotaPath)
+      : {}
+
+    for (const [key, value] of Object.entries(quotaPreset.defaults)) {
+      addIfMissing(quotaConfig, key, value)
+    }
+
+    await fse.ensureDir(quotaDir)
+    await fse.writeJson(quotaPath, quotaConfig, { spaces: 2 })
+
+    success('opencode-quota configured (manual setup)')
+    info('Restart OpenCode and run /quota to verify')
+    return { optedIn: true, installed: true }
+  } catch (err) {
+    error(`Failed to configure opencode-quota: ${err.message}`)
+    return { optedIn: true, installed: false }
+  }
+}

package/src/steps/source/source.test.js

@@ -1,124 +1,124 @@
-import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest'
-import fs from 'node:fs'
-import path from 'node:path'
-import os from 'node:os'
-
-vi.mock('../../utils/exec.js', () => ({
-  header: vi.fn(),
-  info: vi.fn(),
-  success: vi.fn(),
-  warn: vi.fn(),
-}))
-
-vi.mock('@inquirer/prompts', () => ({
-  select: vi.fn(),
-  checkbox: vi.fn(),
-}))
-
-vi.mock('fs-extra', () => ({
-  default: {
-    readJson: vi.fn().mockResolvedValue({
-      message: 'Select source scope',
-      default: 'current',
-      choices: [
-        { name: 'Current folder', value: 'current' },
-        { name: 'Parent folder', value: 'parent' },
-        { name: 'Child folders', value: 'children' },
-      ],
-      parentSelectionMessage: 'Select sibling folders',
-      childrenSelectionMessage: 'Select child folders',
-    }),
-    readdir: vi.fn(),
-    stat: vi.fn().mockResolvedValue({ isDirectory: () => true }),
-  },
-}))
-
-import { select, checkbox } from '@inquirer/prompts'
-import fse from 'fs-extra'
-import { chooseSourceScope } from './index.js'
-
-describe('chooseSourceScope()', () => {
-  let tmpDir, originalCwd
-
-  beforeEach(() => {
-    tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'source-test-'))
-    originalCwd = process.cwd()
-    process.chdir(tmpDir)
-    vi.clearAllMocks()
-  })
-
-  afterEach(() => {
-    process.chdir(originalCwd)
-    fs.rmSync(tmpDir, { recursive: true, force: true })
-  })
-
-  it('returns current folder when user selects current mode', async () => {
-    select.mockResolvedValue('current')
-
-    const result = await chooseSourceScope()
-
-    expect(result.sourceMode).toBe('current')
-    expect(result.sourceRoots).toContain(tmpDir)
-  })
-
-  it('lists parent folders when user selects parent mode', async () => {
-    select.mockResolvedValue('parent')
-    const parentDir = path.dirname(tmpDir)
-    const siblingDir = path.join(parentDir, 'sibling-project')
-    fs.mkdirSync(siblingDir, { recursive: true })
-    fse.readdir.mockResolvedValue(['sibling-project'])
-
-    await chooseSourceScope()
-
-    expect(checkbox).toHaveBeenCalled()
-  })
-
-  it('falls back to current when no siblings found', async () => {
-    select.mockResolvedValue('parent')
-    fse.readdir.mockResolvedValue([])
-
-    const result = await chooseSourceScope()
-
-    expect(result.sourceMode).toBe('current')
-  })
-
-  it('falls back to current when no folders selected', async () => {
-    select.mockResolvedValue('parent')
-    checkbox.mockResolvedValue([])
-
-    const result = await chooseSourceScope()
-
-    expect(result.sourceMode).toBe('current')
-  })
-
-  it('lists child folders when user selects children mode', async () => {
-    select.mockResolvedValue('children')
-    fse.readdir.mockResolvedValue(['packages', 'apps'])
-    checkbox.mockResolvedValue([path.join(tmpDir, 'packages')])
-
-    const result = await chooseSourceScope()
-
-    expect(checkbox).toHaveBeenCalled()
-    expect(result.sourceMode).toBe('children-selected')
-    expect(result.sourceRoots).toContain(path.join(tmpDir, 'packages'))
-  })
-
-  it('falls back to current when no child folders found', async () => {
-    select.mockResolvedValue('children')
-    fse.readdir.mockResolvedValue([])
-
-    const result = await chooseSourceScope()
-
-    expect(result.sourceMode).toBe('current')
-  })
-
-  it('falls back to current when no child folders selected', async () => {
-    select.mockResolvedValue('children')
-    fse.readdir.mockResolvedValue(['packages'])
-    checkbox.mockResolvedValue([])
-
-    const result = await chooseSourceScope()
-
-    expect(result.sourceMode).toBe('current')
-  })
-})
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest'
+import fs from 'node:fs'
+import path from 'node:path'
+import os from 'node:os'
+
+vi.mock('../../utils/exec.js', () => ({
+  header: vi.fn(),
+  info: vi.fn(),
+  success: vi.fn(),
+  warn: vi.fn(),
+}))
+
+vi.mock('@inquirer/prompts', () => ({
+  select: vi.fn(),
+  checkbox: vi.fn(),
+}))
+
+vi.mock('fs-extra', () => ({
+  default: {
+    readJson: vi.fn().mockResolvedValue({
+      message: 'Select source scope',
+      default: 'current',
+      choices: [
+        { name: 'Current folder', value: 'current' },
+        { name: 'Parent folder', value: 'parent' },
+        { name: 'Child folders', value: 'children' },
+      ],
+      parentSelectionMessage: 'Select sibling folders',
+      childrenSelectionMessage: 'Select child folders',
+    }),
+    readdir: vi.fn(),
+    stat: vi.fn().mockResolvedValue({ isDirectory: () => true }),
+  },
+}))
+
+import { select, checkbox } from '@inquirer/prompts'
+import fse from 'fs-extra'
+import { chooseSourceScope } from './index.js'
+
+describe('chooseSourceScope()', () => {
+  let tmpDir, originalCwd
+
+  beforeEach(() => {
+    tmpDir = fs.realpathSync(fs.mkdtempSync(path.join(os.tmpdir(), 'source-test-')))
+    originalCwd = process.cwd()
+    process.chdir(tmpDir)
+    vi.clearAllMocks()
+  })
+
+  afterEach(() => {
+    process.chdir(originalCwd)
+    fs.rmSync(tmpDir, { recursive: true, force: true })
+  })
+
+  it('returns current folder when user selects current mode', async () => {
+    select.mockResolvedValue('current')
+
+    const result = await chooseSourceScope()
+
+    expect(result.sourceMode).toBe('current')
+    expect(result.sourceRoots).toContain(tmpDir)
+  })
+
+  it('lists parent folders when user selects parent mode', async () => {
+    select.mockResolvedValue('parent')
+    const parentDir = path.dirname(tmpDir)
+    const siblingDir = path.join(parentDir, 'sibling-project')
+    fs.mkdirSync(siblingDir, { recursive: true })
+    fse.readdir.mockResolvedValue(['sibling-project'])
+
+    await chooseSourceScope()
+
+    expect(checkbox).toHaveBeenCalled()
+  })
+
+  it('falls back to current when no siblings found', async () => {
+    select.mockResolvedValue('parent')
+    fse.readdir.mockResolvedValue([])
+
+    const result = await chooseSourceScope()
+
+    expect(result.sourceMode).toBe('current')
+  })
+
+  it('falls back to current when no folders selected', async () => {
+    select.mockResolvedValue('parent')
+    checkbox.mockResolvedValue([])
+
+    const result = await chooseSourceScope()
+
+    expect(result.sourceMode).toBe('current')
+  })
+
+  it('lists child folders when user selects children mode', async () => {
+    select.mockResolvedValue('children')
+    fse.readdir.mockResolvedValue(['packages', 'apps'])
+    checkbox.mockResolvedValue([path.join(tmpDir, 'packages')])
+
+    const result = await chooseSourceScope()
+
+    expect(checkbox).toHaveBeenCalled()
+    expect(result.sourceMode).toBe('children-selected')
+    expect(result.sourceRoots).toContain(path.join(tmpDir, 'packages'))
+  })
+
+  it('falls back to current when no child folders found', async () => {
+    select.mockResolvedValue('children')
+    fse.readdir.mockResolvedValue([])
+
+    const result = await chooseSourceScope()
+
+    expect(result.sourceMode).toBe('current')
+  })
+
+  it('falls back to current when no child folders selected', async () => {
+    select.mockResolvedValue('children')
+    fse.readdir.mockResolvedValue(['packages'])
+    checkbox.mockResolvedValue([])
+
+    const result = await chooseSourceScope()
+
+    expect(result.sourceMode).toBe('current')
+  })
+})