opencode-onboard 0.4.2 → 0.4.4

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/index.js

@@ -8,6 +8,26 @@ const __dirname = path.dirname(fileURLToPath(import.meta.url))
 const SOURCE_PRESET_PATH = path.resolve(__dirname, '../../presets/source.json')
 const sourcePreset = await fse.readJson(SOURCE_PRESET_PATH)
 
+async function listChildFolders(cwd) {
+  const entries = await fse.readdir(cwd)
+  const dirs = []
+
+  for (const name of entries) {
+    if (name.startsWith('.')) continue
+    const abs = path.join(cwd, name)
+    try {
+      const stat = await fse.stat(abs)
+      if (!stat.isDirectory()) continue
+      dirs.push({ name, abs })
+    } catch {
+      // ignore invalid entries
+    }
+  }
+
+  dirs.sort((a, b) => a.name.localeCompare(b.name))
+  return dirs
+}
+
 async function listParentFolders(cwd) {
   const parent = path.resolve(cwd, '..')
   const entries = await fse.readdir(parent)
@@ -47,6 +67,34 @@ export async function chooseSourceScope() {
     return { sourceMode: 'current', sourceRoots: [cwd] }
   }
 
+  if (mode === 'children') {
+    const childFolders = await listChildFolders(cwd)
+    if (childFolders.length === 0) {
+      warn('No child folders found in current directory. Falling back to current folder.')
+      success(`Source scope: ${cwd}`)
+      return { sourceMode: 'current', sourceRoots: [cwd] }
+    }
+
+    const selected = await checkbox({
+      message: sourcePreset.childrenSelectionMessage,
+      choices: childFolders.map(d => ({
+        name: `./${d.name}`,
+        value: d.abs,
+        checked: true,
+      })),
+      required: true,
+    })
+
+    if (!selected || selected.length === 0) {
+      warn('No folders selected. Falling back to current folder.')
+      success(`Source scope: ${cwd}`)
+      return { sourceMode: 'current', sourceRoots: [cwd] }
+    }
+
+    success(`Source scope: ${selected.map(p => path.basename(p)).join(', ')}`)
+    return { sourceMode: 'children-selected', sourceRoots: selected }
+  }
+
   const parentFolders = await listParentFolders(cwd)
   if (parentFolders.length === 0) {
     warn('No sibling folders found in parent directory. Falling back to current folder.')

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

@@ -1,91 +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' },
-      ],
-      parentSelectionMessage: 'Select sibling 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')
-  })
-})
+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')
+  })
+})