statewire 0.16.0 → 0.17.1

package/dist/host/socket.js

@@ -1,6 +1,8 @@
+import { encodeDocument, reviveDocumentValue } from "../document.js";
+import { DEFAULT_PROTOCOL, WIRE_VERSION, negotiateProtocols } from "../wire/protocol.js";
+import { applyDocumentOps, mainDocument } from "../wire/documents.js";
 import { StatewireSendError } from "../types.js";
 import { getCommandsHost, getStateHost } from "./host.js";
-import { encodeValue, reviveValue } from "../wire/meta.js";
 import { COMMAND_CAPACITY_RETRY_MS, MAX_INFLIGHT_COMMANDS_PER_HOST } from "../wire/limits.js";
 import { createMacrotaskPoster } from "../util/macrotask.js";
 import { frames, stream } from "./http.js";
@@ -49,15 +51,6 @@ const parseLane = (value) => {
 	return statements;
 };
 const frameEncoder = new TextEncoder();
-/** v1 ctx statements state the context whole: one root replace with an object value. */
-const contextValue = (ops) => {
-	if (!Array.isArray(ops) || ops.length !== 1) throw new Error("context must be a single-op array");
-	const op = ops[0];
-	if (!isPlainObject(op) || op["op"] !== "replace" || !Array.isArray(op["path"]) || op["path"].length !== 0) throw new Error("context op must be a root replace");
-	const value = op["meta"] !== void 0 ? reviveValue(op["value"], op["meta"], { rejectUnknown: true }) : op["value"];
-	if (!isPlainObject(value)) throw new Error("context value must be an object");
-	return value;
-};
 const parseFrameValue = (frame) => {
 	if (!isPlainObject(frame)) return violation("malformed frame");
 	const keys = Object.keys(frame);
@@ -69,16 +62,12 @@ const parseFrameValue = (frame) => {
 		const lane = frame["ctx"];
 		if (Array.isArray(lane) && lane.length > 256) return violation("too many context statements");
 		for (const statement of parseLane(lane)) {
-			let value;
-			try {
-				value = contextValue(statement["ops"]);
-			} catch {
-				return violation("invalid-context");
-			}
-			if (frameEncoder.encode(JSON.stringify(value)).byteLength > 65536) return violation("oversized-context");
+			const ops = statement["ops"];
+			if (!Array.isArray(ops) || !ops.length) return violation("invalid-context");
+			if (frameEncoder.encode(JSON.stringify(ops)).byteLength > 65536) return violation("oversized-context");
 			ctx.push({
 				seq: statement["seq"],
-				value
+				ops
 			});
 		}
 	}
@@ -88,8 +77,9 @@ const parseFrameValue = (frame) => {
 		if (Array.isArray(lane) && lane.length > 256) return violation("too many commands");
 		for (const statement of parseLane(lane)) {
 			const method = statement["method"];
-			if (typeof method !== "string" || !("params" in statement)) return violation("malformed statement");
+			if (typeof method !== "string" || typeof statement["protocol"] !== "string" || !("params" in statement)) return violation("malformed statement");
 			cmd.push({
+				protocol: statement["protocol"],
 				seq: statement["seq"],
 				method,
 				params: statement["params"],
@@ -114,7 +104,7 @@ const parseFrame = (raw) => {
 const frameIsOversized = (raw) => raw.length > 8388608 || frameEncoder.encode(raw).byteLength > 8388608;
 const encodePayload = (payload) => {
 	if (payload === void 0) return {};
-	const { value, meta } = encodeValue(payload);
+	const { value, meta } = encodeDocument(payload);
 	return {
 		payload: value,
 		...meta !== void 0 && { meta }
@@ -186,7 +176,7 @@ const scheduleFlush = (h) => {
 };
 const recordOp = (h, op) => {
 	if (op.op === "replace" || op.op === "add") {
-		const { value, meta } = encodeValue(op.value);
+		const { value, meta } = encodeDocument(op.value);
 		h.pendingOps.push({
 			...op,
 			value,
@@ -277,13 +267,16 @@ const flush = (h, fin) => {
 		}));
 		if (ops.length > 0) {
 			const packet = {
-				ops,
+				ops: conn.selection.protocols.some((p) => p.name === h.protocol.name) ? ops : [],
 				...appliedEntries.length > 0 && { cmd: appliedEntries },
 				...edge && { idle: level }
 			};
 			sendTo(h, conn, JSON.stringify(packet));
 		} else if (appliedEntries.length > 0) sendTo(h, conn, JSON.stringify({ cmd: appliedEntries }));
-		const verdicts = cmds.get(conn.clientId) ?? [];
+		const verdicts = (cmds.get(conn.clientId) ?? []).map((cmd) => conn.selection.protocols.some((p) => p.name === h.protocol.name) ? cmd : {
+			seq: cmd.seq,
+			...cmd.dur === void 0 ? {} : { dur: cmd.dur }
+		});
 		if (verdicts.length > 0) sendTo(h, conn, JSON.stringify({ cmd: verdicts }));
 		if (edge && ops.length === 0) sendTo(h, conn, JSON.stringify({ idle: level }));
 		if (fin !== void 0) {
@@ -311,6 +304,7 @@ const notifyClientRecords = (h) => {
 		ctxSeq: rec.ctxSeq,
 		touched: rec.touched,
 		context: rec.context,
+		contextDocuments: [...rec.contextDocuments.documents],
 		results: [...rec.results].map(([seq, entry]) => ({
 			seq,
 			ts: entry.ts,
@@ -380,7 +374,7 @@ const rejectStatement = (h, rec, adm, message) => settleCmd(h, rec, adm, {
 	code: "invalid-params",
 	message
 });
-const admitCmd = (h, rec, statement, auth) => {
+const admitCmd = (h, rec, statement, auth, selection) => {
 	const adm = {
 		seq: statement.seq,
 		method: statement.method,
@@ -390,13 +384,22 @@ const admitCmd = (h, rec, statement, auth) => {
 	};
 	rec.inflight.set(statement.seq, adm);
 	notifyClientRecords(h);
+	if (statement.protocol !== h.protocol.name || !selection.protocols.some((p) => p.name === statement.protocol)) {
+		settleCmd(h, rec, adm, {
+			seq: adm.seq,
+			type: "rejected",
+			code: "unsupported-protocol",
+			message: "command protocol was not negotiated"
+		});
+		return;
+	}
 	if (!Array.isArray(statement.params)) {
 		rejectStatement(h, rec, adm, "params must be an array");
 		return;
 	}
 	let params = statement.params;
 	if (statement.meta !== void 0) try {
-		params = reviveValue(params, statement.meta, { rejectUnknown: true });
+		params = reviveDocumentValue(params, statement.meta, { rejectUnknown: true });
 	} catch (error) {
 		rejectStatement(h, rec, adm, `invalid params: ${error.message}`);
 		return;
@@ -434,7 +437,13 @@ const admitCmd = (h, rec, statement, auth) => {
 		});
 	});
 };
-const admitFrame = (h, rec, frame, auth) => {
+const admitFrame = (h, rec, frame, auth, selection = {
+	version: h.wireVersion.version,
+	protocols: [{
+		name: h.protocol.name,
+		version: h.protocol.version
+	}]
+}) => {
 	rec.touched = Date.now();
 	const through = admittedThrough(rec);
 	for (const statement of frame.cmd) if (statement.seq <= through) reemit(h, rec, statement.seq);
@@ -476,11 +485,17 @@ const admitFrame = (h, rec, frame, auth) => {
 			};
 		}
 	}
-	for (const statement of ctxNew) {
-		rec.context = statement.value;
-		rec.ctxSeq = statement.seq;
+	let contextDocuments = rec.contextDocuments;
+	try {
+		for (const statement of ctxNew) contextDocuments = applyDocumentOps(contextDocuments, statement.ops, selection.protocols, true);
+		if (frameEncoder.encode(JSON.stringify([...contextDocuments.documents])).byteLength > 65536) return violation("oversized-context");
+	} catch {
+		return violation("invalid-context");
 	}
-	for (const statement of cmdNew) admitCmd(h, rec, statement, auth);
+	rec.contextDocuments = contextDocuments;
+	rec.context = mainDocument(contextDocuments, h.protocol.name)?.value ?? null;
+	if (ctxNew.length) rec.ctxSeq = ctxNew.at(-1).seq;
+	for (const statement of cmdNew) admitCmd(h, rec, statement, auth, selection);
 	if (cmdNew.length > 0) queueCmd(h, rec.id, {
 		seq: cmdNew.at(-1).seq,
 		dur: true
@@ -516,7 +531,19 @@ const message = (h, conn, data) => {
 	}
 	if (frame.ctx.length === 0 && frame.cmd.length === 0) return;
 	const rec = h.clients.get(conn.clientId);
-	const admission = admitFrame(h, rec, frame, conn.auth);
+	let admission;
+	try {
+		admission = admitFrame(h, rec, frame, conn.auth, conn.selection);
+	} catch (error) {
+		const reason = violationReason(error);
+		if (reason === null) throw error;
+		sendTo(h, conn, JSON.stringify({ fin: {
+			reason: "protocol-error",
+			message: reason
+		} }));
+		closeConn(h, conn, 1008, reason);
+		return;
+	}
 	if (admission.type === "refused") sendTo(h, conn, JSON.stringify(admission.syn));
 };
 const closePending = (h, pending, code, reason) => {
@@ -573,7 +600,7 @@ const normalizedHeaders = (headers) => {
 };
 const connectAuthorized = (h, clientId, socket, options) => {
 	const authorize = h.authorize;
-	if (options.authorized === true) return register(h, clientId, socket, options.authorized === true, options.auth)?.connection ?? null;
+	if (options.authorized === true) return register(h, clientId, socket, options.authorized === true, options.auth, options.selection)?.connection ?? null;
 	const refuse = (code, reason) => {
 		try {
 			socket.close(code, reason);
@@ -640,7 +667,10 @@ const connectAuthorized = (h, clientId, socket, options) => {
 			const frame = initial.frame;
 			let headers;
 			try {
-				headers = normalizedHeaders(initial.headers);
+				headers = {
+					...normalizedHeaders(initial.headers),
+					...normalizedHeaders(options.headers ?? {})
+				};
 			} catch {
 				sendPending(pending, JSON.stringify({ fin: {
 					reason: "protocol-error",
@@ -661,17 +691,23 @@ const connectAuthorized = (h, clientId, socket, options) => {
 			pending.authorizing = true;
 			Promise.resolve().then(() => authorize?.({ headers })).then((auth) => {
 				if (!pending.open || h.disposed || h.finished !== null) return;
-				options.onAuthorized?.(auth);
+				const selection = negotiateProtocols(new Headers(headers), h.wireVersion, [h.protocol]);
+				if ("fin" in selection) {
+					sendPending(pending, JSON.stringify(selection));
+					closePending(h, pending, 1008, selection.fin.message);
+					return;
+				}
+				options.onAuthorized?.(auth, selection);
 				pending.open = false;
 				h.pendingConns.delete(pending);
 				if (pending.timer !== null) clearTimeout(pending.timer);
-				const registration = register(h, clientId, socket, true, auth);
+				const registration = register(h, clientId, socket, true, auth, selection);
 				if (registration === null) return;
 				pending.connection = registration.connection;
-				if (frame.ctx.length || frame.cmd.length) {
-					const admission = admitFrame(h, registration.rec, frame, auth);
-					if (admission.type === "refused") sendTo(h, registration.conn, JSON.stringify(admission.syn));
-				}
+				if (frame.ctx.length || frame.cmd.length) message(h, registration.conn, JSON.stringify({
+					...frame.ctx.length ? { ctx: frame.ctx } : {},
+					...frame.cmd.length ? { cmd: frame.cmd } : {}
+				}));
 				for (const data of pending.queued.splice(0)) registration.connection.message(data);
 			}).catch((error) => {
 				if (pending.open) rejectPendingAuthorization(h, pending, error);
@@ -698,7 +734,13 @@ const gcClients = (h) => {
 	}
 	if (changed) notifyClientRecords(h);
 };
-const register = (h, clientId, socket, sawFrame = false, auth = void 0) => {
+const register = (h, clientId, socket, sawFrame = false, auth = void 0, selection = {
+	version: h.wireVersion.version,
+	protocols: [{
+		name: h.protocol.name,
+		version: h.protocol.version
+	}]
+}) => {
 	const refuse = (code, reason) => {
 		try {
 			socket.close(code, reason);
@@ -734,6 +776,10 @@ const register = (h, clientId, socket, sawFrame = false, auth = void 0) => {
 			ctxSeq: 0,
 			touched: Date.now(),
 			context: null,
+			contextDocuments: {
+				mode: "mounted",
+				documents: /* @__PURE__ */ new Map()
+			},
 			results: /* @__PURE__ */ new Map(),
 			inflight: /* @__PURE__ */ new Map(),
 			principal: auth
@@ -743,6 +789,7 @@ const register = (h, clientId, socket, sawFrame = false, auth = void 0) => {
 	})();
 	rec.touched = Date.now();
 	const conn = {
+		selection,
 		clientId,
 		socket,
 		open: true,
@@ -755,20 +802,23 @@ const register = (h, clientId, socket, sawFrame = false, auth = void 0) => {
 	gcClients(h);
 	notifyClientRecords(h);
 	ensureHeartbeat(h);
-	const { value, meta } = encodeValue(h.stateHost.snapshot());
+	const { value, meta } = encodeDocument(h.stateHost.snapshot());
 	const hello = {
 		ops: [{
-			op: "replace",
-			path: [],
+			op: "mount",
+			protocol: h.protocol.name,
+			main: true,
 			value,
 			...meta !== void 0 && { meta }
 		}],
 		syn: {
+			...selection,
 			seq: admittedThrough(rec),
 			ctxSeq: rec.ctxSeq,
 			...h.idleWire && { idle: true }
 		}
 	};
+	if (!selection.protocols.some((p) => p.name === h.protocol.name)) hello.ops = [];
 	sendTo(h, conn, JSON.stringify(hello));
 	return {
 		rec,
@@ -808,6 +858,10 @@ const suspend = (h) => {
 };
 const restoreClientRecords = (records = []) => new Map(records.map((record) => [record.id, {
 	...record,
+	contextDocuments: {
+		mode: "mounted",
+		documents: new Map(record.contextDocuments ?? [])
+	},
 	results: new Map(record.results.map(({ seq, ts, cmd }) => [seq, {
 		ts,
 		cmd
@@ -817,6 +871,11 @@ const restoreClientRecords = (records = []) => new Map(records.map((record) => [
 const useSocketHost = (transport, options, persistence) => {
 	const [h] = useState(() => {
 		const host = {
+			protocol: {
+				name: DEFAULT_PROTOCOL,
+				version: WIRE_VERSION
+			},
+			wireVersion: { version: WIRE_VERSION },
 			heartbeatMs: 15e3,
 			wsInitTimeoutMs: 5e3,
 			maxSubscribers: MAX_SUBSCRIBERS,
@@ -852,6 +911,11 @@ const useSocketHost = (transport, options, persistence) => {
 		};
 		return host;
 	});
+	h.protocol = options?.protocol ?? {
+		name: "default",
+		version: "2026-09-07"
+	};
+	h.wireVersion = options?.wireVersion ?? { version: "2026-09-07" };
 	setHeartbeatMs(h, options?.heartbeatIntervalMs ?? 15e3);
 	h.pushback = options?.pushback;
 	h.wsInitTimeoutMs = options?.wsInitTimeoutMs ?? 5e3;