statewire 0.0.1 → 0.0.2

package/dist/transports/sse.js

@@ -1,41 +1,21 @@
 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;
+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 +23,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 +37,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,13 +69,13 @@ 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 }
 		}),
@@ -108,7 +86,15 @@ const sendCommand = async (c, { seq, 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,9 +112,24 @@ 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 };

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","StatewireSubmitResult","type","response","message","ConnectionState","status","cause","attempt","nextRetryAt","lastError","payload","RemoteLink","connection","Connection","send","seq","command","Promise","baseUrl","options","SSEOptions","url","replace","requestHeaders","extra","Record","headers","merged","Headers","name","value","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","undefined","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;\n  params: readonly unknown[];\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({\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    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":";;;;;;;;AAoEA,MAAMqD,WAAWC,YACfA,QAAQE,IAAIC,QAAQ,OAAO,EAAE;AAE/B,MAAMC,iBAAiB,OACrBJ,SACAK,UACG;CACH,MAAM,EAAEE,YAAYP;CACpB,MAAMQ,SAAS,IAAIC,QACjB,OAAOF,YAAY,aAAa,MAAMA,QAAQ,IAAIA,OACpD;CACA,KAAK,MAAM,CAACG,MAAMC,UAAUC,OAAOC,QAAQR,KAAK,GAAGG,OAAOM,IAAIJ,MAAMC,KAAK;CACzE,OAAOH;AACT;AAEA,MAAMO,aAAa,OACjBf,SACAgB,OACAC,MACAI,YACsB;CACtB,IAAI;EACF,OAAO,OAAOrB,QAAQuB,SAASC,WAAWD,MAAAA,CAAOP,OAAOC,IAAI;CAC9D,QAAQ;EACN,MAAM9D,SAASkE,OAAO;CACxB;AACF;AAEA,MAAMI,UAAU,OACdC,GACAC,QACAG,WACAE,SAC2C;CAC3C,MAAMC,UAAU,IAAIC,YAAY;CAChC,MAAMC,QAAQlF,gBAAgB;CAC9B,SAAS;EACP,MAAMmF,QAAQC,iBACNP,UAAUQ,MAAM,GACtBpF,OAAOqF,kBACT;EACA,IAAIC;EACJ,IAAI;GACFA,SAAS,MAAMb,OAAOe,KAAK;EAC7B,QAAQ;GACN,MAAMvF,SAAS2E,UAAUX,MAAM;EACjC,UAAU;GACRwB,aAAaP,KAAK;EACpB;EACA,IAAII,OAAOI,MAAM,OAAO;EACxB,KAAK,MAAMC,QAAQV,MAAMF,QAAQa,OAAON,OAAO7B,OAAO,EAAEoC,QAAQ,KAAK,CAAC,CAAC,GAAG;GACxE,IAAIf,KAAK,GAAG,OAAO;GACnB,MAAMgB,MAAMC,KAAKd,MAAMU,IAAI;GAC3B,MAAMD,OAAOhG,mBAAmBQ,eAAesE,GAAGsB,GAAG,CAAC;GACtD,IAAIJ,MAAM,OAAOA;EACnB;CACF;AACF;AAEA,MAAMM,gBAAgB,OACpBxB,GACA1B,SACAsC,UAC2C;CAC3C,MAAM/B,UAAU,MAAMH,eAAeJ,SAAS;EAC5C,uBAAuB0B,EAAEyB,IAAInF;EAC7BoF,QAAQ;CACV,CAAC;CACD,MAAMtB,YAAY,IAAIC,gBAAgB;CACtC,MAAMsB,cAAchB,iBACZP,UAAUQ,MAAM,GACtBpF,OAAOqF,kBACT;CACA,IAAIjE;CACJ,IAAI;EACFA,MAAM,MAAMyC,WACVf,SACA,GAAGD,QAAQC,OAAO,EAAC,UACnB;GACEO;GACA+C,OAAO;GACPnC,QAAQC,YAAYmC,IAAI,CAACjB,MAAMnB,QAAQW,UAAUX,MAAM,CAAC;EAC1D,GACAW,UAAUX,MACZ;CACF,UAAU;EACRwB,aAAaU,WAAW;CAC1B;CACA,IAAI/E,IAAIY,WAAW,OAAOZ,IAAIY,WAAW,KAAK,MAAMsE,WAAWlF,GAAG;CAClE,IAAI,CAACA,IAAImF,MAAM,CAACnF,IAAIoF,MAAM,MAAM,IAAIC,MAAM,QAAQrF,IAAIY,QAAQ;CAC9D,MAAMyC,SAASrD,IAAIoF,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,cAAclF,QAClBjB,gBAAgBC,gBAAgBgB,IAAIiC,QAAQyD,IAAI,aAAa,CAAC,CAAC;AAEjE,MAAMC,cAAc,OAClBvC,GACA1B,SACA,EAAEJ,KAAKC,cAC4B;CACnC,MAAMwB,UAAUD,YAAYC,QAAQnE,OAAOqF,kBAAkB;CAC7D,MAAMjE,MAAM,MAAMyC,WAChBf,SACA,GAAGD,QAAQC,OAAO,EAAC,YACnB;EACErB,QAAQ;EACR4B,SAAS,MAAMH,eAAeJ,SAAS;GACrC,gBAAgB;GAChB,uBAAuB0B,EAAEyB,IAAInF;GAC7B,yBAAyBkG,OAAOtE,GAAG;GACnC,GAAI8B,EAAEyC,UAAU,QAAQ,EAAE,mBAAmBzC,EAAEyC,MAAM;EACvD,CAAC;EACDT,MAAMT,KAAKmB,UAAU;GACnBzF,QAAQkB,QAAQlB;GAChB,GAAG5B,aAAa8C,QAAQjB,MAAM;EAChC,CAAC;EACDuC,QAAQC,YAAYmC,IAAI,CACtB7B,EAAE2C,UAAUlD,UAAUC,YAAYkB,MAAM,GACxCjB,OAAO,CACR;CACH,GACAA,OACF;CACA,IAAI/C,IAAImF,IAAI;EACV,IAAIC;EACJ,IAAI;GACFA,OAAQ,MAAMpF,IAAIgG,KAAK;EACzB,QAAQ;GACN,OAAO;IACLxF,MAAM;IACNE,SAAS;GACX;EACF;EACA,OAAO0E,MAAMpF,QAAQiG,KAAAA,IACjB;GAAEzF,MAAM;GAAWC,UAAU/B,eAAe0G,KAAKpF,GAAG;EAAE,IACtD,EAAEQ,MAAM,UAAU;CACxB;CACA,IAAIR,IAAIY,WAAW,OAAOZ,IAAIY,WAAW,OAAOZ,IAAIY,WAAW,KAAK;EAGlE,MAAMsF,UACJlG,IAAIY,WAAW,MACV,YACDZ,IAAIY,WAAW,MACZ,mBACA;EAIT,KAAIwE,MAHgBpF,IAAIgG,KAAK,CAAC,CAACP,YAAY,IAAI,EAAA,EAGrCW,UAAUF,SAAS,OAAO,EAAE1F,MAAM0F,QAAQ;EACpD,MAAM,IAAIb,MAAM,QAAQrF,IAAIY,QAAQ;CACtC;CACA,IAAIZ,IAAIY,WAAW,OAAOZ,IAAIY,WAAW,KAAK,MAAMsE,WAAWlF,GAAG;CAClE,IAAIA,IAAIY,UAAU,OAAOZ,IAAIY,SAAS,KACpC,OAAO;EAAEJ,MAAM;EAAYE,SAAS,QAAQV,IAAIY;CAAS;CAE3D,MAAM,IAAIyE,MAAM,QAAQrF,IAAIY,QAAQ;AACtC;AAEA,MAAMyF,cACJ3E,SACAmD,QACe;CACf,MAAMyB,SAASlI,OAAOsD,OAAO;CAC7B4E,OAAOC,UAAU7E;CACjB,MAAM0B,MAAInE,iBAAiB;EACzB2C,KAAKF,QAAQE;EACb4E,UAAU9E,QAAQ8E;EAClB3B;EACA4B,cAAcrD,GAAGY,UAAUY,cAAcxB,GAAG1B,SAASsC,KAAK;CAC5D,CAAC;CACD,MAAM,CAAC0C,QAAQrI,gBACM;EACjB,IAAI8C,aAAa;GACf,OAAOiC,IAAEjC;EACX;EACAE,OAAOzB,SAAS+F,YAAYvC,KAAGkD,OAAOC,SAAS3G,IAAI;CACrD,EACF;CACA,OAAO8G;AACT;AAEA,MAAMC,mBACJjF,YAC6BlC,mBAAmBkC,SAAS2E,UAAU;AAErE,MAAaO,eAAerI,SAASoI,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,7 +9,7 @@ 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>>;
+  transport: ResourceElement<Statewire<TState | undefined, any, any>>;
   /** Captured `{ method, params }` command bodies, in POST order. */
   commands: {
     method: string;
@@ -27,5 +28,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;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"}

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","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 `{ 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<FakeStreamHandle>();\n  const commands: { method: string; params: unknown[] }[] = [];\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 { 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":";;;;;;AAWA,MAAaQ,aAAa,IAAIC,SAAeC,MAAMC,WAAWD,GAAG,CAAC,CAAC;;AAGnE,MAAaE,mBACXC,OACAE,WACWb,gBAAgBW,OAAOE,MAAM,CAAC,CAACF;;;;;;AAwB5C,IAAImB,YAAY;AAEhB,MAAaC,uBACXC,iBAC0B;CAC1B,IAAIrB,QAAiBqB;CACrB,IAAIJ,WAAW;CACf,IAAIK,WAAW;CACf,MAAMC,0BAAU,IAAIC,IAAoB;CACxC,MAAMC,0BAAU,IAAIC,IAAsB;CAC1C,MAAMpB,WAAoD,CAAA;CAE1D,MAAMM,gBAAgBC,aAAgC;EACpD,KAAK,MAAMc,UAAUF,SAASE,OAAOC,KAAKnC,SAASoB,QAAQ,CAAC;EAC5D,OAAOlB,KAAK;CACd;CAEA,MAAMkC,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,OAAOpD,cACLiC,SACAgB,KACCK,WAAW;IACVA,OAAOlB,KACLnC,SAAS;KACPsD,KAAK,CAAC;MAAEC,IAAI;MAAWC,MAAM,CAAA;MAAIC,OAAOlD;KAAM,CAAC;KAC/CmD,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;GACnDL,SAASoD,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;EACL1D,WAAWf,aAAa;GACtB+C,KAAK,eAAe,EAAElB,UAAS;GAC/B6C,OAAOnC;EACT,CAAC;EACDvB;EACAG,WAAWwD,SAAS;GAClBjE,QAAQiE;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,MAAMxE,KAAK;EACb;EACA,IAAIsB,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;cAsNA,cACX,cACA,kBAAkB,UAAU,WAAW,UAAU,UACjD,cAEA,SAAS,gBAAgB,cACtB,gBAAgB,UAAU,oBAAoB,WAAW"}

package/dist/transports/ws.js

@@ -1,67 +1,42 @@
 import { encodeParams } from "../wire/meta.js";
+import { CLIENT_QUERY_PARAM } from "../host/adapter.js";
 import { timing } from "../client/timing.js";
-import { createGate, handleEnvelope, overloadedError, reconnectConn } from "../client/connection.js";
+import { handleEnvelope, overloadedError, useStatewireConn } from "../client/connection.js";
 import { useStatewireRemote } from "./remote.js";
+import { useState } from "@assistant-ui/tap/react-shim";
 import { flushTapSync, resource } from "@assistant-ui/tap";
 //#region src/transports/ws.ts
 const OPEN = 1;
 const WS_SUBPROTOCOL = "statewire.v1";
-const createConn = (latest, notify) => {
-	const c = {
-		hasSnapshot: false,
-		lease: null,
-		attempt: 0,
-		episode: null,
-		abort: null,
-		socket: null,
-		pending: [],
-		gate: createGate(),
-		lifetime: null,
-		latest,
-		notify,
-		reconnect: () => reconnectConn(c),
-		attemptOnce: (abort) => attemptSocket(c, abort),
-		send: (args) => sendCommand(c, args),
-		connection: {
-			status: "connecting",
-			reconnect: () => c.reconnect()
-		}
-	};
-	return c;
-};
-const socketUrl = async (c) => {
-	const { url, query } = c.latest.current.options;
+const socketUrl = async (options, clientId) => {
+	const { url, query } = options;
 	const params = new URLSearchParams(typeof query === "function" ? await query() : query);
-	params.set("client", c.latest.current.ctx.clientId);
+	params.set(CLIENT_QUERY_PARAM, clientId);
 	return `${url.replace(/\/$/, "").replace(/^http/, "ws")}/ws?${params.toString()}`;
 };
-const settleOverloaded = (c, { seq, retryAfterMs }) => {
+const settleOverloaded = (w, { seq, retryAfterMs }) => {
 	if (seq === void 0) throw overloadedError(retryAfterMs);
-	const entry = c.pending.find((p) => p.seq === seq);
+	const entry = w.pending.find((p) => p.seq === seq);
 	if (!entry) return;
-	c.pending = c.pending.filter((p) => p !== entry);
+	w.pending = w.pending.filter((p) => p !== entry);
 	entry.reject(overloadedError(retryAfterMs));
 };
-const resolveReceipts = (c, ack) => {
-	while (c.pending[0] && c.pending[0].seq <= ack) c.pending.shift().resolve({ type: "receipt" });
-};
-const resolveResponded = (c, seqs) => {
+const resolvePending = (w, matches) => {
 	const remaining = [];
-	for (const entry of c.pending) if (seqs.has(entry.seq)) entry.resolve({ type: "receipt" });
+	for (const entry of w.pending) if (matches(entry)) entry.resolve({ type: "receipt" });
 	else remaining.push(entry);
-	c.pending = remaining;
+	w.pending = remaining;
 };
-const failPending = (c, error) => {
-	const pending = c.pending;
-	c.pending = [];
+const failPending = (w, error) => {
+	const pending = w.pending;
+	w.pending = [];
 	for (const entry of pending) entry.reject(error);
 };
-const runSocket = (c, ws, abort) => new Promise((resolve, reject) => {
+const runSocket = (c, w, ws, abort) => new Promise((resolve, reject) => {
 	let timer;
 	let timedOut = false;
 	let outcome = null;
 	let closed = false;
-	let attached = false;
 	const arm = () => {
 		clearTimeout(timer);
 		timer = setTimeout(() => {
@@ -79,10 +54,15 @@ const runSocket = (c, ws, abort) => new Promise((resolve, reject) => {
 		ws.close();
 	};
 	const processEnv = (env) => {
-		const done = flushTapSync(() => handleEnvelope(c, env, attached));
-		attached = true;
-		if (env.ack !== void 0) resolveReceipts(c, env.ack);
-		if (env.res !== void 0) resolveResponded(c, new Set(env.res.map((r) => r.seq)));
+		const done = flushTapSync(() => handleEnvelope(c, env));
+		if (env.ack !== void 0) {
+			const ack = env.ack;
+			resolvePending(w, (entry) => entry.seq <= ack);
+		}
+		if (env.res !== void 0) {
+			const seqs = new Set(env.res.map((r) => r.seq));
+			resolvePending(w, (entry) => seqs.has(entry.seq));
+		}
 		if (done) {
 			outcome = done;
 			ws.close();
@@ -99,7 +79,7 @@ const runSocket = (c, ws, abort) => new Promise((resolve, reject) => {
 			return;
 		}
 		try {
-			if (frame.overloaded !== void 0) settleOverloaded(c, frame.overloaded);
+			if (frame.overloaded !== void 0) settleOverloaded(w, frame.overloaded);
 			else processEnv(frame);
 		} catch (error) {
 			fail(error);
@@ -108,28 +88,28 @@ const runSocket = (c, ws, abort) => new Promise((resolve, reject) => {
 	ws.onclose = () => {
 		closed = true;
 		clearTimeout(timer);
-		if (c.socket === ws) failPending(c, /* @__PURE__ */ new Error("network"));
+		if (w.socket === ws) failPending(w, /* @__PURE__ */ new Error("network"));
 		if (outcome) resolve(outcome);
 		else reject(/* @__PURE__ */ new Error(timedOut ? "timeout" : "network"));
 	};
 });
-const attemptSocket = async (c, abort) => {
+const attemptSocket = async (c, w, options, abort) => {
 	let ws = null;
 	try {
-		const url = await socketUrl(c);
+		const url = await socketUrl(options, c.ctx.clientId);
 		abort.signal.throwIfAborted();
-		const { protocols, webSocket } = c.latest.current.options;
+		const { protocols, webSocket } = options;
 		const WS = webSocket ?? globalThis.WebSocket;
 		if (!WS) throw new Error("no WebSocket implementation");
 		ws = new WS(url, [WS_SUBPROTOCOL, ...protocols === void 0 ? [] : typeof protocols === "string" ? [protocols] : protocols]);
-		c.socket = ws;
-		return await runSocket(c, ws, abort);
+		w.socket = ws;
+		return await runSocket(c, w, ws, abort);
 	} finally {
-		if (c.socket === ws) c.socket = null;
+		if (w.socket === ws) w.socket = null;
 	}
 };
-const sendCommand = (c, { seq, command }) => {
-	const ws = c.socket;
+const sendCommand = (w, { seq, command }) => {
+	const ws = w.socket;
 	if (!ws || ws.readyState !== OPEN) return Promise.reject(/* @__PURE__ */ new Error("network"));
 	return new Promise((resolve, reject) => {
 		const entry = {
@@ -144,11 +124,11 @@ const sendCommand = (c, { seq, command }) => {
 			}
 		};
 		const timer = setTimeout(() => {
-			c.pending = c.pending.filter((p) => p !== entry);
+			w.pending = w.pending.filter((p) => p !== entry);
 			reject(/* @__PURE__ */ new Error("timeout"));
 			ws.close();
 		}, timing.heartbeatTimeoutMs);
-		c.pending.push(entry);
+		w.pending.push(entry);
 		try {
 			ws.send(JSON.stringify({
 				method: command.method,
@@ -157,14 +137,31 @@ const sendCommand = (c, { seq, command }) => {
 			}));
 		} catch {
 			clearTimeout(timer);
-			c.pending = c.pending.filter((p) => p !== entry);
+			w.pending = w.pending.filter((p) => p !== entry);
 			reject(/* @__PURE__ */ new Error("network"));
 		}
 	});
 };
-const useStatewireWS = (options) => {
-	return useStatewireRemote(options, createConn);
+const useWsLink = (options, ctx) => {
+	const [w] = useState(() => ({
+		socket: null,
+		pending: []
+	}));
+	const c_0 = useStatewireConn({
+		url: options.url,
+		onChange: options.onChange,
+		ctx,
+		attemptOnce: (c, abort) => attemptSocket(c, w, options, abort)
+	});
+	const [link] = useState(() => ({
+		get connection() {
+			return c_0.connection;
+		},
+		send: (args) => sendCommand(w, args)
+	}));
+	return link;
 };
+const useStatewireWS = (options) => useStatewireRemote(options, useWsLink);
 const StatewireWS = resource(useStatewireWS);
 //#endregion
 export { StatewireWS, WS_SUBPROTOCOL };