scaffold.ts 56 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200
  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 dsh-base and dsh-web-app bundle
  4. // patches over the empty profile root through the vendored Loader (the same
  5. // layer stack the profile boot composes), patched the
  6. // snapshot way — so a real chromium exercises the real HTTP uplink/WebSocket
  7. // downlink, api-gateway, agent loop, tools, and persistence. Modes ride $DSH_SNAPSHOT:
  8. // replay (default, keyless: normally disables the llm-deepseek row and
  9. // inserts dsh-llm-replay in providers mode), record (real adapter + key,
  10. // harvests fixtures from live session memory), refresh (keyless replay that
  11. // rewrites goldens). A first-run option keeps the real adapter mounted while
  12. // masking its credential, without making a model call.
  13. //
  14. // Composition divergences from `dsh web`, all deliberate, all via include
  15. // patches after the shipped bundle layers, over the SAME tree (never a
  16. // second yml): temp persistenceRoot; host-level skill roots confined to the
  17. // temp workspace while project skill discovery remains real; agent-instructions
  18. // disabled (recorded fixtures must not embed this repo's AGENTS.md);
  19. // session-title-llm disabled (its fire-and-forget title call would race the
  20. // loop for the session's replay cursor); webserver pinned to port 0 with the
  21. // built dist; ordinary keyless modes disable llm-deepseek and fill the open
  22. // llm seam post-boot with installLlmReplay on the settled root ctx
  23. // (the plugin-row path discards the ReplayHandle; the direct install keeps
  24. // assertConsumed for the teardown fixture-consumption check).
  25. import { existsSync, readFileSync } from 'node:fs'
  26. import { createHash } from 'node:crypto'
  27. import { mkdir, mkdtemp, readFile, readdir, realpath, rm, writeFile } from 'node:fs/promises'
  28. import { tmpdir } from 'node:os'
  29. import { basename, dirname, join, resolve } from 'node:path'
  30. import { pathToFileURL } from 'node:url'
  31. import type { Page } from 'playwright'
  32. import { expect } from 'vitest'
  33. import { Context } from '@deepseek-ai/cordis'
  34. import Loader from '@deepseek-ai/cordis-plugin-loader'
  35. import Include, { type PatchOptions } from '@deepseek-ai/cordis-plugin-include'
  36. import Group from '@deepseek-ai/cordis-plugin-group'
  37. import {
  38. captureExpectedWorkspaceSnapshot,
  39. captureWorkspaceSnapshot,
  40. formatSystemPromptSnapshot,
  41. formatToolSchemasSnapshot,
  42. normalizedSystemPrompts,
  43. normalizedToolSchemas,
  44. parseSnapshotManifest,
  45. redactSessionSnapshotIds,
  46. normalizeSessionSnapshots,
  47. scrubRequestHeaders,
  48. scrubSessionSnapshot,
  49. stabilizeFixtureMessageIds,
  50. type NormalizeContext,
  51. } from '@deepseek-ai/dsh-session-snapshot'
  52. import {
  53. assertEntriesLoaded,
  54. composeEntries,
  55. healProfilesModuleFallback,
  56. loadOverlayPatches,
  57. } from '@deepseek-ai/dsh-app-boot'
  58. import { dshHomePath } from '@deepseek-ai/dsh-home-paths'
  59. import { settingsNamespace } from '@deepseek-ai/dsh-settings'
  60. import { LlmAdapter } from '@deepseek-ai/dsh-llm'
  61. import type {
  62. LlmModelInfo, LlmProviderInfo, LlmResolvedModelInfo, RetryPolicyConfig, StreamChunk,
  63. } from '@deepseek-ai/dsh-llm'
  64. import type { ReplayHandle } from '@deepseek-ai/dsh-llm-replay'
  65. import { installLlmReplay, parseSessionLog } from '@deepseek-ai/dsh-llm-replay'
  66. import SessionStore, {
  67. packChunkRuns,
  68. SESSION_FORMAT_VERSION,
  69. SessionId,
  70. type Session,
  71. type SessionEvent,
  72. type SessionHeader,
  73. } from '@deepseek-ai/dsh-session'
  74. import JsonlSessionPersistence from '@deepseek-ai/dsh-session-persistence-jsonl'
  75. // Empty type imports carry the webServer/agents/sessionPersistence Context merges.
  76. import type {} from '@deepseek-ai/dsh-host-webserver'
  77. import type {} from '@deepseek-ai/dsh-agent'
  78. import { provideCmdline } from '@deepseek-ai/dsh-cmdline'
  79. import { REPO_ROOT, requireDist } from './support.ts'
  80. // Host-side web e2e cannot import a browser package: doing so would pull that
  81. // package's complete TS project into this graph. Mirrored from
  82. // packages/client/ui-settings-models/src/onboarding-copy.ts; drift makes the
  83. // default pre-acknowledgement stop suppressing the notice and fails loudly.
  84. // import {
  85. // WELCOME_NOTICE_ACK_FIELD, WELCOME_NOTICE_SETTINGS_NAMESPACE,
  86. // WELCOME_NOTICE_VERSION, WELCOME_NOTICE_COPY,
  87. // } from '@deepseek-ai/dsh-client-ui-settings-models'
  88. export const WELCOME_NOTICE_SETTINGS_NAMESPACE = 'ui-onboarding'
  89. export const WELCOME_NOTICE_ACK_FIELD = 'welcomeNoticeVersion'
  90. export const WELCOME_NOTICE_VERSION = '2026-08-13.1'
  91. export const WELCOME_NOTICE_COPY = {
  92. zh: {
  93. title: '内测声明',
  94. body: 'DeepSeek Harness 目前的 0.1 版本仍处在面向 Harness 开发者进行测试的阶段,还有许多地方需要持续改进和打磨,希望听取广大开发者的反馈建议。预计 DeepSeek Harness 的核心插件以及基础 API 都会在接下来的一段时间内快速迭代、持续演化。\n\n我们期待与全球开发者一起,在开源、开放、可复用、可组合的基础设施之上,共同探索智能上限。欢迎全球 Harness 开发者加入 DSH 插件生态。',
  95. continueLabel: '继续',
  96. },
  97. } as const
  98. /** Snapshot mode for the lane, from $DSH_SNAPSHOT (same vocabulary as the other snapshot suites). */
  99. export type WebSnapshotMode = 'replay' | 'record' | 'refresh'
  100. /**
  101. * Resolve and validate the lane's snapshot mode.
  102. * @returns the active mode; unset/empty selects replay.
  103. */
  104. export function webSnapshotMode(): WebSnapshotMode {
  105. const value = process.env.DSH_SNAPSHOT
  106. if (value === undefined || value === '' || value === 'replay') return 'replay'
  107. if (value === 'record' || value === 'refresh') return value
  108. throw new Error(`DSH_SNAPSHOT must be replay, record, or refresh; got ${JSON.stringify(value)}`)
  109. }
  110. /**
  111. * Compare a session-driven Web scenario's complete workspace with its committed independent expected state.
  112. * @param scenarioDir - Absolute recorded-session scenario directory.
  113. * @param workspaceRoot - Absolute cwd used by the controlled session.
  114. */
  115. export async function assertFinalWorkspaceSnapshot(scenarioDir: string, workspaceRoot: string): Promise<void> {
  116. const manifestPath = join(scenarioDir, 'snapshot.yml')
  117. const manifest = parseSnapshotManifest(await readFile(manifestPath, 'utf8'), manifestPath)
  118. expect(manifest.workspace?.final, `${manifest.scenario ?? scenarioDir}: mutating Web scenario declares workspace.final`)
  119. .toBe(true)
  120. const actual = await captureWorkspaceSnapshot(workspaceRoot)
  121. const expected = await captureExpectedWorkspaceSnapshot(join(scenarioDir, 'workspace.expected'))
  122. expect(actual, `${manifest.scenario ?? scenarioDir}: complete final workspace`).toEqual(expected)
  123. }
  124. async function ownsReplayFixture(replayFixture: string | undefined): Promise<boolean> {
  125. if (replayFixture === undefined || basename(replayFixture) !== 'session.jsonl') return false
  126. const manifestPath = join(dirname(replayFixture), 'snapshot.yml')
  127. if (!existsSync(manifestPath)) return false
  128. const manifest = parseSnapshotManifest(await readFile(manifestPath, 'utf8'), manifestPath)
  129. return manifest.session === undefined
  130. }
  131. /** The shipped composition under test: the dsh-base and dsh-web-app bundle patches over the empty profile root. */
  132. const BASE_PATCH_PATH = join(REPO_ROOT, 'packages/bundle/base/cordis.patch.yml')
  133. const WEB_PATCH_PATH = join(REPO_ROOT, 'packages/bundle/web-app/cordis.patch.yml')
  134. /** The installation anchor whose dependency surface the profile module fallback mirrors. */
  135. const INSTALL_ANCHOR = join(REPO_ROOT, 'apps/cli/package.json')
  136. // Replay publishes the provider catalog the gateway routes to (providers
  137. // mode, never catch-all: with llm-deepseek disabled no adapter exists, so a
  138. // catch-all would leave resolveModelInfo unroutable and compaction-basic's
  139. // post-step pressure check would warn every step). The published
  140. // contextWindow keeps that pressure path provably inert for small fixtures.
  141. const REPLAY_PROVIDERS = [{
  142. id: 'deepseek-official',
  143. name: 'DeepSeek',
  144. models: [{ id: 'deepseek-v4-flash', name: 'DeepSeek-V4-Flash', contextWindow: 128_000 }],
  145. }]
  146. /**
  147. * The routes a shipped composition always has, with no ability to stream.
  148. * A fixture-less keyless scenario issues no model calls, but its tree must
  149. * still answer `listProviders()` — surfaces legitimately gate on whether any
  150. * adapter serves a session's route, and an empty registry is a test artifact,
  151. * not a product state.
  152. */
  153. class RouteOnlyAdapter extends LlmAdapter {
  154. constructor(private readonly providers: typeof REPLAY_PROVIDERS) {
  155. super()
  156. }
  157. override providerInfo(provider: string): LlmProviderInfo {
  158. return { id: provider, name: this.providers.find(entry => entry.id === provider)?.name ?? provider }
  159. }
  160. override listModels(provider: string): Promise<readonly LlmModelInfo[]> {
  161. return Promise.resolve((this.providers.find(entry => entry.id === provider)?.models ?? [])
  162. .map(model => ({ provider, id: model.id, name: model.name })))
  163. }
  164. override resolveModel(provider: string, model: string): Promise<LlmResolvedModelInfo> {
  165. const listed = this.providers.find(entry => entry.id === provider)?.models
  166. .find(entry => entry.id === model)
  167. return Promise.resolve({
  168. provider,
  169. id: model,
  170. name: listed?.name ?? model,
  171. ...listed?.contextWindow === undefined ? {} : { contextWindow: listed.contextWindow },
  172. })
  173. }
  174. override async *stream(): AsyncIterable<StreamChunk> {
  175. throw new Error(
  176. 'web e2e scaffold: a model call was issued by a scenario that declared no replay fixture'
  177. + ' — pass replayFixture, or keep the scenario free of model calls',
  178. )
  179. }
  180. }
  181. function replayProviders(contextWindow: number | undefined): typeof REPLAY_PROVIDERS {
  182. if (contextWindow === undefined) return REPLAY_PROVIDERS
  183. return REPLAY_PROVIDERS.map(provider => ({
  184. ...provider,
  185. models: provider.models.map(model => ({ ...model, contextWindow })),
  186. }))
  187. }
  188. /** A booted web scaffold: real composition, mode-selected model backend, temp world. */
  189. export interface WebScaffold {
  190. /** The active snapshot mode this scaffold booted under. */
  191. mode: WebSnapshotMode
  192. /** Browser-facing origin for the bound test server. */
  193. baseUrl: string
  194. /** Process-token URL that establishes this scaffold's browser session. */
  195. authenticatedUrl: string
  196. /** Settled root context (the in-process readiness barrier; headless event subscription is its sanctioned use). */
  197. ctx: Context
  198. /** Temp project directory sessions run in (shell/fs tool cwd). */
  199. workspaceCwd: string
  200. /** Temp persistence root (seeded sessions land here through the real API). */
  201. persistenceRoot: string
  202. /** Isolated harness home the settings/credentials rows write ($DSH_HOME double). */
  203. harnessHome: string
  204. /** Send a browser-equivalent Host request with this scaffold's authenticated cookie. */
  205. hostFetch(path: string, init?: RequestInit): Promise<Response>
  206. /** Await a settled turn end: in-process turn/end, then the agent's idle flip (which follows the persistence flush). */
  207. whenTurnSettled(timeoutMs?: number): Promise<SessionId>
  208. /**
  209. * Tear everything down; asserts the replay fixture was fully consumed first
  210. * (replay/refresh), unless booted with replayProvidersOnly (whose fixture
  211. * is validated call-free at boot).
  212. */
  213. close(): Promise<void>
  214. }
  215. /** Options for {@link launchWebScaffold}. */
  216. export interface LaunchOptions {
  217. /** Compare the replayed root session with `replayFixture`; defaults on for a manifest-owned canonical recording. */
  218. compareReplaySession?: boolean
  219. /**
  220. * Optional product overlay applied after the shipped Web surface and before
  221. * the scaffold's hermetic test patches, matching the launcher's `--patch`
  222. * ordering.
  223. */
  224. extraOverlayPath?: string
  225. /**
  226. * Replay fixture (session.jsonl) served by the inserted dsh-llm-replay row
  227. * in replay/refresh modes; ignored in record mode (the real adapter
  228. * answers). Omit for scenarios issuing no model calls — a stray stream then
  229. * fails loud with NO_ADAPTER (llm-deepseek is disabled and no replay row
  230. * mounts). With {@link replayProvidersOnly}, the fixture must record no
  231. * model calls (its header alone mounts the catalog).
  232. */
  233. replayFixture?: string
  234. /**
  235. * Mount the replay provider catalog (the model directory the UI shows)
  236. * without consuming any recorded script: for scenarios that never call a
  237. * model but need the real provider/model labels rendered. Requires
  238. * {@link replayFixture} whose log records no model calls, and rejects
  239. * {@link replayOverride} and {@link replayChildFixtures}; the teardown
  240. * consumption check is skipped for this mode. `replayFixture` without this
  241. * flag keeps the consumption check.
  242. */
  243. replayProvidersOnly?: boolean
  244. /**
  245. * Recorded child logs assigned in child creation order. Each child owns its
  246. * own positional replay cursor across initial and continuation turns.
  247. */
  248. replayChildFixtures?: string[]
  249. /**
  250. * Optional replay.override.json sidecar (whole-script replacement or
  251. * `{ patches }` augmentation) for throw/hang scenarios not expressible as
  252. * recorded chunks; replay/refresh only.
  253. */
  254. replayOverride?: string
  255. /**
  256. * Retry policy registered on every replay provider route, for failure-
  257. * injection scenarios that must exhaust recovery quickly instead of walking
  258. * the shared normal default's five backed-off retries; replay/refresh only.
  259. */
  260. replayRetryPolicy?: RetryPolicyConfig
  261. /** Per-chunk replay pacing (ms) so the browser observes genuinely incremental SSE; replay/refresh only. */
  262. paceMs?: number
  263. /** Synthetic model capacity for UI scenarios whose seeded history must remain uncompacted. */
  264. replayContextWindow?: number
  265. /**
  266. * Tool presentation mode patched onto the shipped `tools` row (`code`
  267. * collapses the wire to run_code + the SDK prompt section). Omit for the
  268. * yml default. The code runtime row is always in the tree, so no extra
  269. * insertion is needed.
  270. */
  271. toolsMode?: 'native' | 'code' | 'both'
  272. /**
  273. * Insert the opt-in model-facing Cordis tool provider into the shipped tree.
  274. * Record and replay use the same tool surface, so captured request headers
  275. * remain reconstructable without making the tools a product default.
  276. */
  277. cordisTools?: boolean
  278. /**
  279. * Keep the shipped DeepSeek adapter mounted while masking the process
  280. * environment's DEEPSEEK_API_KEY for this scaffold lifetime. This is the
  281. * keyless first-run configuration lane; the default disables the adapter.
  282. */
  283. deepSeekMissingCredential?: boolean
  284. /** Leave the current welcome notice pending; ordinary scenarios pre-acknowledge it before browser boot. */
  285. welcomeNoticePending?: boolean
  286. /**
  287. * Patch the shipped DeepSeek search row to a deterministic endpoint and
  288. * credential reference. Browser search scenarios keep the real provider and
  289. * credentials seam while avoiding external search traffic and ambient keys.
  290. */
  291. deepSeekSearch?: {
  292. /** Anthropic-compatible base URL; the provider appends `/messages`. */
  293. baseURL: string
  294. /** Credential reference resolved by the shipped search provider. */
  295. apiKeyEnv: string
  296. }
  297. /**
  298. * Replace the roster row the scaffold pins by default (no configured roots,
  299. * default `standard` — the plugin's own shipped presets). Supply this only
  300. * to change WHICH presets a scenario sees beyond the shipped set — a
  301. * writable user root, a different default. The patch lands after the
  302. * default, so it wins.
  303. */
  304. agentPresets?: {
  305. /** Roots to discover after the plugin's shipped root, in precedence order. */
  306. roots: { path: string; trust: 'system' | 'user' }[]
  307. /** The preset a session that names none is composed from. */
  308. default: string
  309. }
  310. /**
  311. * Mount the shipped telemetry row in FULL mode against this exporter URL
  312. * instead of disabling it. Used to pin a real backend disclosure in
  313. * assembled coverage; point the URL at a local dead endpoint so no record
  314. * leaves the process.
  315. */
  316. telemetryUrl?: string
  317. /**
  318. * Browse through a trusted non-loopback hostname that the browser resolves
  319. * to loopback (for example `*.localhost`). The test server stays bound to
  320. * 127.0.0.1; a non-resolving authority fails before Host trust is exercised.
  321. */
  322. remoteAuthority?: string
  323. /** Reuse an existing harness home so a second Host can verify user settings across origins. */
  324. harnessHome?: string
  325. }
  326. /** Dispose the booted tree and remove both owned temp roots, reporting every independent cleanup failure. */
  327. async function cleanupScaffoldWorld(ctx: Context, workspaceCwd: string, persistenceRoot: string): Promise<unknown[]> {
  328. const failures: unknown[] = []
  329. await Promise.resolve(ctx.fiber.dispose()).catch((error: unknown) => failures.push(error))
  330. await rm(workspaceCwd, { recursive: true, force: true }).catch((error: unknown) => failures.push(error))
  331. await rm(persistenceRoot, { recursive: true, force: true }).catch((error: unknown) => failures.push(error))
  332. return failures
  333. }
  334. /**
  335. * Boot the real web composition under the current snapshot mode.
  336. * @param options - replay fixture selection and pacing.
  337. * @returns the running scaffold.
  338. */
  339. export async function launchWebScaffold(options: LaunchOptions = {}): Promise<WebScaffold> {
  340. requireDist()
  341. const mode = webSnapshotMode()
  342. const compareReplaySession = options.compareReplaySession ?? await ownsReplayFixture(options.replayFixture)
  343. const browserHost = options.remoteAuthority ?? '127.0.0.1'
  344. if (mode === 'record') {
  345. // Both owning vitest configs (web unconditionally, snapshot in record
  346. // mode) load the repo-root .env before this file runs.
  347. if (process.env.DEEPSEEK_API_KEY === undefined || process.env.DEEPSEEK_API_KEY.length === 0) {
  348. throw new Error('web e2e record mode needs DEEPSEEK_API_KEY (env or repo-root .env)')
  349. }
  350. }
  351. if (mode === 'record' && options.deepSeekMissingCredential === true) {
  352. throw new Error('deepSeekMissingCredential is a keyless replay/refresh option')
  353. }
  354. const maskDeepSeekCredential = mode !== 'record' && options.deepSeekMissingCredential === true
  355. const originalDeepSeekCredential = process.env.DEEPSEEK_API_KEY
  356. let credentialEnvironmentRestored = false
  357. const restoreCredentialEnvironment = (): void => {
  358. if (credentialEnvironmentRestored || !maskDeepSeekCredential) return
  359. credentialEnvironmentRestored = true
  360. if (originalDeepSeekCredential === undefined) {
  361. Reflect.deleteProperty(process.env, 'DEEPSEEK_API_KEY')
  362. } else {
  363. process.env.DEEPSEEK_API_KEY = originalDeepSeekCredential
  364. }
  365. }
  366. const workspaceCwd = await realpath(await mkdtemp(join(tmpdir(), 'dsh-web-e2e-ws-')))
  367. // Isolated harness home: the settings/credentials rows resolve $DSH_HOME
  368. // paths at load, and an in-process boot must NEVER touch the developer's
  369. // real ~/.dsh document or credential file.
  370. const harnessHome = options.harnessHome ?? join(workspaceCwd, '.dsh-home')
  371. // Skill discovery is model-visible input, and its roots now resolve inside a
  372. // PRESET — a subtree this lane's include patches cannot reach, because the
  373. // roster mounts it directly per session rather than as a row of the booted
  374. // tree. The row's documented fallback is the environment, so pin that: the
  375. // whole scaffold lifetime, not just the boot, since presets mount when a
  376. // session is created. Without this a developer's real ~/.dsh/skills silently
  377. // enters replay requests and goldens while CI sees none. `DSH_HOME` follows
  378. // the resolved harness home so a scaffold sharing another's home — the
  379. // cross-port persistence scenario — pins the same roots the settings and
  380. // credentials rows were configured with.
  381. const skillRootEnvironment = {
  382. DSH_HOME: harnessHome,
  383. DSH_AGENTS_HOME: join(workspaceCwd, '.agents-home'),
  384. DSH_BUNDLED_SKILL_DIR: join(workspaceCwd, '.bundled-skills'),
  385. }
  386. const originalSkillRootEnvironment = Object.fromEntries(
  387. Object.keys(skillRootEnvironment).map(key => [key, process.env[key]]),
  388. )
  389. let skillRootEnvironmentRestored = false
  390. const restoreSkillRootEnvironment = (): void => {
  391. if (skillRootEnvironmentRestored) return
  392. skillRootEnvironmentRestored = true
  393. for (const [key, value] of Object.entries(originalSkillRootEnvironment)) {
  394. if (value === undefined) Reflect.deleteProperty(process.env, key)
  395. else process.env[key] = value
  396. }
  397. }
  398. Object.assign(process.env, skillRootEnvironment)
  399. let persistenceRoot: string
  400. try {
  401. persistenceRoot = await mkdtemp(join(tmpdir(), 'dsh-web-e2e-sessions-'))
  402. } catch (error) {
  403. const failures: unknown[] = [error]
  404. await rm(workspaceCwd, { recursive: true, force: true }).catch((cleanupError: unknown) => failures.push(cleanupError))
  405. restoreSkillRootEnvironment()
  406. if (failures.length > 1) throw new AggregateError(failures, 'web scaffold temp-root setup failed')
  407. throw error
  408. }
  409. if (maskDeepSeekCredential) Reflect.deleteProperty(process.env, 'DEEPSEEK_API_KEY')
  410. // The include patch set — the same layer stack the profile boot composes
  411. // (bundle patches in dsh.profile.bundles order), applied over the SAME empty root (a
  412. // patch id that stops matching a row fails the boot sweep loudly instead of
  413. // drifting).
  414. const basePatches = loadOverlayPatches('web e2e scaffold', BASE_PATCH_PATH)
  415. const surfacePatches = loadOverlayPatches('web e2e scaffold', WEB_PATCH_PATH)
  416. const extraOverlayPatches = options.extraOverlayPath === undefined
  417. ? []
  418. : loadOverlayPatches('web e2e scaffold', options.extraOverlayPath)
  419. const composedRows = composeEntries([basePatches, surfacePatches, extraOverlayPatches])
  420. const webRuntimeConfig = composedRows.find(row => row.id === 'web-runtime')?.config as {
  421. surfaceContext?: boolean
  422. } | undefined
  423. const surfaceContext = webRuntimeConfig?.surfaceContext !== false
  424. const patches: PatchOptions[] = [
  425. ...basePatches,
  426. ...surfacePatches,
  427. ...extraOverlayPatches,
  428. // The roster's shipped presets are the plugin's own, bundled inside
  429. // `dsh-agent-presets` and prepended by it. Pin only the machine-local
  430. // root away: a developer's own `~/.dsh/.agent-presets` must not be able
  431. // to change a golden.
  432. {
  433. id: 'agent-presets',
  434. config: {
  435. default: 'standard',
  436. includeUserRoot: false,
  437. },
  438. },
  439. { id: 'session-persistence-jsonl', config: { root: persistenceRoot } },
  440. // Content search is enabled here although the shipped bundles default it
  441. // off (`openAt: never`, pinned by apps/cli/tests/lazy-search-startup):
  442. // the seeded-session scenarios navigate by content search, and these e2e
  443. // runs are the assembled coverage for the opt-in search path.
  444. { id: 'session-query-sqlite', config: { path: ':memory:', openAt: 'first-search' } },
  445. // storage-json's yml root is anchored to the real $DSH_HOME; pin the row
  446. // to an absolute temp root (removed with the workspace at close) so tests
  447. // never write the user's harness home.
  448. { id: 'storage-json', config: { root: join(workspaceCwd, '.dsh-storages') } },
  449. // Skill discovery is model-visible input. Pin every host-level root inside
  450. // the owned temp world so ~/.dsh, ~/.agents, and a bundled-root env setting
  451. // cannot change replay requests or conversation goldens. Project roots stay
  452. // enabled against the same empty temp workspace, preserving the real seam.
  453. {
  454. id: 'skill-filesystem',
  455. config: {
  456. dshHome: join(workspaceCwd, '.dsh-home'),
  457. agentsHome: join(workspaceCwd, '.agents-home'),
  458. bundledSkillDir: join(workspaceCwd, '.bundled-skills'),
  459. watch: false,
  460. },
  461. },
  462. // fs/bash cwd default to process.cwd(); the gateway injects the same
  463. // value into session.cwd — chdir below anchors all three to the temp
  464. // workspace, keeping the composition untouched.
  465. { id: 'agent-instructions', disabled: true },
  466. { id: 'session-title-llm', disabled: true },
  467. // Fixture sessions must never leave the process: the shipped row defaults
  468. // to the production OTLP endpoint (or whatever DSH_TELEMETRY_OTLP_URL
  469. // names in the ambient environment). A scenario that pins a real backend
  470. // disclosure passes a local dead endpoint instead of disabling the row.
  471. options.telemetryUrl === undefined
  472. ? { id: 'session-telemetry-otel', disabled: true }
  473. : {
  474. id: 'session-telemetry-otel',
  475. config: {
  476. mode: 'FULL',
  477. exporter: { url: options.telemetryUrl },
  478. shutdownTimeoutMillis: 1_000,
  479. },
  480. },
  481. // Use an ephemeral port while preserving the shipped compression policy;
  482. // a patch replaces the row's complete config.
  483. {
  484. id: 'webserver',
  485. config: {
  486. host: '127.0.0.1', port: 0, compression: 'gzip',
  487. compressionLevel: 1, compressionThresholdBytes: 1024,
  488. },
  489. },
  490. // The bundle's web-runtime row resolves the same built dist under test
  491. // (apps/web IS @deepseek-ai/dsh-web-frontend); native browser opening and the
  492. // URL line are disabled because this scaffold owns its Playwright browser.
  493. // Preserve the composed surface-context choice because a patch replaces
  494. // the row's complete config.
  495. { id: 'web-runtime', config: { openBrowser: false, printUrl: false, surfaceContext } },
  496. ...options.remoteAuthority === undefined
  497. ? []
  498. : [{ id: 'connection', config: { trustedHosts: [options.remoteAuthority] } }],
  499. { id: 'settings', config: { dshHome: harnessHome } },
  500. { id: 'credentials', config: { dshHome: harnessHome } },
  501. // The shipped directory-picker row is the -auto chooser, which resolves
  502. // the interaction from the RUNNING host (display, SSH launch, bind). The
  503. // lane's goldens are interaction-specific (workspace-management drives
  504. // the in-app browse dialog), so pin -browse deterministically on every
  505. // host: patch `name` is an assertion, not an override, hence the
  506. // disable+insert pair.
  507. { id: 'directory-picker', disabled: true },
  508. { insert: [
  509. { id: 'directory-picker-browse', name: '@deepseek-ai/dsh-host-directory-picker-browse' },
  510. { id: 'ui-directory-picker-browse', name: '@deepseek-ai/dsh-client-ui-directory-picker-browse' },
  511. ] },
  512. ...options.agentPresets === undefined
  513. ? []
  514. // Never the derived harness-home root: a developer's own presets must not
  515. // be able to change a golden, whatever roots a scenario asks for.
  516. : [{ id: 'agent-presets', config: { ...options.agentPresets, includeUserRoot: false } }],
  517. ...options.toolsMode === undefined ? [] : [{ id: 'tools', config: { mode: options.toolsMode } }],
  518. // The shipped Web bundle already owns both runners and the Cordis UI. This
  519. // scenario adds only the model-facing tools that exercise those services.
  520. ...options.cordisTools === true
  521. ? [{ insert: [
  522. { id: 'tool-cordis', name: '@deepseek-ai/dsh-tool-cordis' },
  523. ] }]
  524. : [],
  525. ...options.deepSeekSearch === undefined
  526. ? []
  527. : [{
  528. id: 'web-search-deepseek',
  529. config: {
  530. apiKeyEnv: options.deepSeekSearch.apiKeyEnv,
  531. baseURL: options.deepSeekSearch.baseURL,
  532. },
  533. }],
  534. ...mode === 'record' || options.deepSeekMissingCredential === true
  535. ? []
  536. : [{ id: 'llm-deepseek', disabled: true }],
  537. ]
  538. // Sessions inherit the gateway's process.cwd() default; run the boot from
  539. // the temp workspace so tool cwd, session cwd, and fixtures agree.
  540. const originalCwd = process.cwd()
  541. const ctx = new Context()
  542. const observedSessions = new Map<SessionId, Session>()
  543. const stopObservingSessions = ctx.on('session/created', (session) => {
  544. observedSessions.set(session.id, session)
  545. })
  546. let port = 0
  547. let baseUrl = ''
  548. let authenticatedUrl = ''
  549. let cookieHeader = ''
  550. let replayHandle: ReplayHandle | undefined
  551. try {
  552. process.chdir(workspaceCwd)
  553. // The production module-resolution setup: an empty profile root inside the temp
  554. // harness home, with bare plugin names resolving through the flat module
  555. // fallback the launcher heals under <home>/profiles.
  556. await healProfilesModuleFallback(INSTALL_ANCHOR, harnessHome)
  557. const profileDir = join(harnessHome, 'profiles', 'scaffold')
  558. await mkdir(profileDir, { recursive: true })
  559. const rootConfig = join(profileDir, 'cordis.yml')
  560. await writeFile(rootConfig, '[]\n')
  561. ctx.baseUrl = pathToFileURL(profileDir).href + '/'
  562. // This direct Loader harness supplies the same root-path capability as app-boot.
  563. ctx.provide('dshHomePath', dshHomePath)
  564. // A host with no command line still provides one: the web bundle's startup
  565. // row releases the rows waiting on it, and with no arguments each starts on
  566. // the values this scaffold composed above. An exit request can only come
  567. // from a rejected argument, which a fixed empty list has none of.
  568. provideCmdline(ctx, {
  569. args: [],
  570. exit: (code) => {
  571. throw new Error(`web e2e scaffold: the web app requested exit ${String(code)} with no arguments to reject`)
  572. },
  573. })
  574. await ctx.plugin(Loader)
  575. ctx.loader.builtins.include = Include
  576. // `cordis:group` beside it, exactly as `boot()` registers it: a group row is
  577. // how a preset gives one `isolate` realm to a provider and its consumers,
  578. // and a preset resolving package names from its own directory cannot reach
  579. // `@deepseek-ai/cordis-plugin-group` by name.
  580. ctx.loader.builtins.group = Group
  581. await ctx.loader.create({
  582. name: 'cordis:include',
  583. config: { path: pathToFileURL(rootConfig).href, patches },
  584. })
  585. await ctx.loader.await()
  586. assertEntriesLoaded(ctx, 'web e2e scaffold')
  587. if (options.welcomeNoticePending !== true) {
  588. await ctx.settings.mutate(settingsNamespace(WELCOME_NOTICE_SETTINGS_NAMESPACE), [{
  589. op: 'set', path: [WELCOME_NOTICE_ACK_FIELD], value: WELCOME_NOTICE_VERSION,
  590. }])
  591. }
  592. const boundPort = ctx.get('webServer')?.port
  593. if (boundPort === undefined) {
  594. throw new Error('web e2e scaffold: webServer service missing after settled boot')
  595. }
  596. port = boundPort
  597. // Fill the open llm seam on the settled root ctx. Ordinary keyless modes
  598. // disable llm-deepseek; the first-run lane keeps it mounted but has no
  599. // replay fixture and never streams. The direct install, unlike the plugin
  600. // row, returns the ReplayHandle for the teardown consumption check.
  601. if (options.replayProvidersOnly) {
  602. if (options.replayFixture === undefined) {
  603. throw new Error('replayProvidersOnly requires replayFixture (its file supplies the header)')
  604. }
  605. const fixtureText = readFileSync(options.replayFixture, 'utf8')
  606. // The consumption check is skipped for this mode, so no script source
  607. // may carry callable entries: reject override/child sources outright
  608. // and any call-bearing fixture.
  609. if (options.replayOverride !== undefined || options.replayChildFixtures !== undefined) {
  610. throw new Error('replayProvidersOnly cannot combine with replayOverride or replayChildFixtures')
  611. }
  612. // A fixture without a session header row must not mount the catalog
  613. // silently: the consumption-skip assumes the header-only shape.
  614. let headerType: unknown
  615. try {
  616. headerType = (JSON.parse(fixtureText.trimStart().split('\n', 1)[0] ?? '') as { type?: unknown }).type
  617. } catch {
  618. headerType = undefined
  619. }
  620. if (headerType !== 'session') {
  621. throw new Error('replayProvidersOnly fixture must open with a session header row')
  622. }
  623. const recorded = parseSessionLog(fixtureText)
  624. const hasModelCall = recorded.some(event => (
  625. event.type === 'assistant/chunk' || event.type === 'request/header' || event.type === 'tool/call'
  626. ))
  627. if (hasModelCall) {
  628. throw new Error('replayProvidersOnly fixture must record no model calls')
  629. }
  630. }
  631. if (mode !== 'record' && options.replayFixture !== undefined) {
  632. replayHandle = installLlmReplay(ctx, {
  633. file: options.replayFixture,
  634. providers: replayProviders(options.replayContextWindow).map(provider => ({
  635. ...provider,
  636. ...(options.replayRetryPolicy === undefined ? {} : { retryPolicy: options.replayRetryPolicy }),
  637. })),
  638. ...(options.replayOverride === undefined ? {} : { overrideFile: options.replayOverride }),
  639. ...(options.replayChildFixtures === undefined ? {} : { childFiles: options.replayChildFixtures }),
  640. ...(options.paceMs === undefined ? {} : { paceMs: options.paceMs }),
  641. })
  642. } else if (mode !== 'record' && options.deepSeekMissingCredential !== true) {
  643. // No fixture and no shipped adapter would leave the tree with ZERO
  644. // provider routes — a state no product composition has, and one the
  645. // composer refuses to type into. Register the same routes
  646. // a fixture would, with streaming that still fails loud: the scenario
  647. // issues no model calls, and one that slipped in must not pass quietly.
  648. ctx.effect(() => ctx.llm.registerAdapter(
  649. replayProviders(options.replayContextWindow).map(provider => provider.id),
  650. new RouteOnlyAdapter(replayProviders(options.replayContextWindow)),
  651. ), 'web e2e scaffold: route-only adapter')
  652. }
  653. baseUrl = `http://${browserHost}:${String(port)}`
  654. authenticatedUrl = ctx.connection.authenticatedUrl(baseUrl)
  655. const login = await fetch(authenticatedUrl, { redirect: 'manual' })
  656. const setCookie = login.headers.get('set-cookie')
  657. if (login.status !== 303 || login.headers.get('location') !== '/' || setCookie === null) {
  658. throw new Error('web e2e scaffold: browser token exchange did not return its session cookie')
  659. }
  660. cookieHeader = setCookie.split(';', 1)[0] ?? ''
  661. if (cookieHeader.length === 0) {
  662. throw new Error('web e2e scaffold: browser token exchange returned an empty session cookie')
  663. }
  664. } catch (error) {
  665. if (process.cwd() !== originalCwd) process.chdir(originalCwd)
  666. const cleanupFailures = await cleanupScaffoldWorld(ctx, workspaceCwd, persistenceRoot)
  667. restoreCredentialEnvironment()
  668. restoreSkillRootEnvironment()
  669. if (cleanupFailures.length > 0) {
  670. throw new AggregateError([error, ...cleanupFailures], 'web scaffold setup failed and cleanup was incomplete')
  671. }
  672. throw error
  673. } finally {
  674. if (process.cwd() !== originalCwd) process.chdir(originalCwd)
  675. }
  676. return {
  677. harnessHome,
  678. mode,
  679. baseUrl,
  680. authenticatedUrl,
  681. ctx,
  682. workspaceCwd,
  683. persistenceRoot,
  684. hostFetch(path: string, init: RequestInit = {}): Promise<Response> {
  685. const headers = new Headers(init.headers)
  686. headers.set('cookie', cookieHeader)
  687. return fetch(new URL(path, baseUrl), { ...init, headers })
  688. },
  689. // Barrier stack: the in-process turn/end identifies the session, its
  690. // explicit flush makes the transcript durable, and the caller's browser
  691. // settled-poll comes last because host completion strictly precedes render.
  692. whenTurnSettled(timeoutMs = mode === 'record' ? 180_000 : 30_000): Promise<SessionId> {
  693. return new Promise<SessionId>((resolveSettled, reject) => {
  694. const timer = setTimeout(() => {
  695. off()
  696. reject(new Error(`no turn/end within ${timeoutMs}ms`))
  697. }, timeoutMs)
  698. const off = ctx.on('session/event', (session: Session, event: SessionEvent) => {
  699. if (event.type !== 'turn/end') return
  700. clearTimeout(timer)
  701. off()
  702. ctx.sessions.flush(session)
  703. .then(() => { resolveSettled(session.id) }, reject)
  704. })
  705. })
  706. },
  707. async close(): Promise<void> {
  708. const failures: unknown[] = []
  709. if (mode !== 'record'
  710. && options.replayFixture !== undefined
  711. && options.replayProvidersOnly !== true
  712. && compareReplaySession) {
  713. try {
  714. await assertReplaySession(
  715. [...observedSessions.values()],
  716. options.replayFixture,
  717. mode,
  718. `http://${browserHost}:${port}`,
  719. )
  720. } catch (error) {
  721. failures.push(error)
  722. }
  723. }
  724. // Fixture-consumption check first, while the run's binding state is
  725. // still authoritative — a scenario that drove fewer model calls than
  726. // recorded fails here instead of drifting green. Skipped for
  727. // replayProvidersOnly, whose fixture is validated call-free at boot.
  728. if (!options.replayProvidersOnly) {
  729. try {
  730. replayHandle?.assertConsumed()
  731. } catch (error) {
  732. failures.push(error)
  733. }
  734. }
  735. try {
  736. stopObservingSessions()
  737. failures.push(...await cleanupScaffoldWorld(ctx, workspaceCwd, persistenceRoot))
  738. } finally {
  739. restoreCredentialEnvironment()
  740. restoreSkillRootEnvironment()
  741. }
  742. if (failures.length > 0) throw new AggregateError(failures, 'web scaffold teardown failed')
  743. },
  744. }
  745. }
  746. /**
  747. * Serialize a live session to the canonical raw session-JSONL layout — the
  748. * in-memory record-mode harvest, so the on-disk zstd default never matters.
  749. */
  750. function rawSessionLog(session: Session): string {
  751. return [
  752. JSON.stringify({ type: 'session', ...session.header }),
  753. ...packChunkRuns(session.events).map(record => JSON.stringify(record)),
  754. '',
  755. ].join('\n')
  756. }
  757. function normalizeWebSessionVolatiles(log: string): string {
  758. const normalizeValue = (value: unknown): unknown => {
  759. if (typeof value === 'string') {
  760. return value.replace(/Anonymous user: [^.]+(?=\. Session sharing)/g, 'Anonymous user: {{anonymousUserId}}')
  761. }
  762. if (Array.isArray(value)) return value.map(normalizeValue)
  763. if (value !== null && typeof value === 'object') {
  764. return Object.fromEntries(Object.entries(value).map(([key, item]) => [key, normalizeValue(item)]))
  765. }
  766. return value
  767. }
  768. return log.split(/\r?\n/).map((line) => {
  769. if (line.trim() === '') return line
  770. const record = normalizeValue(JSON.parse(line)) as { type?: unknown; data?: { endpoint?: unknown } }
  771. if (record.type === 'web/deepseek-search-llm-request' && typeof record.data?.endpoint === 'string') {
  772. record.data.endpoint = '{{webSearchEndpoint}}'
  773. }
  774. return JSON.stringify(record)
  775. }).join('\n')
  776. }
  777. function stableSessionFixture(session: Session, existing: string, workspaceCwd: string): string {
  778. const fresh = scrubSessionSnapshot(normalizeWebSessionVolatiles(rawSessionLog(session)))
  779. .split(session.id).join('{{sessionId}}')
  780. .split(workspaceCwd).join('{{cwd}}')
  781. const stable = redactSessionSnapshotIds(stabilizeFixtureMessageIds([fresh], [existing]))[0]
  782. if (stable === undefined) throw new Error('session harvest produced no stabilized fixture')
  783. return stable
  784. }
  785. async function assertReplaySession(
  786. sessions: readonly Session[],
  787. fixturePath: string,
  788. mode: WebSnapshotMode,
  789. webUrl: string,
  790. ): Promise<void> {
  791. let expected = await readFile(fixturePath, 'utf8')
  792. const userPrompts = fixtureUserPrompts(expected)
  793. const candidates = sessions.filter((session) => {
  794. if (session.header.parentSession !== undefined) return false
  795. const actual = session.events.flatMap((event) => {
  796. if (event.type !== 'user/message' || event.data.source.kind !== 'user') return []
  797. const text = event.data.content.filter(block => block.type === 'text').map(block => block.text).join('')
  798. return text.length === 0 ? [] : [text]
  799. })
  800. return JSON.stringify(actual) === JSON.stringify(userPrompts)
  801. })
  802. expect(candidates, `Web replay fixture ${fixturePath} must match one live root session`).toHaveLength(1)
  803. const session = candidates[0] as Session
  804. const sessionCwd = session.header.cwd
  805. if (sessionCwd === undefined) throw new Error(`${fixturePath}: replayed session has no cwd`)
  806. const actual = rawSessionLog(session)
  807. if (mode === 'refresh') {
  808. expected = stableSessionFixture(session, expected, sessionCwd)
  809. await writeFile(fixturePath, expected)
  810. }
  811. const expectedHeader = JSON.parse(expected.split('\n').find(line => line.trim() !== '') ?? '{}') as {
  812. id?: unknown
  813. cwd?: unknown
  814. }
  815. const actualContext: NormalizeContext = { sessionIds: [String(session.id)], cwd: sessionCwd }
  816. const expectedContext: NormalizeContext = {
  817. sessionIds: typeof expectedHeader.id === 'string' ? [expectedHeader.id] : [],
  818. cwd: typeof expectedHeader.cwd === 'string' ? expectedHeader.cwd : '\0no-cwd\0',
  819. }
  820. expect(normalizeSessionSnapshots([normalizeWebSessionVolatiles(actual)], actualContext)[0], `${fixturePath}: persisted replay`)
  821. .toBe(normalizeSessionSnapshots([normalizeWebSessionVolatiles(expected)], expectedContext)[0])
  822. const fixtureDir = dirname(fixturePath)
  823. const manifestPath = join(fixtureDir, 'snapshot.yml')
  824. const manifest = parseSnapshotManifest(await readFile(manifestPath, 'utf8'), manifestPath)
  825. if (manifest.header?.pin !== true) return
  826. const normalizePrompt = (value: string): string => value
  827. .split(REPO_ROOT).join('{{sourceRoot}}')
  828. .split(webUrl).join('{{webUrl}}')
  829. const prompts = normalizedSystemPrompts(actual, actualContext).map(normalizePrompt)
  830. const schemas = normalizedToolSchemas(actual, actualContext)
  831. const promptPath = join(fixtureDir, 'system-prompt.expected.md')
  832. const schemaPath = join(fixtureDir, 'tool-schemas.expected.json')
  833. const promptSnapshot = formatSystemPromptSnapshot(prompts[0] as string, prompts.slice(1))
  834. const schemaSnapshot = formatToolSchemasSnapshot(schemas[0] as unknown[], schemas.slice(1))
  835. if (mode === 'refresh') {
  836. await Promise.all([writeFile(promptPath, promptSnapshot), writeFile(schemaPath, schemaSnapshot)])
  837. }
  838. expect(promptSnapshot, `${fixturePath}: system-prompt pin`).toBe(await readFile(promptPath, 'utf8'))
  839. expect(schemaSnapshot, `${fixturePath}: tool-schema pin`).toBe(await readFile(schemaPath, 'utf8'))
  840. }
  841. /**
  842. * Record-mode fixture write-back: harvest the live session, scrub request
  843. * headers to {{system}}/{{tools}}, tokenize the run-local cwd, redact opaque
  844. * identities with typed relationship-preserving tokens, and write the fixture.
  845. * @param scaffold - the record-mode scaffold.
  846. * @param sessionId - the driven session.
  847. * @param fixturePath - the committed session.jsonl target.
  848. */
  849. export async function recordFixture(scaffold: WebScaffold, sessionId: SessionId, fixturePath: string): Promise<void> {
  850. const agent = scaffold.ctx.agents.get(sessionId)
  851. if (agent === undefined) throw new Error(`record harvest: no live agent for ${sessionId}`)
  852. const existing = existsSync(fixturePath) ? await readFile(fixturePath, 'utf8') : ''
  853. await writeFile(fixturePath, stableSessionFixture(agent.session, existing, scaffold.workspaceCwd))
  854. }
  855. /**
  856. * The user prompts recorded in a fixture, in order — the single source tying
  857. * spec drive steps to recorded reality so script and fixture cannot drift.
  858. * @param fixtureText - raw session.jsonl contents.
  859. * @returns the recorded user prompt texts.
  860. */
  861. export function fixtureUserPrompts(fixtureText: string): string[] {
  862. return parseSessionLog(fixtureText).flatMap((event) => {
  863. if (event.type !== 'user/message' || event.data.source.kind !== 'user') return []
  864. const text = event.data.content.filter(block => block.type === 'text').map(block => block.text).join('')
  865. return text.length > 0 ? [text] : []
  866. })
  867. }
  868. /** Deterministic UUID used when a seed fixture's typed identity token is materialized. */
  869. export function fixtureIdentity(
  870. kind: 'message' | 'approval' | 'workflow' | 'command' | 'rpc' | 'retry' | 'id',
  871. ordinal: number,
  872. ): string {
  873. const hex = createHash('sha256').update(`${kind}:${ordinal}`).digest('hex').slice(0, 32).split('')
  874. hex[12] = '4'
  875. hex[16] = ['8', '9', 'a', 'b'][Number.parseInt(hex[16] as string, 16) % 4] as string
  876. return `${hex.slice(0, 8).join('')}-${hex.slice(8, 12).join('')}-${hex.slice(12, 16).join('')}-${hex.slice(16, 20).join('')}-${hex.slice(20).join('')}`
  877. }
  878. /**
  879. * Seed a recorded session fixture into the scaffold's persistence root
  880. * through the REAL backend API (throwaway Context + SessionStore + JSONL
  881. * plugin — the semantic-checkpoint precedent), never raw file writes: no
  882. * knowledge of bucket hashing, filename encoding, or compression, and
  883. * malformed session events fail loud at seed time. The fixture's tokenized identity
  884. * ({{sessionId}}/{{cwd}}) is realized for this world before parsing. Event
  885. * times are materialized from event order against the fixture header's
  886. * creation time, or the seeded creation time when normalization replaced the
  887. * header value with zero.
  888. * @param scaffold - the target scaffold.
  889. * @param fixtureText - raw recorded session.jsonl contents.
  890. * @param id - the seeded session id (stable for deterministic goldens).
  891. * @param agentPreset - the preset the recorded session was composed from,
  892. * for scenarios asserting what a resumed session reports running.
  893. * @returns the seeded id.
  894. */
  895. /**
  896. * Realize a recorded seed fixture against one scaffold: substitute the
  897. * `{{sessionId}}`/`{{cwd}}` placeholders and rewrite the recorded cwd to the
  898. * scaffold's workspace. Idempotent, so a caller may realize early (e.g. to
  899. * price content exactly as the host will fold it) and still pass the result
  900. * through {@link seedSession}.
  901. * @param scaffold - the booted scaffold whose workspace the seed targets.
  902. * @param fixtureText - the committed seed fixture text.
  903. * @param id - the session id the seed is realized for.
  904. * @returns the realized fixture text.
  905. */
  906. export function realizeSeedFixture(scaffold: WebScaffold, fixtureText: string, id: string): string {
  907. const realized = fixtureText
  908. .split('{{sessionId}}').join(id)
  909. .split('{{session:1}}').join(id)
  910. .replace(/\{\{session:([2-9]\d*)\}\}/g, (_token, ordinal: string) => `${id}-child-${ordinal}`)
  911. .replace(/\{\{(message|approval|workflow|command|rpc|retry|id):([1-9]\d*)\}\}/g, (_token, kind: string, ordinal: string) =>
  912. fixtureIdentity(kind as 'message' | 'approval' | 'workflow' | 'command' | 'rpc' | 'retry' | 'id', Number(ordinal)))
  913. .split('{{cwd}}').join(scaffold.workspaceCwd)
  914. const fixtureCwd = (JSON.parse(realized.split('\n', 1)[0]!) as { cwd?: string }).cwd
  915. return fixtureCwd === undefined
  916. ? realized
  917. : realized.split(fixtureCwd).join(scaffold.workspaceCwd)
  918. }
  919. /**
  920. * Parse a committed web seed fixture through the replay reader.
  921. * @param fixtureText - session JSONL fixture contents.
  922. * @returns the original header line, parsed header, and logical events.
  923. */
  924. export function parseSeedFixture(fixtureText: string): {
  925. headerLine: string
  926. header: Record<string, unknown>
  927. events: SessionEvent[]
  928. } {
  929. const headerLine = fixtureText.split(/\r?\n/).find(line => line.trim().length > 0)
  930. if (headerLine === undefined) throw new Error('seed fixture has no session header')
  931. const header = JSON.parse(headerLine) as Record<string, unknown>
  932. if (header.type !== 'session') throw new Error('seed fixture must start with a session header')
  933. return { headerLine, header, events: parseSessionLog(fixtureText) }
  934. }
  935. /**
  936. * Render logical events as an envelope-free web seed fixture.
  937. * @param headerLine - original session header line.
  938. * @param events - logical session events in order.
  939. * @returns projected session JSONL.
  940. */
  941. export function renderSeedFixture(
  942. headerLine: string,
  943. events: readonly ({ readonly seq: number; readonly time: number } & object)[],
  944. ): string {
  945. return [
  946. headerLine,
  947. ...events.map(({ seq: _seq, time: _time, ...event }) => JSON.stringify(event)),
  948. '',
  949. ].join('\n')
  950. }
  951. export async function seedSession(
  952. scaffold: WebScaffold,
  953. fixtureText: string,
  954. id: string,
  955. agentPreset?: string,
  956. ): Promise<SessionId> {
  957. const decoded = parseSeedFixture(realizeSeedFixture(scaffold, fixtureText, id))
  958. const events = decoded.events
  959. if (events.length === 0) throw new Error('seed fixture has no events')
  960. const last = events[events.length - 1]!
  961. // An open final turn would be mutated by resume's crash repair on first
  962. // open; a committed seed must be a closed recording.
  963. if (last.type !== 'turn/end') throw new Error(`seed fixture must end in turn/end, got ${last.type}`)
  964. const meta: SessionHeader = {
  965. version: SESSION_FORMAT_VERSION,
  966. id: SessionId(id),
  967. createdAt: Date.now() - 60_000,
  968. cwd: scaffold.workspaceCwd,
  969. delegationDepth: 0,
  970. ...agentPreset === undefined ? {} : { agentPreset },
  971. }
  972. const fixtureCreatedAt = decoded.header.createdAt
  973. if (typeof fixtureCreatedAt !== 'number') {
  974. throw new Error('seed fixture requires a numeric createdAt header')
  975. }
  976. const timeAnchor = fixtureCreatedAt === 0 ? meta.createdAt : fixtureCreatedAt
  977. const materializedEvents = events.map((event, index) => ({ ...event, time: timeAnchor + index }))
  978. await persistSeedSession(scaffold, meta, materializedEvents)
  979. return meta.id
  980. }
  981. /** Seed one materialized cold Session whose log has no turn/start event. */
  982. export async function seedBlankSession(
  983. scaffold: WebScaffold,
  984. id: string,
  985. cwd: string,
  986. ): Promise<SessionId> {
  987. const meta: SessionHeader = {
  988. version: SESSION_FORMAT_VERSION,
  989. id: SessionId(id),
  990. createdAt: Date.now() - 60_000,
  991. cwd,
  992. delegationDepth: 0,
  993. }
  994. await persistSeedSession(scaffold, meta, [{
  995. type: 'session/end-seed',
  996. seq: 0,
  997. time: meta.createdAt,
  998. data: {},
  999. }])
  1000. return meta.id
  1001. }
  1002. /** Materialize one detached Session fixture through the shipped JSONL provider. */
  1003. async function persistSeedSession(
  1004. scaffold: WebScaffold,
  1005. meta: SessionHeader,
  1006. events: readonly SessionEvent[],
  1007. ): Promise<void> {
  1008. const seeder = new Context()
  1009. try {
  1010. await seeder.plugin(SessionStore)
  1011. // Same root as the booted tree with the plugin's own default compression,
  1012. // so the host's directory-scan list() sees one consistent encoding.
  1013. await seeder.plugin(JsonlSessionPersistence, { root: scaffold.persistenceRoot })
  1014. await seeder.sessionPersistence.create(meta)
  1015. await seeder.sessionPersistence.append(meta.id, events)
  1016. } finally {
  1017. await seeder.fiber.dispose()
  1018. }
  1019. }
  1020. /**
  1021. * Normalize an aria snapshot: uuid, cwd, workspace-basename, duration,
  1022. * decode-throughput, and path-sensitive compaction estimates collapse to
  1023. * stable tokens.
  1024. *
  1025. * Throughput needs a token for the same reason durations do, and no fixture
  1026. * can supply one: the figure divides a replayed step's output tokens by the
  1027. * wall time the local run took to stream them, so it moves between two runs
  1028. * on one machine (measured 69 → 70 tok/s) and swings wildly on a fast replay
  1029. * (26333 tok/s for a 3 ms stream).
  1030. */
  1031. function normalizeAria(snapshot: string, workspaceCwd: string): string {
  1032. // The session heading renders the workspace's basename, not the full
  1033. // path, so both spellings must collapse to the token.
  1034. const base = workspaceCwd.split('/').pop()!
  1035. return snapshot
  1036. .split(workspaceCwd).join('{{cwd}}')
  1037. .split(base).join('{{workspace}}')
  1038. .replace(/[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}/gi, '{{uuid}}')
  1039. // The optional space in `\d+m ?\d+s` covers both minute spellings: the
  1040. // stats line's compact `2m42s` and the message-chrome template's `2m 42s`.
  1041. .replace(
  1042. /~\d+(?:y(?: \d+mo)?|mo(?: \d+d)?)|\b(?:\d+d(?: \d+h(?: \d+m \d+s)?)?|\d+h \d+m \d+s|\d+m ?\d+s|\d+(?:\.\d+)?s|\d+(?:\.\d+)?ms)\b/g,
  1043. duration => duration.startsWith('~') ? duration : '{{duration}}',
  1044. )
  1045. .replace(/\b\d[\d,]*(?:\.\d+)? ms\b/g, '{{duration}}')
  1046. .replace(
  1047. /约\d+(?:年(?:\d+个月)?|个月(?:\d+天)?)|\d+(?:天(?:\d+小时(?:\d+分\d+秒)?)?|小时\d+分\d+秒|分\d+秒|(?:\.\d+)?秒)/g,
  1048. duration => duration.startsWith('约') ? duration : '{{duration}}',
  1049. )
  1050. .replace(/\d+(?:\.\d+)?(?= tok\/s(?!\w))/g, '{{throughput}}')
  1051. // Seeded compaction prices realized file paths, whose length differs
  1052. // between local worktrees and CI scratch directories.
  1053. .replace(/(Compacted \d+ history items \(~)\d+( tokens\))/g, '$1{{tokens}}$2')
  1054. // Session summaries and Message IconActions clocks cross calendar
  1055. // boundaries; collapse every shape so goldens stay stable across them.
  1056. .replace(/\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}(?:\.\d+)?Z/g, '{{timestamp}}')
  1057. .replace(/\d{4}年\d{1,2}月\d{1,2}日 \d{2}:\d{2}/g, '{{clock}}')
  1058. .replace(/\d{1,2}月\d{1,2}日 \d{2}:\d{2}/g, '{{clock}}')
  1059. .replace(/(?<!\d)\d{1,2}:\d{2}:\d{2}(?:\.\d+)?(?:\s*[AP]M)?(?!\d)/gi, '{{clock}}')
  1060. .replace(/(?<!\d)\d{2}:\d{2}(?!\d)/g, '{{clock}}')
  1061. }
  1062. /**
  1063. * Capture the region's aria snapshot at a settled milestone: poll until two
  1064. * consecutive normalized captures are equal — a single-shot capture races the
  1065. * last React commits.
  1066. * @param page - the page under test.
  1067. * @param selector - the region locator selector.
  1068. * @param workspaceCwd - normalization input.
  1069. * @returns the stable normalized snapshot.
  1070. */
  1071. export async function captureStableAria(page: Page, selector: string, workspaceCwd: string): Promise<string> {
  1072. const region = page.locator(selector).first()
  1073. let previous = normalizeAria(await region.ariaSnapshot(), workspaceCwd)
  1074. await expect.poll(async () => {
  1075. const current = normalizeAria(await region.ariaSnapshot(), workspaceCwd)
  1076. const stable = current === previous
  1077. previous = current
  1078. return stable
  1079. }, { timeout: 5_000, message: 'aria snapshot did not stabilize' }).toBe(true)
  1080. return previous
  1081. }
  1082. /**
  1083. * Compare a normalized golden, or rewrite it under refresh. Refresh is the
  1084. * ONLY writer: a missing golden in replay mode fails with the healing command
  1085. * instead of silently self-bootstrapping.
  1086. * @param goldenPath - the committed ui.expected.md path.
  1087. * @param actual - the stable normalized snapshot.
  1088. * @param mode - the active snapshot mode.
  1089. */
  1090. export async function compareOrRefreshGolden(goldenPath: string, actual: string, mode: WebSnapshotMode): Promise<void> {
  1091. const payload = `${actual}\n`
  1092. if (mode === 'refresh') {
  1093. await writeFile(goldenPath, payload)
  1094. return
  1095. }
  1096. if (!existsSync(goldenPath)) {
  1097. throw new Error(`missing golden ${goldenPath} — run DSH_SNAPSHOT=refresh pnpm run test:web to generate it`)
  1098. }
  1099. expect(payload).toBe(await readFile(goldenPath, 'utf8'))
  1100. }
  1101. /**
  1102. * Fixture-inventory guard: the scenario directory holds exactly the expected
  1103. * files and every committed JSONL is a header-scrubbed, typed-redaction fixed point.
  1104. * @param dir - the scenario snapshot directory.
  1105. * @param expected - the exact expected file inventory.
  1106. */
  1107. export async function assertFixtureInventory(dir: string, expected: string[]): Promise<void> {
  1108. const entries = (await readdir(dir)).sort()
  1109. const ownsManifest = entries.includes('snapshot.yml')
  1110. const artifacts = entries.filter(name => name !== 'snapshot.yml')
  1111. expect(artifacts).toEqual([...expected].sort())
  1112. if (ownsManifest) {
  1113. const manifestPath = join(dir, 'snapshot.yml')
  1114. const manifest = parseSnapshotManifest(await readFile(manifestPath, 'utf8'), manifestPath)
  1115. expect(manifest.profile).toBe('web')
  1116. if (manifest.session === undefined) {
  1117. expect(
  1118. artifacts.includes('session.jsonl'),
  1119. `${dir}: session owner must carry session.jsonl`,
  1120. ).toBe(true)
  1121. } else {
  1122. expect(existsSync(resolve(dir, manifest.session.source)), `${dir}: session source`).toBe(true)
  1123. }
  1124. }
  1125. for (const entry of artifacts.filter(name => name.endsWith('.jsonl'))) {
  1126. const content = await readFile(join(dir, entry), 'utf8')
  1127. expect(scrubRequestHeaders(content), `${dir}/${entry} carries request-header bulk`).toBe(content)
  1128. expect(redactSessionSnapshotIds([content]), `${dir}/${entry} carries unredacted identities`).toEqual([content])
  1129. }
  1130. }
  1131. /**
  1132. * Console tripwires: reconnect/gap-repair self-healing or a pageerror must
  1133. * fail the scenario, not mask a dead wire behind eventual consistency.
  1134. * @param page - the page under test.
  1135. * @returns live warning/pageerror collectors to assert empty at scenario end.
  1136. */
  1137. export function watchConsole(page: Page): { warnings: string[]; pageErrors: string[] } {
  1138. const warnings: string[] = []
  1139. const pageErrors: string[] = []
  1140. page.on('console', (message) => {
  1141. const text = message.text()
  1142. if (/connection lost|gap repair|discontinuous/i.test(text)) warnings.push(text)
  1143. })
  1144. page.on('pageerror', (error) => { pageErrors.push(String(error)) })
  1145. return { warnings, pageErrors }
  1146. }
  1147. /**
  1148. * Remove only connection-loss warnings emitted after an intentional reload.
  1149. * Earlier warnings and all gap-repair/discontinuity warnings remain fatal.
  1150. * @param tripwire - the live console-warning collector.
  1151. * @param warningStart - warning count captured immediately before reloading.
  1152. */
  1153. export function acknowledgeReloadConnectionLoss(
  1154. tripwire: ReturnType<typeof watchConsole>,
  1155. warningStart: number,
  1156. ): void {
  1157. const reloadWarnings = tripwire.warnings.splice(warningStart)
  1158. tripwire.warnings.push(...reloadWarnings.filter(text => !/connection lost/i.test(text)))
  1159. }