151 lines
7.5 KiB
TypeScript
151 lines
7.5 KiB
TypeScript
import { existsSync, mkdirSync, mkdtempSync, readFileSync, rmSync, writeFileSync } from 'node:fs'
|
|
import * as os from 'node:os'
|
|
import * as path from 'node:path'
|
|
import { afterEach, describe, expect, it } from 'vitest'
|
|
import { packHandoff, unpackHandoff } from './handoffArchive'
|
|
import { installHandoffBundle, stageHandoffBundle } from './handoffBundle'
|
|
import { HANDOFF_MANIFEST_FILE, parseHandoffManifest } from './handoffManifest'
|
|
|
|
const tempDirs: string[] = []
|
|
afterEach(() => {
|
|
while (tempDirs.length > 0) {
|
|
const d = tempDirs.pop()
|
|
if (d)
|
|
rmSync(d, { recursive: true, force: true })
|
|
}
|
|
})
|
|
function tempDir(prefix: string): string {
|
|
const d = mkdtempSync(path.join(os.tmpdir(), prefix))
|
|
tempDirs.push(d)
|
|
return d
|
|
}
|
|
|
|
const BRAIN = 'aaaaaaaa-0000-4000-8000-000000000001'
|
|
const IMPL = 'bbbbbbbb-0000-4000-8000-000000000002'
|
|
const REVIEW = '019dcb27-58a6-70a1-a5d1-bfc7f3ed9d0a'
|
|
|
|
function senderRoots(): { projectsRoot: string, codexRoot: string } {
|
|
const projectsRoot = tempDir('sender-projects-')
|
|
const codexRoot = tempDir('sender-codex-')
|
|
const proj = path.join(projectsRoot, '-Users-chw-proj')
|
|
const wt = path.join(projectsRoot, '-Users-chw-wt-x')
|
|
mkdirSync(proj, { recursive: true })
|
|
mkdirSync(path.join(wt, IMPL, 'subagents'), { recursive: true })
|
|
writeFileSync(path.join(proj, `${BRAIN}.jsonl`), 'brain\n')
|
|
writeFileSync(path.join(wt, `${IMPL}.jsonl`), 'impl\n')
|
|
writeFileSync(path.join(wt, IMPL, 'subagents', 'agent-1.jsonl'), '{}\n')
|
|
const day = path.join(codexRoot, '2026', '08', '26')
|
|
mkdirSync(day, { recursive: true })
|
|
writeFileSync(path.join(day, `rollout-2026-08-26T01-02-03-${REVIEW}.jsonl`), 'review\n')
|
|
return { projectsRoot, codexRoot }
|
|
}
|
|
|
|
describe('stageHandoffBundle', () => {
|
|
it('收集存在的会话，缺失的不进列表', async () => {
|
|
const roots = senderRoots()
|
|
const staging = tempDir('staging-')
|
|
const manifest = await stageHandoffBundle({
|
|
issueNumber: 7,
|
|
from: 'chw',
|
|
branch: 'feature/x',
|
|
sessions: { sessionId: BRAIN, implementSessionId: IMPL, testSessionId: 'missing-0000-4000-8000-000000000000', reviewSessionId: REVIEW },
|
|
profiles: { profilePath: '/Users/chw/p.json' },
|
|
}, staging, roots, new Date('2026-08-26T00:00:00Z'))
|
|
expect(manifest.claude).toEqual([{ id: BRAIN, kind: 'brainstorm' }, { id: IMPL, kind: 'implement' }])
|
|
expect(manifest.codex).toEqual([{ id: REVIEW, relPath: path.join('2026', '08', '26', `rollout-2026-08-26T01-02-03-${REVIEW}.jsonl`) }])
|
|
expect(manifest.sessions.testSessionId).toBe('missing-0000-4000-8000-000000000000')
|
|
expect(manifest.createdAt).toBe('2026-08-26T00:00:00.000Z')
|
|
expect(existsSync(path.join(staging, 'claude', `${IMPL}.jsonl`))).toBe(true)
|
|
expect(existsSync(path.join(staging, 'claude', IMPL, 'subagents', 'agent-1.jsonl'))).toBe(true)
|
|
expect(existsSync(path.join(staging, 'codex', manifest.codex[0].relPath))).toBe(true)
|
|
expect(parseHandoffManifest(readFileSync(path.join(staging, HANDOFF_MANIFEST_FILE), 'utf8'), 7)).toEqual(manifest)
|
|
})
|
|
})
|
|
|
|
describe('pack → unpack → install', () => {
|
|
it('接管方装到对应 project 目录与 codex 目录', async () => {
|
|
const roots = senderRoots()
|
|
const staging = tempDir('staging-')
|
|
const manifest = await stageHandoffBundle({
|
|
issueNumber: 7,
|
|
from: 'chw',
|
|
branch: 'feature/x',
|
|
sessions: { sessionId: BRAIN, implementSessionId: IMPL, reviewSessionId: REVIEW },
|
|
profiles: {},
|
|
}, staging, roots)
|
|
const archive = path.join(tempDir('archive-'), 'spx-handoff-issue-7.tgz')
|
|
await packHandoff(staging, archive)
|
|
expect(existsSync(archive)).toBe(true)
|
|
|
|
const extracted = tempDir('extracted-')
|
|
await unpackHandoff(archive, extracted)
|
|
const parsed = parseHandoffManifest(readFileSync(path.join(extracted, HANDOFF_MANIFEST_FILE), 'utf8'), 7)
|
|
expect(parsed).toEqual(manifest)
|
|
|
|
const projectsRoot = tempDir('receiver-projects-')
|
|
const codexRoot = tempDir('receiver-codex-')
|
|
const workspaceProjectsDir = path.join(projectsRoot, '-home-me-proj')
|
|
const worktreeProjectsDir = path.join(projectsRoot, '-home-me-wt-x')
|
|
// 本机残留的同 sid 副本必须被清掉
|
|
mkdirSync(path.join(projectsRoot, '-stale'), { recursive: true })
|
|
writeFileSync(path.join(projectsRoot, '-stale', `${IMPL}.jsonl`), 'stale\n')
|
|
await installHandoffBundle(extracted, parsed, { projectsRoot, codexRoot, workspaceProjectsDir, worktreeProjectsDir })
|
|
expect(readFileSync(path.join(workspaceProjectsDir, `${BRAIN}.jsonl`), 'utf8')).toBe('brain\n')
|
|
expect(readFileSync(path.join(worktreeProjectsDir, `${IMPL}.jsonl`), 'utf8')).toBe('impl\n')
|
|
expect(existsSync(path.join(worktreeProjectsDir, IMPL, 'subagents', 'agent-1.jsonl'))).toBe(true)
|
|
expect(existsSync(path.join(projectsRoot, '-stale', `${IMPL}.jsonl`))).toBe(false)
|
|
expect(readFileSync(path.join(codexRoot, parsed.codex[0].relPath), 'utf8')).toBe('review\n')
|
|
})
|
|
|
|
it('无 worktreeProjectsDir 时实施会话也装到工作区目录', async () => {
|
|
const roots = senderRoots()
|
|
const staging = tempDir('staging-')
|
|
const manifest = await stageHandoffBundle({ issueNumber: 7, from: 'chw', sessions: { implementSessionId: IMPL }, profiles: {} }, staging, roots)
|
|
const projectsRoot = tempDir('receiver-projects-')
|
|
const workspaceProjectsDir = path.join(projectsRoot, '-home-me-proj')
|
|
await installHandoffBundle(staging, manifest, { projectsRoot, codexRoot: tempDir('receiver-codex-'), workspaceProjectsDir })
|
|
expect(existsSync(path.join(workspaceProjectsDir, `${IMPL}.jsonl`))).toBe(true)
|
|
})
|
|
|
|
it('带 worktreePath + pathRewrite：装机文件里的发送方路径全部改写成接收方路径', async () => {
|
|
const WORKTREE_FROM = '/Users/chw/wt/x'
|
|
const WORKTREE_TO = '/home/me/wt/x'
|
|
const roots = senderRoots()
|
|
// sender 会话文件里嵌入发送方的 worktree 路径（plain / file:// 两种形式）
|
|
const wt = path.join(roots.projectsRoot, '-Users-chw-wt-x')
|
|
writeFileSync(path.join(wt, `${IMPL}.jsonl`), `{"cwd":"${WORKTREE_FROM}"}\n{"cwd":"file://${WORKTREE_FROM}"}\n`)
|
|
writeFileSync(path.join(wt, IMPL, 'subagents', 'agent-1.jsonl'), `{"cwd":"${WORKTREE_FROM}"}\n`)
|
|
const day = path.join(roots.codexRoot, '2026', '08', '26')
|
|
writeFileSync(path.join(day, `rollout-2026-08-26T01-02-03-${REVIEW}.jsonl`), `{"cwd":"${WORKTREE_FROM}"}\n{"cwd":"file://${WORKTREE_FROM}"}\n`)
|
|
|
|
const staging = tempDir('staging-')
|
|
const manifest = await stageHandoffBundle({
|
|
issueNumber: 7,
|
|
from: 'chw',
|
|
worktreePath: WORKTREE_FROM,
|
|
sessions: { implementSessionId: IMPL, reviewSessionId: REVIEW },
|
|
profiles: {},
|
|
}, staging, roots)
|
|
expect(manifest.worktreePath).toBe(WORKTREE_FROM)
|
|
|
|
const projectsRoot = tempDir('receiver-projects-')
|
|
const codexRoot = tempDir('receiver-codex-')
|
|
const worktreeProjectsDir = path.join(projectsRoot, '-home-me-wt-x')
|
|
await installHandoffBundle(staging, manifest, {
|
|
projectsRoot,
|
|
codexRoot,
|
|
workspaceProjectsDir: path.join(projectsRoot, '-home-me-proj'),
|
|
worktreeProjectsDir,
|
|
pathRewrite: { from: WORKTREE_FROM, to: WORKTREE_TO },
|
|
})
|
|
|
|
const installedJsonl = readFileSync(path.join(worktreeProjectsDir, `${IMPL}.jsonl`), 'utf8')
|
|
const installedSubagent = readFileSync(path.join(worktreeProjectsDir, IMPL, 'subagents', 'agent-1.jsonl'), 'utf8')
|
|
const installedRollout = readFileSync(path.join(codexRoot, manifest.codex[0].relPath), 'utf8')
|
|
for (const content of [installedJsonl, installedSubagent, installedRollout]) {
|
|
expect(content).toContain(WORKTREE_TO)
|
|
expect(content).not.toContain(WORKTREE_FROM)
|
|
}
|
|
})
|
|
})
|