243 lines
13 KiB
TypeScript
243 lines
13 KiB
TypeScript
import { expect, mock, test } from 'claude-code/testing'
|
||
|
||
import { parseWikiPages, parseWikiSource } from './extract'
|
||
import type { Extracted } from './extract'
|
||
|
||
const char = (over: Record<string, unknown> = {}, tone = '溫柔'): Extracted['characters'][number] =>
|
||
({
|
||
name: '小葵', nameEn: 'Aoi', slug: 'aoi',
|
||
persona: { selfRef: '我', address: '你', tone, catchphrase: '加油', values: '誠實', taboos: '背叛' },
|
||
axes: { gentle: 7, tsundere: 2, lively: 5, formal: 4, talkative: 5, humor: 3, proactive: 6, expressive: 5 },
|
||
anime: { moe: '認真', speech: '柔和', emotions: '會臉紅', kaomoji: '(^_^)', themeColor: '#88c' },
|
||
state: { mood: '緊張', dialogueStyle: '客氣', monologue: '想變強', relationships: [{ with: '阿零', kind: 'friend', note: '同伴' }], age: 16, place: '學院' },
|
||
...over,
|
||
}) as never
|
||
|
||
const reply = (c: Extracted['characters'][number]) =>
|
||
JSON.stringify({ work: { name: '測試物語', nameEn: 'Test Story', slug: 'ts', timeline: [{ time: '春', place: '學院' }], terms: [{ term: '魔核', definition: '力量來源' }] }, characters: [c] })
|
||
|
||
type Setup = { files: Map<string, string>; cmds: string[][]; prompts: string[] }
|
||
|
||
let cloneMissing = false
|
||
let registered: string[] = []
|
||
const probed = new Set<string>()
|
||
let sessionCwd = '/repo'
|
||
const web = new Map<string, string>()
|
||
const setup = (on: never, answers: string[], files = new Map<string, string>()): Setup => {
|
||
const s: Setup = { files, cmds: [], prompts: [] }
|
||
const o = on as unknown as (n: string, h: (...a: never[]) => unknown) => void
|
||
mock.store(on)
|
||
o('command.register', ((_$: unknown, e: { name: string }) => { registered.push(e.name); return { value: { name: e.name } } }) as never)
|
||
o('session.start', () => ({ cwd: '/repo' }))
|
||
o('session.cwd', (() => ({ value: sessionCwd })) as never)
|
||
mock.env(on, { HOME: '/home/u' })
|
||
o('fs.exists', ((_$: unknown, e: { path: string }) => (e.path.endsWith('/characters/index.json') && probed.add(e.path), { value: s.files.has(e.path) || (e.path === '/home/u/.cache/jsc-anima/repo/.git' && !cloneMissing) })) as never)
|
||
o('fs.read', ((_$: unknown, e: { path: string }) => ({ value: s.files.get(e.path) ?? '' })) as never)
|
||
o('fs.write', ((_$: unknown, e: { path: string; text: string }) => { s.files.set(e.path, e.text); return { value: undefined } }) as never)
|
||
o('model.complete', ((_$: unknown, e: { prompt: string }) => {
|
||
s.prompts.push(e.prompt)
|
||
return { value: { isAnswered: true, text: answers.shift() ?? '', usage: {} } }
|
||
}) as never)
|
||
o('process.run', ((_$: unknown, e: { argv: string[]; }) => {
|
||
s.cmds.push(e.argv)
|
||
const j = e.argv.join(' ')
|
||
if (e.argv[0] === 'curl') {
|
||
const body = web.get(e.argv[e.argv.length - 1]!)
|
||
return { value: { exitCode: body === undefined ? 22 : 0, stdout: body ?? '', stderr: '' } }
|
||
}
|
||
const out = j.includes('--abbrev-ref') ? 'issue-4' : j.includes('symbolic-ref') ? 'origin/main' : j.includes('get-url') ? 'https://gitea.jsc.idv.tw/jiantw83/jsc-anima.git' : j.startsWith('tea logins') ? 'admin https://gitea.jsc.idv.tw u' : j.startsWith('tea pulls') ? 'PR#1' : ''
|
||
return { value: { exitCode: 0, stdout: out, stderr: '' } }
|
||
}) as never)
|
||
return s
|
||
}
|
||
|
||
const run = ($: never, args: string) => ($ as any).command.run({ command: 'anima-extract', args }).then((r: { text: string }) => r.text)
|
||
|
||
test('萃取：先確認代號、再看變更摘要、最後寫檔並建立 PR', async ($, on) => {
|
||
const s = setup(on as never, [reply(char())], new Map([['/repo/ch1.txt', '第一段\n\n第二段']]))
|
||
await $.session.start({ source: 'startup', cwd: '/repo' } as never)
|
||
const t1 = await run($ as never, 'ch1.txt')
|
||
expect(t1).toContain('ts')
|
||
expect(t1).toContain('aoi')
|
||
expect(t1).toContain('尚未建立')
|
||
expect(s.files.has('/home/u/.cache/jsc-anima/repo/characters/ts/aoi.json')).toBe(false)
|
||
|
||
const t2 = await run($ as never, 'confirm work=testy aoi=ao')
|
||
expect(t2).toContain('變更摘要')
|
||
expect(s.files.has('/home/u/.cache/jsc-anima/repo/characters/testy/ao.json')).toBe(false)
|
||
|
||
const t3 = await run($ as never, 'confirm')
|
||
expect(t3).toContain('已建立 PR')
|
||
const pack = JSON.parse(s.files.get('/home/u/.cache/jsc-anima/repo/characters/testy/ao.json')!)
|
||
expect(pack.timeline[0].chapter).toBe('ch1')
|
||
expect(pack.timeline[0].age).toBe(16)
|
||
expect(JSON.parse(s.files.get('/home/u/.cache/jsc-anima/repo/characters/testy/work.json')!).terms[0].term).toBe('魔核')
|
||
const pr = s.cmds.find(c => c[0] === 'tea' && c[1] === 'pulls')!
|
||
expect(pr).toContain('main')
|
||
expect(pr).toContain('feat/extract-testy-ch1')
|
||
expect(s.cmds.some(c => c[0] === 'git' && c[1] === 'commit' && c[3]!.startsWith('feat: '))).toBe(true)
|
||
})
|
||
|
||
test('同章節重跑覆蓋、新章節追加、基礎設定不同標記待確認', async ($, on) => {
|
||
const files = new Map([['/repo/ch1.txt', 'a'], ['/repo/ch2.txt', 'b']])
|
||
const s = setup(on as never, [reply(char()), reply(char()), reply(char({ state: { mood: '開心', dialogueStyle: '', monologue: '', relationships: [], age: 17, place: '' } }, '傲嬌'))], files)
|
||
await $.session.start({ source: 'startup', cwd: '/repo' } as never)
|
||
const write = async (f: string) => {
|
||
await run($ as never, f)
|
||
await run($ as never, 'confirm')
|
||
return run($ as never, 'confirm')
|
||
}
|
||
await write('ch1.txt')
|
||
await write('ch1.txt')
|
||
expect(JSON.parse(s.files.get('/home/u/.cache/jsc-anima/repo/characters/ts/aoi.json')!).timeline.length).toBe(1)
|
||
|
||
await run($ as never, 'ch2.txt')
|
||
const summary = await run($ as never, 'confirm')
|
||
expect(summary).toContain('追加')
|
||
expect(summary).toContain('待確認')
|
||
await run($ as never, 'confirm')
|
||
const pack = JSON.parse(s.files.get('/home/u/.cache/jsc-anima/repo/characters/ts/aoi.json')!)
|
||
expect(pack.timeline.map((t: { chapter: string }) => t.chapter)).toEqual(['ch1', 'ch2'])
|
||
expect(pack.persona.tone).toBe('溫柔')
|
||
expect(pack.conflicts[0]).toMatchObject({ field: 'persona.tone', proposed: '傲嬌', status: '待確認' })
|
||
})
|
||
|
||
test('schema 驗證失敗不寫檔、不建 PR；cancel 清除待確認', async ($, on) => {
|
||
const bad = char({ axes: { gentle: 99, tsundere: 2, lively: 5, formal: 4, talkative: 5, humor: 3, proactive: 6, expressive: 5 } })
|
||
const s = setup(on as never, [reply(bad)], new Map([['/repo/ch1.txt', 'a']]))
|
||
await $.session.start({ source: 'startup', cwd: '/repo' } as never)
|
||
await run($ as never, 'ch1.txt')
|
||
expect(await run($ as never, 'confirm')).toContain('schema 驗證失敗')
|
||
expect(s.cmds.some(c => c[0] === 'tea')).toBe(false)
|
||
expect([...s.files.keys()].some(k => k.includes('characters'))).toBe(false)
|
||
expect(await run($ as never, 'cancel')).toContain('已取消')
|
||
})
|
||
|
||
test('貼上章節內容：paste <章節名稱> 加下一行起的內容，並寫入語言欄位與 index.json', async ($, on) => {
|
||
const s = setup(on as never, [reply(char({ language: '繁體中文，少量原作口癖' }))])
|
||
await $.session.start({ source: 'startup', cwd: '/repo' } as never)
|
||
expect(await run($ as never, 'paste')).toContain('貼上章節的用法')
|
||
const t1 = await run($ as never, 'paste 第一章\n這是貼上的章節內容。')
|
||
expect(t1).toContain('aoi')
|
||
expect(s.prompts[0]).toContain('這是貼上的章節內容。')
|
||
await run($ as never, 'confirm')
|
||
await run($ as never, 'confirm')
|
||
const pack = JSON.parse(s.files.get('/home/u/.cache/jsc-anima/repo/characters/ts/aoi.json')!)
|
||
expect(pack.timeline[0].chapter).toBe('第一章')
|
||
expect(pack.language).toBe('繁體中文，少量原作口癖')
|
||
expect(JSON.parse(s.files.get('/home/u/.cache/jsc-anima/repo/characters/index.json')!).packs[0]).toEqual({ work: 'ts', workName: '測試物語', slug: 'aoi' })
|
||
})
|
||
|
||
test('欄位含長段引號內容視為抄錄原作台詞，不寫檔', async ($, on) => {
|
||
const quoted = char({ persona: { selfRef: '我', address: '你', tone: '「這是一段很長很長的原作台詞內容，不應該出現在角色包裡面」', catchphrase: '加油', values: '誠實', taboos: '背叛' } })
|
||
const s = setup(on as never, [reply(quoted)], new Map([['/repo/ch1.txt', 'a']]))
|
||
await $.session.start({ source: 'startup', cwd: '/repo' } as never)
|
||
await run($ as never, 'ch1.txt')
|
||
expect(await run($ as never, 'confirm')).toContain('疑似抄錄原作台詞')
|
||
expect([...s.files.keys()].some(k => k.includes('characters'))).toBe(false)
|
||
})
|
||
|
||
test('在非 jsc-anima 目錄執行：git 與 PR 都走 jsc-anima 的快取 clone，不碰 session 目錄', async ($, on) => {
|
||
sessionCwd = '/other/proj'
|
||
try {
|
||
const s = setup(on as never, [reply(char())], new Map([['/other/proj/ch1.txt', '第一段']]))
|
||
await $.session.start({ source: 'startup', cwd: '/other/proj' } as never)
|
||
await run($ as never, '/other/proj/ch1.txt')
|
||
await run($ as never, 'confirm')
|
||
expect(await run($ as never, 'confirm')).toContain('已建立 PR')
|
||
expect(s.cmds.some(c => c.includes('fetch'))).toBe(true)
|
||
const pr = s.cmds.find(c => c[0] === 'tea' && c[1] === 'pulls')!
|
||
expect(pr).toContain('jiantw83/jsc-anima')
|
||
expect(s.files.has('/other/proj/characters/index.json')).toBe(false)
|
||
expect(JSON.parse(s.files.get('/home/u/.cache/jsc-anima/repo/characters/index.json')!).packs[0].slug).toBe('aoi')
|
||
} finally {
|
||
sessionCwd = '/repo'
|
||
}
|
||
})
|
||
|
||
test('快取 clone 不存在時先 clone jsc-anima', async ($, on) => {
|
||
cloneMissing = true
|
||
try {
|
||
const s = setup(on as never, [reply(char())], new Map([['/repo/ch1.txt', 'a']]))
|
||
await $.session.start({ source: 'startup', cwd: '/repo' } as never)
|
||
await run($ as never, 'ch1.txt')
|
||
const clone = s.cmds.find(c => c.includes('clone'))!
|
||
expect(clone).toContain('https://gitea.jsc.idv.tw/jiantw83/jsc-anima.git')
|
||
} finally {
|
||
cloneMissing = false
|
||
}
|
||
})
|
||
|
||
test('paste：CRLF 換行也能分出章節名稱與內容', async ($, on) => {
|
||
const s = setup(on as never, [reply(char())])
|
||
await $.session.start({ source: 'startup', cwd: '/repo' } as never)
|
||
await run($ as never, 'paste 第二章\r\n第一行\r\n第二行')
|
||
expect(s.prompts[0]).toContain('第一行')
|
||
})
|
||
|
||
test('合併並更新外掛後：啟動時從外掛目錄的 index.json 註冊 {作品代號}-{角色代號} 並可啟動', async ($, on) => {
|
||
const s = setup(on as never, [reply(char())], new Map([['/repo/ch1.txt', 'a']]))
|
||
registered = []
|
||
await $.session.start({ source: 'startup', cwd: '/repo' } as never)
|
||
await run($ as never, 'ch1.txt')
|
||
await run($ as never, 'confirm')
|
||
await run($ as never, 'confirm')
|
||
// 模擬 PR 合併、外掛更新：clone 內的 characters/ 內容出現在外掛目錄。
|
||
const root = ([...probed].find(q => !q.startsWith('/home/u/')) ?? '').replace('/characters/index.json', '')
|
||
for (const [k, v] of [...s.files]) if (k.startsWith('/home/u/.cache/jsc-anima/repo/characters/')) s.files.set(k.replace('/home/u/.cache/jsc-anima/repo', root), v)
|
||
await $.session.start({ source: 'startup', cwd: '/anywhere' } as never)
|
||
expect(registered).toContain('ts-aoi')
|
||
})
|
||
|
||
const WIKI = 'https://gitea.jsc.idv.tw/jiantw83/sao/wiki'
|
||
const listHtml = ['Home', '第10章', '第2章', '第1章', '_pages']
|
||
.map(n => `<a href="/jiantw83/sao/wiki/${encodeURIComponent(n)}">${n}</a>`)
|
||
.join('') + '<a href="/jiantw83/sao">repo</a><a href="/jiantw83/sao/wiki/第1章/_revision/abc">rev</a>'
|
||
|
||
test('parseWikiSource：從自然語言中取出 wiki URL，頁面清單與單頁分開判斷', () => {
|
||
const src = parseWikiSource(`請幫我依照順序逐章處理 ${WIKI}/?action=_pages 謝謝`)
|
||
expect(src).toMatchObject({ kind: 'list', base: WIKI })
|
||
expect(parseWikiSource(`處理 ${WIKI}/第2章。`)).toMatchObject({ kind: 'page', page: '第2章', base: WIKI })
|
||
expect(parseWikiSource('ch1.txt')).toBeUndefined()
|
||
expect(parseWikiSource('請處理 ch 1.txt')).toBeUndefined()
|
||
})
|
||
|
||
test('parseWikiPages：略過 Home／_pages／子路徑，並依章節數字自然排序', () => {
|
||
expect(parseWikiPages(listHtml, WIKI)).toEqual(['第1章', '第2章', '第10章'])
|
||
})
|
||
|
||
test('anima-extract 帶 URL 與自然語言：不當檔案路徑，依序逐章處理', async ($, on) => {
|
||
web.clear()
|
||
web.set(`${WIKI}/?action=_pages`, listHtml)
|
||
for (const n of ['第1章', '第2章', '第10章']) web.set(`${WIKI}/raw/${encodeURIComponent(n)}`, `${n}內容`)
|
||
const s = setup(on as never, [reply(char()), reply(char()), reply(char())])
|
||
await $.session.start({ source: 'startup', cwd: '/repo' } as never)
|
||
const t1 = await run($ as never, `請幫我依照順序逐章處理 ${WIKI}/?action=_pages`)
|
||
expect(t1).not.toContain('找不到檔案')
|
||
expect(t1).toContain('共 3 章：第1章、第2章、第10章')
|
||
expect(t1).toContain('章節「第1章」')
|
||
expect(s.prompts[0]).toContain('第1章內容')
|
||
await run($ as never, 'confirm')
|
||
const t3 = await run($ as never, 'confirm')
|
||
expect(t3).toContain('已建立 PR')
|
||
expect(t3).toContain('剩餘 1 章')
|
||
expect(t3).toContain('章節「第2章」')
|
||
expect(s.prompts[1]).toContain('第2章內容')
|
||
await run($ as never, 'confirm')
|
||
const t5 = await run($ as never, 'confirm')
|
||
expect(t5).toContain('章節「第10章」')
|
||
await run($ as never, 'confirm')
|
||
const t7 = await run($ as never, 'confirm')
|
||
expect(t7).toContain('已建立 PR')
|
||
expect(t7).not.toContain('接著處理')
|
||
})
|
||
|
||
test('wiki 章節讀不到時回報原因與剩餘章數；cancel 清除佇列', async ($, on) => {
|
||
web.clear()
|
||
web.set(`${WIKI}/?action=_pages`, listHtml)
|
||
setup(on as never, [])
|
||
await $.session.start({ source: 'startup', cwd: '/repo' } as never)
|
||
expect(await run($ as never, `${WIKI}/?action=_pages`)).toContain('無法讀取 wiki 章節「第1章」')
|
||
expect(await run($ as never, 'cancel')).toContain('已取消')
|
||
})
|