import { describe, expect, it } from 'vitest'
import { headLines } from './pathAskContext'
import { PATH_ASK_HEAD_LINES, pathAskPrompt } from './pathAskPrompt'

const FILE = 'app/[[...section]]/page.tsx'
const FOLDER = 'app/[[...section]]'

describe('问路径时发给模型的话', () => {
  it('文件：写明是文件，带相对路径和开头', () => {
    const prompt = pathAskPrompt({
      kind: 'file',
      relativePath: FILE,
      head: { lines: ['export default function Page() {', '  return null', '}'], truncated: false },
    }, '这个名字是什么意思')
    expect(prompt).toContain(`这是一个文件，相对路径：${FILE}`)
    expect(prompt).toContain('1 | export default function Page() {')
    expect(prompt).toContain('3 | }')
    expect(prompt).toContain('问题：这个名字是什么意思')
    expect(prompt).toContain('只读，不要改任何东西')
    expect(prompt).not.toContain('剩下的需要时自己读')
  })

  it('文件开头超过上限时说明只贴了开头，相对路径仍在', () => {
    const lines = Array.from({ length: PATH_ASK_HEAD_LINES }, (_, index) => `line ${index}`)
    const prompt = pathAskPrompt({
      kind: 'file',
      relativePath: FILE,
      head: { lines, truncated: true },
    }, '?')
    expect(prompt).toContain(`这是一个文件，相对路径：${FILE}`)
    expect(prompt).toContain(`上面只贴了开头 ${PATH_ASK_HEAD_LINES} 行`)
    expect(prompt).toContain(`80 | line ${PATH_ASK_HEAD_LINES - 1}`)
    expect(prompt).not.toContain('line 80')
  })

  it('文件夹：写明是文件夹，带相对路径，只列这一层的名字', () => {
    const prompt = pathAskPrompt({
      kind: 'folder',
      relativePath: FOLDER,
      children: [
        { name: 'components', kind: 'folder' },
        { name: 'page.tsx', kind: 'file' },
      ],
    }, '[[...section]] 是啥意思')
    expect(prompt).toContain(`这是一个文件夹，相对路径：${FOLDER}`)
    expect(prompt).toContain('- 文件夹 components\n- 文件 page.tsx')
    expect(prompt).toContain('只列了这一层的名字')
    expect(prompt).toContain('问题：[[...section]] 是啥意思')
    expect(prompt).not.toContain('export default')
  })
})

describe('文件开头截取', () => {
  it('超过行数上限时只留前 80 行，并标明没贴完', () => {
    const text = Array.from({ length: 100 }, (_, index) => `L${index}`).join('\n')
    expect(headLines(text, false, PATH_ASK_HEAD_LINES)).toEqual({
      lines: Array.from({ length: PATH_ASK_HEAD_LINES }, (_, index) => `L${index}`),
      truncated: true,
    })
  })

  it('字节没读完时丢掉被截断的最后一行', () => {
    expect(headLines('alpha\nbeta-cut', true, PATH_ASK_HEAD_LINES)).toEqual({
      lines: ['alpha'],
      truncated: true,
    })
  })

  it('空文件没有行', () => {
    expect(headLines('', false, PATH_ASK_HEAD_LINES)).toEqual({ lines: [], truncated: false })
  })
})