import { test } from 'node:test'
import assert from 'node:assert/strict'
import path from 'node:path'
import { promises as fs } from 'node:fs'
import {
  readBatch,
  stagedFacts,
  stageChapter,
  judgeStop,
  judgeBatchQuality,
  rejectFrom,
  restageReview,
  discardBatch,
  章状态,
} from '../../src/staging/index.js'
import { repoCtx } from '../commands/_helper.js'

// —— fixture：已有 2 章定稿的书，批次从第 3 章起 ——

const 定稿章 = (num) =>
  [
    '---',
    `章号: ${num}`,
    `标题: 第${num}章`,
    '卷: 1',
    `书内时间: 1023春月初${num}`,
    '字数: 100',
    '章定位: 推进',
    '钩子: 危机钩-强',
    '情绪定位: 铺垫',
    '---',
    '',
    `第${num}章正文。`,
  ].join('\n')

function bookFiles({ 批次大小 = 3, 卷纲 = true } = {}) {
  const files = {
    'book.yaml': `spec_version: "7.0"\n书名: 连写测试\n类型: 玄幻\n每章目标字数: 3000\n卷规模: 40\n连写批次大小: ${批次大小}\n`,
    '定稿/正文/0001-第1章.md': 定稿章(1),
    '定稿/正文/0002-第2章.md': 定稿章(2),
  }
  if (卷纲) files['大纲/卷纲/第01卷.md'] = '# 第01卷\n第三章到第五章：追查黑影。\n'
  return files
}

const 审稿单 = (num) => `# 第 ${num} 章审稿单\n\n> 完整两审模式（事实审查/编辑审各自独立上下文）。\n> 共 0 个问题：0 阻断。\n`

function mkPayload(num, { 标题 = `第${num}章`, 钩子 = '危机钩-强', 书内时间 = `1023夏月初${num}`, 伏笔 = null, 收卷 = null, body = null } = {}) {
  const frontMatter = {
    章号: num,
    标题,
    卷: 1,
    视角: '林晚',
    字数: 100,
    章定位: '推进',
    钩子,
    情绪定位: '铺垫',
  }
  if (书内时间) frontMatter.书内时间 = 书内时间
  if (伏笔) frontMatter.伏笔 = 伏笔
  if (收卷) frontMatter.收卷 = 收卷
  return {
    frontMatter,
    body: body ?? `林晚在第${num}章继续追查，线索一路指向后山。夜色渐深，她的脚步没有停。`,
    summary: `第${num}章摘要。`,
    commitLines: {},
    workspaceFiles: [],
  }
}

async function stage(ctx, num, opts) {
  await fs.mkdir(path.join(ctx.repoPath, '工作区'), { recursive: true })
  await fs.writeFile(path.join(ctx.repoPath, '工作区', '审稿.md'), 审稿单(num), 'utf8')
  return stageChapter(ctx, { chapterNum: num, payload: mkPayload(num, opts) })
}

test('stage 首章：三件套落位、meta 记录、工作区清、停止未命中', async () => {
  const { ctx, cleanup } = await repoCtx(null, bookFiles())
  try {
    const r = await stage(ctx, 3, { 伏笔: ['推进 伏笔-001'] })
    assert.equal(r.ok, true, r.error)
    assert.equal(r.停止.stop, false, JSON.stringify(r.停止))

    const dir = path.join(ctx.repoPath, '工作区', '待定稿', '0003-第3章')
    for (const f of ['草稿.md', '定稿包.json', '审稿单.md']) {
      await fs.access(path.join(dir, f))
    }
    const batch = await readBatch(ctx.repoPath)
    assert.equal(batch.exists, true)
    assert.deepEqual(
      batch.章列表.map((x) => [x.章号, x.状态]),
      [[3, 章状态.待审收]]
    )
    // 本章工作区文件已清（审稿单搬进批次目录）
    await assert.rejects(() => fs.access(path.join(ctx.repoPath, '工作区', '审稿.md')))
  } finally {
    await cleanup()
  }
})

test('stage 跳章拒绝、重章覆盖（改标题清旧目录）', async () => {
  const { ctx, cleanup } = await repoCtx(null, bookFiles())
  try {
    const skip = await stage(ctx, 5)
    assert.equal(skip.ok, false)
    assert.match(skip.error, /连续.*第 3 章/)

    assert.equal((await stage(ctx, 3)).ok, true)
    const again = await stage(ctx, 3, { 标题: '改名章' })
    assert.equal(again.ok, true, again.error)
    const batch = await readBatch(ctx.repoPath)
    assert.equal(batch.章列表.length, 1)
    assert.equal(batch.章列表[0].标题, '改名章')
    await assert.rejects(() => fs.access(path.join(ctx.repoPath, '工作区', '待定稿', '0003-第3章')))
  } finally {
    await cleanup()
  }
})

test('stage 无审稿单拒绝，零写入', async () => {
  const { ctx, cleanup } = await repoCtx(null, bookFiles())
  try {
    const r = await stageChapter(ctx, { chapterNum: 3, payload: mkPayload(3) })
    assert.equal(r.ok, false)
    assert.match(r.error, /审稿/)
    await assert.rejects(() => fs.access(path.join(ctx.repoPath, '工作区', '待定稿')))
  } finally {
    await cleanup()
  }
})

test('停止条件：批次写满', async () => {
  const { ctx, cleanup } = await repoCtx(null, bookFiles({ 批次大小: 3 }))
  try {
    assert.equal((await stage(ctx, 3, { 伏笔: ['推进 伏笔-001'] })).停止.stop, false)
    assert.equal((await stage(ctx, 4, { 伏笔: ['推进 伏笔-001'] })).停止.stop, false)
    const r = await stage(ctx, 5, { 伏笔: ['推进 伏笔-001'] })
    assert.equal(r.停止.stop, true)
    assert.ok(r.停止.reasons.some((x) => x.includes('写满')), JSON.stringify(r.停止))
  } finally {
    await cleanup()
  }
})

test('停止条件：收卷声明终止批次', async () => {
  const { ctx, cleanup } = await repoCtx(null, bookFiles({ 批次大小: 8 }))
  try {
    const r = await stage(ctx, 3, { 收卷: '是', 伏笔: ['推进 伏笔-001'] })
    assert.equal(r.停止.stop, true)
    assert.ok(r.停止.reasons.some((x) => x.includes('收卷')), JSON.stringify(r.停止))
  } finally {
    await cleanup()
  }
})

test('停止条件：卷纲耗尽（当前卷无卷纲文件）', async () => {
  const { ctx, cleanup } = await repoCtx(null, bookFiles({ 批次大小: 8, 卷纲: false }))
  try {
    const r = await stage(ctx, 3, { 伏笔: ['推进 伏笔-001'] })
    assert.equal(r.停止.stop, true)
    assert.ok(r.停止.reasons.some((x) => x.includes('卷纲')), JSON.stringify(r.停止))
  } finally {
    await cleanup()
  }
})

test('停止条件：连续 3 章无条目变动（默认上限），有变动章重置计数', async () => {
  const { ctx, cleanup } = await repoCtx(null, bookFiles({ 批次大小: 8 }))
  try {
    assert.equal((await stage(ctx, 3)).停止.stop, false)
    assert.equal((await stage(ctx, 4)).停止.stop, false)
    const r5 = await stage(ctx, 5)
    assert.equal(r5.停止.stop, true)
    assert.ok(r5.停止.reasons.some((x) => x.includes('无条目变动')), JSON.stringify(r5.停止))
    // 第 5 章重暂存并带条目变动 → 尾部计数归零，不再停
    const r5b = await stage(ctx, 5, { 伏笔: ['推进 伏笔-001'] })
    assert.equal(r5b.停止.stop, false, JSON.stringify(r5b.停止))
  } finally {
    await cleanup()
  }
})

// —— 批次质检（纯函数）——

const 质检章 = (num, { 钩子 = '危机钩-强', 书内时间 = 'x', body = '正文。' } = {}) => ({
  章号: num,
  frontMatter: { 钩子, 书内时间 },
  body,
})

test('批次质检：弱钩尾达上限', async () => {
  const staged = [1, 2, 3].map((n) => 质检章(n, { 钩子: '悬念钩-弱' }))
  const q = judgeBatchQuality(staged, null, { 连续弱钩上限: 3 })
  assert.equal(q.过线, false)
  assert.ok(q.原因.some((x) => x.includes('弱钩')), JSON.stringify(q))
  // 尾部只有 2 章弱钩 → 过线
  const staged2 = [质检章(1), 质检章(2, { 钩子: '悬念钩-弱' }), 质检章(3, { 钩子: '悬念钩-弱' })]
  assert.equal(judgeBatchQuality(staged2, null, { 连续弱钩上限: 3 }).过线, true)
})

test('批次质检：缺书内时间列出章号', async () => {
  const q = judgeBatchQuality([质检章(3), 质检章(4, { 书内时间: '' })], null, {})
  assert.equal(q.过线, false)
  assert.ok(q.原因.some((x) => x.includes('第 4 章') && x.includes('书内时间')), JSON.stringify(q))
})

test('批次质检：句式 vs 基线 29% 不停 31% 停；无基线跳过', async () => {
  const 字池 =
    '天地玄黄宇宙洪荒日月盈昃辰宿列张寒来暑往秋收冬藏闰余成岁律吕调阳云腾致雨露结为霜金生丽水玉出昆冈剑号巨阙珠称夜光果珍李柰菜重芥姜海咸河淡鳞潜羽翔龙师火帝鸟官人皇始制文字乃服衣裳推位让国有虞陶唐吊民伐罪周发殷汤坐朝问道垂拱平章爱育黎首臣伏戎羌遐迩一体率宾归王鸣凤在竹白驹食场'
  const bodyOf = (lengths) => {
    let pos = 0
    const parts = []
    for (const n of lengths) {
      parts.push(字池.slice(pos, pos + n))
      pos += n
    }
    return parts.join('。') + '。'
  }
  const base = { avg_sentence_length: 10, sentence_length_variance: 0 }
  const at29 = [质检章(3, { body: bodyOf([12, 13, 13, 13, 13, 13, 13, 13, 13, 13]) })]
  assert.equal(judgeBatchQuality(at29, base, {}).过线, true, '偏 29% 不停')
  const at31 = [质检章(3, { body: bodyOf([13, 13, 13, 13, 13, 13, 13, 13, 13, 14]) })]
  const q = judgeBatchQuality(at31, base, {})
  assert.equal(q.过线, false, '偏 31% 停')
  assert.ok(q.原因.some((x) => x.includes('平均句长')), JSON.stringify(q))
  assert.equal(judgeBatchQuality(at31, null, {}).过线, true, '无基线句式判据跳过')
})

// —— 打回 / 重审 / 丢弃 / 对账 ——

test('rejectFrom：K 打回清工件，K+1..N 受影响；restageReview 回待审收', async () => {
  const { ctx, cleanup } = await repoCtx(null, bookFiles({ 批次大小: 8 }))
  try {
    for (const n of [3, 4, 5]) await stage(ctx, n, { 伏笔: ['推进 伏笔-001'] })
    const r = await rejectFrom(ctx.repoPath, 4)
    assert.equal(r.ok, true, r.error)
    assert.deepEqual(r.受影响, [5])
    const batch = await readBatch(ctx.repoPath)
    assert.deepEqual(
      batch.章列表.map((x) => [x.章号, x.状态]),
      [
        [3, 章状态.待审收],
        [4, 章状态.打回],
        [5, 章状态.受影响],
      ]
    )
    // 打回章工件已清
    await assert.rejects(() =>
      fs.access(path.join(ctx.repoPath, '工作区', '待定稿', '0004-第4章', '草稿.md'))
    )
    // 打回章不能只重审
    const bad = await restageReview(ctx.repoPath, 4)
    assert.equal(bad.ok, false)
    assert.match(bad.error, /重写/)
    // 受影响章重审：新审稿单 → 待审收
    await fs.writeFile(path.join(ctx.repoPath, '工作区', '审稿.md'), 审稿单(5), 'utf8')
    const ok5 = await restageReview(ctx.repoPath, 5)
    assert.equal(ok5.ok, true, ok5.error)
    assert.equal((await readBatch(ctx.repoPath)).章列表.find((x) => x.章号 === 5).状态, 章状态.待审收)
  } finally {
    await cleanup()
  }
})

test('批次.json 损坏 → 按目录重建，保守标记受影响', async () => {
  const { ctx, cleanup } = await repoCtx(null, bookFiles())
  try {
    await stage(ctx, 3, { 伏笔: ['推进 伏笔-001'] })
    await fs.writeFile(path.join(ctx.repoPath, '工作区', '待定稿', '批次.json'), '{{{', 'utf8')
    const batch = await readBatch(ctx.repoPath)
    assert.equal(batch.exists, true)
    assert.equal(batch.章列表[0].状态, 章状态.受影响)
    assert.ok(batch.warnings.some((w) => w.includes('批次.json')), JSON.stringify(batch.warnings))
  } finally {
    await cleanup()
  }
})

test('discardBatch：整批丢弃，工作区批次消失', async () => {
  const { ctx, cleanup } = await repoCtx(null, bookFiles())
  try {
    await stage(ctx, 3)
    const r = await discardBatch(ctx.repoPath)
    assert.equal(r.ok, true)
    assert.equal(r.章数, 1)
    assert.equal((await readBatch(ctx.repoPath)).exists, false)
    await assert.rejects(() => fs.access(path.join(ctx.repoPath, '工作区', '待定稿')))
  } finally {
    await cleanup()
  }
})

test('stagedFacts：声明与定稿包合并出条目/名册/时间线/信息差事实', async () => {
  const { ctx, cleanup } = await repoCtx(null, bookFiles({ 批次大小: 8 }))
  try {
    await fs.mkdir(path.join(ctx.repoPath, '工作区'), { recursive: true })
    await fs.writeFile(path.join(ctx.repoPath, '工作区', '审稿.md'), 审稿单(3), 'utf8')
    const p = mkPayload(3, { 伏笔: ['埋下 伏笔-009'] })
    p.threadCreates = [
      { id: '伏笔-009', 短题: '古钟', frontMatter: { 强度: '中', 状态: '进行', 开启章: 3 }, body: '## 描述\n古钟。\n' },
    ]
    p.rosterUpserts = [{ 正名: '古钟长老', 别名: '钟叟, 老钟', 类型: 'character', 首现章: 3 }]
    p.characterUpdates = [{ name: '林晚', updates: { 最后变更章: 3 } }]
    p.timelineRows = [{ volumeNum: 1, row: { 章: 3, 书内时间: '夏月初三', 一句话事件: '闻古钟', 在场: '林晚' } }]
    p.secretWrites = [{ id: '信息差-002-钟声', frontMatter: { 读者已知: false, 登记章: 3, 关键词: ['古钟'] }, content: '## 内容\n钟声有古怪。\n' }]
    const r = await stageChapter(ctx, { chapterNum: 3, payload: p })
    assert.equal(r.ok, true, r.error)

    const facts = await stagedFacts(ctx.repoPath)
    assert.equal(facts.exists, true)
    assert.equal(facts.chapters.length, 1)
    const t = facts.threads.get('伏笔-009')
    assert.ok(t, '声明+定稿包都指向的新条目必须在')
    assert.equal(t.新开, true)
    assert.equal(t.状态, '进行')
    assert.ok(facts.newEntities.has('古钟长老'))
    assert.ok(facts.newAliases.has('钟叟') && facts.newAliases.has('老钟'))
    assert.deepEqual(facts.characterUpdates.get('林晚'), { 最后变更章: 3 })
    assert.equal(facts.timelineRows.length, 1)
    assert.equal(facts.secretWrites.length, 1)
    assert.equal(facts.总字数, 100)
  } finally {
    await cleanup()
  }
})