@alumbwe/anvil 1.0.0 → 1.0.1

package/src/utils/__tests__/chat-history.test.ts

@@ -1,203 +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', () => ({
-  logger: {
-    debug: () => {},
-    info: () => {},
-    warn: () => {},
-    error: () => {},
-    fatal: () => {},
-  },
-}))
-
-import { deleteChatSession, getAllChats } from '../chat-history'
-
-let tempDataDir = ''
-
-function writeChat(chatId: string, prompt: string) {
-  const chatDir = path.join(tempDataDir, 'chats', chatId)
-  fs.mkdirSync(chatDir, { recursive: true })
-  fs.writeFileSync(
-    path.join(chatDir, 'chat-messages.json'),
-    JSON.stringify([
-      {
-        id: `${chatId}-message`,
-        variant: 'user',
-        content: prompt,
-        timestamp: new Date().toISOString(),
-        blocks: [],
-      },
-    ]),
-  )
-}
-
-describe('chat-history', () => {
-  beforeEach(() => {
-    tempDataDir = fs.mkdtempSync(path.join(os.tmpdir(), 'anvil-history-'))
-  })
-
-  afterEach(() => {
-    fs.rmSync(tempDataDir, { recursive: true, force: true })
-  })
-
-  test('deleteChatSession removes a saved chat directory', () => {
-    writeChat('chat-a', 'hello from chat a')
-    writeChat('chat-b', 'hello from chat b')
-
-    expect(deleteChatSession('chat-a', tempDataDir)).toBe(true)
-
-    expect(fs.existsSync(path.join(tempDataDir, 'chats', 'chat-a'))).toBe(false)
-    expect(fs.existsSync(path.join(tempDataDir, 'chats', 'chat-b'))).toBe(true)
-    expect(
-      getAllChats(500, tempDataDir).map((chat) => chat.chatId),
-    ).toEqual(['chat-b'])
-  })
-
-  test('deleteChatSession rejects invalid chat ids', () => {
-    const outsideDir = path.join(tempDataDir, 'outside')
-    fs.mkdirSync(outsideDir, { recursive: true })
-
-    expect(deleteChatSession('../outside', tempDataDir)).toBe(false)
-    expect(deleteChatSession('..', tempDataDir)).toBe(false)
-
-    expect(fs.existsSync(outsideDir)).toBe(true)
-  })
-
-  test('deleteChatSession returns false when the chat does not exist', () => {
-    expect(deleteChatSession('missing-chat', tempDataDir)).toBe(false)
-  })
-
-  test('getAllChats lists corrupt chats as unreadable instead of hiding them', () => {
-    writeChat('chat-good', 'hello from a healthy chat')
-
-    // Simulate a chat-messages.json truncated by a crash mid-write
-    const corruptDir = path.join(tempDataDir, 'chats', 'chat-corrupt')
-    fs.mkdirSync(corruptDir, { recursive: true })
-    fs.writeFileSync(
-      path.join(corruptDir, 'chat-messages.json'),
-      '[{"id":"msg-1","variant":"user","content":"truncat',
-    )
-
-    const chats = getAllChats(500, tempDataDir)
-
-    const good = chats.find((chat) => chat.chatId === 'chat-good')
-    expect(good).toBeDefined()
-    expect(good?.unreadable).toBeUndefined()
-    expect(good?.lastPrompt).toBe('hello from a healthy chat')
-
-    const corrupt = chats.find((chat) => chat.chatId === 'chat-corrupt')
-    expect(corrupt).toBeDefined()
-    expect(corrupt?.unreadable).toBe(true)
-    expect(corrupt?.lastPrompt).toBe('(unreadable chat)')
-  })
-
-  test('getAllChats lists non-array chat-messages.json as unreadable', () => {
-    const badDir = path.join(tempDataDir, 'chats', 'chat-not-array')
-    fs.mkdirSync(badDir, { recursive: true })
-    fs.writeFileSync(
-      path.join(badDir, 'chat-messages.json'),
-      '{"not":"an array"}',
-    )
-
-    const chats = getAllChats(500, tempDataDir)
-
-    expect(chats).toHaveLength(1)
-    expect(chats[0].chatId).toBe('chat-not-array')
-    expect(chats[0].unreadable).toBe(true)
-  })
-
-  test('getAllChats prefers the chat-meta.json sidecar over parsing messages', () => {
-    writeChat('chat-with-meta', 'prompt from messages file')
-    const chatDir = path.join(tempDataDir, 'chats', 'chat-with-meta')
-    const stats = fs.statSync(path.join(chatDir, 'chat-messages.json'))
-    fs.writeFileSync(
-      path.join(chatDir, 'chat-meta.json'),
-      JSON.stringify({
-        messageCount: 42,
-        firstPrompt: 'prompt from meta',
-        messagesSize: stats.size,
-        messagesMtimeMs: stats.mtimeMs,
-      }),
-    )
-
-    const chats = getAllChats(500, tempDataDir)
-
-    expect(chats).toHaveLength(1)
-    expect(chats[0].lastPrompt).toBe('prompt from meta')
-    expect(chats[0].messageCount).toBe(42)
-  })
-
-  test('getAllChats falls back to parsing messages when meta is stale', () => {
-    writeChat('chat-stale-meta', 'prompt from messages file')
-    const chatDir = path.join(tempDataDir, 'chats', 'chat-stale-meta')
-    // Sidecar recorded before the messages file was last rewritten: its
-    // size/mtime no longer match, so it must be ignored
-    fs.writeFileSync(
-      path.join(chatDir, 'chat-meta.json'),
-      JSON.stringify({
-        messageCount: 42,
-        firstPrompt: 'stale prompt from meta',
-        messagesSize: 1,
-        messagesMtimeMs: 1,
-      }),
-    )
-
-    const chats = getAllChats(500, tempDataDir)
-
-    expect(chats).toHaveLength(1)
-    expect(chats[0].lastPrompt).toBe('prompt from messages file')
-    expect(chats[0].messageCount).toBe(1)
-  })
-
-  test('getAllChats still flags a chat unreadable when messages are corrupted after the meta was written', () => {
-    writeChat('chat-corrupt-after-meta', 'healthy prompt')
-    const chatDir = path.join(tempDataDir, 'chats', 'chat-corrupt-after-meta')
-    const messagesPath = path.join(chatDir, 'chat-messages.json')
-    const stats = fs.statSync(messagesPath)
-    fs.writeFileSync(
-      path.join(chatDir, 'chat-meta.json'),
-      JSON.stringify({
-        messageCount: 1,
-        firstPrompt: 'healthy prompt',
-        messagesSize: stats.size,
-        messagesMtimeMs: stats.mtimeMs,
-      }),
-    )
-    // Transcript truncated later (crash mid-write by an older CLI, disk
-    // issue): the sidecar no longer matches, so the corruption must not be
-    // masked by it
-    fs.writeFileSync(messagesPath, '[{"id":"msg-1","variant":"user","con')
-
-    const chats = getAllChats(500, tempDataDir)
-
-    expect(chats).toHaveLength(1)
-    expect(chats[0].unreadable).toBe(true)
-    expect(chats[0].lastPrompt).toBe('(unreadable chat)')
-  })
-
-  test('getAllChats falls back to parsing messages when meta is corrupt', () => {
-    writeChat('chat-bad-meta', 'prompt from messages file')
-    fs.writeFileSync(
-      path.join(tempDataDir, 'chats', 'chat-bad-meta', 'chat-meta.json'),
-      '{"messageCount": tru',
-    )
-
-    const chats = getAllChats(500, tempDataDir)
-
-    expect(chats).toHaveLength(1)
-    expect(chats[0].lastPrompt).toBe('prompt from messages file')
-    expect(chats[0].messageCount).toBe(1)
-    expect(chats[0].unreadable).toBeUndefined()
-  })
-
-  test('getAllChats still hides empty chats', () => {
-    const emptyDir = path.join(tempDataDir, 'chats', 'chat-empty')
-    fs.mkdirSync(emptyDir, { recursive: true })
-    fs.writeFileSync(path.join(emptyDir, 'chat-messages.json'), '[]')
-
-    expect(getAllChats(500, tempDataDir)).toHaveLength(0)
-  })
-})

package/src/utils/__tests__/chat-input-key-intercept.test.ts

@@ -1,40 +0,0 @@
-import { describe, expect, test } from 'bun:test'
-
-import { shouldInterceptChatInputKey } from '../chat-input-key-intercept'
-
-const baseState = {
-  hasSlashSuggestions: false,
-  hasMentionSuggestions: false,
-  lastEditDueToNav: false,
-  cursorPosition: 1,
-  inputLength: 3,
-}
-
-describe('shouldInterceptChatInputKey', () => {
-  test('intercepts keypad Enter while slash suggestions are visible', () => {
-    expect(
-      shouldInterceptChatInputKey(
-        { name: 'kpenter', sequence: '\x1b[57414u' },
-        { ...baseState, hasSlashSuggestions: true },
-      ),
-    ).toBe(true)
-  })
-
-  test('intercepts raw application keypad Enter while mention suggestions are visible', () => {
-    expect(
-      shouldInterceptChatInputKey(
-        { sequence: '\x1bOM' },
-        { ...baseState, hasMentionSuggestions: true },
-      ),
-    ).toBe(true)
-  })
-
-  test('does not intercept keypad Enter without visible suggestions', () => {
-    expect(
-      shouldInterceptChatInputKey(
-        { name: 'kpenter', sequence: '\x1b[57414u' },
-        baseState,
-      ),
-    ).toBe(false)
-  })
-})

package/src/utils/__tests__/chat-meta.test.ts

@@ -1,130 +0,0 @@
-import { describe, test, expect, beforeEach, afterEach } from 'bun:test'
-import * as fs from 'fs'
-import * as os from 'os'
-import * as path from 'path'
-
-import {
-  CHAT_MESSAGES_FILENAME,
-  CHAT_META_FILENAME,
-  getFirstUserPrompt,
-  readChatMeta,
-  writeChatMeta,
-} from '../chat-meta'
-import {
-  clearChatState,
-  saveChatState,
-  setChatDirOverrideForTesting,
-} from '../run-state-storage'
-
-import type { ChatMessage } from '../../types/chat'
-import type { RunState } from '@anvil/sdk'
-
-let chatDir = ''
-
-function userMessage(content: string): ChatMessage {
-  return {
-    id: `msg-${content}`,
-    variant: 'user',
-    content,
-    timestamp: new Date().toISOString(),
-    blocks: [],
-  }
-}
-
-/** Write chat-messages.json so writeChatMeta can stat it. */
-function writeMessagesFile(messages: ChatMessage[]): void {
-  fs.writeFileSync(
-    path.join(chatDir, CHAT_MESSAGES_FILENAME),
-    JSON.stringify(messages),
-  )
-}
-
-describe('chat-meta', () => {
-  beforeEach(() => {
-    chatDir = fs.mkdtempSync(path.join(os.tmpdir(), 'anvil-meta-'))
-    setChatDirOverrideForTesting(chatDir)
-  })
-
-  afterEach(() => {
-    setChatDirOverrideForTesting(undefined)
-    fs.rmSync(chatDir, { recursive: true, force: true })
-  })
-
-  test('getFirstUserPrompt returns first user message, truncated', () => {
-    const messages: ChatMessage[] = [
-      { ...userMessage(''), variant: 'agent' },
-      userMessage('hello there'),
-      userMessage('second prompt'),
-    ]
-    expect(getFirstUserPrompt(messages)).toBe('hello there')
-
-    const long = 'x'.repeat(150)
-    expect(getFirstUserPrompt([userMessage(long)])).toBe(
-      'x'.repeat(97) + '...',
-    )
-    expect(getFirstUserPrompt([])).toBe('(empty chat)')
-  })
-
-  test('writeChatMeta / readChatMeta round trip', () => {
-    const messages = [userMessage('round trip prompt'), userMessage('two')]
-    writeMessagesFile(messages)
-    writeChatMeta(chatDir, messages)
-
-    expect(readChatMeta(chatDir)).toMatchObject({
-      messageCount: 2,
-      firstPrompt: 'round trip prompt',
-    })
-  })
-
-  test('readChatMeta returns null for missing or invalid meta', () => {
-    writeMessagesFile([userMessage('hi')])
-    expect(readChatMeta(chatDir)).toBeNull()
-
-    const metaPath = path.join(chatDir, CHAT_META_FILENAME)
-    fs.writeFileSync(metaPath, '{"messageCount": "not a nu')
-    expect(readChatMeta(chatDir)).toBeNull()
-
-    fs.writeFileSync(metaPath, JSON.stringify({ messageCount: 3 }))
-    expect(readChatMeta(chatDir)).toBeNull()
-  })
-
-  test('readChatMeta returns null when chat-messages.json no longer matches', () => {
-    const messages = [userMessage('original prompt')]
-    writeMessagesFile(messages)
-    writeChatMeta(chatDir, messages)
-    expect(readChatMeta(chatDir)).not.toBeNull()
-
-    // Simulate another writer (e.g. an older CLI) rewriting the transcript
-    // without refreshing the sidecar
-    writeMessagesFile([
-      userMessage('original prompt'),
-      userMessage('a new message the sidecar does not know about'),
-    ])
-    expect(readChatMeta(chatDir)).toBeNull()
-
-    // Messages file deleted entirely
-    fs.rmSync(path.join(chatDir, CHAT_MESSAGES_FILENAME))
-    expect(readChatMeta(chatDir)).toBeNull()
-  })
-
-  test('saveChatState writes the meta sidecar and clearChatState removes it', () => {
-    const runState = { output: undefined } as unknown as RunState
-    const messages = [userMessage('saved prompt')]
-
-    saveChatState(runState, messages)
-
-    const metaPath = path.join(chatDir, CHAT_META_FILENAME)
-    expect(fs.existsSync(metaPath)).toBe(true)
-    expect(readChatMeta(chatDir)).toMatchObject({
-      messageCount: 1,
-      firstPrompt: 'saved prompt',
-    })
-
-    clearChatState()
-    expect(fs.existsSync(metaPath)).toBe(false)
-    expect(fs.existsSync(path.join(chatDir, 'run-state.json'))).toBe(false)
-    expect(fs.existsSync(path.join(chatDir, CHAT_MESSAGES_FILENAME))).toBe(
-      false,
-    )
-  })
-})