statewire 0.0.1 → 0.0.3

package/dist/transports/sse.d.ts

@@ -1,9 +1,11 @@
-import { StatewireFinish, StatewireResponse } from "../client/connection.js";
-import { StatewireSessionStore } from "../client/sessionStore.js";
-import { StatewireCommands, StatewireTransport } from "../types.js";
+import { StatewireMetaEntry } from "../wire/meta.js";
+import { StatewireHost } from "../host.js";
+import { StatewireResponse } from "../client/connection.js";
+import { StatewireClient } from "../statewire.js";
+import { Statewire } from "../types.js";
 import { ResourceElement } from "@assistant-ui/tap";
 //#region src/transports/sse.d.ts
-type StatewireTransportProps = {
+type StatewireRemoteProps = {
   clientId: string;
   onSnapshot: (args: {
     ops: unknown[];
@@ -14,11 +16,32 @@ type StatewireTransportProps = {
     res?: StatewireResponse[];
     ack?: number;
   }) => void;
-  onFinish: (args: StatewireFinish) => void;
+  onFinish: (args: StatewireHost.Fin) => void;
 };
 type StatewireWireCommand = {
   method: string;
   params: readonly unknown[];
+  context?: never;
+} | {
+  method?: never;
+  params?: never;
+  context: Record<string, unknown>;
+};
+/** The wire shape of a client context submission: one root replace op. */
+declare const contextOps: (value: Record<string, unknown>) => {
+  op: string;
+  path: never[];
+  value: Record<string, unknown>;
+}[];
+declare const encodeWireCommand: (command: StatewireWireCommand) => {
+  context?: {
+    op: string;
+    path: never[];
+    value: Record<string, unknown>;
+  }[];
+  params?: unknown[];
+  meta?: StatewireMetaEntry[];
+  method?: string;
 };
 type StatewireSubmitResult = {
   type: "receipt";
@@ -26,7 +49,6 @@ type StatewireSubmitResult = {
 } | {
   type: "rejected";
   message: string;
-  payload?: unknown;
 } | {
   type: "seq-gap";
 } | {
@@ -40,7 +62,7 @@ type ConnectionState = {
   status: "live";
 } | {
   status: "reconnecting";
-  cause: "dropped" | "ended" | "error";
+  cause: "dropped" | "evicted" | "error";
   attempt: number;
   nextRetryAt: number;
   lastError?: string;
@@ -50,25 +72,15 @@ type ConnectionState = {
   message?: string;
   payload?: unknown;
 };
-type StatewireConnection = ConnectionState & {
-  reconnect: () => void;
-};
 /** The live wire a remote session drives: connection status plus command POST. */
 type RemoteLink = {
-  connection: StatewireConnection;
+  connection: StatewireClient.Connection;
   send: (args: {
     seq: number;
     command: StatewireWireCommand;
   }) => Promise<StatewireSubmitResult>;
 };
-type StatewireSSEOptions = {
-  url: string;
-  headers?: HeadersInit | (() => HeadersInit | Promise<HeadersInit>);
-  fetch?: typeof globalThis.fetch;
-  onChange?: (connection: StatewireConnection) => void;
-  sessionStore?: StatewireSessionStore;
-};
-declare const StatewireSSE: <TState = any, TCommands extends StatewireCommands = StatewireCommands, TEvent = any>(options: StatewireSSEOptions) => ResourceElement<StatewireTransport<TState | undefined, TCommands, TEvent>>;
+declare const StatewireSSE: <TState = any, TCommands extends Statewire.Commands = Statewire.Commands, TEvent = any>(options: StatewireClient.SSEOptions) => ResourceElement<Statewire<TState | undefined, TCommands, TEvent>>;
 //#endregion
-export { ConnectionState, RemoteLink, StatewireConnection, StatewireSSE, StatewireSSEOptions, StatewireSubmitResult, StatewireTransportProps, StatewireWireCommand };
+export { ConnectionState, RemoteLink, StatewireRemoteProps, StatewireSSE, StatewireSubmitResult, StatewireWireCommand, contextOps, encodeWireCommand };
 //# sourceMappingURL=sse.d.ts.map

package/dist/transports/sse.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"sse.d.ts","names":[],"sources":["../../src/transports/sse.ts"],"mappings":";;;;;KAwBY;EACV;EACA,aAAa;IAAQ;IAAgB;;EACrC,QAAQ;IACN;IACA,MAAM;IACN;;EAEF,WAAW,MAAM;;KAGP;EACV;EACA;;KAGU;EACN;EAAiB,WAAW;;EAC5B;EAAkB;EAAiB;;EACnC;;EACA;;EACA;;KAEM;EACN;;EACA;;EAEA;EACA;EACA;EACA;EACA;EACA;;EAEA;EAAgB;EAAkB;;KAE5B,sBAAsB;EAAoB;;;KAG1C;EACV,YAAY;EACZ,OAAO;IACL;IACA,SAAS;QACL,QAAQ;;KAGJ;EACV;EACA,UAAU,qBAAqB,cAAc,QAAQ;EACrD,eAAe,WAAW;EAC1B,YAAY,YAAY;EACxB,eAAe;;cAqMJ,eACX,cACA,kBAAkB,oBAAoB,mBACtC,cAEA,SAAS,wBACN,gBAAgB,mBAAmB,oBAAoB,WAAW"}
+{"version":3,"file":"sse.d.ts","names":[],"sources":["../../src/transports/sse.ts"],"mappings":";;;;;;;KAuBY;EACV;EACA,aAAa;IAAQ;IAAgB;;EACrC,QAAQ;IACN;IACA,MAAM;IACN;;EAEF,WAAW,MAAM,cAAc;;KAGrB;EACN;EAAgB;EAA4B;;EAC5C;EAAgB;EAAgB,SAAS;;;cAGlC,aAAc,OAAO;;;SAAA;;cAIrB,oBAAqB,SAAS;;;;;;;;;;KAQ/B;EACN;EAAiB,WAAW;;EAC5B;EAAkB;;EAClB;;EACA;;EACA;;KAEM;EACN;;EACA;;EAEA;EACA;EACA;EACA;EACA;EACA;;EAEA;EAAgB;EAAkB;;;KAG5B;EACV,YAAY,gBAAgB;EAC5B,OAAO;IACL;IACA,SAAS;QACL,QAAQ;;cAgMH,eACX,cACA,kBAAkB,UAAU,WAAW,UAAU,UACjD,cAEA,SAAS,gBAAgB,eACtB,gBAAgB,UAAU,oBAAoB,WAAW"}

package/dist/transports/sse.js

@@ -1,41 +1,34 @@
 import { encodeParams, reviveResponse } from "../wire/meta.js";
 import { createSseParser } from "./sse-parser.js";
 import { timing } from "../client/timing.js";
-import { classify, createGate, handleEnvelope, overloadedError, parseRetryAfter, reconnectConn } from "../client/connection.js";
+import { classify, handleEnvelope, overloadedError, parseRetryAfter, useStatewireConn } from "../client/connection.js";
 import { useStatewireRemote } from "./remote.js";
+import { useRef, useState } from "@assistant-ui/tap/react-shim";
 import { flushTapSync, resource } from "@assistant-ui/tap";
 //#region src/transports/sse.ts
-const createConn = (latest, notify) => {
-	const c = {
-		hasSnapshot: false,
-		lease: null,
-		attempt: 0,
-		episode: null,
-		abort: null,
-		gate: createGate(),
-		lifetime: null,
-		latest,
-		notify,
-		reconnect: () => reconnectConn(c),
-		attemptOnce: (abort) => attemptStream(c, abort),
-		send: (args) => sendCommand(c, args),
-		connection: {
-			status: "connecting",
-			reconnect: () => c.reconnect()
-		}
-	};
-	return c;
-};
-const baseUrl = (c) => c.latest.current.options.url.replace(/\/$/, "");
-const requestHeaders = async (c, extra) => {
-	const { headers } = c.latest.current.options;
+/** The wire shape of a client context submission: one root replace op. */
+const contextOps = (value) => [{
+	op: "replace",
+	path: [],
+	value
+}];
+const encodeWireCommand = (command) => ({
+	...command.method !== void 0 && {
+		method: command.method,
+		...encodeParams(command.params)
+	},
+	...command.context !== void 0 && { context: contextOps(command.context) }
+});
+const baseUrl = (options) => options.url.replace(/\/$/, "");
+const requestHeaders = async (options, extra) => {
+	const { headers } = options;
 	const merged = new Headers(typeof headers === "function" ? await headers() : headers);
 	for (const [name, value] of Object.entries(extra)) merged.set(name, value);
 	return merged;
 };
-const timedFetch = async (c, input, init, timeout) => {
+const timedFetch = async (options, input, init, timeout) => {
 	try {
-		return await (c.latest.current.options.fetch ?? globalThis.fetch)(input, init);
+		return await (options.fetch ?? globalThis.fetch)(input, init);
 	} catch {
 		throw classify(timeout);
 	}
@@ -43,7 +36,6 @@ const timedFetch = async (c, input, init, timeout) => {
 const consume = async (c, reader, heartbeat, dead) => {
 	const decoder = new TextDecoder();
 	const parse = createSseParser();
-	let attached = false;
 	for (;;) {
 		const timer = setTimeout(() => heartbeat.abort(), timing.heartbeatTimeoutMs);
 		let result;
@@ -58,22 +50,21 @@ const consume = async (c, reader, heartbeat, dead) => {
 		for (const data of parse(decoder.decode(result.value, { stream: true }))) {
 			if (dead()) return "ended";
 			const env = JSON.parse(data);
-			const done = flushTapSync(() => handleEnvelope(c, env, attached));
-			attached = true;
+			const done = flushTapSync(() => handleEnvelope(c, env));
 			if (done) return done;
 		}
 	}
 };
-const attemptStream = async (c, abort) => {
-	const headers = await requestHeaders(c, {
-		"Statewire-Client-Id": c.latest.current.ctx.clientId,
+const attemptStream = async (c, options, abort) => {
+	const headers = await requestHeaders(options, {
+		"Statewire-Client-Id": c.ctx.clientId,
 		accept: "text/event-stream"
 	});
 	const heartbeat = new AbortController();
 	const headerTimer = setTimeout(() => heartbeat.abort(), timing.heartbeatTimeoutMs);
 	let res;
 	try {
-		res = await timedFetch(c, `${baseUrl(c)}/stream`, {
+		res = await timedFetch(options, `${baseUrl(options)}/stream`, {
 			headers,
 			cache: "no-store",
 			signal: AbortSignal.any([abort.signal, heartbeat.signal])
@@ -91,24 +82,29 @@ const attemptStream = async (c, abort) => {
 	}
 };
 const overloaded = (res) => overloadedError(parseRetryAfter(res.headers.get("Retry-After")));
-const sendCommand = async (c, { seq, command }) => {
+const sendCommand = async (c, options, { seq, command }) => {
 	const timeout = AbortSignal.timeout(timing.heartbeatTimeoutMs);
-	const res = await timedFetch(c, `${baseUrl(c)}/commands`, {
+	const res = await timedFetch(options, `${baseUrl(options)}/commands`, {
 		method: "POST",
-		headers: await requestHeaders(c, {
+		headers: await requestHeaders(options, {
 			"content-type": "application/json",
-			"Statewire-Client-Id": c.latest.current.ctx.clientId,
+			"Statewire-Client-Id": c.ctx.clientId,
 			"Statewire-Command-Seq": String(seq),
 			...c.lease !== null && { "Statewire-Lease": c.lease }
 		}),
-		body: JSON.stringify({
-			method: command.method,
-			...encodeParams(command.params)
-		}),
+		body: JSON.stringify(encodeWireCommand(command)),
 		signal: AbortSignal.any([c.lifetime?.signal ?? AbortSignal.abort(), timeout])
 	}, timeout);
 	if (res.ok) {
-		const body = await res.json().catch(() => null);
+		let body;
+		try {
+			body = await res.json();
+		} catch {
+			return {
+				type: "rejected",
+				message: "statewire: protocol error: malformed /commands receipt body"
+			};
+		}
 		return body?.res !== void 0 ? {
 			type: "receipt",
 			response: reviveResponse(body.res)
@@ -126,11 +122,26 @@ const sendCommand = async (c, { seq, command }) => {
 	};
 	throw new Error(`HTTP ${res.status}`);
 };
-const useStatewireSSE = (options) => {
-	return useStatewireRemote(options, createConn);
+const useSseLink = (options, ctx) => {
+	const latest = useRef(options);
+	latest.current = options;
+	const c_0 = useStatewireConn({
+		url: options.url,
+		onChange: options.onChange,
+		ctx,
+		attemptOnce: (c, abort) => attemptStream(c, options, abort)
+	});
+	const [link] = useState(() => ({
+		get connection() {
+			return c_0.connection;
+		},
+		send: (args) => sendCommand(c_0, latest.current, args)
+	}));
+	return link;
 };
+const useStatewireSSE = (options) => useStatewireRemote(options, useSseLink);
 const StatewireSSE = resource(useStatewireSSE);
 //#endregion
-export { StatewireSSE };
+export { StatewireSSE, contextOps, encodeWireCommand };
 
 //# sourceMappingURL=sse.js.map

package/dist/transports/sse.js.map

@@ -1 +1 @@
-{"version":3,"file":"sse.js","names":["flushTapSync","resource","ResourceElement","encodeParams","reviveResponse","createSseParser","timing","classify","createGate","handleEnvelope","overloadedError","parseRetryAfter","reconnectConn","ConnBase","StatewireEnvelope","StatewireFinish","StatewireResponse","StatewireTransport","useStatewireRemote","StatewireSessionStore","StatewireCommands","StatewireTransportProps","clientId","onSnapshot","args","ops","lastSeq","onOps","res","ack","onFinish","StatewireWireCommand","method","params","StatewireSubmitResult","type","response","message","payload","ConnectionState","status","cause","attempt","nextRetryAt","lastError","StatewireConnection","reconnect","RemoteLink","connection","send","seq","command","Promise","StatewireSSEOptions","url","headers","HeadersInit","fetch","globalThis","onChange","sessionStore","Latest","options","ctx","Conn","latest","current","createConn","notify","c","hasSnapshot","lease","episode","abort","gate","lifetime","attemptOnce","attemptStream","sendCommand","baseUrl","replace","requestHeaders","extra","Record","merged","Headers","name","value","Object","entries","set","timedFetch","input","init","RequestInit","signal","AbortSignal","timeout","Response","consume","reader","ReadableStreamDefaultReader","Uint8Array","heartbeat","AbortController","dead","decoder","TextDecoder","parse","attached","timer","setTimeout","heartbeatTimeoutMs","result","ReadableStreamReadResult","read","clearTimeout","done","data","decode","stream","env","JSON","accept","headerTimer","cache","any","overloaded","ok","body","Error","getReader","aborted","cancel","catch","get","String","stringify","json","undefined","verdict","const","error","useStatewireSSE","StatewireSSE","TState","TCommands","TEvent"],"sources":["../../src/transports/sse.ts"],"sourcesContent":["import { flushTapSync, resource } from \"@assistant-ui/tap\";\nimport type { ResourceElement } from \"@assistant-ui/tap\";\nimport { encodeParams, reviveResponse } from \"../wire/meta\";\nimport { createSseParser } from \"./sse-parser\";\nimport { timing } from \"../client/timing\";\nimport {\n  classify,\n  createGate,\n  handleEnvelope,\n  overloadedError,\n  parseRetryAfter,\n  reconnectConn,\n} from \"../client/connection\";\nimport type {\n  ConnBase,\n  StatewireEnvelope,\n  StatewireFinish,\n  StatewireResponse,\n} from \"../client/connection\";\nimport type { StatewireTransport } from \"../types\";\nimport { useStatewireRemote } from \"./remote\";\nimport type { StatewireSessionStore } from \"../client/sessionStore\";\nimport type { StatewireCommands } from \"../types\";\n\nexport type StatewireTransportProps = {\n  clientId: string;\n  onSnapshot: (args: { ops: unknown[]; lastSeq: number }) => void;\n  onOps: (args: {\n    ops: unknown[];\n    res?: StatewireResponse[];\n    ack?: number;\n  }) => void;\n  onFinish: (args: StatewireFinish) => void;\n};\n\nexport type StatewireWireCommand = {\n  method: string;\n  params: readonly unknown[];\n};\n\nexport type StatewireSubmitResult =\n  | { type: \"receipt\"; response?: StatewireResponse }\n  | { type: \"rejected\"; message: string; payload?: unknown }\n  | { type: \"seq-gap\" }\n  | { type: \"unknown-client\" }\n  | { type: \"stale-lease\" };\n\nexport type ConnectionState =\n  | { status: \"connecting\" }\n  | { status: \"live\" }\n  | {\n      status: \"reconnecting\";\n      cause: \"dropped\" | \"ended\" | \"error\";\n      attempt: number;\n      nextRetryAt: number;\n      lastError?: string;\n      message?: string;\n    }\n  | { status: \"gone\"; message?: string; payload?: unknown };\n\nexport type StatewireConnection = ConnectionState & { reconnect: () => void };\n\n/** The live wire a remote session drives: connection status plus command POST. */\nexport type RemoteLink = {\n  connection: StatewireConnection;\n  send: (args: {\n    seq: number;\n    command: StatewireWireCommand;\n  }) => Promise<StatewireSubmitResult>;\n};\n\nexport type StatewireSSEOptions = {\n  url: string;\n  headers?: HeadersInit | (() => HeadersInit | Promise<HeadersInit>);\n  fetch?: typeof globalThis.fetch;\n  onChange?: (connection: StatewireConnection) => void;\n  sessionStore?: StatewireSessionStore;\n};\n\ntype Latest = {\n  options: StatewireSSEOptions;\n  ctx: StatewireTransportProps;\n};\n\ntype Conn = ConnBase & {\n  latest: { readonly current: Latest };\n  send: RemoteLink[\"send\"];\n};\n\nconst createConn = (\n  latest: { readonly current: Latest },\n  notify: (connection: StatewireConnection) => void,\n): Conn => {\n  const c: Conn = {\n    hasSnapshot: false,\n    lease: null,\n    attempt: 0,\n    episode: null,\n    abort: null,\n    gate: createGate(),\n    lifetime: null,\n    latest,\n    notify,\n    reconnect: () => reconnectConn(c),\n    attemptOnce: (abort) => attemptStream(c, abort),\n    send: (args) => sendCommand(c, args),\n    connection: { status: \"connecting\", reconnect: () => c.reconnect() },\n  };\n  return c;\n};\n\nconst baseUrl = (c: Conn) => c.latest.current.options.url.replace(/\\/$/, \"\");\n\nconst requestHeaders = async (c: Conn, extra: Record<string, string>) => {\n  const { headers } = c.latest.current.options;\n  const merged = new Headers(\n    typeof headers === \"function\" ? await headers() : headers,\n  );\n  for (const [name, value] of Object.entries(extra)) merged.set(name, value);\n  return merged;\n};\n\nconst timedFetch = async (\n  c: Conn,\n  input: string,\n  init: RequestInit & { signal: AbortSignal },\n  timeout: AbortSignal,\n): Promise<Response> => {\n  try {\n    return await (c.latest.current.options.fetch ?? globalThis.fetch)(\n      input,\n      init,\n    );\n  } catch {\n    throw classify(timeout);\n  }\n};\n\nconst consume = async (\n  c: Conn,\n  reader: ReadableStreamDefaultReader<Uint8Array>,\n  heartbeat: AbortController,\n  dead: () => boolean,\n): Promise<\"stop\" | \"reattach\" | \"ended\"> => {\n  const decoder = new TextDecoder();\n  const parse = createSseParser();\n  let attached = false;\n  for (;;) {\n    const timer = setTimeout(\n      () => heartbeat.abort(),\n      timing.heartbeatTimeoutMs,\n    );\n    let result: ReadableStreamReadResult<Uint8Array>;\n    try {\n      result = await reader.read();\n    } catch {\n      throw classify(heartbeat.signal);\n    } finally {\n      clearTimeout(timer);\n    }\n    if (result.done) return \"ended\";\n    for (const data of parse(decoder.decode(result.value, { stream: true }))) {\n      if (dead()) return \"ended\";\n      const env = JSON.parse(data) as StatewireEnvelope;\n      const done = flushTapSync(() => handleEnvelope(c, env, attached));\n      attached = true;\n      if (done) return done;\n    }\n  }\n};\n\nconst attemptStream = async (\n  c: Conn,\n  abort: AbortController,\n): Promise<\"stop\" | \"reattach\" | \"ended\"> => {\n  const headers = await requestHeaders(c, {\n    \"Statewire-Client-Id\": c.latest.current.ctx.clientId,\n    accept: \"text/event-stream\",\n  });\n  const heartbeat = new AbortController();\n  const headerTimer = setTimeout(\n    () => heartbeat.abort(),\n    timing.heartbeatTimeoutMs,\n  );\n  let res: Response;\n  try {\n    res = await timedFetch(\n      c,\n      `${baseUrl(c)}/stream`,\n      {\n        headers,\n        cache: \"no-store\",\n        signal: AbortSignal.any([abort.signal, heartbeat.signal]),\n      },\n      heartbeat.signal,\n    );\n  } finally {\n    clearTimeout(headerTimer);\n  }\n  if (res.status === 429 || res.status === 503) throw overloaded(res);\n  if (!res.ok || !res.body) throw new Error(`HTTP ${res.status}`);\n  const reader = res.body.getReader();\n  try {\n    return await consume(c, reader, heartbeat, () => abort.signal.aborted);\n  } finally {\n    reader.cancel().catch(() => {});\n  }\n};\n\nconst overloaded = (res: Response) =>\n  overloadedError(parseRetryAfter(res.headers.get(\"Retry-After\")));\n\nconst sendCommand = async (\n  c: Conn,\n  { seq, command }: { seq: number; command: StatewireWireCommand },\n): Promise<StatewireSubmitResult> => {\n  const timeout = AbortSignal.timeout(timing.heartbeatTimeoutMs);\n  const res = await timedFetch(\n    c,\n    `${baseUrl(c)}/commands`,\n    {\n      method: \"POST\",\n      headers: await requestHeaders(c, {\n        \"content-type\": \"application/json\",\n        \"Statewire-Client-Id\": c.latest.current.ctx.clientId,\n        \"Statewire-Command-Seq\": String(seq),\n        ...(c.lease !== null && { \"Statewire-Lease\": c.lease }),\n      }),\n      body: JSON.stringify({\n        method: command.method,\n        ...encodeParams(command.params),\n      }),\n      signal: AbortSignal.any([\n        c.lifetime?.signal ?? AbortSignal.abort(),\n        timeout,\n      ]),\n    },\n    timeout,\n  );\n  if (res.ok) {\n    const body = (await res.json().catch(() => null)) as {\n      res?: StatewireResponse;\n    } | null;\n    return body?.res !== undefined\n      ? { type: \"receipt\", response: reviveResponse(body.res) }\n      : { type: \"receipt\" };\n  }\n  if (res.status === 409 || res.status === 412 || res.status === 423) {\n    // Only a body naming the same verdict as the status is statewire's own;\n    // a bare status is middleware noise — a transport fault, retried.\n    const verdict =\n      res.status === 409\n        ? (\"seq-gap\" as const)\n        : res.status === 412\n          ? (\"unknown-client\" as const)\n          : (\"stale-lease\" as const);\n    const body = (await res.json().catch(() => null)) as {\n      error?: string;\n    } | null;\n    if (body?.error === verdict) return { type: verdict };\n    throw new Error(`HTTP ${res.status}`);\n  }\n  if (res.status === 429 || res.status === 503) throw overloaded(res);\n  if (res.status >= 400 && res.status < 500) {\n    return { type: \"rejected\", message: `HTTP ${res.status}` };\n  }\n  throw new Error(`HTTP ${res.status}`);\n};\n\nconst useStatewireSSE = (\n  options: StatewireSSEOptions,\n): StatewireTransport<any, any, any> => useStatewireRemote(options, createConn);\n\nexport const StatewireSSE = resource(useStatewireSSE) as <\n  TState = any,\n  TCommands extends StatewireCommands = StatewireCommands,\n  TEvent = any,\n>(\n  options: StatewireSSEOptions,\n) => ResourceElement<StatewireTransport<TState | undefined, TCommands, TEvent>>;\n"],"mappings":";;;;;;;AAyFA,MAAMmE,cACJF,QACAG,WACS;CACT,MAAMC,IAAU;EACdC,aAAa;EACbC,OAAO;EACP7B,SAAS;EACT8B,SAAS;EACTC,OAAO;EACPC,MAAMlE,WAAW;EACjBmE,UAAU;EACVV;EACAG;EACAtB,iBAAiBlC,cAAcyD,CAAC;EAChCO,cAAcH,UAAUI,cAAcR,GAAGI,KAAK;EAC9CxB,OAAOzB,SAASsD,YAAYT,GAAG7C,IAAI;EACnCwB,YAAY;GAAER,QAAQ;GAAcM,iBAAiBuB,EAAEvB,UAAU;EAAE;CACrE;CACA,OAAOuB;AACT;AAEA,MAAMU,WAAWV,MAAYA,EAAEJ,OAAOC,QAAQJ,QAAQR,IAAI0B,QAAQ,OAAO,EAAE;AAE3E,MAAMC,iBAAiB,OAAOZ,GAASa,UAAkC;CACvE,MAAM,EAAE3B,YAAYc,EAAEJ,OAAOC,QAAQJ;CACrC,MAAMsB,SAAS,IAAIC,QACjB,OAAO9B,YAAY,aAAa,MAAMA,QAAQ,IAAIA,OACpD;CACA,KAAK,MAAM,CAAC+B,MAAMC,UAAUC,OAAOC,QAAQP,KAAK,GAAGE,OAAOM,IAAIJ,MAAMC,KAAK;CACzE,OAAOH;AACT;AAEA,MAAMO,aAAa,OACjBtB,GACAuB,OACAC,MACAI,YACsB;CACtB,IAAI;EACF,OAAO,OAAO5B,EAAEJ,OAAOC,QAAQJ,QAAQL,SAASC,WAAWD,MAAAA,CACzDmC,OACAC,IACF;CACF,QAAQ;EACN,MAAMtF,SAAS0F,OAAO;CACxB;AACF;AAEA,MAAME,UAAU,OACd9B,GACA+B,QACAG,WACAE,SAC2C;CAC3C,MAAMC,UAAU,IAAIC,YAAY;CAChC,MAAMC,QAAQvG,gBAAgB;CAC9B,IAAIwG,WAAW;CACf,SAAS;EACP,MAAMC,QAAQC,iBACNR,UAAU9B,MAAM,GACtBnE,OAAO0G,kBACT;EACA,IAAIC;EACJ,IAAI;GACFA,SAAS,MAAMb,OAAOe,KAAK;EAC7B,QAAQ;GACN,MAAM5G,SAASgG,UAAUR,MAAM;EACjC,UAAU;GACRqB,aAAaN,KAAK;EACpB;EACA,IAAIG,OAAOI,MAAM,OAAO;EACxB,KAAK,MAAMC,QAAQV,MAAMF,QAAQa,OAAON,OAAO1B,OAAO,EAAEiC,QAAQ,KAAK,CAAC,CAAC,GAAG;GACxE,IAAIf,KAAK,GAAG,OAAO;GACnB,MAAMgB,MAAMC,KAAKd,MAAMU,IAAI;GAC3B,MAAMD,OAAOrH,mBAAmBS,eAAe4D,GAAGoD,KAAKZ,QAAQ,CAAC;GAChEA,WAAW;GACX,IAAIQ,MAAM,OAAOA;EACnB;CACF;AACF;AAEA,MAAMxC,gBAAgB,OACpBR,GACAI,UAC2C;CAC3C,MAAMlB,UAAU,MAAM0B,eAAeZ,GAAG;EACtC,uBAAuBA,EAAEJ,OAAOC,QAAQH,IAAIzC;EAC5CqG,QAAQ;CACV,CAAC;CACD,MAAMpB,YAAY,IAAIC,gBAAgB;CACtC,MAAMoB,cAAcb,iBACZR,UAAU9B,MAAM,GACtBnE,OAAO0G,kBACT;CACA,IAAIpF;CACJ,IAAI;EACFA,MAAM,MAAM+D,WACVtB,GACA,GAAGU,QAAQV,CAAC,EAAC,UACb;GACEd;GACAsE,OAAO;GACP9B,QAAQC,YAAY8B,IAAI,CAACrD,MAAMsB,QAAQQ,UAAUR,MAAM,CAAC;EAC1D,GACAQ,UAAUR,MACZ;CACF,UAAU;EACRqB,aAAaQ,WAAW;CAC1B;CACA,IAAIhG,IAAIY,WAAW,OAAOZ,IAAIY,WAAW,KAAK,MAAMuF,WAAWnG,GAAG;CAClE,IAAI,CAACA,IAAIoG,MAAM,CAACpG,IAAIqG,MAAM,MAAM,IAAIC,MAAM,QAAQtG,IAAIY,QAAQ;CAC9D,MAAM4D,SAASxE,IAAIqG,KAAKE,UAAU;CAClC,IAAI;EACF,OAAO,MAAMhC,QAAQ9B,GAAG+B,QAAQG,iBAAiB9B,MAAMsB,OAAOqC,OAAO;CACvE,UAAU;EACRhC,OAAOiC,OAAO,CAAC,CAACC,YAAY,CAAC,CAAC;CAChC;AACF;AAEA,MAAMP,cAAcnG,QAClBlB,gBAAgBC,gBAAgBiB,IAAI2B,QAAQgF,IAAI,aAAa,CAAC,CAAC;AAEjE,MAAMzD,cAAc,OAClBT,GACA,EAAEnB,KAAKC,cAC4B;CACnC,MAAM8C,UAAUD,YAAYC,QAAQ3F,OAAO0G,kBAAkB;CAC7D,MAAMpF,MAAM,MAAM+D,WAChBtB,GACA,GAAGU,QAAQV,CAAC,EAAC,YACb;EACErC,QAAQ;EACRuB,SAAS,MAAM0B,eAAeZ,GAAG;GAC/B,gBAAgB;GAChB,uBAAuBA,EAAEJ,OAAOC,QAAQH,IAAIzC;GAC5C,yBAAyBkH,OAAOtF,GAAG;GACnC,GAAImB,EAAEE,UAAU,QAAQ,EAAE,mBAAmBF,EAAEE,MAAM;EACvD,CAAC;EACD0D,MAAMP,KAAKe,UAAU;GACnBzG,QAAQmB,QAAQnB;GAChB,GAAG7B,aAAagD,QAAQlB,MAAM;EAChC,CAAC;EACD8D,QAAQC,YAAY8B,IAAI,CACtBzD,EAAEM,UAAUoB,UAAUC,YAAYvB,MAAM,GACxCwB,OAAO,CACR;CACH,GACAA,OACF;CACA,IAAIrE,IAAIoG,IAAI;EACV,MAAMC,OAAQ,MAAMrG,IAAI8G,KAAK,CAAC,CAACJ,YAAY,IAAI;EAG/C,OAAOL,MAAMrG,QAAQ+G,KAAAA,IACjB;GAAExG,MAAM;GAAWC,UAAUhC,eAAe6H,KAAKrG,GAAG;EAAE,IACtD,EAAEO,MAAM,UAAU;CACxB;CACA,IAAIP,IAAIY,WAAW,OAAOZ,IAAIY,WAAW,OAAOZ,IAAIY,WAAW,KAAK;EAGlE,MAAMoG,UACJhH,IAAIY,WAAW,MACV,YACDZ,IAAIY,WAAW,MACZ,mBACA;EAIT,KAAIyF,MAHgBrG,IAAI8G,KAAK,CAAC,CAACJ,YAAY,IAAI,EAAA,EAGrCQ,UAAUF,SAAS,OAAO,EAAEzG,MAAMyG,QAAQ;EACpD,MAAM,IAAIV,MAAM,QAAQtG,IAAIY,QAAQ;CACtC;CACA,IAAIZ,IAAIY,WAAW,OAAOZ,IAAIY,WAAW,KAAK,MAAMuF,WAAWnG,GAAG;CAClE,IAAIA,IAAIY,UAAU,OAAOZ,IAAIY,SAAS,KACpC,OAAO;EAAEL,MAAM;EAAYE,SAAS,QAAQT,IAAIY;CAAS;CAE3D,MAAM,IAAI0F,MAAM,QAAQtG,IAAIY,QAAQ;AACtC;AAEA,MAAMuG,mBAAkBjF,YAAA;CAAA,OAEgB5C,mBAAmB4C,SAASK,UAAU;AAAC;AAE/E,MAAa6E,eAAe/I,SAAS8I,eAAe"}
+{"version":3,"file":"sse.js","names":["useRef","useState","flushTapSync","resource","ResourceElement","encodeParams","reviveResponse","createSseParser","timing","classify","handleEnvelope","overloadedError","parseRetryAfter","useStatewireConn","Conn","StatewireEnvelope","StatewireResponse","Statewire","StatewireClient","StatewireHost","useStatewireRemote","StatewireRemoteProps","clientId","onSnapshot","args","ops","lastSeq","onOps","res","ack","onFinish","Fin","StatewireWireCommand","method","params","context","Record","contextOps","value","op","path","encodeWireCommand","command","undefined","StatewireSubmitResult","type","response","message","ConnectionState","status","cause","attempt","nextRetryAt","lastError","payload","RemoteLink","connection","Connection","send","seq","Promise","baseUrl","options","SSEOptions","url","replace","requestHeaders","extra","headers","merged","Headers","name","Object","entries","set","timedFetch","input","init","RequestInit","signal","AbortSignal","timeout","Response","fetch","globalThis","consume","c","reader","ReadableStreamDefaultReader","Uint8Array","heartbeat","AbortController","dead","decoder","TextDecoder","parse","timer","setTimeout","abort","heartbeatTimeoutMs","result","ReadableStreamReadResult","read","clearTimeout","done","data","decode","stream","env","JSON","attemptStream","ctx","accept","headerTimer","cache","any","overloaded","ok","body","Error","getReader","aborted","cancel","catch","get","sendCommand","String","lease","stringify","lifetime","json","verdict","const","error","useSseLink","latest","current","onChange","attemptOnce","link","useStatewireSSE","StatewireSSE","Commands","TState","TCommands","TEvent"],"sources":["../../src/transports/sse.ts"],"sourcesContent":["import { useRef, useState } from \"react\";\nimport { flushTapSync, resource } from \"@assistant-ui/tap\";\nimport type { ResourceElement } from \"@assistant-ui/tap\";\nimport { encodeParams, reviveResponse } from \"../wire/meta\";\nimport { createSseParser } from \"./sse-parser\";\nimport { timing } from \"../client/timing\";\nimport {\n  classify,\n  handleEnvelope,\n  overloadedError,\n  parseRetryAfter,\n  useStatewireConn,\n} from \"../client/connection\";\nimport type {\n  Conn,\n  StatewireEnvelope,\n  StatewireResponse,\n} from \"../client/connection\";\nimport type { Statewire } from \"../types\";\nimport type { StatewireClient } from \"../statewire\";\nimport type { StatewireHost } from \"../host\";\nimport { useStatewireRemote } from \"./remote\";\n\nexport type StatewireRemoteProps = {\n  clientId: string;\n  onSnapshot: (args: { ops: unknown[]; lastSeq: number }) => void;\n  onOps: (args: {\n    ops: unknown[];\n    res?: StatewireResponse[];\n    ack?: number;\n  }) => void;\n  onFinish: (args: StatewireHost.Fin) => void;\n};\n\nexport type StatewireWireCommand =\n  | { method: string; params: readonly unknown[]; context?: never }\n  | { method?: never; params?: never; context: Record<string, unknown> };\n\n/** The wire shape of a client context submission: one root replace op. */\nexport const contextOps = (value: Record<string, unknown>) => [\n  { op: \"replace\", path: [], value },\n];\n\nexport const encodeWireCommand = (command: StatewireWireCommand) => ({\n  ...(command.method !== undefined && {\n    method: command.method,\n    ...encodeParams(command.params),\n  }),\n  ...(command.context !== undefined && { context: contextOps(command.context) }),\n});\n\nexport type StatewireSubmitResult =\n  | { type: \"receipt\"; response?: StatewireResponse }\n  | { type: \"rejected\"; message: string }\n  | { type: \"seq-gap\" }\n  | { type: \"unknown-client\" }\n  | { type: \"stale-lease\" };\n\nexport type ConnectionState =\n  | { status: \"connecting\" }\n  | { status: \"live\" }\n  | {\n      status: \"reconnecting\";\n      cause: \"dropped\" | \"evicted\" | \"error\";\n      attempt: number;\n      nextRetryAt: number;\n      lastError?: string;\n      message?: string;\n    }\n  | { status: \"gone\"; message?: string; payload?: unknown };\n\n/** The live wire a remote session drives: connection status plus command POST. */\nexport type RemoteLink = {\n  connection: StatewireClient.Connection;\n  send: (args: {\n    seq: number;\n    command: StatewireWireCommand;\n  }) => Promise<StatewireSubmitResult>;\n};\n\nconst baseUrl = (options: StatewireClient.SSEOptions) =>\n  options.url.replace(/\\/$/, \"\");\n\nconst requestHeaders = async (\n  options: StatewireClient.SSEOptions,\n  extra: Record<string, string>,\n) => {\n  const { headers } = options;\n  const merged = new Headers(\n    typeof headers === \"function\" ? await headers() : headers,\n  );\n  for (const [name, value] of Object.entries(extra)) merged.set(name, value);\n  return merged;\n};\n\nconst timedFetch = async (\n  options: StatewireClient.SSEOptions,\n  input: string,\n  init: RequestInit & { signal: AbortSignal },\n  timeout: AbortSignal,\n): Promise<Response> => {\n  try {\n    return await (options.fetch ?? globalThis.fetch)(input, init);\n  } catch {\n    throw classify(timeout);\n  }\n};\n\nconst consume = async (\n  c: Conn,\n  reader: ReadableStreamDefaultReader<Uint8Array>,\n  heartbeat: AbortController,\n  dead: () => boolean,\n): Promise<\"stop\" | \"reattach\" | \"ended\"> => {\n  const decoder = new TextDecoder();\n  const parse = createSseParser();\n  for (;;) {\n    const timer = setTimeout(\n      () => heartbeat.abort(),\n      timing.heartbeatTimeoutMs,\n    );\n    let result: ReadableStreamReadResult<Uint8Array>;\n    try {\n      result = await reader.read();\n    } catch {\n      throw classify(heartbeat.signal);\n    } finally {\n      clearTimeout(timer);\n    }\n    if (result.done) return \"ended\";\n    for (const data of parse(decoder.decode(result.value, { stream: true }))) {\n      if (dead()) return \"ended\";\n      const env = JSON.parse(data) as StatewireEnvelope;\n      const done = flushTapSync(() => handleEnvelope(c, env));\n      if (done) return done;\n    }\n  }\n};\n\nconst attemptStream = async (\n  c: Conn,\n  options: StatewireClient.SSEOptions,\n  abort: AbortController,\n): Promise<\"stop\" | \"reattach\" | \"ended\"> => {\n  const headers = await requestHeaders(options, {\n    \"Statewire-Client-Id\": c.ctx.clientId,\n    accept: \"text/event-stream\",\n  });\n  const heartbeat = new AbortController();\n  const headerTimer = setTimeout(\n    () => heartbeat.abort(),\n    timing.heartbeatTimeoutMs,\n  );\n  let res: Response;\n  try {\n    res = await timedFetch(\n      options,\n      `${baseUrl(options)}/stream`,\n      {\n        headers,\n        cache: \"no-store\",\n        signal: AbortSignal.any([abort.signal, heartbeat.signal]),\n      },\n      heartbeat.signal,\n    );\n  } finally {\n    clearTimeout(headerTimer);\n  }\n  if (res.status === 429 || res.status === 503) throw overloaded(res);\n  if (!res.ok || !res.body) throw new Error(`HTTP ${res.status}`);\n  const reader = res.body.getReader();\n  try {\n    return await consume(c, reader, heartbeat, () => abort.signal.aborted);\n  } finally {\n    reader.cancel().catch(() => {});\n  }\n};\n\nconst overloaded = (res: Response) =>\n  overloadedError(parseRetryAfter(res.headers.get(\"Retry-After\")));\n\nconst sendCommand = async (\n  c: Conn,\n  options: StatewireClient.SSEOptions,\n  { seq, command }: { seq: number; command: StatewireWireCommand },\n): Promise<StatewireSubmitResult> => {\n  const timeout = AbortSignal.timeout(timing.heartbeatTimeoutMs);\n  const res = await timedFetch(\n    options,\n    `${baseUrl(options)}/commands`,\n    {\n      method: \"POST\",\n      headers: await requestHeaders(options, {\n        \"content-type\": \"application/json\",\n        \"Statewire-Client-Id\": c.ctx.clientId,\n        \"Statewire-Command-Seq\": String(seq),\n        ...(c.lease !== null && { \"Statewire-Lease\": c.lease }),\n      }),\n      body: JSON.stringify(encodeWireCommand(command)),\n      signal: AbortSignal.any([\n        c.lifetime?.signal ?? AbortSignal.abort(),\n        timeout,\n      ]),\n    },\n    timeout,\n  );\n  if (res.ok) {\n    let body: { res?: StatewireResponse } | null;\n    try {\n      body = (await res.json()) as { res?: StatewireResponse } | null;\n    } catch {\n      return {\n        type: \"rejected\",\n        message: \"statewire: protocol error: malformed /commands receipt body\",\n      };\n    }\n    return body?.res !== undefined\n      ? { type: \"receipt\", response: reviveResponse(body.res) }\n      : { type: \"receipt\" };\n  }\n  if (res.status === 409 || res.status === 412 || res.status === 423) {\n    // Only a body naming the same verdict as the status is statewire's own;\n    // a bare status is middleware noise — a transport fault, retried.\n    const verdict =\n      res.status === 409\n        ? (\"seq-gap\" as const)\n        : res.status === 412\n          ? (\"unknown-client\" as const)\n          : (\"stale-lease\" as const);\n    const body = (await res.json().catch(() => null)) as {\n      error?: string;\n    } | null;\n    if (body?.error === verdict) return { type: verdict };\n    throw new Error(`HTTP ${res.status}`);\n  }\n  if (res.status === 429 || res.status === 503) throw overloaded(res);\n  if (res.status >= 400 && res.status < 500) {\n    return { type: \"rejected\", message: `HTTP ${res.status}` };\n  }\n  throw new Error(`HTTP ${res.status}`);\n};\n\nconst useSseLink = (\n  options: StatewireClient.SSEOptions,\n  ctx: StatewireRemoteProps,\n): RemoteLink => {\n  const latest = useRef(options);\n  latest.current = options;\n  const c = useStatewireConn({\n    url: options.url,\n    onChange: options.onChange,\n    ctx,\n    attemptOnce: (c, abort) => attemptStream(c, options, abort),\n  });\n  const [link] = useState(\n    (): RemoteLink => ({\n      get connection() {\n        return c.connection;\n      },\n      send: (args) => sendCommand(c, latest.current, args),\n    }),\n  );\n  return link;\n};\n\nconst useStatewireSSE = (\n  options: StatewireClient.SSEOptions,\n): Statewire<any, any, any> => useStatewireRemote(options, useSseLink);\n\nexport const StatewireSSE = resource(useStatewireSSE) as <\n  TState = any,\n  TCommands extends Statewire.Commands = Statewire.Commands,\n  TEvent = any,\n>(\n  options: StatewireClient.SSEOptions,\n) => ResourceElement<Statewire<TState | undefined, TCommands, TEvent>>;\n"],"mappings":";;;;;;;;;AAuCA,MAAaqC,cAAcC,UAAmC,CAC5D;CAAEC,IAAI;CAAWC,MAAM,CAAA;CAAIF;AAAM,CAAC;AAGpC,MAAaG,qBAAqBC,aAAmC;CACnE,GAAIA,QAAQT,WAAWU,KAAAA,KAAa;EAClCV,QAAQS,QAAQT;EAChB,GAAG5B,aAAaqC,QAAQR,MAAM;CAChC;CACA,GAAIQ,QAAQP,YAAYQ,KAAAA,KAAa,EAAER,SAASE,WAAWK,QAAQP,OAAO,EAAE;AAC9E;AA+BA,MAAM0B,WAAWC,YACfA,QAAQE,IAAIC,QAAQ,OAAO,EAAE;AAE/B,MAAMC,iBAAiB,OACrBJ,SACAK,UACG;CACH,MAAM,EAAEC,YAAYN;CACpB,MAAMO,SAAS,IAAIC,QACjB,OAAOF,YAAY,aAAa,MAAMA,QAAQ,IAAIA,OACpD;CACA,KAAK,MAAM,CAACG,MAAMjC,UAAUkC,OAAOC,QAAQN,KAAK,GAAGE,OAAOK,IAAIH,MAAMjC,KAAK;CACzE,OAAO+B;AACT;AAEA,MAAMM,aAAa,OACjBb,SACAc,OACAC,MACAI,YACsB;CACtB,IAAI;EACF,OAAO,OAAOnB,QAAQqB,SAASC,WAAWD,MAAAA,CAAOP,OAAOC,IAAI;CAC9D,QAAQ;EACN,MAAMpE,SAASwE,OAAO;CACxB;AACF;AAEA,MAAMI,UAAU,OACdC,GACAC,QACAG,WACAE,SAC2C;CAC3C,MAAMC,UAAU,IAAIC,YAAY;CAChC,MAAMC,QAAQxF,gBAAgB;CAC9B,SAAS;EACP,MAAMyF,QAAQC,iBACNP,UAAUQ,MAAM,GACtB1F,OAAO2F,kBACT;EACA,IAAIC;EACJ,IAAI;GACFA,SAAS,MAAMb,OAAOe,KAAK;EAC7B,QAAQ;GACN,MAAM7F,SAASiF,UAAUX,MAAM;EACjC,UAAU;GACRwB,aAAaP,KAAK;EACpB;EACA,IAAII,OAAOI,MAAM,OAAO;EACxB,KAAK,MAAMC,QAAQV,MAAMF,QAAQa,OAAON,OAAO9D,OAAO,EAAEqE,QAAQ,KAAK,CAAC,CAAC,GAAG;GACxE,IAAIf,KAAK,GAAG,OAAO;GACnB,MAAMgB,MAAMC,KAAKd,MAAMU,IAAI;GAC3B,MAAMD,OAAOtG,mBAAmBQ,eAAe4E,GAAGsB,GAAG,CAAC;GACtD,IAAIJ,MAAM,OAAOA;EACnB;CACF;AACF;AAEA,MAAMM,gBAAgB,OACpBxB,GACAxB,SACAoC,UAC2C;CAC3C,MAAM9B,UAAU,MAAMF,eAAeJ,SAAS;EAC5C,uBAAuBwB,EAAEyB,IAAIzF;EAC7B0F,QAAQ;CACV,CAAC;CACD,MAAMtB,YAAY,IAAIC,gBAAgB;CACtC,MAAMsB,cAAchB,iBACZP,UAAUQ,MAAM,GACtB1F,OAAO2F,kBACT;CACA,IAAIvE;CACJ,IAAI;EACFA,MAAM,MAAM+C,WACVb,SACA,GAAGD,QAAQC,OAAO,EAAC,UACnB;GACEM;GACA8C,OAAO;GACPnC,QAAQC,YAAYmC,IAAI,CAACjB,MAAMnB,QAAQW,UAAUX,MAAM,CAAC;EAC1D,GACAW,UAAUX,MACZ;CACF,UAAU;EACRwB,aAAaU,WAAW;CAC1B;CACA,IAAIrF,IAAIqB,WAAW,OAAOrB,IAAIqB,WAAW,KAAK,MAAMmE,WAAWxF,GAAG;CAClE,IAAI,CAACA,IAAIyF,MAAM,CAACzF,IAAI0F,MAAM,MAAM,IAAIC,MAAM,QAAQ3F,IAAIqB,QAAQ;CAC9D,MAAMsC,SAAS3D,IAAI0F,KAAKE,UAAU;CAClC,IAAI;EACF,OAAO,MAAMnC,QAAQC,GAAGC,QAAQG,iBAAiBQ,MAAMnB,OAAO0C,OAAO;CACvE,UAAU;EACRlC,OAAOmC,OAAO,CAAC,CAACC,YAAY,CAAC,CAAC;CAChC;AACF;AAEA,MAAMP,cAAcxF,QAClBjB,gBAAgBC,gBAAgBgB,IAAIwC,QAAQwD,IAAI,aAAa,CAAC,CAAC;AAEjE,MAAMC,cAAc,OAClBvC,GACAxB,SACA,EAAEH,KAAKjB,cAC4B;CACnC,MAAMuC,UAAUD,YAAYC,QAAQzE,OAAO2F,kBAAkB;CAC7D,MAAMvE,MAAM,MAAM+C,WAChBb,SACA,GAAGD,QAAQC,OAAO,EAAC,YACnB;EACE7B,QAAQ;EACRmC,SAAS,MAAMF,eAAeJ,SAAS;GACrC,gBAAgB;GAChB,uBAAuBwB,EAAEyB,IAAIzF;GAC7B,yBAAyBwG,OAAOnE,GAAG;GACnC,GAAI2B,EAAEyC,UAAU,QAAQ,EAAE,mBAAmBzC,EAAEyC,MAAM;EACvD,CAAC;EACDT,MAAMT,KAAKmB,UAAUvF,kBAAkBC,OAAO,CAAC;EAC/CqC,QAAQC,YAAYmC,IAAI,CACtB7B,EAAE2C,UAAUlD,UAAUC,YAAYkB,MAAM,GACxCjB,OAAO,CACR;CACH,GACAA,OACF;CACA,IAAIrD,IAAIyF,IAAI;EACV,IAAIC;EACJ,IAAI;GACFA,OAAQ,MAAM1F,IAAIsG,KAAK;EACzB,QAAQ;GACN,OAAO;IACLrF,MAAM;IACNE,SAAS;GACX;EACF;EACA,OAAOuE,MAAM1F,QAAQe,KAAAA,IACjB;GAAEE,MAAM;GAAWC,UAAUxC,eAAegH,KAAK1F,GAAG;EAAE,IACtD,EAAEiB,MAAM,UAAU;CACxB;CACA,IAAIjB,IAAIqB,WAAW,OAAOrB,IAAIqB,WAAW,OAAOrB,IAAIqB,WAAW,KAAK;EAGlE,MAAMkF,UACJvG,IAAIqB,WAAW,MACV,YACDrB,IAAIqB,WAAW,MACZ,mBACA;EAIT,KAAIqE,MAHgB1F,IAAIsG,KAAK,CAAC,CAACP,YAAY,IAAI,EAAA,EAGrCU,UAAUF,SAAS,OAAO,EAAEtF,MAAMsF,QAAQ;EACpD,MAAM,IAAIZ,MAAM,QAAQ3F,IAAIqB,QAAQ;CACtC;CACA,IAAIrB,IAAIqB,WAAW,OAAOrB,IAAIqB,WAAW,KAAK,MAAMmE,WAAWxF,GAAG;CAClE,IAAIA,IAAIqB,UAAU,OAAOrB,IAAIqB,SAAS,KACpC,OAAO;EAAEJ,MAAM;EAAYE,SAAS,QAAQnB,IAAIqB;CAAS;CAE3D,MAAM,IAAIsE,MAAM,QAAQ3F,IAAIqB,QAAQ;AACtC;AAEA,MAAMqF,cACJxE,SACAiD,QACe;CACf,MAAMwB,SAASvI,OAAO8D,OAAO;CAC7ByE,OAAOC,UAAU1E;CACjB,MAAMwB,MAAIzE,iBAAiB;EACzBmD,KAAKF,QAAQE;EACbyE,UAAU3E,QAAQ2E;EAClB1B;EACA2B,cAAcpD,GAAGY,UAAUY,cAAcxB,GAAGxB,SAASoC,KAAK;CAC5D,CAAC;CACD,MAAM,CAACyC,QAAQ1I,gBACM;EACjB,IAAIuD,aAAa;GACf,OAAO8B,IAAE9B;EACX;EACAE,OAAOlC,SAASqG,YAAYvC,KAAGiD,OAAOC,SAAShH,IAAI;CACrD,EACF;CACA,OAAOmH;AACT;AAEA,MAAMC,mBACJ9E,YAC6B1C,mBAAmB0C,SAASwE,UAAU;AAErE,MAAaO,eAAe1I,SAASyI,eAAe"}

package/dist/transports/testing.d.ts

@@ -1,5 +1,6 @@
 import { StatewireEnvelope } from "../client/connection.js";
-import { StatewireTransport } from "../types.js";
+import { Statewire } from "../types.js";
+import { SocketRecorder, commandFrame, createCounterElement, createSocketRecorder, settle } from "./fixtures.js";
 import { ResourceElement } from "@assistant-ui/tap";
 //#region src/transports/testing.d.ts
 /** Let queued microtasks and op batches drain through the statewire client. */
@@ -8,11 +9,12 @@ declare const tick: () => Promise<void>;
 declare const applyOptimistic: <TState>(state: TState, mutate: (draft: TState) => void) => TState;
 type MockTransport<TState> = {
   /** Pass as the `transport:` option of useStatewire / Harness. */
-  transport: ResourceElement<StatewireTransport<TState | undefined, any, any>>;
-  /** Captured `{ method, params }` command bodies, in POST order. */
+  transport: ResourceElement<Statewire<TState | undefined, any, any>>;
+  /** Captured `/commands` POST bodies, in POST order. */
   commands: {
-    method: string;
-    params: unknown[];
+    method?: string;
+    params?: unknown[];
+    context?: unknown;
   }[];
   /** Replace the replicated state and broadcast it as a root snapshot op. */
   setState: (state: TState) => Promise<void>;
@@ -27,5 +29,5 @@ type MockTransport<TState> = {
 };
 declare const createMockTransport: <TState>(initialState: TState) => MockTransport<TState>;
 //#endregion
-export { MockTransport, applyOptimistic, createMockTransport, tick };
+export { MockTransport, SocketRecorder, applyOptimistic, commandFrame, createCounterElement, createMockTransport, createSocketRecorder, settle, tick };
 //# sourceMappingURL=testing.d.ts.map

package/dist/transports/testing.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"testing.d.ts","names":[],"sources":["../../src/transports/testing.ts"],"mappings":";;;;;cAOa,YAAI;;cAGJ,kBAAmB,QAC9B,OAAO,QACP,SAAS,OAAO,oBACf;KAQS,cAAc;;EAExB,WAAW,gBAAgB,mBAAmB;;EAE9C;IAAY;IAAgB;;;EAE5B,WAAW,OAAO,WAAW;;EAE7B,MAAM,gBAAgB;;EAEtB,eAAe,UAAU,sBAAsB;;EAE/C,SAAS,gBAAgB,qBAAqB;WACrC;WACA;;cAUE,sBAAuB,QAClC,cAAc,WACb,cAAc"}
+{"version":3,"file":"testing.d.ts","names":[],"sources":["../../src/transports/testing.ts"],"mappings":";;;;;;cAWa,YAAI;;cAGJ,kBAAmB,QAC9B,OAAO,QACP,SAAS,OAAO,oBACf;KAES,cAAc;;EAExB,WAAW,gBAAgB,UAAU;;EAErC;IAAY;IAAiB;IAAoB;;;EAEjD,WAAW,OAAO,WAAW;;EAE7B,MAAM,gBAAgB;;EAEtB,eAAe,UAAU,sBAAsB;;EAE/C,SAAS,gBAAgB,qBAAqB;WACrC;WACA;;cAUE,sBAAuB,QAClC,cAAc,WACb,cAAc"}

package/dist/transports/testing.js

@@ -1,5 +1,7 @@
-import { recordMutations } from "../client/optimistic.js";
+import { recordMutations } from "../util/recordMutations.js";
 import { StatewireSSE } from "./sse.js";
+import { openSSEStream, sseFrame } from "./fake-sse.js";
+import { commandFrame, createCounterElement, createSocketRecorder, settle } from "./fixtures.js";
 //#region src/transports/testing.ts
 /** Let queued microtasks and op batches drain through the statewire client. */
 const tick = () => new Promise((r) => setTimeout(r, 0));
@@ -19,32 +21,20 @@ const createMockTransport = (initialState) => {
 	const streams = /* @__PURE__ */ new Set();
 	const commands = [];
 	const sendEnvelope = (envelope) => {
-		for (const stream of streams) stream.send(`data: ${JSON.stringify(envelope)}\n\n`);
+		for (const stream of streams) stream.send(sseFrame(envelope));
 		return tick();
 	};
-	const handleStream = (id, signal) => {
-		clientId = id;
-		const lastSeq = clients.get(id) ?? 0;
-		clients.set(id, lastSeq);
-		const encoder = new TextEncoder();
-		let handle;
-		const body = new ReadableStream({
-			start: (controller) => {
-				let open = true;
-				handle = {
-					clientId: id,
-					send: (text) => {
-						if (open) controller.enqueue(encoder.encode(text));
-					},
-					close: () => {
-						if (!open) return;
-						open = false;
-						streams.delete(handle);
-						controller.close();
-					}
-				};
-				streams.add(handle);
-				handle.send(`data: ${JSON.stringify({
+	const fakeFetch = (async (input, init) => {
+		if (finished) throw new TypeError("fetch failed (finished)");
+		const url = String(input);
+		const headers = new Headers(init?.headers);
+		const id = headers.get("Statewire-Client-Id") ?? "";
+		if (url.endsWith("/stream")) {
+			clientId = id;
+			const lastSeq = clients.get(id) ?? 0;
+			clients.set(id, lastSeq);
+			return openSSEStream(streams, id, (handle) => {
+				handle.send(sseFrame({
 					ops: [{
 						op: "replace",
 						path: [],
@@ -54,26 +44,9 @@ const createMockTransport = (initialState) => {
 						lastSeq,
 						lease: "mock-lease"
 					}
-				})}\n\n`);
-			},
-			cancel: () => {
-				streams.delete(handle);
-			}
-		});
-		signal?.addEventListener("abort", () => {
-			for (const stream of [...streams]) if (stream.clientId === id) stream.close();
-		});
-		return new Response(body, {
-			status: 200,
-			headers: { "content-type": "text/event-stream" }
-		});
-	};
-	const fakeFetch = (async (input, init) => {
-		if (finished) throw new TypeError("fetch failed (finished)");
-		const url = String(input);
-		const headers = new Headers(init?.headers);
-		const id = headers.get("Statewire-Client-Id") ?? "";
-		if (url.endsWith("/stream")) return handleStream(id, init?.signal ?? void 0);
+				}));
+			}, init?.signal ?? void 0);
+		}
 		if (url.endsWith("/commands")) {
 			const seq = Number(headers.get("Statewire-Command-Seq"));
 			clients.set(id, Math.max(clients.get(id) ?? 0, seq));
@@ -116,6 +89,6 @@ const createMockTransport = (initialState) => {
 	};
 };
 //#endregion
-export { applyOptimistic, createMockTransport, tick };
+export { applyOptimistic, commandFrame, createCounterElement, createMockTransport, createSocketRecorder, settle, tick };
 
 //# sourceMappingURL=testing.js.map

package/dist/transports/testing.js.map

@@ -1 +1 @@
-{"version":3,"file":"testing.js","names":["ResourceElement","StatewireEnvelope","recordMutations","StatewireSSE","StatewireTransport","tick","Promise","r","setTimeout","applyOptimistic","state","TState","mutate","draft","StreamHandle","clientId","send","text","close","MockTransport","transport","commands","method","params","setState","ack","seq","sendEnvelope","envelope","finish","reason","message","attached","mockCount","createMockTransport","initialState","finished","clients","Map","streams","Set","stream","JSON","stringify","handleStream","id","signal","AbortSignal","lastSeq","get","set","encoder","TextEncoder","handle","body","ReadableStream","Uint8Array","start","controller","open","enqueue","encode","delete","add","ops","op","path","value","syn","lease","cancel","addEventListener","Response","status","headers","fakeFetch","input","RequestInfo","URL","init","RequestInit","TypeError","url","String","Headers","endsWith","undefined","Number","Math","max","push","parse","json","fetch","next","fin","size"],"sources":["../../src/transports/testing.ts"],"sourcesContent":["import type { ResourceElement } from \"@assistant-ui/tap\";\nimport type { StatewireEnvelope } from \"../client/connection\";\nimport { recordMutations } from \"../client/optimistic\";\nimport { StatewireSSE } from \"./sse\";\nimport type { StatewireTransport } from \"../types\";\n\n/** Let queued microtasks and op batches drain through the statewire client. */\nexport const tick = () => new Promise<void>((r) => setTimeout(r, 0));\n\n/** Run a mutator against a recording draft, as the client does for `optimistic`. */\nexport const applyOptimistic = <TState>(\n  state: TState,\n  mutate: (draft: TState) => void,\n): TState => recordMutations(state, mutate).state;\n\ntype StreamHandle = {\n  clientId: string;\n  send: (text: string) => void;\n  close: () => void;\n};\n\nexport type MockTransport<TState> = {\n  /** Pass as the `transport:` option of useStatewire / Harness. */\n  transport: ResourceElement<StatewireTransport<TState | undefined, any, any>>;\n  /** Captured `{ method, params }` command bodies, in POST order. */\n  commands: { method: string; params: unknown[] }[];\n  /** Replace the replicated state and broadcast it as a root snapshot op. */\n  setState: (state: TState) => Promise<void>;\n  /** Deliver an ack watermark on the stream. */\n  ack: (seq: number) => Promise<void>;\n  /** Deliver an arbitrary envelope on the stream. */\n  sendEnvelope: (envelope: StatewireEnvelope) => Promise<void>;\n  /** End the stream with a fin; further attaches fail like a dead server. */\n  finish: (reason: string, message?: string) => Promise<void>;\n  readonly clientId: string;\n  readonly attached: boolean;\n};\n\n/**\n * A mock statewire backend: serves `initialState` as the attach snapshot,\n * records command POSTs (receipt only — acks and results are delivered from\n * test code), and exposes the wire as a transport ResourceElement.\n */\nlet mockCount = 0;\n\nexport const createMockTransport = <TState>(\n  initialState: TState,\n): MockTransport<TState> => {\n  let state: unknown = initialState;\n  let clientId = \"\";\n  let finished = false;\n  const clients = new Map<string, number>();\n  const streams = new Set<StreamHandle>();\n  const commands: { method: string; params: unknown[] }[] = [];\n\n  const sendEnvelope = (envelope: StatewireEnvelope) => {\n    for (const stream of streams) {\n      stream.send(`data: ${JSON.stringify(envelope)}\\n\\n`);\n    }\n    return tick();\n  };\n\n  const handleStream = (id: string, signal: AbortSignal | undefined) => {\n    clientId = id;\n    const lastSeq = clients.get(id) ?? 0;\n    clients.set(id, lastSeq);\n    const encoder = new TextEncoder();\n    let handle: StreamHandle;\n    const body = new ReadableStream<Uint8Array>({\n      start: (controller) => {\n        let open = true;\n        handle = {\n          clientId: id,\n          send: (text) => {\n            if (open) controller.enqueue(encoder.encode(text));\n          },\n          close: () => {\n            if (!open) return;\n            open = false;\n            streams.delete(handle);\n            controller.close();\n          },\n        };\n        streams.add(handle);\n        handle.send(\n          `data: ${JSON.stringify({\n            ops: [{ op: \"replace\", path: [], value: state }],\n            syn: { lastSeq, lease: \"mock-lease\" },\n          })}\\n\\n`,\n        );\n      },\n      cancel: () => {\n        streams.delete(handle);\n      },\n    });\n    signal?.addEventListener(\"abort\", () => {\n      for (const stream of [...streams]) {\n        if (stream.clientId === id) stream.close();\n      }\n    });\n    return new Response(body, {\n      status: 200,\n      headers: { \"content-type\": \"text/event-stream\" },\n    });\n  };\n\n  const fakeFetch = (async (\n    input: RequestInfo | URL,\n    init?: RequestInit,\n  ): Promise<Response> => {\n    if (finished) throw new TypeError(\"fetch failed (finished)\");\n    const url = String(input);\n    const headers = new Headers(init?.headers);\n    const id = headers.get(\"Statewire-Client-Id\") ?? \"\";\n    if (url.endsWith(\"/stream\"))\n      return handleStream(id, init?.signal ?? undefined);\n    if (url.endsWith(\"/commands\")) {\n      const seq = Number(headers.get(\"Statewire-Command-Seq\"));\n      clients.set(id, Math.max(clients.get(id) ?? 0, seq));\n      commands.push(\n        JSON.parse(String(init?.body)) as { method: string; params: unknown[] },\n      );\n      return Response.json({});\n    }\n    return new Response(null, { status: 404 });\n  }) as typeof fetch;\n\n  return {\n    transport: StatewireSSE({\n      url: `http://mock-${++mockCount}/api`,\n      fetch: fakeFetch,\n    }),\n    commands,\n    setState: (next) => {\n      state = next;\n      return sendEnvelope({ ops: [{ op: \"replace\", path: [], value: next }] });\n    },\n    ack: (seq) => sendEnvelope({ ack: seq }),\n    sendEnvelope,\n    finish: async (reason, message) => {\n      await sendEnvelope({\n        fin: { reason, ...(message !== undefined && { message }) },\n      });\n      finished = true;\n      for (const stream of [...streams]) stream.close();\n      await tick();\n    },\n    get clientId() {\n      return clientId;\n    },\n    get attached() {\n      return streams.size > 0;\n    },\n  };\n};\n"],"mappings":";;;;AAOA,MAAaK,aAAa,IAAIC,SAAeC,MAAMC,WAAWD,GAAG,CAAC,CAAC;;AAGnE,MAAaE,mBACXC,OACAE,WACWV,gBAAgBQ,OAAOE,MAAM,CAAC,CAACF;;;;;;AA8B5C,IAAIuB,YAAY;AAEhB,MAAaC,uBACXC,iBAC0B;CAC1B,IAAIzB,QAAiByB;CACrB,IAAIpB,WAAW;CACf,IAAIqB,WAAW;CACf,MAAMC,0BAAU,IAAIC,IAAoB;CACxC,MAAMC,0BAAU,IAAIC,IAAkB;CACtC,MAAMnB,WAAoD,CAAA;CAE1D,MAAMM,gBAAgBC,aAAgC;EACpD,KAAK,MAAMa,UAAUF,SACnBE,OAAOzB,KAAK,SAAS0B,KAAKC,UAAUf,QAAQ,EAAC,KAAM;EAErD,OAAOvB,KAAK;CACd;CAEA,MAAMuC,gBAAgBC,IAAYC,WAAoC;EACpE/B,WAAW8B;EACX,MAAMG,UAAUX,QAAQY,IAAIJ,EAAE,KAAK;EACnCR,QAAQa,IAAIL,IAAIG,OAAO;EACvB,MAAMG,UAAU,IAAIC,YAAY;EAChC,IAAIC;EACJ,MAAMC,OAAO,IAAIC,eAA2B;GAC1CE,QAAQC,eAAe;IACrB,IAAIC,OAAO;IACXN,SAAS;KACPtC,UAAU8B;KACV7B,OAAOC,SAAS;MACd,IAAI0C,MAAMD,WAAWE,QAAQT,QAAQU,OAAO5C,IAAI,CAAC;KACnD;KACAC,aAAa;MACX,IAAI,CAACyC,MAAM;MACXA,OAAO;MACPpB,QAAQuB,OAAOT,MAAM;MACrBK,WAAWxC,MAAM;KACnB;IACF;IACAqB,QAAQwB,IAAIV,MAAM;IAClBA,OAAOrC,KACL,SAAS0B,KAAKC,UAAU;KACtBqB,KAAK,CAAC;MAAEC,IAAI;MAAWC,MAAM,CAAA;MAAIC,OAAOzD;KAAM,CAAC;KAC/C0D,KAAK;MAAEpB;MAASqB,OAAO;KAAa;IACtC,CAAC,EAAC,KACJ;GACF;GACAC,cAAc;IACZ/B,QAAQuB,OAAOT,MAAM;GACvB;EACF,CAAC;EACDP,QAAQyB,iBAAiB,eAAe;GACtC,KAAK,MAAM9B,UAAU,CAAC,GAAGF,OAAO,GAC9B,IAAIE,OAAO1B,aAAa8B,IAAIJ,OAAOvB,MAAM;EAE7C,CAAC;EACD,OAAO,IAAIsD,SAASlB,MAAM;GACxBmB,QAAQ;GACRC,SAAS,EAAE,gBAAgB,oBAAoB;EACjD,CAAC;CACH;CAEA,MAAMC,aAAa,OACjBC,OACAG,SACsB;EACtB,IAAI3C,UAAU,MAAM,IAAI6C,UAAU,yBAAyB;EAC3D,MAAMC,MAAMC,OAAOP,KAAK;EACxB,MAAMF,UAAU,IAAIU,QAAQL,MAAML,OAAO;EACzC,MAAM7B,KAAK6B,QAAQzB,IAAI,qBAAqB,KAAK;EACjD,IAAIiC,IAAIG,SAAS,SAAS,GACxB,OAAOzC,aAAaC,IAAIkC,MAAMjC,UAAUwC,KAAAA,CAAS;EACnD,IAAIJ,IAAIG,SAAS,WAAW,GAAG;GAC7B,MAAM3D,MAAM6D,OAAOb,QAAQzB,IAAI,uBAAuB,CAAC;GACvDZ,QAAQa,IAAIL,IAAI2C,KAAKC,IAAIpD,QAAQY,IAAIJ,EAAE,KAAK,GAAGnB,GAAG,CAAC;GACnDL,SAASqE,KACPhD,KAAKiD,MAAMR,OAAOJ,MAAMzB,IAAI,CAAC,CAC/B;GACA,OAAOkB,SAASoB,KAAK,CAAC,CAAC;EACzB;EACA,OAAO,IAAIpB,SAAS,MAAM,EAAEC,QAAQ,IAAI,CAAC;CAC3C;CAEA,OAAO;EACLrD,WAAWjB,aAAa;GACtB+E,KAAK,eAAe,EAAEjD,UAAS;GAC/B4D,OAAOlB;EACT,CAAC;EACDtD;EACAG,WAAWsE,SAAS;GAClBpF,QAAQoF;GACR,OAAOnE,aAAa,EAAEqC,KAAK,CAAC;IAAEC,IAAI;IAAWC,MAAM,CAAA;IAAIC,OAAO2B;GAAK,CAAC,EAAE,CAAC;EACzE;EACArE,MAAMC,QAAQC,aAAa,EAAEF,KAAKC,IAAI,CAAC;EACvCC;EACAE,QAAQ,OAAOC,QAAQC,YAAY;GACjC,MAAMJ,aAAa,EACjBoE,KAAK;IAAEjE;IAAQ,GAAIC,YAAYuD,KAAAA,KAAa,EAAEvD,QAAQ;GAAG,EAC3D,CAAC;GACDK,WAAW;GACX,KAAK,MAAMK,UAAU,CAAC,GAAGF,OAAO,GAAGE,OAAOvB,MAAM;GAChD,MAAMb,KAAK;EACb;EACA,IAAIU,WAAW;GACb,OAAOA;EACT;EACA,IAAIiB,WAAW;GACb,OAAOO,QAAQyD,OAAO;EACxB;CACF;AACF"}
+{"version":3,"file":"testing.js","names":["ResourceElement","StatewireEnvelope","recordMutations","StatewireSSE","Statewire","openSSEStream","sseFrame","FakeStreamHandle","tick","Promise","r","setTimeout","applyOptimistic","state","TState","mutate","draft","MockTransport","transport","commands","method","params","context","setState","ack","seq","sendEnvelope","envelope","finish","reason","message","clientId","attached","mockCount","createMockTransport","initialState","finished","clients","Map","streams","Set","stream","send","fakeFetch","input","RequestInfo","URL","init","RequestInit","Response","TypeError","url","String","headers","Headers","id","get","endsWith","lastSeq","set","handle","ops","op","path","value","syn","lease","signal","undefined","Number","Math","max","push","JSON","parse","body","json","status","fetch","next","fin","close","size"],"sources":["../../src/transports/testing.ts"],"sourcesContent":["import type { ResourceElement } from \"@assistant-ui/tap\";\nimport type { StatewireEnvelope } from \"../client/connection\";\nimport { recordMutations } from \"../util/recordMutations\";\nimport { StatewireSSE } from \"./sse\";\nimport type { Statewire } from \"../types\";\nimport { openSSEStream, sseFrame } from \"./fake-sse\";\nimport type { FakeStreamHandle } from \"./fake-sse\";\n\nexport * from \"./fixtures\";\n\n/** Let queued microtasks and op batches drain through the statewire client. */\nexport const tick = () => new Promise<void>((r) => setTimeout(r, 0));\n\n/** Run a mutator against a recording draft, as the client does for `optimistic`. */\nexport const applyOptimistic = <TState>(\n  state: TState,\n  mutate: (draft: TState) => void,\n): TState => recordMutations(state, mutate).state;\n\nexport type MockTransport<TState> = {\n  /** Pass as the `transport:` option of useStatewire / Harness. */\n  transport: ResourceElement<Statewire<TState | undefined, any, any>>;\n  /** Captured `/commands` POST bodies, in POST order. */\n  commands: { method?: string; params?: unknown[]; context?: unknown }[];\n  /** Replace the replicated state and broadcast it as a root snapshot op. */\n  setState: (state: TState) => Promise<void>;\n  /** Deliver an ack watermark on the stream. */\n  ack: (seq: number) => Promise<void>;\n  /** Deliver an arbitrary envelope on the stream. */\n  sendEnvelope: (envelope: StatewireEnvelope) => Promise<void>;\n  /** End the stream with a fin; further attaches fail like a dead server. */\n  finish: (reason: string, message?: string) => Promise<void>;\n  readonly clientId: string;\n  readonly attached: boolean;\n};\n\n/**\n * A mock statewire backend: serves `initialState` as the attach snapshot,\n * records command POSTs (receipt only — acks and results are delivered from\n * test code), and exposes the wire as a transport ResourceElement.\n */\nlet mockCount = 0;\n\nexport const createMockTransport = <TState>(\n  initialState: TState,\n): MockTransport<TState> => {\n  let state: unknown = initialState;\n  let clientId = \"\";\n  let finished = false;\n  const clients = new Map<string, number>();\n  const streams = new Set<FakeStreamHandle>();\n  const commands: MockTransport<unknown>[\"commands\"] = [];\n\n  const sendEnvelope = (envelope: StatewireEnvelope) => {\n    for (const stream of streams) stream.send(sseFrame(envelope));\n    return tick();\n  };\n\n  const fakeFetch = (async (\n    input: RequestInfo | URL,\n    init?: RequestInit,\n  ): Promise<Response> => {\n    if (finished) throw new TypeError(\"fetch failed (finished)\");\n    const url = String(input);\n    const headers = new Headers(init?.headers);\n    const id = headers.get(\"Statewire-Client-Id\") ?? \"\";\n    if (url.endsWith(\"/stream\")) {\n      clientId = id;\n      const lastSeq = clients.get(id) ?? 0;\n      clients.set(id, lastSeq);\n      return openSSEStream(\n        streams,\n        id,\n        (handle) => {\n          handle.send(\n            sseFrame({\n              ops: [{ op: \"replace\", path: [], value: state }],\n              syn: { lastSeq, lease: \"mock-lease\" },\n            }),\n          );\n        },\n        init?.signal ?? undefined,\n      );\n    }\n    if (url.endsWith(\"/commands\")) {\n      const seq = Number(headers.get(\"Statewire-Command-Seq\"));\n      clients.set(id, Math.max(clients.get(id) ?? 0, seq));\n      commands.push(\n        JSON.parse(String(init?.body)) as MockTransport<unknown>[\"commands\"][number],\n      );\n      return Response.json({});\n    }\n    return new Response(null, { status: 404 });\n  }) as typeof fetch;\n\n  return {\n    transport: StatewireSSE({\n      url: `http://mock-${++mockCount}/api`,\n      fetch: fakeFetch,\n    }),\n    commands,\n    setState: (next) => {\n      state = next;\n      return sendEnvelope({ ops: [{ op: \"replace\", path: [], value: next }] });\n    },\n    ack: (seq) => sendEnvelope({ ack: seq }),\n    sendEnvelope,\n    finish: async (reason, message) => {\n      await sendEnvelope({\n        fin: { reason, ...(message !== undefined && { message }) },\n      });\n      finished = true;\n      for (const stream of [...streams]) stream.close();\n      await tick();\n    },\n    get clientId() {\n      return clientId;\n    },\n    get attached() {\n      return streams.size > 0;\n    },\n  };\n};\n"],"mappings":";;;;;;AAWA,MAAaQ,aAAa,IAAIC,SAAeC,MAAMC,WAAWD,GAAG,CAAC,CAAC;;AAGnE,MAAaE,mBACXC,OACAE,WACWb,gBAAgBW,OAAOE,MAAM,CAAC,CAACF;;;;;;AAwB5C,IAAIoB,YAAY;AAEhB,MAAaC,uBACXC,iBAC0B;CAC1B,IAAItB,QAAiBsB;CACrB,IAAIJ,WAAW;CACf,IAAIK,WAAW;CACf,MAAMC,0BAAU,IAAIC,IAAoB;CACxC,MAAMC,0BAAU,IAAIC,IAAsB;CAC1C,MAAMrB,WAA+C,CAAA;CAErD,MAAMO,gBAAgBC,aAAgC;EACpD,KAAK,MAAMc,UAAUF,SAASE,OAAOC,KAAKpC,SAASqB,QAAQ,CAAC;EAC5D,OAAOnB,KAAK;CACd;CAEA,MAAMmC,aAAa,OACjBC,OACAG,SACsB;EACtB,IAAIX,UAAU,MAAM,IAAIc,UAAU,yBAAyB;EAC3D,MAAMC,MAAMC,OAAOR,KAAK;EACxB,MAAMS,UAAU,IAAIC,QAAQP,MAAMM,OAAO;EACzC,MAAME,KAAKF,QAAQG,IAAI,qBAAqB,KAAK;EACjD,IAAIL,IAAIM,SAAS,SAAS,GAAG;GAC3B1B,WAAWwB;GACX,MAAMG,UAAUrB,QAAQmB,IAAID,EAAE,KAAK;GACnClB,QAAQsB,IAAIJ,IAAIG,OAAO;GACvB,OAAOrD,cACLkC,SACAgB,KACCK,WAAW;IACVA,OAAOlB,KACLpC,SAAS;KACPuD,KAAK,CAAC;MAAEC,IAAI;MAAWC,MAAM,CAAA;MAAIC,OAAOnD;KAAM,CAAC;KAC/CoD,KAAK;MAAEP;MAASQ,OAAO;KAAa;IACtC,CAAC,CACH;GACF,GACAnB,MAAMoB,UAAUC,KAAAA,CAClB;EACF;EACA,IAAIjB,IAAIM,SAAS,WAAW,GAAG;GAC7B,MAAMhC,MAAM4C,OAAOhB,QAAQG,IAAI,uBAAuB,CAAC;GACvDnB,QAAQsB,IAAIJ,IAAIe,KAAKC,IAAIlC,QAAQmB,IAAID,EAAE,KAAK,GAAG9B,GAAG,CAAC;GACnDN,SAASqD,KACPC,KAAKC,MAAMtB,OAAOL,MAAM4B,IAAI,CAAC,CAC/B;GACA,OAAO1B,SAAS2B,KAAK,CAAC,CAAC;EACzB;EACA,OAAO,IAAI3B,SAAS,MAAM,EAAE4B,QAAQ,IAAI,CAAC;CAC3C;CAEA,OAAO;EACL3D,WAAWf,aAAa;GACtBgD,KAAK,eAAe,EAAElB,UAAS;GAC/B6C,OAAOnC;EACT,CAAC;EACDxB;EACAI,WAAWwD,SAAS;GAClBlE,QAAQkE;GACR,OAAOrD,aAAa,EAAEmC,KAAK,CAAC;IAAEC,IAAI;IAAWC,MAAM,CAAA;IAAIC,OAAOe;GAAK,CAAC,EAAE,CAAC;EACzE;EACAvD,MAAMC,QAAQC,aAAa,EAAEF,KAAKC,IAAI,CAAC;EACvCC;EACAE,QAAQ,OAAOC,QAAQC,YAAY;GACjC,MAAMJ,aAAa,EACjBsD,KAAK;IAAEnD;IAAQ,GAAIC,YAAYsC,KAAAA,KAAa,EAAEtC,QAAQ;GAAG,EAC3D,CAAC;GACDM,WAAW;GACX,KAAK,MAAMK,UAAU,CAAC,GAAGF,OAAO,GAAGE,OAAOwC,MAAM;GAChD,MAAMzE,KAAK;EACb;EACA,IAAIuB,WAAW;GACb,OAAOA;EACT;EACA,IAAIC,WAAW;GACb,OAAOO,QAAQ2C,OAAO;EACxB;CACF;AACF"}

package/dist/transports/ws.d.ts

@@ -1,18 +1,9 @@
-import { StatewireSessionStore } from "../client/sessionStore.js";
-import { StatewireConnection } from "./sse.js";
-import { StatewireCommands, StatewireTransport } from "../types.js";
+import { StatewireClient } from "../statewire.js";
+import { Statewire } from "../types.js";
 import { ResourceElement } from "@assistant-ui/tap";
 //#region src/transports/ws.d.ts
-type StatewireWSOptions = {
-  url: string;
-  protocols?: string | string[];
-  query?: Record<string, string> | (() => Record<string, string> | Promise<Record<string, string>>);
-  webSocket?: new (url: string, protocols?: string | string[]) => WebSocket;
-  onChange?: (connection: StatewireConnection) => void;
-  sessionStore?: StatewireSessionStore;
-};
 declare const WS_SUBPROTOCOL = "statewire.v1";
-declare const StatewireWS: <TState = any, TCommands extends StatewireCommands = StatewireCommands, TEvent = any>(options: StatewireWSOptions) => ResourceElement<StatewireTransport<TState | undefined, TCommands, TEvent>>;
+declare const StatewireWS: <TState = any, TCommands extends Statewire.Commands = Statewire.Commands, TEvent = any>(options: StatewireClient.WSOptions) => ResourceElement<Statewire<TState | undefined, TCommands, TEvent>>;
 //#endregion
-export { StatewireWS, StatewireWSOptions, WS_SUBPROTOCOL };
+export { StatewireWS, WS_SUBPROTOCOL };
 //# sourceMappingURL=ws.d.ts.map

package/dist/transports/ws.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ws.d.ts","names":[],"sources":["../../src/transports/ws.ts"],"mappings":";;;;;KAuBY;EACV;EACA;EACA,QACI,gCACO,yBAAyB,QAAQ;EAC5C,iBAAiB,aAAa,kCAAkC;EAChE,YAAY,YAAY;EACxB,eAAe;;cAuBJ;cAmNA,cACX,cACA,kBAAkB,oBAAoB,mBACtC,cAEA,SAAS,uBACN,gBAAgB,mBAAmB,oBAAoB,WAAW"}
+{"version":3,"file":"ws.d.ts","names":[],"sources":["../../src/transports/ws.ts"],"mappings":";;;;cAmCa;cAgNA,cACX,cACA,kBAAkB,UAAU,WAAW,UAAU,UACjD,cAEA,SAAS,gBAAgB,cACtB,gBAAgB,UAAU,oBAAoB,WAAW"}