@alumbwe/anvil 1.0.0 → 1.0.2

package/src/utils/__tests__/trace-writer.test.ts

@@ -1,223 +0,0 @@
-import { describe, test, expect, beforeEach, afterEach, mock } from 'bun:test'
-import * as fs from 'fs'
-import * as os from 'os'
-import * as path from 'path'
-
-// Force tracing on so the test is deterministic
-mock.module('@anvil/common/env', () => ({
-  IS_DEV: true,
-}))
-
-import { createTraceWriter } from '../trace-writer'
-import { setProjectRoot, tryGetProjectRoot } from '../../project-files'
-
-import type { Message } from '@anvil/common/types/messages/anvil-message'
-
-let tempDir = ''
-
-function tracePathFor(dir: string): string {
-  return path.join(dir, 'debug', 'trace.jsonl')
-}
-
-function readTraceLines(): any[] {
-  const tracePath = tracePathFor(tempDir)
-  if (!fs.existsSync(tracePath)) return []
-  return fs
-    .readFileSync(tracePath, 'utf8')
-    .trim()
-    .split('\n')
-    .map((line) => JSON.parse(line))
-}
-
-const userMessage = (text: string): Message => ({
-  role: 'user',
-  content: [{ type: 'text', text }],
-})
-const assistantMessage = (text: string): Message => ({
-  role: 'assistant',
-  content: [{ type: 'text', text }],
-})
-
-const baseParams = {
-  agentId: 'agent-1',
-  agentType: 'base',
-  runId: 'run-1',
-  userInputId: 'input-1',
-  system: undefined,
-}
-
-describe('createTraceWriter', () => {
-  let originalProjectRoot: string | undefined
-
-  beforeEach(() => {
-    tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'anvil-trace-'))
-    originalProjectRoot = tryGetProjectRoot()
-  })
-
-  afterEach(() => {
-    fs.rmSync(tempDir, { recursive: true, force: true })
-    if (originalProjectRoot !== undefined) {
-      setProjectRoot(originalProjectRoot)
-    }
-  })
-
-  test('falls back to the real project-root-based path when no resolver is injected', () => {
-    setProjectRoot(tempDir)
-    const writer = createTraceWriter()!
-
-    writer.recordStep({
-      ...baseParams,
-      step: 1,
-      messages: [userMessage('hello')],
-    })
-
-    const lines = readTraceLines()
-    expect(lines).toHaveLength(1)
-    expect(lines[0].message.role).toBe('user')
-  })
-
-  test('writes each message exactly once across steps', () => {
-    const writer = createTraceWriter(() => tracePathFor(tempDir))!
-    const history = [userMessage('hello')]
-
-    writer.recordStep({ ...baseParams, step: 1, messages: history })
-    history.push(assistantMessage('hi there'))
-    writer.recordStep({ ...baseParams, step: 1, messages: history })
-    // Re-recording the same history adds nothing
-    writer.recordStep({ ...baseParams, step: 2, messages: history })
-
-    const lines = readTraceLines()
-    expect(lines).toHaveLength(2)
-    expect(lines.map((l) => l.message.role)).toEqual(['user', 'assistant'])
-    expect(lines.map((l) => l.index)).toEqual([0, 1])
-  })
-
-  test('records system prompt once and again only when it changes', () => {
-    const writer = createTraceWriter(() => tracePathFor(tempDir))!
-
-    writer.recordStep({
-      ...baseParams,
-      step: 1,
-      system: 'be helpful',
-      messages: [userMessage('hello')],
-    })
-    writer.recordStep({
-      ...baseParams,
-      step: 2,
-      system: 'be helpful',
-      messages: [userMessage('hello'), assistantMessage('hi')],
-    })
-    writer.recordStep({
-      ...baseParams,
-      step: 3,
-      system: 'be terse',
-      messages: [
-        userMessage('hello'),
-        assistantMessage('hi'),
-        userMessage('more'),
-      ],
-    })
-
-    const systemLines = readTraceLines().filter((l) => l.type === 'system')
-    expect(systemLines.map((l) => l.system)).toEqual([
-      'be helpful',
-      'be terse',
-    ])
-  })
-
-  test('detects history rewrites and re-dumps the new history', () => {
-    const writer = createTraceWriter(() => tracePathFor(tempDir))!
-
-    writer.recordStep({
-      ...baseParams,
-      step: 1,
-      messages: [
-        userMessage('a'),
-        assistantMessage('b'),
-        userMessage('c'),
-      ],
-    })
-    // Compaction: history replaced with a shorter one
-    writer.recordStep({
-      ...baseParams,
-      step: 2,
-      messages: [userMessage('summary of earlier conversation')],
-    })
-
-    const lines = readTraceLines()
-    const marker = lines.find((l) => l.type === 'history_rewritten')
-    expect(marker).toBeDefined()
-    expect(marker.previousMessageCount).toBe(3)
-    expect(marker.messageCount).toBe(1)
-    // 3 original + 1 re-dumped after the marker
-    expect(lines.filter((l) => l.type === 'message')).toHaveLength(4)
-  })
-
-  test('tracks agents independently', () => {
-    const writer = createTraceWriter(() => tracePathFor(tempDir))!
-
-    writer.recordStep({
-      ...baseParams,
-      step: 1,
-      messages: [userMessage('main')],
-    })
-    writer.recordStep({
-      ...baseParams,
-      agentId: 'agent-2',
-      agentType: 'researcher',
-      step: 1,
-      messages: [userMessage('sub')],
-    })
-
-    const lines = readTraceLines()
-    expect(lines.filter((l) => l.type === 'history_rewritten')).toHaveLength(0)
-    expect(lines.map((l) => l.agentId)).toEqual(['agent-1', 'agent-2'])
-  })
-
-  test('writes base64 image data in full', () => {
-    const base64 = 'a'.repeat(10_000)
-    const writer = createTraceWriter(() => tracePathFor(tempDir))!
-
-    writer.recordStep({
-      ...baseParams,
-      step: 1,
-      messages: [
-        {
-          role: 'user',
-          content: [
-            { type: 'text', text: 'look at this' },
-            { type: 'image', image: base64, mediaType: 'image/png' },
-          ],
-        } as Message,
-      ],
-    })
-
-    const lines = readTraceLines()
-    expect(lines).toHaveLength(1)
-    expect(lines[0].message.content[1].image).toBe(base64)
-  })
-
-  test('breaks cycles in traced message content instead of throwing', () => {
-    const writer = createTraceWriter(() => tracePathFor(tempDir))!
-    const cyclic: any = { type: 'text', text: 'boom' }
-    cyclic.self = cyclic
-
-    writer.recordStep({
-      ...baseParams,
-      step: 1,
-      messages: [
-        {
-          role: 'user',
-          content: [{ type: 'text', text: 'plain' }, cyclic],
-        } as unknown as Message,
-      ],
-    })
-
-    const lines = readTraceLines()
-    expect(lines).toHaveLength(1)
-    const message = lines[0].message
-    expect(message.content[0].text).toBe('plain')
-    expect(message.content[1].text).toBe('boom')
-    expect(message.content[1].self).toBe('[Circular]')
-  })
-})

package/src/utils/__tests__/trim-chat-logs.test.ts

@@ -1,78 +0,0 @@
-import { describe, test, expect, beforeEach, afterEach, mock } from 'bun:test'
-import * as fs from 'fs'
-import * as os from 'os'
-import * as path from 'path'
-
-mock.module('../logger', () => ({
-  CHAT_LOG_FILENAME: 'log.jsonl',
-  logger: {
-    debug: () => {},
-    info: () => {},
-    warn: () => {},
-    error: () => {},
-    fatal: () => {},
-  },
-}))
-
-import { trimOversizedChatLogs } from '../chat-history'
-
-let tempDataDir = ''
-
-function writeLog(
-  chatId: string,
-  sizeBytes: number,
-  { ageDays = 0 }: { ageDays?: number } = {},
-) {
-  const chatDir = path.join(tempDataDir, 'chats', chatId)
-  fs.mkdirSync(chatDir, { recursive: true })
-  const logFile = path.join(chatDir, 'log.jsonl')
-  fs.writeFileSync(logFile, Buffer.alloc(sizeBytes, 'a'))
-  if (ageDays > 0) {
-    const mtime = new Date(Date.now() - ageDays * 24 * 60 * 60 * 1000)
-    fs.utimesSync(logFile, mtime, mtime)
-  }
-  return logFile
-}
-
-describe('trimOversizedChatLogs', () => {
-  beforeEach(() => {
-    tempDataDir = fs.mkdtempSync(path.join(os.tmpdir(), 'anvil-logs-'))
-  })
-
-  afterEach(() => {
-    fs.rmSync(tempDataDir, { recursive: true, force: true })
-  })
-
-  test('deletes oversized old log files and keeps small ones', () => {
-    const bigOldLog = writeLog('chat-big', 11 * 1024 * 1024, { ageDays: 15 })
-    const smallOldLog = writeLog('chat-small', 1024, { ageDays: 15 })
-    const messagesFile = path.join(
-      tempDataDir,
-      'chats',
-      'chat-big',
-      'chat-messages.json',
-    )
-    fs.writeFileSync(messagesFile, '[]')
-
-    trimOversizedChatLogs(tempDataDir)
-
-    expect(fs.existsSync(bigOldLog)).toBe(false)
-    expect(fs.existsSync(smallOldLog)).toBe(true)
-    // Chat history files are never touched
-    expect(fs.existsSync(messagesFile)).toBe(true)
-  })
-
-  test('keeps oversized log files from recent chats', () => {
-    const bigRecentLog = writeLog('chat-recent', 11 * 1024 * 1024, {
-      ageDays: 5,
-    })
-
-    trimOversizedChatLogs(tempDataDir)
-
-    expect(fs.existsSync(bigRecentLog)).toBe(true)
-  })
-
-  test('does nothing when chats directory does not exist', () => {
-    expect(() => trimOversizedChatLogs(tempDataDir)).not.toThrow()
-  })
-})

package/src/utils/__tests__/usage-banner-state.test.ts

@@ -1,298 +0,0 @@
-import { describe, test, expect } from 'bun:test'
-
-import {
-  getBannerColorLevel,
-  getThresholdInfo,
-  generateLoadingBannerText,
-  shouldAutoShowBanner,
-} from '../usage-banner-state'
-
-describe('usage-banner-state', () => {
-  describe('getThresholdInfo', () => {
-    test('returns high tier for >= 1000 credits', () => {
-      expect(getThresholdInfo(1000)).toEqual({
-        tier: 'high',
-        colorLevel: 'success',
-        threshold: 1000,
-      })
-      expect(getThresholdInfo(5000)).toEqual({
-        tier: 'high',
-        colorLevel: 'success',
-        threshold: 1000,
-      })
-    })
-
-    test('returns medium tier for 500-999 credits', () => {
-      expect(getThresholdInfo(999)).toEqual({
-        tier: 'medium',
-        colorLevel: 'warning',
-        threshold: 500,
-      })
-      expect(getThresholdInfo(500)).toEqual({
-        tier: 'medium',
-        colorLevel: 'warning',
-        threshold: 500,
-      })
-    })
-
-    test('returns low tier for 100-499 credits', () => {
-      expect(getThresholdInfo(499)).toEqual({
-        tier: 'low',
-        colorLevel: 'warning',
-        threshold: 100,
-      })
-      expect(getThresholdInfo(100)).toEqual({
-        tier: 'low',
-        colorLevel: 'warning',
-        threshold: 100,
-      })
-    })
-
-    test('returns out tier for < 100 credits', () => {
-      expect(getThresholdInfo(99)).toEqual({
-        tier: 'out',
-        colorLevel: 'error',
-        threshold: 0,
-      })
-      expect(getThresholdInfo(0)).toEqual({
-        tier: 'out',
-        colorLevel: 'error',
-        threshold: 0,
-      })
-      expect(getThresholdInfo(-50)).toEqual({
-        tier: 'out',
-        colorLevel: 'error',
-        threshold: 0,
-      })
-    })
-
-    test('returns medium tier when balance is unknown', () => {
-      expect(getThresholdInfo(null)).toEqual({
-        tier: 'medium',
-        colorLevel: 'warning',
-        threshold: 500,
-      })
-    })
-  })
-
-  describe('getBannerColorLevel', () => {
-    test('shows success for healthy credit balance (>= 1000)', () => {
-      expect(getBannerColorLevel(1000)).toBe('success')
-      expect(getBannerColorLevel(5000)).toBe('success')
-    })
-
-    test('shows warning for moderate credit balance (100-999)', () => {
-      expect(getBannerColorLevel(999)).toBe('warning')
-      expect(getBannerColorLevel(500)).toBe('warning')
-      expect(getBannerColorLevel(100)).toBe('warning')
-    })
-
-    test('shows error for low credit balance (< 100)', () => {
-      expect(getBannerColorLevel(99)).toBe('error')
-      expect(getBannerColorLevel(0)).toBe('error')
-      expect(getBannerColorLevel(-50)).toBe('error')
-    })
-
-    test('shows warning when balance is unknown', () => {
-      expect(getBannerColorLevel(null)).toBe('warning')
-    })
-  })
-
-  describe('generateLoadingBannerText', () => {
-    test('shows session usage while loading', () => {
-      const text = generateLoadingBannerText(150)
-      expect(text).toContain('150')
-    })
-
-    test('indicates loading state', () => {
-      const text = generateLoadingBannerText(0)
-      expect(text.toLowerCase()).toContain('loading')
-    })
-  })
-
-  describe('shouldAutoShowBanner', () => {
-    describe('when banner should NOT auto-show', () => {
-      test('during active AI response chain', () => {
-        const result = shouldAutoShowBanner(true, true, 50, null)
-        expect(result.shouldShow).toBe(false)
-        expect(result.newWarningThreshold).toBe(null)
-      })
-
-      test('when user is not authenticated', () => {
-        const result = shouldAutoShowBanner(false, false, 50, null)
-        expect(result.shouldShow).toBe(false)
-        expect(result.newWarningThreshold).toBe(null)
-      })
-
-      test('when balance data is unavailable', () => {
-        const result = shouldAutoShowBanner(false, true, null, null)
-        expect(result.shouldShow).toBe(false)
-        expect(result.newWarningThreshold).toBe(null)
-      })
-
-      test('when user has healthy credits (>= 1000)', () => {
-        const result = shouldAutoShowBanner(false, true, 1500, null)
-        expect(result.shouldShow).toBe(false)
-        expect(result.newWarningThreshold).toBe(null)
-      })
-
-      test('when staying within the same threshold bucket', () => {
-        // Already warned about 500, current is 400 (still in < 500 bucket and > 100)
-        const result = shouldAutoShowBanner(false, true, 400, 500)
-        expect(result.shouldShow).toBe(false)
-        expect(result.newWarningThreshold).toBe(500)
-      })
-    })
-
-    describe('when banner SHOULD auto-show', () => {
-      test('when crossing HIGH threshold (< 1000)', () => {
-        const result = shouldAutoShowBanner(false, true, 999, null)
-        expect(result.shouldShow).toBe(true)
-        expect(result.newWarningThreshold).toBe(1000)
-      })
-
-      test('when crossing MEDIUM threshold (< 500)', () => {
-        const result = shouldAutoShowBanner(false, true, 499, null)
-        expect(result.shouldShow).toBe(true)
-        expect(result.newWarningThreshold).toBe(500)
-      })
-
-      test('when crossing LOW threshold (< 100)', () => {
-        const result = shouldAutoShowBanner(false, true, 99, null)
-        expect(result.shouldShow).toBe(true)
-        expect(result.newWarningThreshold).toBe(100)
-      })
-
-      test('when crossing multiple thresholds at once (e.g. dropping huge amount)', () => {
-        // Dropping from >1000 to <100
-        const result = shouldAutoShowBanner(false, true, 50, null)
-        expect(result.shouldShow).toBe(true)
-        expect(result.newWarningThreshold).toBe(100)
-      })
-
-      test('when crossing to a lower threshold than previously warned', () => {
-        // Previously warned at 500, now dropped below 100
-        const result = shouldAutoShowBanner(false, true, 50, 500)
-        expect(result.shouldShow).toBe(true)
-        expect(result.newWarningThreshold).toBe(100)
-      })
-    })
-
-    describe('auto top-up enabled behavior', () => {
-      test('does not auto-show for auto-top-up users above 0 credits', () => {
-        // Even at low credits, auto-top-up users shouldn't see warnings
-        const result = shouldAutoShowBanner(false, true, 50, null, true)
-        expect(result.shouldShow).toBe(false)
-        expect(result.newWarningThreshold).toBe(null)
-      })
-
-      test('does not auto-show for auto-top-up users at any positive threshold', () => {
-        // At 500 credits - would normally warn
-        let result = shouldAutoShowBanner(false, true, 499, null, true)
-        expect(result.shouldShow).toBe(false)
-
-        // At 100 credits - would normally warn
-        result = shouldAutoShowBanner(false, true, 99, null, true)
-        expect(result.shouldShow).toBe(false)
-
-        // Even at 1 credit
-        result = shouldAutoShowBanner(false, true, 1, null, true)
-        expect(result.shouldShow).toBe(false)
-      })
-
-      test('DOES auto-show for auto-top-up users when truly out (0 credits)', () => {
-        const result = shouldAutoShowBanner(false, true, 0, null, true)
-        expect(result.shouldShow).toBe(true)
-        expect(result.newWarningThreshold).toBe(100)
-      })
-
-      test('DOES auto-show for auto-top-up users when in debt (negative credits)', () => {
-        const result = shouldAutoShowBanner(false, true, -50, null, true)
-        expect(result.shouldShow).toBe(true)
-        expect(result.newWarningThreshold).toBe(100)
-      })
-
-      test('non-auto-top-up users still get warnings as normal', () => {
-        // Without auto-top-up, should warn at low credits
-        const result = shouldAutoShowBanner(false, true, 50, null, false)
-        expect(result.shouldShow).toBe(true)
-        expect(result.newWarningThreshold).toBe(100)
-      })
-
-      test('defaults autoTopUpEnabled to false when omitted', () => {
-        // When autoTopUpEnabled parameter is omitted, should behave like false
-        const result = shouldAutoShowBanner(false, true, 50, null)
-        expect(result.shouldShow).toBe(true)
-        expect(result.newWarningThreshold).toBe(100)
-      })
-    })
-
-    describe('combined scenarios', () => {
-      test('chain in progress takes precedence over auto-top-up status', () => {
-        // Even with auto-top-up disabled and low credits, chain in progress blocks showing
-        const result = shouldAutoShowBanner(true, true, 50, null, false)
-        expect(result.shouldShow).toBe(false)
-      })
-
-      test('unauthenticated takes precedence over auto-top-up status', () => {
-        // Even with auto-top-up disabled and low credits, no auth token blocks showing
-        const result = shouldAutoShowBanner(false, false, 50, null, false)
-        expect(result.shouldShow).toBe(false)
-      })
-
-      test('null balance takes precedence over auto-top-up status', () => {
-        // Even with auto-top-up disabled, null balance blocks showing
-        const result = shouldAutoShowBanner(false, true, null, null, false)
-        expect(result.shouldShow).toBe(false)
-      })
-
-      test('auto-top-up user with previous warning threshold and now at 0 credits', () => {
-        // Auto-top-up user who was previously warned at 500, now at 0 - should show
-        const result = shouldAutoShowBanner(false, true, 0, 500, true)
-        expect(result.shouldShow).toBe(true)
-        expect(result.newWarningThreshold).toBe(100)
-      })
-
-      test('auto-top-up user with healthy balance clears warning state', () => {
-        // Auto-top-up user who now has healthy balance should have cleared state
-        const result = shouldAutoShowBanner(false, true, 1500, 100, true)
-        expect(result.shouldShow).toBe(false)
-        expect(result.newWarningThreshold).toBe(null)
-      })
-    })
-
-    describe('state reset behavior', () => {
-      test('clears warning state when credits return to healthy', () => {
-        const result = shouldAutoShowBanner(false, true, 1500, 100)
-        expect(result.shouldShow).toBe(false)
-        expect(result.newWarningThreshold).toBe(null)
-      })
-
-      test('re-warns after refill and subsequent drop', () => {
-        // First: warned about low credits
-        let result = shouldAutoShowBanner(false, true, 50, null)
-        expect(result.shouldShow).toBe(true)
-        expect(result.newWarningThreshold).toBe(100)
-
-        // Then: refilled
-        result = shouldAutoShowBanner(
-          false,
-          true,
-          1500,
-          result.newWarningThreshold,
-        )
-        expect(result.newWarningThreshold).toBe(null) // cleared
-
-        // Finally: dropped again - should warn again
-        result = shouldAutoShowBanner(
-          false,
-          true,
-          50,
-          result.newWarningThreshold,
-        )
-        expect(result.shouldShow).toBe(true)
-        expect(result.newWarningThreshold).toBe(100)
-      })
-    })
-  })
-})