import { describe, expect, it } from 'vitest'

import { fuzzyRank } from './fuzzy'
import { modelSearchText } from './model-search-text'

describe('modelSearchText', () => {
  it('keeps ordinary model ids unchanged', () => {
    expect(modelSearchText('kimi-k2.6')).toBe('kimi-k2.6')
    expect(modelSearchText('glm-5.2')).toBe('glm-5.2')
  })

  it('adds kimi aliases for the bare Kimi Coding k3 wire id', () => {
    expect(modelSearchText('k3')).toBe('k3 kimi-k3 kimi')
    expect(modelSearchText('K3')).toBe('K3 kimi-k3 kimi')
  })

  it('adds ox-alpha aliases for the Ox Alpha preview wire id', () => {
    expect(modelSearchText('x-preview-f-free')).toBe('x-preview-f-free ox-alpha ox')
    expect(modelSearchText('X-Preview-F-Free')).toBe('X-Preview-F-Free ox-alpha ox')
  })
})

describe('model picker search with aliases', () => {
  const models = ['kimi-k2.6', 'kimi-k2.5', 'k3', 'kimi-for-coding']

  it('surfaces k3 when the user searches kimi', () => {
    const ranked = fuzzyRank(models, 'kimi', modelSearchText).map(r => r.item)
    expect(ranked).toContain('k3')
  })

  it('still finds k3 by its wire id', () => {
    const ranked = fuzzyRank(models, 'k3', modelSearchText).map(r => r.item)
    expect(ranked).toEqual(['k3'])
  })

  it('does not invent k3 for unrelated queries', () => {
    const ranked = fuzzyRank(models, 'glm', modelSearchText).map(r => r.item)
    expect(ranked).toEqual([])
  })

  it('surfaces the Ox Alpha preview slug when the user searches ox', () => {
    const zenModels = ['x-preview-f-free', 'gpt-5.6-sol', 'kimi-k3']
    const ranked = fuzzyRank(zenModels, 'ox', modelSearchText).map(r => r.item)
    expect(ranked).toContain('x-preview-f-free')
    const rankedFull = fuzzyRank(zenModels, 'ox-alpha', modelSearchText).map(r => r.item)
    expect(rankedFull).toContain('x-preview-f-free')
  })
})