scaffold.ts 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610
  1. // Shared scaffold for the keyless browser e2e lane (Agent Note:
  2. // .agents/notes/implemented/testing/2026-07-24-web-gui-browser-e2e-lane.md).
  3. // Boots the REAL web composition — the shipped base plus web overlay through
  4. // the vendored Loader (the same include boot AppCLIEntry drives), patched the
  5. // snapshot way — so a real chromium exercises the real HTTP/SSE wire, the
  6. // api-gateway, agent loop, tools, and persistence. Modes ride $DSH_SNAPSHOT:
  7. // replay (default, keyless: normally disables the llm-deepseek row and
  8. // inserts dsh-llm-replay in providers mode), record (real adapter + key,
  9. // harvests fixtures from live session memory), refresh (keyless replay that
  10. // rewrites goldens). A first-run option keeps the real adapter mounted while
  11. // masking its credential, without making a model call.
  12. //
  13. // Composition divergences from `dsh web`, all deliberate, all via include
  14. // patches after the shipped surface overlay, over the SAME tree (never a
  15. // second yml): temp persistenceRoot; host-level skill roots confined to the
  16. // temp workspace while project skill discovery remains real; workspace-context
  17. // disabled (recorded fixtures must not embed this repo's AGENTS.md);
  18. // session-title-llm disabled (its fire-and-forget title call would race the
  19. // loop for the session's replay cursor); webserver pinned to port 0 with the
  20. // built dist; ordinary keyless modes disable llm-deepseek and fill the open
  21. // llm seam post-boot with installLlmReplay on the settled root ctx
  22. // (the plugin-row path discards the ReplayHandle; the direct install keeps
  23. // assertConsumed for the teardown fixture-consumption check).
  24. import { existsSync } from 'node:fs'
  25. import { mkdtemp, readFile, readdir, realpath, rm, utimes, writeFile } from 'node:fs/promises'
  26. import { tmpdir } from 'node:os'
  27. import { join, resolve } from 'node:path'
  28. import { pathToFileURL } from 'node:url'
  29. import type { Page } from 'playwright'
  30. import { expect } from 'vitest'
  31. import { Context } from 'cordis'
  32. import Loader from '@cordisjs/plugin-loader'
  33. import Include, { type PatchOptions } from '@cordisjs/plugin-include'
  34. import { scrubRequestHeaders } from '@deepseek-ai/dsh-acp-snapshot'
  35. import { assertEntriesLoaded, loadOverlayPatches } from '@deepseek-ai/dsh-app-boot'
  36. import { dshHomePath } from '@deepseek-ai/dsh-paths'
  37. import {
  38. WELCOME_NOTICE_ACK_FIELD, WELCOME_NOTICE_SETTINGS_NAMESPACE, WELCOME_NOTICE_VERSION,
  39. } from '@deepseek-ai/dsh-client-ui-settings-general'
  40. import { settingsNamespace } from '@deepseek-ai/dsh-settings'
  41. import type { ReplayHandle } from '@deepseek-ai/dsh-llm-replay'
  42. import { installLlmReplay, parseSessionLog } from '@deepseek-ai/dsh-llm-replay'
  43. import SessionStore, {
  44. packChunkRuns,
  45. SESSION_FORMAT_VERSION,
  46. SessionId,
  47. type Session,
  48. type SessionEvent,
  49. type SessionHeader,
  50. } from '@deepseek-ai/dsh-session'
  51. import SessionPersistenceJsonl from '@deepseek-ai/dsh-session-persistence-jsonl'
  52. import * as ToolCordis from '@deepseek-ai/dsh-tool-cordis'
  53. // Empty type imports carry the httpServer/agents/sessionPersistence Context merges.
  54. import type {} from '@deepseek-ai/dsh-host-webserver'
  55. import type {} from '@deepseek-ai/dsh-agent'
  56. import { prepareWebRuntimeContext } from '../../cli/src/web.ts'
  57. import { DIST_INDEX, REPO_ROOT, requireDist } from './support.ts'
  58. /** Snapshot mode for the lane, from $DSH_SNAPSHOT (same vocabulary as the ACP/TUI suites). */
  59. export type WebSnapshotMode = 'replay' | 'record' | 'refresh'
  60. /**
  61. * Resolve and validate the lane's snapshot mode.
  62. * @returns the active mode; unset/empty selects replay.
  63. */
  64. export function webSnapshotMode(): WebSnapshotMode {
  65. const value = process.env.DSH_SNAPSHOT
  66. if (value === undefined || value === '' || value === 'replay') return 'replay'
  67. if (value === 'record' || value === 'refresh') return value
  68. throw new Error(`DSH_SNAPSHOT must be replay, record, or refresh; got ${JSON.stringify(value)}`)
  69. }
  70. /** The shipped composition under test: apps/cli's shared base and web overlay. */
  71. const CONFIG_PATH = join(REPO_ROOT, 'apps/cli/config/base.cordis.yml')
  72. const WEB_OVERLAY_PATH = join(REPO_ROOT, 'apps/cli/config/web.cordis.yml')
  73. // Replay publishes the provider catalog the gateway routes to (providers
  74. // mode, never catch-all: with llm-deepseek disabled no adapter exists, so a
  75. // catch-all would leave resolveModelInfo unroutable and compact-basic's
  76. // post-step pressure check would warn every step). The published
  77. // contextWindow keeps that pressure path provably inert for small fixtures.
  78. const REPLAY_PROVIDERS = [{
  79. id: 'deepseek-official',
  80. name: 'DeepSeek',
  81. models: [{ id: 'deepseek-v4-flash', name: 'DeepSeek-V4-Flash', contextWindow: 128_000 }],
  82. }]
  83. /** A booted web scaffold: real composition, mode-selected model backend, temp world. */
  84. export interface WebScaffold {
  85. /** The active snapshot mode this scaffold booted under. */
  86. mode: WebSnapshotMode
  87. /** Browser-facing origin (http://127.0.0.1:<bound port>). */
  88. baseUrl: string
  89. /** Settled root context (the in-process barrier seam; headless event subscription is its sanctioned use). */
  90. ctx: Context
  91. /** Temp project directory sessions run in (bash/fs tool cwd). */
  92. workspaceCwd: string
  93. /** Temp persistence root (seeded sessions land here through the real API). */
  94. persistenceRoot: string
  95. /** Isolated harness home the settings/credentials rows write ($DSH_HOME double). */
  96. harnessHome: string
  97. /** Await a settled turn end: in-process turn/end, then the agent's idle flip (which follows the persistence flush). */
  98. whenTurnSettled(timeoutMs?: number): Promise<SessionId>
  99. /** Tear everything down; asserts the replay fixture was fully consumed first (replay/refresh). */
  100. close(): Promise<void>
  101. }
  102. /** Options for {@link launchWebScaffold}. */
  103. export interface LaunchOptions {
  104. /**
  105. * Optional product overlay applied after the shipped Web surface and before
  106. * the scaffold's hermetic test patches, matching AppCLIEntry's `--config`
  107. * ordering.
  108. */
  109. extraOverlayPath?: string
  110. /**
  111. * Replay fixture (session.jsonl) served by the inserted dsh-llm-replay row
  112. * in replay/refresh modes; ignored in record mode (the real adapter
  113. * answers). Omit for scenarios issuing no model calls — a stray stream then
  114. * fails loud with NO_ADAPTER (llm-deepseek is disabled and no replay row
  115. * mounts).
  116. */
  117. replayFixture?: string
  118. /**
  119. * Recorded child logs assigned in child creation order. Each child owns its
  120. * own positional replay cursor across initial and continuation turns.
  121. */
  122. replayChildFixtures?: string[]
  123. /**
  124. * Optional replay.override.json sidecar (whole-script replacement or
  125. * `{ patches }` augmentation) for throw/hang scenarios not expressible as
  126. * recorded chunks; replay/refresh only.
  127. */
  128. replayOverride?: string
  129. /** Per-chunk replay pacing (ms) so the browser observes genuinely incremental SSE; replay/refresh only. */
  130. paceMs?: number
  131. /**
  132. * Tool presentation mode patched onto the shipped `tools` row (`code`
  133. * collapses the wire to run_code + the SDK prompt section). Omit for the
  134. * yml default. The code runtime row is always in the tree, so no extra
  135. * insertion is needed.
  136. */
  137. toolsMode?: 'native' | 'code' | 'both'
  138. /**
  139. * Insert the opt-in self-referential Cordis tools into the shipped tree.
  140. * Record and replay use the same tool surface, so captured request headers
  141. * remain reconstructable without making the tools a product default.
  142. */
  143. cordisTools?: boolean
  144. /**
  145. * Keep the shipped DeepSeek adapter mounted while masking the process
  146. * environment's DEEPSEEK_API_KEY for this scaffold lifetime. This is the
  147. * keyless first-run configuration lane; the default disables the adapter.
  148. */
  149. deepSeekMissingCredential?: boolean
  150. /**
  151. * Patch the shipped DeepSeek search row to a deterministic endpoint and
  152. * credential reference. Browser search scenarios keep the real provider and
  153. * credentials seam while avoiding external search traffic and ambient keys.
  154. */
  155. deepSeekSearch?: {
  156. /** Anthropic-compatible base URL; the provider appends `/messages`. */
  157. baseURL: string
  158. /** Credential reference resolved by the shipped search provider. */
  159. apiKeyEnv: string
  160. }
  161. /** Leave the current welcome notice unacknowledged; ordinary scenarios publish it as complete before browser boot. */
  162. welcomeNoticePending?: boolean
  163. }
  164. /** Dispose the booted tree and remove both owned temp roots, reporting every independent cleanup failure. */
  165. async function cleanupScaffoldWorld(ctx: Context, workspaceCwd: string, persistenceRoot: string): Promise<unknown[]> {
  166. const failures: unknown[] = []
  167. await Promise.resolve(ctx.fiber.dispose()).catch((error: unknown) => failures.push(error))
  168. await rm(workspaceCwd, { recursive: true, force: true }).catch((error: unknown) => failures.push(error))
  169. await rm(persistenceRoot, { recursive: true, force: true }).catch((error: unknown) => failures.push(error))
  170. return failures
  171. }
  172. /**
  173. * Boot the real web composition under the current snapshot mode.
  174. * @param options - replay fixture selection and pacing.
  175. * @returns the running scaffold.
  176. */
  177. export async function launchWebScaffold(options: LaunchOptions = {}): Promise<WebScaffold> {
  178. requireDist()
  179. const mode = webSnapshotMode()
  180. if (mode === 'record') {
  181. // Both owning vitest configs (web unconditionally, snapshot in record
  182. // mode) load the repo-root .env before this file runs.
  183. if (process.env.DEEPSEEK_API_KEY === undefined || process.env.DEEPSEEK_API_KEY.length === 0) {
  184. throw new Error('web e2e record mode needs DEEPSEEK_API_KEY (env or repo-root .env)')
  185. }
  186. }
  187. if (mode === 'record' && options.deepSeekMissingCredential === true) {
  188. throw new Error('deepSeekMissingCredential is a keyless replay/refresh option')
  189. }
  190. const maskDeepSeekCredential = mode !== 'record' && options.deepSeekMissingCredential === true
  191. const originalDeepSeekCredential = process.env.DEEPSEEK_API_KEY
  192. let credentialEnvironmentRestored = false
  193. const restoreCredentialEnvironment = (): void => {
  194. if (credentialEnvironmentRestored || !maskDeepSeekCredential) return
  195. credentialEnvironmentRestored = true
  196. if (originalDeepSeekCredential === undefined) {
  197. Reflect.deleteProperty(process.env, 'DEEPSEEK_API_KEY')
  198. } else {
  199. process.env.DEEPSEEK_API_KEY = originalDeepSeekCredential
  200. }
  201. }
  202. const workspaceCwd = await realpath(await mkdtemp(join(tmpdir(), 'dsh-web-e2e-ws-')))
  203. // Isolated harness home: the settings/credentials rows resolve $DSH_HOME
  204. // paths at load, and an in-process boot must NEVER touch the developer's
  205. // real ~/.dsh document or credential file.
  206. const harnessHome = join(workspaceCwd, '.dsh-home')
  207. let persistenceRoot: string
  208. try {
  209. persistenceRoot = await mkdtemp(join(tmpdir(), 'dsh-web-e2e-sessions-'))
  210. } catch (error) {
  211. const failures: unknown[] = [error]
  212. await rm(workspaceCwd, { recursive: true, force: true }).catch((cleanupError: unknown) => failures.push(cleanupError))
  213. if (failures.length > 1) throw new AggregateError(failures, 'web scaffold temp-root setup failed')
  214. throw error
  215. }
  216. if (maskDeepSeekCredential) Reflect.deleteProperty(process.env, 'DEEPSEEK_API_KEY')
  217. // The include patch set — the same mechanism AppCLIEntry and the ACP
  218. // snapshot overlay use, applied over the SAME shipped tree (a patch id that
  219. // stops matching a row fails the boot sweep loudly instead of drifting).
  220. const surfacePatches = loadOverlayPatches('web e2e scaffold', WEB_OVERLAY_PATH)
  221. const extraOverlayPatches = options.extraOverlayPath === undefined
  222. ? []
  223. : loadOverlayPatches('web e2e scaffold', options.extraOverlayPath)
  224. const patches: PatchOptions[] = [
  225. ...surfacePatches,
  226. ...extraOverlayPatches,
  227. { id: 'session-persistence-jsonl', config: { root: persistenceRoot } },
  228. { id: 'session-query-sqlite', config: { path: ':memory:', openAt: 'first-search' } },
  229. // storage-json's yml root is anchored to the real $DSH_HOME; pin the row
  230. // to an absolute temp root (removed with the workspace at close) so tests
  231. // never write the user's harness home.
  232. { id: 'storage-json', config: { root: join(workspaceCwd, '.dsh-storages') } },
  233. // Skill discovery is model-visible input. Pin every host-level root inside
  234. // the owned temp world so ~/.dsh, ~/.agents, and a bundled-root env setting
  235. // cannot change replay requests or conversation goldens. Project roots stay
  236. // enabled against the same empty temp workspace, preserving the real seam.
  237. {
  238. id: 'skill-local',
  239. config: {
  240. dshHome: join(workspaceCwd, '.dsh-home'),
  241. agentsHome: join(workspaceCwd, '.agents-home'),
  242. bundledSkillDir: join(workspaceCwd, '.bundled-skills'),
  243. watch: false,
  244. },
  245. },
  246. // fs/bash cwd default to process.cwd(); the gateway injects the same
  247. // value into session.cwd — chdir below anchors all three to the temp
  248. // workspace, keeping the composition untouched.
  249. { id: 'workspace-context', disabled: true },
  250. { id: 'session-title-llm', disabled: true },
  251. // Fixture sessions must never leave the process: the shipped row defaults
  252. // to the production OTLP endpoint (or whatever DSH_TELEMETRY_OTLP_URL
  253. // names in the ambient environment).
  254. { id: 'telemetry-otel', disabled: true },
  255. { id: 'webserver', config: { host: '127.0.0.1', port: 0, distIndex: DIST_INDEX } },
  256. { id: 'settings', config: { dshHome: harnessHome } },
  257. { id: 'credentials', config: { dshHome: harnessHome } },
  258. // The shipped directory-picker row is the -auto chooser, which resolves
  259. // the interaction from the RUNNING host (display, SSH launch, bind). The
  260. // lane's goldens are interaction-specific (workspace-management drives
  261. // the in-app browse dialog), so pin -browse deterministically on every
  262. // host: patch `name` is an assertion, not an override, hence the
  263. // disable+insert pair.
  264. { id: 'directory-picker', disabled: true },
  265. { insert: [{ id: 'directory-picker-browse', name: '@deepseek-ai/dsh-host-directory-picker-browse' }] },
  266. ...options.toolsMode === undefined ? [] : [{ id: 'tools', config: { mode: options.toolsMode } }],
  267. ...options.cordisTools === true
  268. ? [{ insert: [{ id: 'tool-cordis', name: 'cordis:tool-cordis' }] }]
  269. : [],
  270. ...options.deepSeekSearch === undefined
  271. ? []
  272. : [{
  273. id: 'web-search-deepseek',
  274. config: {
  275. apiKeyEnv: options.deepSeekSearch.apiKeyEnv,
  276. baseURL: options.deepSeekSearch.baseURL,
  277. },
  278. }],
  279. ...mode === 'record' || options.deepSeekMissingCredential === true
  280. ? []
  281. : [{ id: 'llm-deepseek', disabled: true }],
  282. ]
  283. // Sessions inherit the gateway's process.cwd() default; run the boot from
  284. // the temp workspace so tool cwd, session cwd, and fixtures agree.
  285. const originalCwd = process.cwd()
  286. const ctx = new Context()
  287. let port = 0
  288. let replayHandle: ReplayHandle | undefined
  289. try {
  290. process.chdir(workspaceCwd)
  291. ctx.baseUrl = pathToFileURL(join(resolve(CONFIG_PATH), '..')).href + '/'
  292. // This direct Loader harness supplies the same root-path capability as app-boot.
  293. ctx.provide('dshHomePath', dshHomePath)
  294. await ctx.plugin(Loader)
  295. ctx.loader.builtins.include = Include
  296. // The shipped CLI deliberately has no dependency on this opt-in package.
  297. // Keep the Loader row real without broadening the product installation.
  298. if (options.cordisTools === true) ctx.loader.builtins['tool-cordis'] = ToolCordis
  299. prepareWebRuntimeContext(ctx, REPO_ROOT, 'production')
  300. await ctx.loader.create({
  301. name: 'cordis:include',
  302. config: { path: pathToFileURL(resolve(CONFIG_PATH)).href, patches },
  303. })
  304. await ctx.loader.await()
  305. assertEntriesLoaded(ctx, 'web e2e scaffold')
  306. if (options.welcomeNoticePending !== true) {
  307. await ctx.settings.mutate(settingsNamespace(WELCOME_NOTICE_SETTINGS_NAMESPACE), [{
  308. op: 'set', path: [WELCOME_NOTICE_ACK_FIELD], value: WELCOME_NOTICE_VERSION,
  309. }])
  310. }
  311. const boundPort = ctx.get('httpServer')?.port
  312. if (boundPort === undefined) {
  313. throw new Error('web e2e scaffold: httpServer service missing after settled boot')
  314. }
  315. port = boundPort
  316. // Fill the open llm seam on the settled root ctx. Ordinary keyless modes
  317. // disable llm-deepseek; the first-run lane keeps it mounted but has no
  318. // replay fixture and never streams. The direct install, unlike the plugin
  319. // row, returns the ReplayHandle for the teardown consumption check.
  320. if (mode !== 'record' && options.replayFixture !== undefined) {
  321. replayHandle = installLlmReplay(ctx, {
  322. file: options.replayFixture,
  323. providers: REPLAY_PROVIDERS,
  324. ...(options.replayOverride === undefined ? {} : { overrideFile: options.replayOverride }),
  325. ...(options.replayChildFixtures === undefined ? {} : { childFiles: options.replayChildFixtures }),
  326. ...(options.paceMs === undefined ? {} : { paceMs: options.paceMs }),
  327. })
  328. }
  329. } catch (error) {
  330. if (process.cwd() !== originalCwd) process.chdir(originalCwd)
  331. const cleanupFailures = await cleanupScaffoldWorld(ctx, workspaceCwd, persistenceRoot)
  332. restoreCredentialEnvironment()
  333. if (cleanupFailures.length > 0) {
  334. throw new AggregateError([error, ...cleanupFailures], 'web scaffold setup failed and cleanup was incomplete')
  335. }
  336. throw error
  337. } finally {
  338. if (process.cwd() !== originalCwd) process.chdir(originalCwd)
  339. }
  340. return {
  341. harnessHome,
  342. mode,
  343. baseUrl: `http://127.0.0.1:${port}`,
  344. ctx,
  345. workspaceCwd,
  346. persistenceRoot,
  347. // Barrier stack: the in-process turn/end identifies the session, then
  348. // agent.whenIdle() covers the persistence flush (the idle flip follows
  349. // the flush), and the caller's browser settled-poll comes last because
  350. // host completion strictly precedes render.
  351. whenTurnSettled(timeoutMs = mode === 'record' ? 180_000 : 30_000): Promise<SessionId> {
  352. return new Promise<SessionId>((resolveSettled, reject) => {
  353. const timer = setTimeout(() => {
  354. off()
  355. reject(new Error(`no turn/end within ${timeoutMs}ms`))
  356. }, timeoutMs)
  357. const off = ctx.on('session/event', (session: { id: SessionId }, event: SessionEvent) => {
  358. if (event.type !== 'turn/end') return
  359. clearTimeout(timer)
  360. off()
  361. const agent = ctx.agents.get(session.id)
  362. if (agent === undefined) {
  363. reject(new Error(`turn/end for ${session.id} but no live agent`))
  364. return
  365. }
  366. agent.whenIdle().then(() => { resolveSettled(session.id) }, reject)
  367. })
  368. })
  369. },
  370. async close(): Promise<void> {
  371. const failures: unknown[] = []
  372. // Fixture-consumption check first, while the run's binding state is
  373. // still authoritative — a scenario that drove fewer model calls than
  374. // recorded fails here instead of drifting green.
  375. try {
  376. replayHandle?.assertConsumed()
  377. } catch (error) {
  378. failures.push(error)
  379. }
  380. try {
  381. failures.push(...await cleanupScaffoldWorld(ctx, workspaceCwd, persistenceRoot))
  382. } finally {
  383. restoreCredentialEnvironment()
  384. }
  385. if (failures.length > 0) throw new AggregateError(failures, 'web scaffold teardown failed')
  386. },
  387. }
  388. }
  389. /**
  390. * Serialize a live session to the canonical raw session-JSONL layout — the
  391. * in-memory record-mode harvest, so the on-disk zstd default never matters.
  392. */
  393. function rawSessionLog(session: Session): string {
  394. return [
  395. JSON.stringify({ type: 'session', ...session.header }),
  396. ...packChunkRuns(session.events).map(record => JSON.stringify(record)),
  397. '',
  398. ].join('\n')
  399. }
  400. /**
  401. * Record-mode fixture write-back: harvest the live session, scrub request
  402. * headers to {{system}}/{{tools}} (TODO(web-header-pin): the web lane pins no
  403. * header class — a deliberate deviation logged in the Agent Note's deferred
  404. * work), tokenize the run-local session id, cwd, and browser RPC id
  405. * ({{sessionId}}/{{cwd}}/{{rpcId}}, the committed fixture convention —
  406. * re-records then diff only on real content), and write the fixture.
  407. * @param scaffold - the record-mode scaffold.
  408. * @param sessionId - the driven session.
  409. * @param fixturePath - the committed session.jsonl / seed.jsonl target.
  410. */
  411. export async function recordFixture(scaffold: WebScaffold, sessionId: SessionId, fixturePath: string): Promise<void> {
  412. const agent = scaffold.ctx.agents.get(sessionId)
  413. if (agent === undefined) throw new Error(`record harvest: no live agent for ${sessionId}`)
  414. const tokenized = scrubRequestHeaders(rawSessionLog(agent.session))
  415. .split(sessionId).join('{{sessionId}}')
  416. .split(scaffold.workspaceCwd).join('{{cwd}}')
  417. .replace(/"rpcId":"[^"]+"/g, '"rpcId":"{{rpcId}}"')
  418. await writeFile(fixturePath, tokenized)
  419. }
  420. /**
  421. * The user prompts recorded in a fixture, in order — the single source tying
  422. * spec drive steps to recorded reality so script and fixture cannot drift.
  423. * @param fixtureText - raw session.jsonl contents.
  424. * @returns the recorded user prompt texts.
  425. */
  426. export function fixtureUserPrompts(fixtureText: string): string[] {
  427. return parseSessionLog(fixtureText).flatMap((event) => {
  428. if (event.type !== 'user/message' || event.data.source.kind !== 'user') return []
  429. const text = event.data.content.filter(block => block.type === 'text').map(block => block.text).join('')
  430. return text.length > 0 ? [text] : []
  431. })
  432. }
  433. /**
  434. * Seed a recorded session fixture into the scaffold's persistence root
  435. * through the REAL backend API (throwaway Context + SessionStore + JSONL
  436. * plugin — the semantic-checkpoint precedent), never raw file writes: no
  437. * knowledge of bucket hashing, filename encoding, or compression, and
  438. * malformed shapes fail loud at seed time. The fixture's tokenized identity
  439. * ({{sessionId}}/{{cwd}}) is realized for this world before parsing.
  440. * @param scaffold - the target scaffold.
  441. * @param fixtureText - raw recorded session.jsonl contents.
  442. * @param id - the seeded session id (stable for deterministic goldens).
  443. * @returns the seeded id.
  444. */
  445. export async function seedSession(scaffold: WebScaffold, fixtureText: string, id: string): Promise<SessionId> {
  446. const realized = fixtureText
  447. .split('{{sessionId}}').join(id)
  448. .split('{{cwd}}').join(scaffold.workspaceCwd)
  449. const fixtureCwd = (JSON.parse(realized.split('\n', 1)[0]!) as { cwd?: string }).cwd
  450. const rewritten = fixtureCwd === undefined
  451. ? realized
  452. : realized.split(fixtureCwd).join(scaffold.workspaceCwd)
  453. const events = parseSessionLog(rewritten)
  454. if (events.length === 0) throw new Error('seed fixture has no events')
  455. const last = events[events.length - 1]!
  456. // An open final turn would be mutated by resume's crash repair on first
  457. // open; a committed seed must be a closed recording.
  458. if (last.type !== 'turn/end') throw new Error(`seed fixture must end in turn/end, got ${last.type}`)
  459. const meta: SessionHeader = {
  460. version: SESSION_FORMAT_VERSION,
  461. id: SessionId(id),
  462. createdAt: Date.now() - 60_000,
  463. cwd: scaffold.workspaceCwd,
  464. delegationDepth: 0,
  465. }
  466. const seeder = new Context()
  467. try {
  468. await seeder.plugin(SessionStore)
  469. // Same root as the booted tree with the plugin's own default compression,
  470. // so the host's directory-scan list() sees one consistent encoding.
  471. await seeder.plugin(SessionPersistenceJsonl, { root: scaffold.persistenceRoot })
  472. await seeder.sessionPersistence.create(meta)
  473. await seeder.sessionPersistence.append(meta.id, events)
  474. // Deterministic sidebar order: cold summaries take updatedAt from mtime.
  475. const located = seeder.sessionPersistence.locate(meta)
  476. if (located !== undefined) {
  477. const backdated = new Date(meta.createdAt)
  478. await utimes(located.path, backdated, backdated)
  479. }
  480. } finally {
  481. await seeder.fiber.dispose()
  482. }
  483. return meta.id
  484. }
  485. /**
  486. * Normalize an aria snapshot: uuid, cwd, workspace-basename, and duration
  487. * volatility collapse to stable tokens.
  488. */
  489. function normalizeAria(snapshot: string, workspaceCwd: string): string {
  490. // The session heading renders the workspace's basename, not the full
  491. // path, so both spellings must collapse to the token.
  492. const base = workspaceCwd.split('/').pop()!
  493. return snapshot
  494. .split(workspaceCwd).join('{{cwd}}')
  495. .split(base).join('{{workspace}}')
  496. .replace(/[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}/gi, '{{uuid}}')
  497. .replace(/\b\d+(?:\.\d+)?(?:ms|s|秒)\b/g, '{{duration}}')
  498. // Message IconActions clocks widen by calendar day/year; collapse every
  499. // shape so goldens stay stable across midnight and year boundaries.
  500. .replace(/\d{4}年\d{1,2}月\d{1,2}日 \d{2}:\d{2}/g, '{{clock}}')
  501. .replace(/\d{1,2}月\d{1,2}日 \d{2}:\d{2}/g, '{{clock}}')
  502. .replace(/(?<!\d)\d{2}:\d{2}(?!\d)/g, '{{clock}}')
  503. }
  504. /**
  505. * Capture the region's aria snapshot at a settled milestone: poll until two
  506. * consecutive normalized captures are equal — a single-shot capture races the
  507. * last React commits.
  508. * @param page - the page under test.
  509. * @param selector - the region locator selector.
  510. * @param workspaceCwd - normalization input.
  511. * @returns the stable normalized snapshot.
  512. */
  513. export async function captureStableAria(page: Page, selector: string, workspaceCwd: string): Promise<string> {
  514. const region = page.locator(selector).first()
  515. let previous = normalizeAria(await region.ariaSnapshot(), workspaceCwd)
  516. await expect.poll(async () => {
  517. const current = normalizeAria(await region.ariaSnapshot(), workspaceCwd)
  518. const stable = current === previous
  519. previous = current
  520. return stable
  521. }, { timeout: 5_000, message: 'aria snapshot did not stabilize' }).toBe(true)
  522. return previous
  523. }
  524. /**
  525. * Compare a normalized golden, or rewrite it under refresh. Refresh is the
  526. * ONLY writer: a missing golden in replay mode fails with the healing command
  527. * instead of silently self-bootstrapping.
  528. * @param goldenPath - the committed ui.expected.md path.
  529. * @param actual - the stable normalized snapshot.
  530. * @param mode - the active snapshot mode.
  531. */
  532. export async function compareOrRefreshGolden(goldenPath: string, actual: string, mode: WebSnapshotMode): Promise<void> {
  533. const payload = `${actual}\n`
  534. if (mode === 'refresh') {
  535. await writeFile(goldenPath, payload)
  536. return
  537. }
  538. if (!existsSync(goldenPath)) {
  539. throw new Error(`missing golden ${goldenPath} — run DSH_SNAPSHOT=refresh pnpm run test:web to generate it`)
  540. }
  541. expect(payload).toBe(await readFile(goldenPath, 'utf8'))
  542. }
  543. /**
  544. * Fixture-inventory guard (the TUI afterAll shape): the scenario directory
  545. * holds exactly the expected files and every committed JSONL is a scrub
  546. * fixed-point without a run-local browser RPC id.
  547. * @param dir - the scenario snapshot directory.
  548. * @param expected - the exact expected file inventory.
  549. */
  550. export async function assertFixtureInventory(dir: string, expected: string[]): Promise<void> {
  551. const entries = (await readdir(dir)).sort()
  552. expect(entries).toEqual([...expected].sort())
  553. for (const entry of entries.filter(name => name.endsWith('.jsonl'))) {
  554. const content = await readFile(join(dir, entry), 'utf8')
  555. expect(scrubRequestHeaders(content), `${dir}/${entry} carries request-header bulk`).toBe(content)
  556. expect(content, `${dir}/${entry} carries a run-local rpcId`)
  557. .not.toMatch(/"rpcId":"(?!\{\{rpcId\}\})[^"]+"/)
  558. }
  559. }
  560. /**
  561. * Console tripwires: reconnect/gap-repair self-healing or a pageerror must
  562. * fail the scenario, not mask a dead wire behind eventual consistency.
  563. * @param page - the page under test.
  564. * @returns live warning/pageerror collectors to assert empty at scenario end.
  565. */
  566. export function watchConsole(page: Page): { warnings: string[]; pageErrors: string[] } {
  567. const warnings: string[] = []
  568. const pageErrors: string[] = []
  569. page.on('console', (message) => {
  570. const text = message.text()
  571. if (/connection lost|gap repair|discontinuous/i.test(text)) warnings.push(text)
  572. })
  573. page.on('pageerror', (error) => { pageErrors.push(String(error)) })
  574. return { warnings, pageErrors }
  575. }
  576. /**
  577. * Remove only connection-loss warnings emitted after an intentional reload.
  578. * Earlier warnings and all gap-repair/discontinuity warnings remain fatal.
  579. * @param tripwire - the live console-warning collector.
  580. * @param warningStart - warning count captured immediately before reloading.
  581. */
  582. export function acknowledgeReloadConnectionLoss(
  583. tripwire: ReturnType<typeof watchConsole>,
  584. warningStart: number,
  585. ): void {
  586. const reloadWarnings = tripwire.warnings.splice(warningStart)
  587. tripwire.warnings.push(...reloadWarnings.filter(text => !/connection lost/i.test(text)))
  588. }