import { describe, expect, test } from 'claude-code/testing'

import { formatExamples } from './examples'
import { retrieve, type IndexEntry } from './retrieval'

const e = (id: string, fn: IndexEntry['function'], mode: string | null = null): IndexEntry => ({
  id,
  function: fn,
  emotion: 'neutral',
  mode,
  text: `句${id}`,
})
const INDEX = [e('c', 'comfort'), e('a', 'comfort'), e('b', 'comfort'), e('d', 'tease'), e('x', 'comfort', 'leafa')]

describe('retrieve', () => {
  test('只取同功能標籤，依 id 排序', () => {
    expect(retrieve(INDEX, 'comfort', { k: 10 }).map(r => r.id)).toEqual(['a', 'b', 'c', 'x'])
  })

  test('預設取 3 句', () => {
    expect(retrieve(INDEX, 'comfort')).toHaveLength(3)
  })

  test('沒有同標籤的對白回傳空陣列', () => {
    expect(retrieve(INDEX, 'greet')).toEqual([])
  })

  test('依模式過濾', () => {
    expect(retrieve(INDEX, 'comfort', { mode: 'leafa' }).map(r => r.id)).toEqual(['x'])
  })

  test('同一輪結果固定，不同輪輪流取', () => {
    const ids = (turn: number) => retrieve(INDEX, 'comfort', { k: 2, turn }).map(r => r.id)
    expect(ids(0)).toEqual(ids(0))
    expect(ids(0)).toEqual(['a', 'b'])
    expect(ids(1)).toEqual(['b', 'c'])
    expect(ids(3)).toEqual(['x', 'a'])
  })

  test('k 大於池子時不重複', () => {
    expect(new Set(retrieve(INDEX, 'tease', { k: 5 }).map(r => r.id)).size).toBe(1)
  })
})

describe('formatExamples', () => {
  test('沒有對白不注入', () => {
    expect(formatExamples('亞絲娜', [])).toBeNull()
  })

  test('列出對白並說明只學語氣', () => {
    const text = formatExamples('亞絲娜', [e('a', 'comfort')])!
    expect(text).toContain('句a')
    expect(text).toContain('不要照抄')
    expect(text).toContain('亞絲娜')
  })

  test('過長的句子截斷', () => {
    const long = { ...e('a', 'comfort'), text: 'あ'.repeat(200) }
    expect(formatExamples('亞絲娜', [long])!.length).toBeLessThan(300)
  })
})