Spaces:
Paused
Paused
| // Simple type to replace removed WebSearchSource | |
| type SimpleSource = { | |
| title?: string; | |
| link: string; | |
| }; | |
| import { processBlocks, fallbackBlocks, type BlockToken } from "$lib/utils/marked"; | |
| export type IncomingMessage = { | |
| type: "process"; | |
| content: string; | |
| sources: SimpleSource[]; | |
| requestId: number; | |
| streaming?: boolean; | |
| }; | |
| export type OutgoingMessage = { | |
| type: "processed"; | |
| blocks: BlockToken[]; | |
| requestId: number; | |
| }; | |
| // Stateless request/response. All flow control (one job at a time per worker, per-client | |
| // coalescing) lives in the shared pool (markdownWorkerPool.ts), so the worker just | |
| // processes whatever it is handed and replies once. The pool never posts a second job to | |
| // a busy worker, hence no internal queue/buffer here. | |
| onmessage = async (event) => { | |
| const data = event.data as IncomingMessage; | |
| if (data.type !== "process") { | |
| return; | |
| } | |
| const { content, sources, requestId, streaming } = data; | |
| let blocks: BlockToken[]; | |
| try { | |
| blocks = await processBlocks(content, sources ?? [], streaming ?? false); | |
| } catch { | |
| // Never strand the pool: it waits for a reply to free this worker, so on failure | |
| // degrade to the lightweight fallback rendering instead of going silent. | |
| blocks = fallbackBlocks(content); | |
| } | |
| try { | |
| // Blocks are plain data (strings/booleans/arrays) and structured-clone | |
| // directly; the JSON round-trip below is only needed for exotic values | |
| // (e.g. an unresolved promise-valued html) that would fail to clone. | |
| postMessage({ type: "processed", blocks, requestId } as OutgoingMessage); | |
| } catch { | |
| try { | |
| postMessage( | |
| JSON.parse(JSON.stringify({ type: "processed", blocks, requestId })) as OutgoingMessage | |
| ); | |
| } catch { | |
| // A block somehow isn't JSON-safe either. Reply with the fallback so the | |
| // pool always gets a reply and the shared worker is never left wedged. | |
| // fallbackBlocks output is plain data and always serializes. | |
| postMessage({ | |
| type: "processed", | |
| blocks: fallbackBlocks(content), | |
| requestId, | |
| } as OutgoingMessage); | |
| } | |
| } | |
| }; | |