@alumbwe/anvil 1.0.0

package/src/__tests__/unit/create-run-config.test.ts

@@ -0,0 +1,66 @@
+import { describe, test, expect } from 'bun:test'
+
+import { isSensitiveFile } from '../../utils/create-run-config'
+
+describe('isSensitiveFile', () => {
+  test.each([
+    // Env files (blocked)
+    ['.env', true],
+    ['.ENV', true],
+    ['.env.local', true],
+    ['.env/./', true],
+    ['.env ', true],
+    ['.env:$DATA', true],
+    ['config\\.Env.Production', true],
+    ['config/.env.production', true],
+
+    // Env templates (allowed)
+    ['.env.example', false],
+    ['.ENV.EXAMPLE', false],
+    ['.env.sample', false],
+    ['.env.template', false],
+
+    // Sensitive extensions
+    ['private.pem', true],
+    ['server.key', true],
+    ['cert.p12', true],
+    ['app.keystore', true],
+    ['server.crt', true],
+
+    // Sensitive basenames
+    ['.htpasswd', true],
+    ['.netrc', true],
+    ['credentials', true],
+    ['.npmrc', true],
+    ['.yarnrc.yml', true],
+    ['auth.json', true],
+    ['terraform.tfvars', true],
+
+    // SSH keys (prefix pattern)
+    ['id_rsa', true],
+    ['id_ed25519', true],
+    ['id_rsa_github', true],
+    ['id_rsa.pub', false], // public keys allowed
+
+    // Credentials suffix pattern
+    ['aws_credentials', true],
+    ['db_credentials', true],
+
+    // Substring patterns
+    ['kubeconfig', true],
+    ['my-kubeconfig.yaml', true],
+    ['terraform.tfstate', true],
+    ['prod.tfstate.backup', true],
+
+    // Non-sensitive (should NOT be blocked)
+    ['package.json', false],
+    ['README.md', false],
+    ['src/index.ts', false],
+    ['.envrc', false],
+    ['credentials.ts', false],
+    ['terraform.tf', false],
+    ['kube-config.ts', false],
+  ])('%s → %s', (file, expected) => {
+    expect(isSensitiveFile(file)).toBe(expected)
+  })
+})

package/src/__tests__/unit/publish-confirmation.test.ts

@@ -0,0 +1,70 @@
+import { describe, expect, test } from 'bun:test'
+
+import { getAllPublishAgentIds } from '../../components/publish-confirmation'
+
+import type { LocalAgentInfo } from '../../utils/local-agent-registry'
+
+const makeAgent = (id: string, isBundled = false): LocalAgentInfo => ({
+  id,
+  displayName: id,
+  filePath: `/agents/${id}.ts`,
+  isBundled,
+})
+
+describe('getAllPublishAgentIds', () => {
+  test('ignores bundled agents even if selected', () => {
+    const base = makeAgent('base', true)
+    const userA = makeAgent('user-a')
+    const agents = [base, userA]
+    const defs = new Map<string, { spawnableAgents?: string[] }>()
+
+    const ids = getAllPublishAgentIds([base, userA], agents, defs)
+
+    expect(ids).toEqual(['user-a'])
+  })
+
+  test('does not include bundled dependencies discovered via spawns', () => {
+    const base = makeAgent('base', true)
+    const userA = makeAgent('user-a')
+    const agents = [base, userA]
+    const defs = new Map<string, { spawnableAgents?: string[] }>([
+      ['user-a', { spawnableAgents: ['base'] }],
+    ])
+
+    const ids = getAllPublishAgentIds([userA], agents, defs)
+
+    expect(ids).toEqual(['user-a'])
+  })
+
+  test('only adds publishable dependents when includeDependents is true', () => {
+    const base = makeAgent('base', true)
+    const userA = makeAgent('user-a')
+    const userB = makeAgent('user-b')
+    const agents = [base, userA, userB]
+    const defs = new Map<string, { spawnableAgents?: string[] }>([
+      ['user-a', { spawnableAgents: [] }],
+      ['user-b', { spawnableAgents: ['user-a'] }],
+      ['base', { spawnableAgents: ['user-a'] }],
+    ])
+
+    const ids = getAllPublishAgentIds([userA], agents, defs, true)
+
+    expect(ids).toEqual(['user-a', 'user-b'])
+  })
+
+  test('includes transitive dependents when includeDependents is true', () => {
+    const userA = makeAgent('user-a')
+    const userB = makeAgent('user-b')
+    const userC = makeAgent('user-c')
+    const agents = [userA, userB, userC]
+    const defs = new Map<string, { spawnableAgents?: string[] }>([
+      ['user-a', { spawnableAgents: [] }],
+      ['user-b', { spawnableAgents: ['user-a'] }],
+      ['user-c', { spawnableAgents: ['user-b'] }],
+    ])
+
+    const ids = getAllPublishAgentIds([userA], agents, defs, true)
+
+    expect(ids).toEqual(['user-a', 'user-b', 'user-c'])
+  })
+})

package/src/__tests__/unit/segmented-control.test.ts

@@ -0,0 +1,152 @@
+import { describe, test, expect } from 'bun:test'
+import stringWidth from 'string-width'
+
+import {
+  type Segment,
+  processSegments,
+  type ProcessedSegment,
+} from '../../components/segmented-control'
+
+import type { ChatTheme } from '../../types/theme-system'
+
+const theme: ChatTheme = {
+  name: 'dark',
+  // Core
+  primary: '#ff0',
+  secondary: '#888',
+  success: '#0f0',
+  error: '#f00',
+  warning: '#fa0',
+  info: '#0af',
+  // Neutral
+  foreground: '#fff',
+  background: 'transparent',
+  muted: '#777',
+  border: '#333',
+  surface: '#000',
+  surfaceHover: '#222',
+  // Context
+  aiLine: '#0a0',
+  userLine: '#08f',
+  agentToggleHeaderBg: '#f60',
+  agentToggleExpandedBg: '#14e',
+  agentFocusedBg: '#334155',
+  agentContentBg: '#000000',
+  // Input
+  inputFg: '#fff',
+  inputFocusedFg: '#fff',
+  // Modes
+  modeFastBg: '#f60',
+  modeFastText: '#f60',
+  modeMaxBg: '#d22',
+  modeMaxText: '#d22',
+  modePlanBg: '#14e',
+  modePlanText: '#14e',
+  // Link
+  link: '#3B82F6',
+  // Directory
+  directory: '#9CA3AF',
+  // Image card
+  imageCardBorder: '#3B82F6',
+  // Markdown
+  markdown: {
+    codeBackground: '#111',
+    codeHeaderFg: '#555',
+    inlineCodeFg: '#fff',
+    codeTextFg: '#fff',
+    headingFg: {
+      1: '#ff0',
+      2: '#ff0',
+      3: '#ff0',
+      4: '#ff0',
+      5: '#ff0',
+      6: '#ff0',
+    },
+    listBulletFg: '#aaa',
+    blockquoteBorderFg: '#444',
+    blockquoteTextFg: '#eee',
+    dividerFg: '#333',
+    codeMonochrome: true,
+  },
+  messageTextAttributes: 0,
+}
+
+describe('SegmentedControl - processSegments', () => {
+  test('computes width from label using string-width', () => {
+    const segments: Segment[] = [
+      { id: 'DEFAULT', label: 'DEFAULT' },
+      { id: 'MAX', label: 'MAX' },
+      { id: 'PLAN', label: 'PLAN' },
+    ]
+
+    const processed = processSegments(segments, null, false, theme)
+    const widths = processed.map((s) => s.width)
+    expect(widths).toEqual([
+      stringWidth(' DEFAULT '),
+      stringWidth(' MAX '),
+      stringWidth(' PLAN '),
+    ])
+  })
+
+  test('applies defaultHighlighted when nothing hovered', () => {
+    const segments: Segment[] = [
+      { id: 'DEFAULT', label: 'DEFAULT' },
+      { id: 'MAX', label: 'MAX' },
+      {
+        id: 'active-DEFAULT',
+        label: '> DEFAULT',
+        isSelected: true,
+        defaultHighlighted: true,
+      },
+    ]
+
+    const processed = processSegments(segments, null, false, theme)
+    const map: Record<string, ProcessedSegment> = Object.fromEntries(
+      processed.map((p) => [p.id, p]),
+    )
+
+    expect(map['active-DEFAULT'].leftBorderColor).toBe(theme.foreground)
+    expect(map['DEFAULT'].leftBorderColor).toBe(theme.border)
+  })
+
+  test('hovering a segment highlights it', () => {
+    const segments: Segment[] = [
+      { id: 'DEFAULT', label: 'DEFAULT' },
+      { id: 'MAX', label: 'MAX' },
+      { id: 'PLAN', label: 'PLAN' },
+    ]
+    const processed = processSegments(segments, 'MAX', true, theme)
+    const map: Record<string, ProcessedSegment> = Object.fromEntries(
+      processed.map((p) => [p.id, p]),
+    )
+
+    expect(map.MAX.isHovered).toBe(true)
+    expect(map.MAX.leftBorderColor).toBe(theme.foreground)
+    expect(map.DEFAULT.leftBorderColor).toBe(theme.border)
+  })
+
+  test('disabled segments use muted text and border colors', () => {
+    const segments: Segment[] = [
+      { id: 'DEFAULT', label: 'DEFAULT', disabled: true },
+      { id: 'MAX', label: 'MAX' },
+    ]
+    const [fast, max] = processSegments(segments, null, false, theme)
+    expect(fast.textColor).toBe(theme.muted)
+    expect(fast.leftBorderColor).toBe(theme.border)
+    expect(max.textColor).toBe(theme.foreground)
+  })
+})
+
+describe('SegmentedControl - string width with double-width glyphs', () => {
+  test('emoji and CJK characters increase width correctly', () => {
+    const segments: Segment[] = [
+      { id: 'E', label: '🚀MAX' },
+      { id: 'CJK', label: '计划' },
+    ]
+
+    const processed = processSegments(segments, null, false, theme)
+
+    expect(processed[0].width).toBe(stringWidth(' 🚀MAX '))
+    expect(processed[1].width).toBe(stringWidth(' 计划 '))
+  })
+})

package/src/__tests__/utils/env.test.ts

@@ -0,0 +1,169 @@
+import { describe, test, expect, afterEach } from 'bun:test'
+
+import { createTestCliEnv } from '../../testing/env'
+import { getCliEnv } from '../../utils/env'
+
+describe('cli/utils/env', () => {
+  describe('getCliEnv', () => {
+    const originalEnv = { ...process.env }
+
+    afterEach(() => {
+      // Restore original env
+      Object.keys(process.env).forEach((key) => {
+        if (!(key in originalEnv)) {
+          delete process.env[key]
+        }
+      })
+      Object.assign(process.env, originalEnv)
+    })
+
+    test('returns current process.env values for base vars', () => {
+      process.env.SHELL = '/bin/zsh'
+      process.env.HOME = '/Users/testuser'
+      const env = getCliEnv()
+      expect(env.SHELL).toBe('/bin/zsh')
+      expect(env.HOME).toBe('/Users/testuser')
+    })
+
+    test('returns current process.env values for terminal detection vars', () => {
+      process.env.TERM_PROGRAM = 'iTerm.app'
+      process.env.KITTY_WINDOW_ID = '12345'
+      const env = getCliEnv()
+      expect(env.TERM_PROGRAM).toBe('iTerm.app')
+      expect(env.KITTY_WINDOW_ID).toBe('12345')
+    })
+
+    test('returns current process.env values for VS Code detection', () => {
+      process.env.VSCODE_PID = '1234'
+      process.env.VSCODE_THEME_KIND = 'dark'
+      const env = getCliEnv()
+      expect(env.VSCODE_PID).toBe('1234')
+      expect(env.VSCODE_THEME_KIND).toBe('dark')
+    })
+
+    test('returns current process.env values for Cursor detection', () => {
+      process.env.CURSOR_PORT = '5678'
+      process.env.CURSOR = 'true'
+      const env = getCliEnv()
+      expect(env.CURSOR_PORT).toBe('5678')
+      expect(env.CURSOR).toBe('true')
+    })
+
+    test('returns current process.env values for JetBrains detection', () => {
+      process.env.TERMINAL_EMULATOR = 'JetBrains-JediTerm'
+      process.env.IDE_CONFIG_DIR = '/path/to/idea'
+      const env = getCliEnv()
+      expect(env.TERMINAL_EMULATOR).toBe('JetBrains-JediTerm')
+      expect(env.IDE_CONFIG_DIR).toBe('/path/to/idea')
+    })
+
+    test('returns current process.env values for editor preferences', () => {
+      process.env.EDITOR = 'vim'
+      process.env.ANVIL_CLI_EDITOR = 'code'
+      const env = getCliEnv()
+      expect(env.EDITOR).toBe('vim')
+      expect(env.ANVIL_CLI_EDITOR).toBe('code')
+    })
+
+    test('returns current process.env values for theme preferences', () => {
+      process.env.OPEN_TUI_THEME = 'dark'
+      const env = getCliEnv()
+      expect(env.OPEN_TUI_THEME).toBe('dark')
+    })
+
+    test('returns current process.env values for binary build config', () => {
+      process.env.ANVIL_IS_BINARY = 'true'
+      process.env.ANVIL_CLI_VERSION = '1.0.0'
+      const env = getCliEnv()
+      expect(env.ANVIL_IS_BINARY).toBe('true')
+      expect(env.ANVIL_CLI_VERSION).toBe('1.0.0')
+    })
+
+    test('returns undefined for unset env vars', () => {
+      delete process.env.KITTY_WINDOW_ID
+      delete process.env.VSCODE_PID
+      const env = getCliEnv()
+      expect(env.KITTY_WINDOW_ID).toBeUndefined()
+      expect(env.VSCODE_PID).toBeUndefined()
+    })
+
+    test('returns a snapshot that does not change when process.env changes', () => {
+      process.env.TERM_PROGRAM = 'iTerm.app'
+      const env = getCliEnv()
+      process.env.TERM_PROGRAM = 'vscode'
+      expect(env.TERM_PROGRAM).toBe('iTerm.app')
+    })
+  })
+
+  describe('createTestCliEnv', () => {
+    test('returns a CliEnv with default test values', () => {
+      const env = createTestCliEnv()
+      expect(env.HOME).toBe('/home/test')
+      expect(env.NODE_ENV).toBe('test')
+      expect(env.TERM).toBe('xterm-256color')
+      expect(env.PATH).toBe('/usr/bin')
+    })
+
+    test('returns undefined for CLI-specific vars by default', () => {
+      const env = createTestCliEnv()
+      expect(env.KITTY_WINDOW_ID).toBeUndefined()
+      expect(env.VSCODE_PID).toBeUndefined()
+      expect(env.CURSOR_PORT).toBeUndefined()
+      expect(env.IDE_CONFIG_DIR).toBeUndefined()
+      expect(env.ANVIL_IS_BINARY).toBeUndefined()
+    })
+
+    test('allows overriding terminal detection vars', () => {
+      const env = createTestCliEnv({
+        TERM_PROGRAM: 'iTerm.app',
+        KITTY_WINDOW_ID: '12345',
+        SIXEL_SUPPORT: 'true',
+      })
+      expect(env.TERM_PROGRAM).toBe('iTerm.app')
+      expect(env.KITTY_WINDOW_ID).toBe('12345')
+      expect(env.SIXEL_SUPPORT).toBe('true')
+    })
+
+    test('allows overriding VS Code detection vars', () => {
+      const env = createTestCliEnv({
+        VSCODE_PID: '1234',
+        VSCODE_THEME_KIND: 'dark',
+        VSCODE_GIT_IPC_HANDLE: '/tmp/vscode-git',
+      })
+      expect(env.VSCODE_PID).toBe('1234')
+      expect(env.VSCODE_THEME_KIND).toBe('dark')
+      expect(env.VSCODE_GIT_IPC_HANDLE).toBe('/tmp/vscode-git')
+    })
+
+    test('allows overriding editor preferences', () => {
+      const env = createTestCliEnv({
+        EDITOR: 'vim',
+        VISUAL: 'code',
+        ANVIL_CLI_EDITOR: 'cursor',
+      })
+      expect(env.EDITOR).toBe('vim')
+      expect(env.VISUAL).toBe('code')
+      expect(env.ANVIL_CLI_EDITOR).toBe('cursor')
+    })
+
+    test('allows overriding binary build config', () => {
+      const env = createTestCliEnv({
+        ANVIL_IS_BINARY: 'true',
+        ANVIL_CLI_VERSION: '2.0.0',
+        ANVIL_CLI_TARGET: 'darwin-arm64',
+      })
+      expect(env.ANVIL_IS_BINARY).toBe('true')
+      expect(env.ANVIL_CLI_VERSION).toBe('2.0.0')
+      expect(env.ANVIL_CLI_TARGET).toBe('darwin-arm64')
+    })
+
+    test('allows overriding default values', () => {
+      const env = createTestCliEnv({
+        HOME: '/custom/home',
+        NODE_ENV: 'production',
+      })
+      expect(env.HOME).toBe('/custom/home')
+      expect(env.NODE_ENV).toBe('production')
+    })
+  })
+})

package/src/__tests__/utils/project-picker.test.ts

@@ -0,0 +1,39 @@
+import path from 'path'
+
+import { describe, test, expect } from 'bun:test'
+
+import { shouldShowProjectPicker } from '../../utils/project-picker'
+
+describe('cli/utils/project-picker', () => {
+  test('returns true when start cwd is home directory', () => {
+    const root = path.parse(process.cwd()).root
+    const homeDir = path.join(root, 'home', 'test-user')
+
+    expect(shouldShowProjectPicker(homeDir, homeDir)).toBe(true)
+  })
+
+  test('returns true when start cwd is a parent of home directory', () => {
+    const root = path.parse(process.cwd()).root
+    const homeDir = path.join(root, 'home', 'test-user')
+    const parentDir = path.dirname(homeDir)
+
+    expect(shouldShowProjectPicker(parentDir, homeDir)).toBe(true)
+    expect(shouldShowProjectPicker(root, homeDir)).toBe(true)
+  })
+
+  test('returns false when start cwd is a child of home directory', () => {
+    const root = path.parse(process.cwd()).root
+    const homeDir = path.join(root, 'home', 'test-user')
+    const childDir = path.join(homeDir, 'projects')
+
+    expect(shouldShowProjectPicker(childDir, homeDir)).toBe(false)
+  })
+
+  test('returns false when start cwd is a sibling of home directory', () => {
+    const root = path.parse(process.cwd()).root
+    const homeDir = path.join(root, 'home', 'test-user')
+    const siblingDir = path.join(root, 'home', 'other-user')
+
+    expect(shouldShowProjectPicker(siblingDir, homeDir)).toBe(false)
+  })
+})

package/src/agents/bundled-agents.generated.d.ts

@@ -0,0 +1,14 @@
+/**
+ * Type declarations for the auto-generated bundled agents module.
+ *
+ * The actual file (bundled-agents.generated.ts) is created by
+ * cli/scripts/prebuild-agents.ts and is gitignored. This declaration
+ * file lets TypeScript resolve the module when the generated file
+ * has not been built yet.
+ */
+import type { LocalAgentInfo } from '../utils/local-agent-registry'
+
+export declare const bundledAgents: Record<string, any>
+export declare function getBundledAgentsAsLocalInfo(): LocalAgentInfo[]
+export declare function getBundledAgentIds(): string[]
+export declare function isBundledAgent(agentId: string): boolean