import { describe, it, expect, afterEach } from 'vitest'
import { DatabaseSync } from 'node:sqlite'
import { migrate } from '../../src/main/db/migrate'
import { VolumeRepository } from '../../src/main/db/repositories/volume.repo'
import { ChapterRepository } from '../../src/main/db/repositories/chapter.repo'
import { KnowledgeRepository } from '../../src/main/db/repositories/knowledge.repo'
import { KnowledgeRetrievalService } from '../../src/main/services/knowledge-retrieval.service'
import type { SqliteDb } from '../../src/main/db/types'

describe('KnowledgeRetrievalService', () => {
  let db: SqliteDb
  afterEach(() => db?.close())

  it('UT-CTX-01: ranks by importance + chapter proximity + goal match', () => {
    db = new DatabaseSync(':memory:')
    migrate(db)
    const volId = new VolumeRepository(db).create('卷一', 0).id
    const chapterRepo = new ChapterRepository(db)
    const cur = chapterRepo.create(volId, '当前章', 5, '<p>x</p>')
    const near = chapterRepo.create(volId, '近章', 4, '<p>x</p>')
    chapterRepo.create(volId, '远章', 0, '<p>x</p>')
    const repo = new KnowledgeRepository(db)
    const low = repo.create({
      type: 'fact',
      title: '无关事实',
      content: '普通内容',
      importance: 2,
      status: 'approved'
    })
    const high = repo.create({
      type: 'foreshadow',
      title: '测灵石异常',
      content: '测灵石在入门考核中发热',
      importance: 5,
      status: 'approved',
      foreshadowStatus: 'buried',
      sourceChapterId: near.id
    })
    const service = new KnowledgeRetrievalService(db)
    const result = service.suggest({
      currentChapterId: cur.id,
      goalText: '测灵石入门考核',
      topN: 5
    })
    expect(result[0].entry.id).toBe(high.id)
    expect(result[0].score).toBeGreaterThan(
      result.find((r) => r.entry.id === low.id)!.score
    )
    expect(result[0].reasons.length).toBeGreaterThan(0)
  })

  it('UT-CTX-04: resolved foreshadow scores lower than buried', () => {
    db = new DatabaseSync(':memory:')
    migrate(db)
    const volId = new VolumeRepository(db).create('卷一', 0).id
    const chId = new ChapterRepository(db).create(volId, '章', 0).id
    const repo = new KnowledgeRepository(db)
    const buried = repo.create({
      type: 'foreshadow',
      title: '埋设伏笔',
      importance: 3,
      status: 'approved',
      foreshadowStatus: 'buried',
      sourceChapterId: chId
    })
    const resolved = repo.create({
      type: 'foreshadow',
      title: '已回收伏笔',
      importance: 3,
      status: 'approved',
      foreshadowStatus: 'resolved',
      sourceChapterId: chId
    })
    const service = new KnowledgeRetrievalService(db)
    const scores = service.suggest({ currentChapterId: chId, topN: 10 })
    const buriedScore = scores.find((s) => s.entry.id === buried.id)!.score
    const resolvedScore = scores.find((s) => s.entry.id === resolved.id)!.score
    expect(buriedScore).toBeGreaterThan(resolvedScore)
  })
})