mirror of
https://github.com/deepseek-ai/deepseek-harness.git
synced 2026-08-29 04:26:38 +00:00
884 lines
36 KiB
TypeScript
884 lines
36 KiB
TypeScript
/**
|
|
* Node half of the client module system (`dsh.client` dual-face package): scans
|
|
* the host Loader's entries for packages declaring `dsh.client`, composes the
|
|
* `window.__DSH_BOOT__` entry graph (wire single source: {@link WebBootEntry}
|
|
* in `./client/manifest.ts`) in module-graph order, serves one-or-more-plugin
|
|
* combo scripts plus their source maps,
|
|
* contributes the registration facade, application preloads, bootstrap scripts,
|
|
* and graph to the webserver's index injection table, and provides the
|
|
* `clientModuleHost` service (the HMR node half's registration/notification
|
|
* face).
|
|
*
|
|
* Scanning is incremental per package — there is no full-rescan code path.
|
|
* Every cordis `internal/plugin` emission (fiber construction/disposal) marks
|
|
* the fiber's entry name dirty; a microtask flush reconciles each dirty name
|
|
* against the live loader entries. The activation pass seeds the same dirty
|
|
* set with all current entries and flushes synchronously, so first scan and
|
|
* steady state share one implementation. Package metadata (including the
|
|
* negative "not a client package" verdict) is cached per name and never
|
|
* expires — plugin-set changes take effect on restart; bundle content
|
|
* changes reach the graph only through
|
|
* {@link ClientModuleRegistry.rebuilt}.
|
|
* @module @deepseek-ai/dsh-client-modules
|
|
*/
|
|
|
|
import { createHash, randomBytes } from 'node:crypto'
|
|
import { readFileSync, statSync } from 'node:fs'
|
|
import type { IncomingMessage, ServerResponse } from 'node:http'
|
|
import { createRequire } from 'node:module'
|
|
import { dirname, join } from 'node:path'
|
|
import { Service } from '@deepseek-ai/cordis'
|
|
import type { Context } from '@deepseek-ai/cordis'
|
|
import type {} from '@deepseek-ai/cordis-plugin-loader'
|
|
import type { IndexInjection } from '@deepseek-ai/dsh-host-webserver'
|
|
import { optionalStringArray, stripClientSuffix } from './client/manifest.ts'
|
|
import type { WebBootBatch, WebBootBatchPhase, WebBootEntry, WebBootGraph } from './client/manifest.ts'
|
|
|
|
export { stripClientSuffix } from './client/manifest.ts'
|
|
export type {
|
|
BootManifest, BootModuleRow, BootPluginRow, WebBootBatch, WebBootBatchPhase, WebBootEntry, WebBootGraph,
|
|
} from './client/manifest.ts'
|
|
|
|
declare module '@deepseek-ai/cordis' {
|
|
interface Context {
|
|
/** The web plugin table (provided by the client-modules node half). */
|
|
clientModules: ClientModuleRegistry
|
|
}
|
|
}
|
|
|
|
/** package.json `dsh.client` declaration fields, validated one by one after reading the file. */
|
|
interface DshClientDeclaration {
|
|
inject?: string[]
|
|
platform: string
|
|
/** Boot phase-one registration barrier; absent rows still ride the shared application batch. */
|
|
immediately?: boolean
|
|
/**
|
|
* Exact module-table requests beyond the implicit client baseline. Any
|
|
* specifier is valid, including subpaths such as `<pkg>/client`; each
|
|
* importing package declares its own exceptional requests. A type-only
|
|
* import is not a request because the transform erases it before resolution.
|
|
* Absent means the package uses only the baseline externals.
|
|
*/
|
|
external?: string[]
|
|
}
|
|
|
|
/** The declared fields a graph row carries, normalized (absent array declarations become empty). */
|
|
interface WebBootRowFields {
|
|
inject?: string[]
|
|
/** Module specifiers the package requests from the module table. */
|
|
external: string[]
|
|
immediately: boolean
|
|
}
|
|
|
|
/** Filesystem baseline captured before a client artifact snapshot is read. */
|
|
export interface ClientArtifactBaseline {
|
|
/** Absolute path of the client bundle. */
|
|
readonly path: string
|
|
/** Bundle modification time in milliseconds. */
|
|
readonly mtimeMs: number
|
|
/** Bundle size in bytes. */
|
|
readonly size: number
|
|
}
|
|
|
|
/** Resolved package metadata for one `dsh.client` package (cached per name, never expires). */
|
|
interface PkgMeta extends WebBootRowFields {
|
|
clientPath: string
|
|
}
|
|
|
|
/** Recovery instruction shared by grouped startup and steady-state bundle diagnostics. */
|
|
const CLIENT_BUNDLE_BUILD_INSTRUCTION = 'run `pnpm run build` before launch'
|
|
|
|
/** Missing built client export, retained as structured data for activation-error grouping. */
|
|
class MissingClientBundleError extends Error {
|
|
constructor(
|
|
readonly packageName: string,
|
|
readonly clientPath: string,
|
|
cause: unknown,
|
|
) {
|
|
super(
|
|
[
|
|
`client-modules: client bundle not found; ${CLIENT_BUNDLE_BUILD_INSTRUCTION}:`,
|
|
` package: ${packageName}`,
|
|
` path: ${clientPath}`,
|
|
].join('\n'),
|
|
{ cause },
|
|
)
|
|
}
|
|
}
|
|
|
|
/** Activation failures grouped by actionable package-build errors and unrelated failures. */
|
|
class ClientPackageCompositionError extends AggregateError {
|
|
constructor(failures: Error[]) {
|
|
const missingBundles = failures.filter((error): error is MissingClientBundleError => error instanceof MissingClientBundleError)
|
|
const otherFailures = failures.filter(error => !(error instanceof MissingClientBundleError))
|
|
const packageNoun = failures.length === 1 ? 'package' : 'packages'
|
|
const lines = [`client-modules: ${String(failures.length)} client ${packageNoun} failed to compose:`]
|
|
if (missingBundles.length > 0) {
|
|
lines.push(` client bundles not found; ${CLIENT_BUNDLE_BUILD_INSTRUCTION}:`)
|
|
for (const error of missingBundles) {
|
|
lines.push(` - package: ${error.packageName}`, ` path: ${error.clientPath}`)
|
|
}
|
|
}
|
|
if (otherFailures.length > 0) {
|
|
lines.push(' other failures:', ...otherFailures.map(error => ` - ${error.message}`))
|
|
}
|
|
super(failures, lines.join('\n'))
|
|
}
|
|
}
|
|
|
|
/** One composed table row: the wire entry plus the resolved package metadata behind it. */
|
|
interface WebPluginRecord {
|
|
entry: WebBootEntry
|
|
meta: PkgMeta
|
|
/** Exact build artifact included in the startup batches. */
|
|
bundle: Buffer
|
|
/** Pre-read filesystem baseline handed to the HMR watcher. */
|
|
baseline: ClientArtifactBaseline
|
|
/** Optional authored source map snapshot; generated-file identity mapping is the fallback. */
|
|
sourceMap?: { body: Buffer; parsed: Record<string, unknown> }
|
|
}
|
|
|
|
/** Fields shared by every generated combo response. */
|
|
interface ComboArtifactBase {
|
|
url: string
|
|
rev: string
|
|
entries: string[]
|
|
script: Buffer
|
|
}
|
|
|
|
/** One generated combo response over an ordered list of plugin resources. */
|
|
interface ComboArtifact extends ComboArtifactBase {
|
|
sourceMap: Buffer
|
|
sourceMapUrl: string
|
|
}
|
|
|
|
/** One generated initial-load response and its wire descriptor. */
|
|
type BatchArtifact = ComboArtifact & { descriptor: WebBootBatch }
|
|
|
|
/** Versioned code is immutable; mismatched revisions are rejected instead of serving newer bytes. */
|
|
const IMMUTABLE_CACHE = 'public, max-age=31536000, immutable'
|
|
/** Generated request URLs stay below conservative browser and intermediary request-target limits. */
|
|
const MAX_COMBO_URL_BYTES = 3 * 1024
|
|
const HASH_REVISION_LENGTH = 12
|
|
const COMBO_REVISION_PLACEHOLDER = '0'.repeat(HASH_REVISION_LENGTH)
|
|
|
|
/** Source-map trailer emitted by tsdown at the end of every client bundle. */
|
|
const SOURCE_MAP_TRAILER = /(?:\r?\n)?\/\/# sourceMappingURL=[^\r\n]*(?:\r?\n)?$/
|
|
/** Debugger source name appended to page bundles in the WebWorker image. */
|
|
const SOURCE_URL_TRAILER = /(?:\r?\n)?\/\/# sourceURL=([^\r\n]+)(?:\r?\n)?$/
|
|
|
|
/** Narrow an unknown parsed JSON value to the `dsh.client` declaration, throwing on malformed fields. */
|
|
function parseDshClient(pkgName: string, value: unknown): DshClientDeclaration | undefined {
|
|
if (value === undefined) return undefined
|
|
if (typeof value !== 'object' || value === null) {
|
|
throw new Error(`client-modules: ${pkgName} has a non-object dsh.client declaration`)
|
|
}
|
|
const decl = value as Record<string, unknown>
|
|
if (typeof decl.platform !== 'string') {
|
|
throw new Error(`client-modules: ${pkgName} dsh.client.platform must be a string`)
|
|
}
|
|
const inject = optionalStringArray(pkgName, 'dsh.client.inject', decl.inject)
|
|
const external = optionalStringArray(pkgName, 'dsh.client.external', decl.external)
|
|
if (decl.immediately !== undefined && typeof decl.immediately !== 'boolean') {
|
|
throw new Error(`client-modules: ${pkgName} dsh.client.immediately must be a boolean`)
|
|
}
|
|
return {
|
|
platform: decl.platform,
|
|
...(inject !== undefined ? { inject } : {}),
|
|
...(external !== undefined ? { external } : {}),
|
|
...(decl.immediately !== undefined ? { immediately: decl.immediately } : {}),
|
|
}
|
|
}
|
|
|
|
/** Resolve `exports["./client"]` to a relative path, accepting the string and one-level conditional forms. */
|
|
function clientExportOf(pkgName: string, exportsField: unknown): string | undefined {
|
|
if (typeof exportsField !== 'object' || exportsField === null) return undefined
|
|
const client = (exportsField as Record<string, unknown>)['./client']
|
|
if (client === undefined) return undefined
|
|
if (typeof client === 'string') return client
|
|
if (typeof client === 'object' && client !== null) {
|
|
const fallback = (client as Record<string, unknown>).default
|
|
if (typeof fallback === 'string') return fallback
|
|
}
|
|
throw new Error(`client-modules: ${pkgName} exports["./client"] must be a string or an object with a string default`)
|
|
}
|
|
|
|
/** sha1 content hash shortened to 12 hex chars (combo / graph / rebuilt-artifact rev). */
|
|
function shortHash(input: string | Buffer): string {
|
|
return createHash('sha1').update(input).digest('hex').slice(0, HASH_REVISION_LENGTH)
|
|
}
|
|
|
|
/** Hash several response fields without allowing bytes to move across field boundaries. */
|
|
function framedHash(domain: string, parts: readonly Buffer[]): string {
|
|
const hash = createHash('sha1').update(domain).update('\0')
|
|
for (const part of parts) hash.update(`${String(part.byteLength)}:`).update(part)
|
|
return hash.digest('hex').slice(0, HASH_REVISION_LENGTH)
|
|
}
|
|
|
|
/** Hash every artifact input served after HMR observes one plugin change. */
|
|
function artifactRevision(bundle: Buffer, sourceMap: WebPluginRecord['sourceMap']): string {
|
|
return framedHash('plugin-artifact', sourceMap === undefined ? [bundle] : [bundle, sourceMap.body])
|
|
}
|
|
|
|
/** Address one ordered plugin-file list through the shared combo route. */
|
|
function comboUrl(ids: readonly string[], rev: string, sourceMap = false): string {
|
|
const resources = ids.map(id => `${id}/client.js${sourceMap ? '.map' : ''}`).join(',')
|
|
return `/plugins/??${resources}&rev=${rev}`
|
|
}
|
|
|
|
/** Measure the longer map-form URL used to partition a startup resource list. */
|
|
function projectedComboUrlBytes(records: readonly WebPluginRecord[]): number {
|
|
return Buffer.byteLength(comboUrl(
|
|
records.map(record => record.entry.id),
|
|
COMBO_REVISION_PLACEHOLDER,
|
|
true,
|
|
))
|
|
}
|
|
|
|
/** Partition one phase in graph order without allowing a generated URL above the protocol limit. */
|
|
function partitionComboRecords(records: readonly WebPluginRecord[]): WebPluginRecord[][] {
|
|
const chunks: WebPluginRecord[][] = []
|
|
let current: WebPluginRecord[] = []
|
|
for (const record of records) {
|
|
const candidate = [...current, record]
|
|
if (projectedComboUrlBytes(candidate) <= MAX_COMBO_URL_BYTES) {
|
|
current = candidate
|
|
continue
|
|
}
|
|
if (current.length === 0) {
|
|
throw new Error(
|
|
`client-modules: ${record.entry.id} exceeds the ${String(MAX_COMBO_URL_BYTES)}-byte combo URL limit`,
|
|
)
|
|
}
|
|
chunks.push(current)
|
|
current = [record]
|
|
if (projectedComboUrlBytes(current) > MAX_COMBO_URL_BYTES) {
|
|
throw new Error(
|
|
`client-modules: ${record.entry.id} exceeds the ${String(MAX_COMBO_URL_BYTES)}-byte combo URL limit`,
|
|
)
|
|
}
|
|
}
|
|
if (current.length > 0) chunks.push(current)
|
|
return chunks
|
|
}
|
|
|
|
/** Executable source plus the generated-file name used when no authored map exists. */
|
|
interface ComboSource {
|
|
source: string
|
|
fallbackSource: string
|
|
}
|
|
|
|
/** Remove bundle-local debug directives and retain their stable generated-file name. */
|
|
function comboSource(record: WebPluginRecord): ComboSource {
|
|
let source = record.bundle.toString('utf8')
|
|
const sourceUrl = SOURCE_URL_TRAILER.exec(source)?.[1]
|
|
source = source.replace(SOURCE_URL_TRAILER, '').replace(SOURCE_MAP_TRAILER, '')
|
|
if (!source.endsWith('\n')) source += '\n'
|
|
const fallbackSource = sourceUrl === undefined
|
|
? `/plugins/${record.entry.id}/client.js`
|
|
: /^(?:[A-Za-z][A-Za-z\d+.-]*:|\/)/.test(sourceUrl) ? sourceUrl : `/${sourceUrl}`
|
|
return { source, fallbackSource }
|
|
}
|
|
|
|
/** Stamp a combo script's absolute indexed-map URL onto its executable bytes. */
|
|
function comboScript(input: string, sourceMapUrl?: string): Buffer {
|
|
return Buffer.from(sourceMapUrl === undefined ? input : `${input}//# sourceMappingURL=${sourceMapUrl}\n`)
|
|
}
|
|
|
|
/** Parse an optional source-map artifact; missing maps do not prevent plugin execution. */
|
|
function sourceMapSnapshot(clientPath: string): WebPluginRecord['sourceMap'] {
|
|
let body: Buffer
|
|
try {
|
|
body = readFileSync(`${clientPath}.map`)
|
|
} catch (error) {
|
|
if ((error as NodeJS.ErrnoException).code === 'ENOENT') return undefined
|
|
throw error
|
|
}
|
|
const value = JSON.parse(body.toString('utf8')) as unknown
|
|
const parsed = typeof value === 'object' && value !== null ? value as Record<string, unknown> : undefined
|
|
if (
|
|
parsed === undefined
|
|
|| parsed.version !== 3
|
|
|| !Array.isArray(parsed.sources)
|
|
|| parsed.sources.some(source => typeof source !== 'string')
|
|
|| !Array.isArray(parsed.names)
|
|
|| parsed.names.some(name => typeof name !== 'string')
|
|
|| typeof parsed.mappings !== 'string'
|
|
) {
|
|
throw new Error(`client-modules: ${clientPath}.map is not a regular Source Map v3 object`)
|
|
}
|
|
return { body, parsed }
|
|
}
|
|
|
|
/** Count generated lines while assembling indexed-map section offsets. */
|
|
function newlineCount(value: string): number {
|
|
let count = 0
|
|
for (const char of value) if (char === '\n') count += 1
|
|
return count
|
|
}
|
|
|
|
/** Resolve section sources against their original per-plugin map URL before combo relocation. */
|
|
function comboSectionMap(record: WebPluginRecord): Record<string, unknown> {
|
|
const original = record.sourceMap?.parsed
|
|
/* v8 ignore next -- callers add sections only for records with a source map. */
|
|
if (original === undefined) throw new Error(`client-modules: source map missing for ${record.entry.id}`)
|
|
const sourcePaths = original.sources as string[]
|
|
const sourceRoot = typeof original.sourceRoot === 'string' ? original.sourceRoot : ''
|
|
const base = new URL(`/plugins/${record.entry.id}/client.js.map`, 'http://dsh.invalid')
|
|
const relocated = sourcePaths.map((source) => {
|
|
const separator = sourceRoot !== '' && !sourceRoot.endsWith('/') && !source.startsWith('/') ? '/' : ''
|
|
const resolved = new URL(`${sourceRoot}${separator}${source}`, base)
|
|
return resolved.origin === base.origin
|
|
? `${resolved.pathname}${resolved.search}${resolved.hash}`
|
|
: resolved.href
|
|
})
|
|
const section: Record<string, unknown> = { ...original, sources: relocated }
|
|
delete section.sourceRoot
|
|
return section
|
|
}
|
|
|
|
/** Map each generated line to the same line in a bundled JavaScript source. */
|
|
function identitySectionMap(source: string, sourceUrl: string): Record<string, unknown> {
|
|
const mappings = Array.from({ length: newlineCount(source) }, (_, index) => index === 0 ? 'AAAA' : 'AACA')
|
|
.join(';')
|
|
return {
|
|
version: 3,
|
|
names: [],
|
|
sources: [sourceUrl],
|
|
sourcesContent: [source],
|
|
mappings,
|
|
}
|
|
}
|
|
|
|
/** Concatenate one or more factory registrations and compose their maps as indexed sections. */
|
|
function buildCombo(records: readonly WebPluginRecord[], revision?: string): ComboArtifact {
|
|
let source = ''
|
|
const sections: { offset: { line: number; column: 0 }; map: Record<string, unknown> }[] = []
|
|
let line = 0
|
|
for (const record of records) {
|
|
const prepared = comboSource(record)
|
|
const section = record.sourceMap === undefined
|
|
? identitySectionMap(prepared.source, prepared.fallbackSource)
|
|
: comboSectionMap(record)
|
|
sections.push({ offset: { line, column: 0 }, map: section })
|
|
const bundle = `${prepared.source};\n`
|
|
source += bundle
|
|
line += newlineCount(bundle)
|
|
}
|
|
const sourceMap = Buffer.from(`${JSON.stringify({ version: 3, file: 'client.js', sections })}\n`)
|
|
const sourceBytes = Buffer.from(source)
|
|
const rev = revision ?? framedHash('combo', [sourceBytes, sourceMap])
|
|
const entries = records.map(record => record.entry.id)
|
|
const url = comboUrl(entries, rev)
|
|
const sourceMapUrl = comboUrl(entries, rev, true)
|
|
return { url, rev, entries, script: comboScript(source, sourceMapUrl), sourceMap, sourceMapUrl }
|
|
}
|
|
|
|
/** Add initial-load scheduling metadata to a combo artifact. */
|
|
function buildBatch(phase: WebBootBatchPhase, records: readonly WebPluginRecord[]): BatchArtifact {
|
|
const artifact = buildCombo(records)
|
|
return {
|
|
...artifact,
|
|
descriptor: { phase, url: artifact.url, rev: artifact.rev, entries: artifact.entries },
|
|
}
|
|
}
|
|
|
|
/** Graph row for one bundle rev (url carries the rev as its cache-busting query). */
|
|
function graphRow(id: string, rev: string, fields: WebBootRowFields): WebBootEntry {
|
|
return {
|
|
id,
|
|
url: comboUrl([id], rev),
|
|
rev,
|
|
...(fields.inject !== undefined ? { inject: fields.inject } : {}),
|
|
...(fields.immediately ? { immediately: true } : {}),
|
|
...(fields.external.length > 0 ? { external: fields.external } : {}),
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Order composed rows so every requested dynamic package precedes its
|
|
* consumers. An `external` specifier is either the package row it names
|
|
* (`<pkg>/client` aliases the bare package) or a static-table name that adds no
|
|
* graph edge.
|
|
* @param entries - composed rows in scan order.
|
|
* @returns the same rows reordered; scan order breaks every tie.
|
|
* @throws {Error} when a row requests itself or when the module graph has a
|
|
* cycle; the message lists the packages on it.
|
|
*/
|
|
export function orderByModuleGraph(entries: readonly WebBootEntry[]): WebBootEntry[] {
|
|
const rowsById = new Map<string, WebBootEntry>()
|
|
for (const entry of entries) rowsById.set(entry.id, entry)
|
|
const ordered: WebBootEntry[] = []
|
|
const placed = new Set<string>()
|
|
const open: string[] = []
|
|
const visit = (entry: WebBootEntry): void => {
|
|
if (placed.has(entry.id)) return
|
|
const cycleStart = open.indexOf(entry.id)
|
|
if (cycleStart !== -1) {
|
|
throw new Error(
|
|
`client-modules: module graph cycle ${[...open.slice(cycleStart), entry.id].join(' -> ')} `
|
|
+ '— a requested package row must precede its consumers, and factory-form CJS cannot deliver partial exports',
|
|
)
|
|
}
|
|
open.push(entry.id)
|
|
for (const name of entry.external ?? []) {
|
|
const dependency = rowsById.get(name) ?? rowsById.get(stripClientSuffix(name))
|
|
if (dependency === entry) {
|
|
throw new Error(
|
|
`client-modules: "${entry.id}" requests module "${name}" that it answers itself `
|
|
+ '— a row must not declare its own package in dsh.client.external',
|
|
)
|
|
}
|
|
if (dependency !== undefined) visit(dependency)
|
|
}
|
|
open.pop()
|
|
placed.add(entry.id)
|
|
ordered.push(entry)
|
|
}
|
|
for (const entry of entries) visit(entry)
|
|
return ordered
|
|
}
|
|
|
|
/** Bootstrap package whose ordinary client bundle supplies the module-system implementation. */
|
|
const CLIENT_MODULES_ID = '@deepseek-ai/dsh-client-modules'
|
|
|
|
/** Dynamic bundles grouped into the parser bootstrap batch before the Vite shell. */
|
|
const PARSER_PRELOAD_IDS = [CLIENT_MODULES_ID] as const
|
|
|
|
/**
|
|
* The boot protocol as index injection rows. The inline registration queue
|
|
* precedes the application-batch preload and the blocking bootstrap batch. Its
|
|
* `create()` method materializes the modules
|
|
* bundle, delegates construction to that bundle, and leaves the same facade
|
|
* in live-registration mode. The graph global follows before the shell reads
|
|
* it.
|
|
* @param graph - the composed entry graph.
|
|
* @returns head rows in execution order: queue script, application preloads,
|
|
* blocking bootstrap scripts, graph global.
|
|
*/
|
|
export function bootInjections(graph: WebBootGraph): IndexInjection[] {
|
|
const bootstrapId = JSON.stringify(CLIENT_MODULES_ID)
|
|
const queue = `(()=>{
|
|
const pendingQueue=[]
|
|
window.__ModuleLoader__={
|
|
mode:"queue",
|
|
pendingQueue,
|
|
load(registration){pendingQueue.push(registration)},
|
|
create(options){
|
|
if(this.mode!=="queue")throw new Error("client-modules: window.__ModuleLoader__.create called after module-system boot")
|
|
const index=pendingQueue.findIndex(registration=>registration.id===${bootstrapId})
|
|
const registration=pendingQueue[index]
|
|
if(registration===undefined)throw new Error("client-modules: HTML did not preload ${CLIENT_MODULES_ID}/client.js")
|
|
pendingQueue.splice(index,1)
|
|
const exports=registration.factory(specifier=>{
|
|
throw new Error('client-modules: ${CLIENT_MODULES_ID}/client.js requested external "'+specifier+'" before the module system existed')
|
|
})
|
|
if(typeof exports!=="object"||exports===null||typeof exports.createClientModuleSystem!=="function"||typeof exports.apply!=="function"){
|
|
throw new Error("client-modules: ${CLIENT_MODULES_ID}/client.js did not export the bootstrap module face")
|
|
}
|
|
return exports.createClientModuleSystem(this,{id:registration.id,exports},options)
|
|
}
|
|
}
|
|
})()`
|
|
const bootstrap = graph.batches.filter(batch => batch.phase === 'bootstrap')
|
|
const application = graph.batches.filter(batch => batch.phase === 'application')
|
|
const rows: IndexInjection[] = [{ kind: 'script', placement: 'head', text: queue }]
|
|
for (const batch of application) {
|
|
rows.push({ kind: 'script-preload', src: batch.url })
|
|
}
|
|
for (const batch of bootstrap) {
|
|
rows.push({ kind: 'script-src', placement: 'head', src: batch.url })
|
|
}
|
|
rows.push({ kind: 'global', name: '__DSH_BOOT__', value: graph })
|
|
return rows
|
|
}
|
|
|
|
/**
|
|
* The web plugin table service: incremental `dsh.client` scan + wire composition
|
|
* + bundle route + index injection rows. Construction runs the activation scan
|
|
* synchronously — a malformed declaration or missing bundle among the
|
|
* already-loaded entries aggregates into one loud throw (FAILED fiber; the
|
|
* boot activation audit reports it).
|
|
*/
|
|
export class ClientModuleRegistry extends Service {
|
|
static inject = ['webServer', 'loader']
|
|
|
|
private readonly table = new Map<string, WebPluginRecord>()
|
|
// Negative verdicts (unresolvable specifier — builtins like cordis:include,
|
|
// subpath rows — or a package without a web `dsh.client` declaration) are
|
|
// cached as null and never expire: plugin-set changes take effect on restart.
|
|
private readonly pkgMeta = new Map<string, PkgMeta | null>()
|
|
private readonly rebuildListeners = new Set<(id: string, rev: string) => void>()
|
|
private readonly graphListeners = new Set<() => void>()
|
|
private readonly dirty = new Set<string>()
|
|
private readonly resolvePkgJson: (spec: string) => string
|
|
private readonly initialRevisionNonce = randomBytes(8).toString('hex')
|
|
private nextInitialRevision = 0
|
|
private responses = new Map<string, { body: Buffer; contentType: string }>()
|
|
private batchResponses = new Map<string, { body: Buffer; contentType: string }>()
|
|
/** One prior graph generation covers a request racing the HMR recomposition that replaced its URL. */
|
|
private previousBatchResponses = new Map<string, { body: Buffer; contentType: string }>()
|
|
private flushQueued = false
|
|
private composed: WebBootGraph
|
|
|
|
/**
|
|
* Build the service: subscribe, seed, and run the activation flush.
|
|
* @param ctx - plugin context carrying webServer and loader.
|
|
*/
|
|
constructor(ctx: Context) {
|
|
super(ctx, 'clientModules')
|
|
// Resolution anchor: the config tree's baseUrl (the cordis.yml directory,
|
|
// whose package declares every composed plugin as a dependency). The
|
|
// modules package's own URL would miss sibling packages under pnpm's
|
|
// isolated node_modules.
|
|
if (ctx.baseUrl === undefined) {
|
|
throw new Error('client-modules: ctx.baseUrl is unset — the node half needs the config-tree anchor to resolve plugin packages')
|
|
}
|
|
const require = createRequire(ctx.baseUrl)
|
|
this.resolvePkgJson = spec => require.resolve(`${spec}/package.json`)
|
|
|
|
// Subscribe before seeding so a fiber arriving mid-activation lands in the
|
|
// same dirty set (Set idempotence makes the overlap harmless). An entry-less
|
|
// fiber is a child plugin or a manual mount — never a loader row; O(1) drop.
|
|
ctx.on('internal/plugin', (fiber) => {
|
|
const entryName = fiber.entry?.options.name
|
|
if (entryName === undefined) return
|
|
this.dirty.add(entryName)
|
|
if (this.flushQueued) return
|
|
this.flushQueued = true
|
|
queueMicrotask(() => {
|
|
this.flushQueued = false
|
|
this.flush((err) => { ctx.logger.warn(err) })
|
|
})
|
|
})
|
|
|
|
// Activation pass: the initial scan IS the incremental path over the
|
|
// current entries, flushed synchronously (nothing async between subscribe,
|
|
// seed, and flush).
|
|
for (const entry of ctx.loader.entries()) this.dirty.add(entry.options.name)
|
|
this.composed = this.compose()
|
|
const failures: Error[] = []
|
|
this.flush(err => failures.push(err))
|
|
if (failures.length > 0) {
|
|
throw new ClientPackageCompositionError(failures)
|
|
}
|
|
|
|
ctx.effect(
|
|
() => ctx.webServer.register({ kind: 'prefix', path: '/plugins', handler: this.serveBundle }),
|
|
'client-modules: bundle route',
|
|
)
|
|
ctx.on('webserver/index-inject', (table) => {
|
|
table.push(...bootInjections(this.composed))
|
|
})
|
|
}
|
|
|
|
/**
|
|
* Current composed entry graph (stable object between changes).
|
|
* @returns the graph served as `window.__DSH_BOOT__`.
|
|
*/
|
|
graph(): WebBootGraph {
|
|
return this.composed
|
|
}
|
|
|
|
/**
|
|
* Absolute path of an entry's client bundle.
|
|
* @param id - entry id (package name).
|
|
* @returns the path, or undefined for an unknown id.
|
|
*/
|
|
clientPath(id: string): string | undefined {
|
|
return this.table.get(id)?.meta.clientPath
|
|
}
|
|
|
|
/**
|
|
* Filesystem baseline captured before an entry's current bytes were read.
|
|
* HMR compares it with the live files when installing a watch, so a write
|
|
* between startup composition and watch installation cannot disappear into
|
|
* the watcher's initial state.
|
|
* @param id - entry id (package name).
|
|
* @returns the path and baseline, or undefined for an unknown id.
|
|
*/
|
|
artifactBaseline(id: string): ClientArtifactBaseline | undefined {
|
|
const baseline = this.table.get(id)?.baseline
|
|
return baseline === undefined ? undefined : { ...baseline }
|
|
}
|
|
|
|
/**
|
|
* Re-hash one bundle (the HMR watch's registration hook — the only entry
|
|
* point through which bundle content changes reach the graph).
|
|
* @param id - entry id (package name).
|
|
* @returns the new rev, or undefined for an unknown id.
|
|
*/
|
|
rebuilt(id: string): string | undefined {
|
|
const record = this.table.get(id)
|
|
if (record === undefined) return undefined
|
|
const baseline = this.captureArtifactBaseline(record.meta.clientPath)
|
|
const bundle = readFileSync(record.meta.clientPath)
|
|
const sourceMap = this.readSourceMapSnapshot(record.meta.clientPath)
|
|
const rev = artifactRevision(bundle, sourceMap)
|
|
record.baseline = baseline
|
|
if (rev === record.entry.rev) return rev
|
|
record.entry = graphRow(id, rev, record.meta)
|
|
record.bundle = bundle
|
|
if (sourceMap === undefined) delete record.sourceMap
|
|
else record.sourceMap = sourceMap
|
|
this.composed = this.compose()
|
|
for (const notify of this.rebuildListeners) {
|
|
// Containment: rebuilt() runs inside the HMR watch callback — a
|
|
// throwing subscriber must not kill the poll or skip later subscribers.
|
|
try {
|
|
notify(id, rev)
|
|
} catch (error) {
|
|
this.ctx.logger.error(error)
|
|
}
|
|
}
|
|
this.notifyGraphChanged()
|
|
return rev
|
|
}
|
|
|
|
/**
|
|
* Subscribe to bundle rebuilds; fires only when the re-hash changed the rev.
|
|
* @param listener - receives the entry id and its new bundle rev.
|
|
* @returns the unsubscriber.
|
|
*/
|
|
onRebuilt(listener: (id: string, rev: string) => void): () => void {
|
|
this.rebuildListeners.add(listener)
|
|
return () => { this.rebuildListeners.delete(listener) }
|
|
}
|
|
|
|
/**
|
|
* Fires after any flush that recomposed the graph (row added/removed, or a
|
|
* rebuilt rev change). Pull model: listeners re-read {@link graph}.
|
|
* @param listener - notified with no payload.
|
|
* @returns the unsubscriber.
|
|
*/
|
|
onGraphChanged(listener: () => void): () => void {
|
|
this.graphListeners.add(listener)
|
|
return () => { this.graphListeners.delete(listener) }
|
|
}
|
|
|
|
private compose(): WebBootGraph {
|
|
const entries = orderByModuleGraph([...this.table.values()].map(record => record.entry))
|
|
const bootstrap = PARSER_PRELOAD_IDS
|
|
.map(id => this.table.get(id))
|
|
.filter((record): record is WebPluginRecord => record !== undefined)
|
|
const bootstrapIds = new Set(bootstrap.map(record => record.entry.id))
|
|
const application = entries
|
|
.filter(entry => !bootstrapIds.has(entry.id))
|
|
.map(entry => this.table.get(entry.id))
|
|
.filter((record): record is WebPluginRecord => record !== undefined)
|
|
const artifacts: BatchArtifact[] = []
|
|
for (const records of partitionComboRecords(bootstrap)) {
|
|
artifacts.push(buildBatch('bootstrap', records))
|
|
}
|
|
for (const records of partitionComboRecords(application)) {
|
|
artifacts.push(buildBatch('application', records))
|
|
}
|
|
|
|
const batchResponses = new Map<string, { body: Buffer; contentType: string }>()
|
|
for (const artifact of artifacts) {
|
|
batchResponses.set(artifact.descriptor.url, {
|
|
body: artifact.script,
|
|
contentType: 'text/javascript; charset=utf-8',
|
|
})
|
|
batchResponses.set(artifact.sourceMapUrl, {
|
|
body: artifact.sourceMap,
|
|
contentType: 'application/json; charset=utf-8',
|
|
})
|
|
}
|
|
const responses = new Map(batchResponses)
|
|
for (const record of this.table.values()) {
|
|
const artifact = buildCombo([record], record.entry.rev)
|
|
responses.set(artifact.url, {
|
|
body: artifact.script,
|
|
contentType: 'text/javascript; charset=utf-8',
|
|
})
|
|
responses.set(artifact.sourceMapUrl, {
|
|
body: artifact.sourceMap,
|
|
contentType: 'application/json; charset=utf-8',
|
|
})
|
|
}
|
|
this.previousBatchResponses = this.batchResponses
|
|
this.batchResponses = batchResponses
|
|
this.responses = responses
|
|
const batches = artifacts.map(artifact => artifact.descriptor)
|
|
return { rev: shortHash(JSON.stringify({ entries, batches })), entries, batches }
|
|
}
|
|
|
|
private notifyGraphChanged(): void {
|
|
for (const listener of this.graphListeners) {
|
|
// A throwing subscriber must not skip later subscribers (or escape into
|
|
// whatever triggered the flush — possibly an fs.watchFile callback).
|
|
try {
|
|
listener()
|
|
} catch (error) {
|
|
this.ctx.logger.error(error)
|
|
}
|
|
}
|
|
}
|
|
|
|
private resolveMeta(pkgName: string): PkgMeta | null {
|
|
const cached = this.pkgMeta.get(pkgName)
|
|
if (cached !== undefined) return cached
|
|
let pkgPath: string
|
|
try {
|
|
pkgPath = this.resolvePkgJson(pkgName)
|
|
} catch {
|
|
// Not a resolvable package root: loader builtins (cordis:include) and
|
|
// subpath entries (…/gateway) land here — permanently not a client row.
|
|
this.pkgMeta.set(pkgName, null)
|
|
return null
|
|
}
|
|
const pkg = JSON.parse(readFileSync(pkgPath, 'utf8')) as Record<string, unknown>
|
|
const dsh = pkg.dsh
|
|
const decl = parseDshClient(
|
|
pkgName,
|
|
dsh !== null && typeof dsh === 'object' ? (dsh as Record<string, unknown>).client : undefined,
|
|
)
|
|
if (decl === undefined || decl.platform !== 'web') {
|
|
this.pkgMeta.set(pkgName, null)
|
|
return null
|
|
}
|
|
const clientRel = clientExportOf(pkgName, pkg.exports)
|
|
if (clientRel === undefined) {
|
|
throw new Error(`client-modules: ${pkgName} declares dsh.client but exports no "./client" bundle`)
|
|
}
|
|
const meta: PkgMeta = {
|
|
clientPath: join(dirname(pkgPath), clientRel),
|
|
...(decl.inject !== undefined ? { inject: decl.inject } : {}),
|
|
external: decl.external ?? [],
|
|
immediately: decl.immediately === true,
|
|
}
|
|
this.pkgMeta.set(pkgName, meta)
|
|
return meta
|
|
}
|
|
|
|
/** Capture the bundle stats before reading its bytes. */
|
|
private captureArtifactBaseline(clientPath: string): ClientArtifactBaseline {
|
|
const bundle = statSync(clientPath)
|
|
return {
|
|
path: clientPath,
|
|
mtimeMs: bundle.mtimeMs,
|
|
size: bundle.size,
|
|
}
|
|
}
|
|
|
|
/** Allocate an opaque initial row revision without inspecting artifact bytes. */
|
|
private allocateInitialRevision(): string {
|
|
return `${this.initialRevisionNonce}-${String(this.nextInitialRevision++)}`
|
|
}
|
|
|
|
/**
|
|
* Read the activation-time bundle and optional source-map snapshots.
|
|
* @param pkgName - package that declares the client bundle.
|
|
* @param clientPath - absolute path of the built client artifact.
|
|
* @returns the immutable bytes plus the pre-read filesystem baseline.
|
|
* @throws {MissingClientBundleError} when the read fails with `ENOENT`; other filesystem errors are rethrown unchanged.
|
|
*/
|
|
private initialBundleSnapshot(pkgName: string, clientPath: string): {
|
|
bundle: Buffer
|
|
baseline: ClientArtifactBaseline
|
|
sourceMap?: WebPluginRecord['sourceMap']
|
|
} {
|
|
try {
|
|
const baseline = this.captureArtifactBaseline(clientPath)
|
|
const bundle = readFileSync(clientPath)
|
|
const sourceMap = this.readSourceMapSnapshot(clientPath)
|
|
return { bundle, baseline, ...(sourceMap === undefined ? {} : { sourceMap }) }
|
|
} catch (error) {
|
|
if ((error as NodeJS.ErrnoException).code !== 'ENOENT') throw error
|
|
throw new MissingClientBundleError(pkgName, clientPath, error)
|
|
}
|
|
}
|
|
|
|
/** Treat a missing, torn, or malformed development map as an identity-mapped artifact revision. */
|
|
private readSourceMapSnapshot(clientPath: string): WebPluginRecord['sourceMap'] {
|
|
try {
|
|
return sourceMapSnapshot(clientPath)
|
|
} catch (error) {
|
|
this.ctx.logger.warn(error)
|
|
return undefined
|
|
}
|
|
}
|
|
|
|
/** Reconcile one entry name against the live loader entries. @returns whether the table changed. */
|
|
private processOne(entryName: string): boolean {
|
|
let qualifies = false
|
|
for (const entry of this.ctx.loader.entries()) {
|
|
if (entry.options.name === entryName && entry.fiber !== undefined && !entry.disabled) {
|
|
qualifies = true
|
|
break
|
|
}
|
|
}
|
|
if (!qualifies) return this.table.delete(entryName)
|
|
if (this.table.has(entryName)) return false
|
|
const meta = this.resolveMeta(entryName)
|
|
if (meta === null) return false
|
|
// The opaque initial rev rides the row until HMR observes a file change;
|
|
// a fiber restart reuses the existing row without inspecting bytes.
|
|
const snapshot = this.initialBundleSnapshot(entryName, meta.clientPath)
|
|
const rev = this.allocateInitialRevision()
|
|
this.table.set(entryName, {
|
|
entry: graphRow(entryName, rev, meta),
|
|
meta,
|
|
bundle: snapshot.bundle,
|
|
baseline: snapshot.baseline,
|
|
...(snapshot.sourceMap === undefined ? {} : { sourceMap: snapshot.sourceMap }),
|
|
})
|
|
return true
|
|
}
|
|
|
|
private flush(onError: (err: Error) => void): void {
|
|
let changed = false
|
|
for (const entryName of [...this.dirty]) {
|
|
this.dirty.delete(entryName)
|
|
try {
|
|
if (this.processOne(entryName)) changed = true
|
|
} catch (error) {
|
|
// Steady state: one broken package must not poison the others; the
|
|
// activation pass aggregates these into a loud throw instead.
|
|
onError(error instanceof Error ? error : new Error(String(error)))
|
|
}
|
|
}
|
|
if (!changed) return
|
|
let composed: WebBootGraph
|
|
try {
|
|
composed = this.compose()
|
|
} catch (error) {
|
|
// An unorderable module graph is a property of the whole table, not of
|
|
// the arriving package, so it surfaces here: aggregated into the
|
|
// activation throw, or warned in steady state while the last orderable
|
|
// graph stays served.
|
|
onError(error as Error)
|
|
return
|
|
}
|
|
this.composed = composed
|
|
this.notifyGraphChanged()
|
|
}
|
|
|
|
private readonly serveBundle = (req: IncomingMessage, res: ServerResponse): void => {
|
|
if (req.method !== 'GET' && req.method !== 'HEAD') {
|
|
res.writeHead(405)
|
|
res.end()
|
|
return
|
|
}
|
|
/* v8 ignore next -- `?? '/'` arm: node:http always sets url on server requests. */
|
|
const requestUrl = new URL(req.url ?? '/', 'http://x')
|
|
const resourceUrl = `${requestUrl.pathname}${requestUrl.search}`
|
|
const response = this.responses.get(resourceUrl) ?? this.previousBatchResponses.get(resourceUrl)
|
|
if (response !== undefined) {
|
|
res.writeHead(200, {
|
|
'content-type': response.contentType,
|
|
'cache-control': IMMUTABLE_CACHE,
|
|
})
|
|
res.end(req.method === 'HEAD' ? undefined : response.body)
|
|
return
|
|
}
|
|
// Anything else under /plugins (including unadvertised combinations and
|
|
// /plugins/events when the HMR row is absent) is an unknown resource.
|
|
res.writeHead(404)
|
|
res.end()
|
|
}
|
|
}
|
|
|
|
export default ClientModuleRegistry
|