webnovel-writer/packages/core/tests/gate.spec.ts

import { afterAll, describe, expect, it } from 'vitest'
import * as fs from 'node:fs'
import * as path from 'node:path'
import * as os from 'node:os'
import { decidePreExecute } from '../src/gate/preExecute'
import { reportDraftReadiness } from '../src/gate/writeDraft'
import { serializeDocument } from '../src/repo/frontmatter'

const roots: string[] = []
function mkBook(): string {
  const dir = fs.mkdtempSync(path.join(os.tmpdir(), 'webnovel-gate-'))
  roots.push(dir)
  return dir
}
function put(root: string, rel: string, content: string): void {
  fs.mkdirSync(path.dirname(path.join(root, rel)), { recursive: true })
  fs.writeFileSync(path.join(root, rel), content, 'utf-8')
}
afterAll(() => { for (const r of roots) { try { fs.rmSync(r, { recursive: true, force: true, maxRetries: 10, retryDelay: 200 }) } catch { /* Windows 句柄延迟释放，%TEMP% 残留无害 */ } } })

function write(file_path: string) {
  return { name: 'write', arguments: { file_path } }
}

describe('文件门禁(不变量 4/5)', () => {
  it('允许写入草稿区/章细纲/', () => {
    const root = mkBook()
    const d = decidePreExecute(root, write(path.join(root, '草稿区/章细纲/卷01-初见.md')))
    expect(d).toEqual({ kind: 'allow' })
  })

  it('拒绝写入定稿/', () => {
    const root = mkBook()
    const d = decidePreExecute(root, write(path.join(root, '定稿/卷01/0001-初见.md')))
    expect(d.kind).toBe('deny')
    if (d.kind === 'deny') expect(d.reason).toMatch(/不变量 4/)
  })

  it('拒绝写入构想/', () => {
    const root = mkBook()
    const d = decidePreExecute(root, write(path.join(root, '构想/构想快照.md')))
    expect(d.kind).toBe('deny')
    if (d.kind === 'deny') expect(d.reason).toMatch(/不变量 5/)
  })

  it('拒绝写入大纲/(真源,非文件工具)', () => {
    const root = mkBook()
    const d = decidePreExecute(root, write(path.join(root, '大纲/故事骨架.md')))
    expect(d.kind).toBe('deny')
    if (d.kind === 'deny') expect(d.reason).toMatch(/不变量 4/)
  })

  it('路径逃逸拒绝', () => {
    const root = mkBook()
    const escaped = decidePreExecute(root, write(path.join(root, '..', '越界.md')))
    expect(escaped.kind).toBe('deny')
    if (escaped.kind === 'deny') expect(escaped.reason).toMatch(/逃逸|不变量 4/)
  })

  it('改写工具缺路径→拒绝', () => {
    const root = mkBook()
    expect(decidePreExecute(root, { name: 'write', arguments: {} }).kind).toBe('deny')
    expect(decidePreExecute(root, { name: 'edit', arguments: { path: '草稿区/x.md' } }).kind).toBe('deny')
  })
})

describe('写稿门槛(R1 落码:播报不拒绝)', () => {
  it('未确认细纲:放行写入,readiness 如实播报', () => {
    const root = mkBook()
    const rel = '草稿区/草稿/卷01-初见/稿1.md'
    const d = decidePreExecute(root, write(path.join(root, rel)))
    expect(d).toEqual({ kind: 'allow' })
    const r = reportDraftReadiness(root, rel)
    expect(r?.细纲已确认).toBe(false)
    expect(r?.说明).toMatch(/细纲未确认/)
  })

  it('材料包组装中:放行写入,readiness 播报材料包状态', () => {
    const root = mkBook()
    put(root, '大纲/卷规划/卷01/章细纲/0003-初见.md', serializeDocument({ 状态: '已确认' }, '细纲'))
    put(root, '草稿区/材料包/卷01-初见/材料清单.json', JSON.stringify({ schemaVersion: 1, 状态: '组装中' }))
    const rel = '草稿区/草稿/卷01-初见/稿1.md'
    expect(decidePreExecute(root, write(path.join(root, rel)))).toEqual({ kind: 'allow' })
    const r = reportDraftReadiness(root, rel)
    expect(r?.细纲已确认).toBe(true)
    expect(r?.材料包状态).toBe('组装中')
  })

  it('草稿路径无法解析章节键→拒绝(路径形状校验保留)', () => {
    const root = mkBook()
    const d = decidePreExecute(root, write(path.join(root, '草稿区/草稿/随意路径/稿1.md')))
    expect(d.kind).toBe('deny')
    if (d.kind === 'deny') expect(d.reason).toMatch(/章节键/)
  })

  it('确认细纲+材料包可写→允许写草稿,readiness 全绿', () => {
    const root = mkBook()
    put(root, '大纲/卷规划/卷01/章细纲/0003-初见.md', serializeDocument({ 状态: '已确认' }, '细纲'))
    put(root, '草稿区/材料包/卷01-初见/材料清单.json', JSON.stringify({ schemaVersion: 1, 状态: '可写' }))
    const rel = '草稿区/草稿/卷01-初见/稿1.md'
    const d = decidePreExecute(root, write(path.join(root, rel)))
    expect(d).toEqual({ kind: 'allow' })
    const r = reportDraftReadiness(root, rel)
    expect(r?.细纲已确认).toBe(true)
    expect(r?.材料包状态).toBe('可写')
  })
})

describe('str_replace_editor 命令分流', () => {
  it('view 允许读定稿;str_replace 拒绝', () => {
    const root = mkBook()
    const dest = path.join(root, '定稿/卷01/0001-初见.md')
    const view = decidePreExecute(root, { name: 'str_replace_editor', arguments: { command: 'view', path: dest } })
    expect(view).toEqual({ kind: 'allow' })
    const edit = decidePreExecute(root, { name: 'str_replace_editor', arguments: { command: 'str_replace', path: dest } })
    expect(edit.kind).toBe('deny')
  })
})