@rocicorp/zero 0.26.1 → 0.26.2-canary.4

package/out/zero-cache/src/workers/syncer-ws-message-handler.js

@@ -1,207 +1,125 @@
-import { trace } from "@opentelemetry/api";
-import { Lock } from "@rocicorp/lock";
-import { startAsyncSpan, startSpan } from "../../../otel/src/span.js";
-import { version } from "../../../otel/src/version.js";
-import { unreachable, assert } from "../../../shared/src/asserts.js";
+import { assert, unreachable } from "../../../shared/src/asserts.js";
 import { InvalidPush } from "../../../zero-protocol/src/error-kind-enum.js";
 import { ZeroCache } from "../../../zero-protocol/src/error-origin-enum.js";
+import { startAsyncSpan, startSpan } from "../../../otel/src/span.js";
+import { version } from "../../../otel/src/version.js";
 import "../services/view-syncer/view-syncer.js";
-const tracer = trace.getTracer("syncer-ws-server", version);
-class SyncerWsMessageHandler {
-  #viewSyncer;
-  #mutagen;
-  #mutationLock;
-  #lc;
-  #clientGroupID;
-  #syncContext;
-  #pusher;
-  constructor(lc, connectParams, viewSyncer, mutagen, pusher) {
-    const {
-      clientGroupID,
-      clientID,
-      profileID,
-      wsID,
-      baseCookie,
-      protocolVersion,
-      httpCookie,
-      origin,
-      userID
-    } = connectParams;
-    this.#viewSyncer = viewSyncer;
-    this.#mutagen = mutagen;
-    this.#mutationLock = new Lock();
-    this.#lc = lc.withContext("connection").withContext("clientID", clientID).withContext("clientGroupID", clientGroupID).withContext("wsID", wsID);
-    this.#clientGroupID = clientGroupID;
-    this.#pusher = pusher;
-    this.#syncContext = {
-      clientID,
-      profileID,
-      wsID,
-      baseCookie,
-      protocolVersion,
-      httpCookie,
-      origin,
-      userID
-    };
-  }
-  async handleMessage(msg) {
-    const lc = this.#lc;
-    const msgType = msg[0];
-    const viewSyncer = this.#viewSyncer;
-    switch (msgType) {
-      case "ping":
-        lc.error?.("Ping is not supported at this layer by Zero");
-        break;
-      case "pull":
-        lc.error?.("Pull is not supported by Zero");
-        break;
-      case "push": {
-        return startAsyncSpan(
-          tracer,
-          "connection.push",
-          async () => {
-            const { clientGroupID, mutations, auth: pushAuth } = msg[1];
-            if (clientGroupID !== this.#clientGroupID) {
-              return [
-                {
-                  type: "fatal",
-                  error: {
-                    kind: InvalidPush,
-                    message: `clientGroupID in mutation "${clientGroupID}" does not match clientGroupID of connection "${this.#clientGroupID}`,
-                    origin: ZeroCache
-                  }
-                }
-              ];
-            }
-            if (pushAuth) {
-              await viewSyncer.updateAuth(this.#syncContext, [
-                "updateAuth",
-                { auth: pushAuth }
-              ]);
-            }
-            if (mutations.length === 0) {
-              return [
-                {
-                  type: "ok"
-                }
-              ];
-            }
-            if (mutations[0].type === "custom") {
-              assert(
-                this.#pusher,
-                "A ZERO_MUTATE_URL must be set in order to process custom mutations."
-              );
-              return [
-                this.#pusher.enqueuePush(
-                  this.#syncContext.clientID,
-                  msg[1],
-                  viewSyncer.auth?.raw,
-                  this.#syncContext.httpCookie,
-                  this.#syncContext.origin
-                )
-              ];
-            }
-            const auth = viewSyncer.auth;
-            assert(
-              auth?.type !== "opaque",
-              "Only JWT auth is supported for CRUD mutations"
-            );
-            const ret = await this.#mutationLock.withLock(async () => {
-              const errors = [];
-              for (const mutation of mutations) {
-                const maybeError = await this.#mutagen.processMutation(
-                  mutation,
-                  auth?.decoded,
-                  this.#pusher !== void 0
-                );
-                if (maybeError !== void 0) {
-                  errors.push({
-                    kind: maybeError[0],
-                    message: maybeError[1],
-                    origin: ZeroCache
-                  });
-                }
-              }
-              if (errors.length > 0) {
-                return { type: "transient", errors };
-              }
-              return { type: "ok" };
-            });
-            return [ret];
-          }
-        );
-      }
-      case "changeDesiredQueries":
-        await startAsyncSpan(
-          tracer,
-          "connection.changeDesiredQueries",
-          () => viewSyncer.changeDesiredQueries(this.#syncContext, msg)
-        );
-        break;
-      case "updateAuth":
-        await startAsyncSpan(tracer, "connection.updateAuth", async () => {
-          await viewSyncer.updateAuth(this.#syncContext, msg);
-        });
-        break;
-      case "deleteClients": {
-        const deletedClientIDs = await startAsyncSpan(
-          tracer,
-          "connection.deleteClients",
-          () => viewSyncer.deleteClients(this.#syncContext, msg)
-        );
-        if (this.#pusher && deletedClientIDs.length > 0) {
-          await this.#pusher.deleteClientMutations(deletedClientIDs);
-        }
-        break;
-      }
-      case "initConnection": {
-        const ret = [
-          {
-            type: "stream",
-            source: "viewSyncer",
-            stream: startSpan(
-              tracer,
-              "connection.initConnection",
-              () => viewSyncer.initConnection(this.#syncContext, msg)
-            )
-          }
-        ];
-        if (this.#pusher) {
-          ret.push({
-            type: "stream",
-            source: "pusher",
-            stream: this.#pusher.initConnection(
-              this.#syncContext.clientID,
-              this.#syncContext.wsID,
-              msg[1].userPushURL,
-              msg[1].userPushHeaders,
-              () => viewSyncer.clearAuth()
-            )
-          });
-        }
-        return ret;
-      }
-      case "closeConnection":
-        break;
-      case "inspect":
-        await startAsyncSpan(
-          tracer,
-          "connection.inspect",
-          () => viewSyncer.inspect(this.#syncContext, msg)
-        );
-        break;
-      case "ackMutationResponses":
-        if (this.#pusher) {
-          await this.#pusher.ackMutationResponses(msg[1]);
-        }
-        break;
-      default:
-        unreachable();
-    }
-    return [{ type: "ok" }];
-  }
-}
-export {
-  SyncerWsMessageHandler
+import { Lock } from "@rocicorp/lock";
+import { trace } from "@opentelemetry/api";
+//#region ../zero-cache/src/workers/syncer-ws-message-handler.ts
+var tracer = trace.getTracer("syncer-ws-server", version);
+var SyncerWsMessageHandler = class {
+	#viewSyncer;
+	#mutagen;
+	#mutationLock;
+	#lc;
+	#clientGroupID;
+	#syncContext;
+	#pusher;
+	constructor(lc, connectParams, viewSyncer, mutagen, pusher) {
+		const { clientGroupID, clientID, profileID, wsID, baseCookie, protocolVersion, httpCookie, origin, userID } = connectParams;
+		this.#viewSyncer = viewSyncer;
+		this.#mutagen = mutagen;
+		this.#mutationLock = new Lock();
+		this.#lc = lc.withContext("connection").withContext("clientID", clientID).withContext("clientGroupID", clientGroupID).withContext("wsID", wsID);
+		this.#clientGroupID = clientGroupID;
+		this.#pusher = pusher;
+		this.#syncContext = {
+			clientID,
+			profileID,
+			wsID,
+			baseCookie,
+			protocolVersion,
+			httpCookie,
+			origin,
+			userID
+		};
+	}
+	async handleMessage(msg) {
+		const lc = this.#lc;
+		const msgType = msg[0];
+		const viewSyncer = this.#viewSyncer;
+		switch (msgType) {
+			case "ping":
+				lc.error?.("Ping is not supported at this layer by Zero");
+				break;
+			case "pull":
+				lc.error?.("Pull is not supported by Zero");
+				break;
+			case "push": return startAsyncSpan(tracer, "connection.push", async () => {
+				const { clientGroupID, mutations, auth: pushAuth } = msg[1];
+				if (clientGroupID !== this.#clientGroupID) return [{
+					type: "fatal",
+					error: {
+						kind: InvalidPush,
+						message: `clientGroupID in mutation "${clientGroupID}" does not match clientGroupID of connection "${this.#clientGroupID}`,
+						origin: ZeroCache
+					}
+				}];
+				if (pushAuth) await viewSyncer.updateAuth(this.#syncContext, ["updateAuth", { auth: pushAuth }]);
+				if (mutations.length === 0) return [{ type: "ok" }];
+				if (mutations[0].type === "custom") {
+					assert(this.#pusher, "A ZERO_MUTATE_URL must be set in order to process custom mutations.");
+					return [this.#pusher.enqueuePush(this.#syncContext.clientID, msg[1], viewSyncer.auth?.raw, this.#syncContext.httpCookie, this.#syncContext.origin)];
+				}
+				const auth = viewSyncer.auth;
+				assert(auth?.type !== "opaque", "Only JWT auth is supported for CRUD mutations");
+				return [await this.#mutationLock.withLock(async () => {
+					const errors = [];
+					for (const mutation of mutations) {
+						const maybeError = await this.#mutagen.processMutation(mutation, auth?.decoded, this.#pusher !== void 0);
+						if (maybeError !== void 0) errors.push({
+							kind: maybeError[0],
+							message: maybeError[1],
+							origin: ZeroCache
+						});
+					}
+					if (errors.length > 0) return {
+						type: "transient",
+						errors
+					};
+					return { type: "ok" };
+				})];
+			});
+			case "changeDesiredQueries":
+				await startAsyncSpan(tracer, "connection.changeDesiredQueries", () => viewSyncer.changeDesiredQueries(this.#syncContext, msg));
+				break;
+			case "updateAuth":
+				await startAsyncSpan(tracer, "connection.updateAuth", async () => {
+					await viewSyncer.updateAuth(this.#syncContext, msg);
+				});
+				break;
+			case "deleteClients": {
+				const deletedClientIDs = await startAsyncSpan(tracer, "connection.deleteClients", () => viewSyncer.deleteClients(this.#syncContext, msg));
+				if (this.#pusher && deletedClientIDs.length > 0) await this.#pusher.deleteClientMutations(deletedClientIDs);
+				break;
+			}
+			case "initConnection": {
+				const ret = [{
+					type: "stream",
+					source: "viewSyncer",
+					stream: startSpan(tracer, "connection.initConnection", () => viewSyncer.initConnection(this.#syncContext, msg))
+				}];
+				if (this.#pusher) ret.push({
+					type: "stream",
+					source: "pusher",
+					stream: this.#pusher.initConnection(this.#syncContext.clientID, this.#syncContext.wsID, msg[1].userPushURL, msg[1].userPushHeaders, () => viewSyncer.clearAuth())
+				});
+				return ret;
+			}
+			case "closeConnection": break;
+			case "inspect":
+				await startAsyncSpan(tracer, "connection.inspect", () => viewSyncer.inspect(this.#syncContext, msg));
+				break;
+			case "ackMutationResponses":
+				if (this.#pusher) await this.#pusher.ackMutationResponses(msg[1]);
+				break;
+			default: unreachable(msgType);
+		}
+		return [{ type: "ok" }];
+	}
 };
-//# sourceMappingURL=syncer-ws-message-handler.js.map
+//#endregion
+export { SyncerWsMessageHandler };
+
+//# sourceMappingURL=syncer-ws-message-handler.js.map

package/out/zero-cache/src/workers/syncer-ws-message-handler.js.map

@@ -1 +1 @@
-{"version":3,"file":"syncer-ws-message-handler.js","sources":["../../../../../zero-cache/src/workers/syncer-ws-message-handler.ts"],"sourcesContent":["import {trace} from '@opentelemetry/api';\nimport {Lock} from '@rocicorp/lock';\nimport type {LogContext} from '@rocicorp/logger';\nimport {startAsyncSpan, startSpan} from '../../../otel/src/span.ts';\nimport {version} from '../../../otel/src/version.ts';\nimport {assert, unreachable} from '../../../shared/src/asserts.ts';\nimport {ErrorKind} from '../../../zero-protocol/src/error-kind.ts';\nimport {ErrorOrigin} from '../../../zero-protocol/src/error-origin.ts';\nimport type {ErrorBody} from '../../../zero-protocol/src/error.ts';\nimport type {Upstream} from '../../../zero-protocol/src/up.ts';\nimport type {Mutagen} from '../services/mutagen/mutagen.ts';\nimport type {Pusher} from '../services/mutagen/pusher.ts';\nimport {\n  type SyncContext,\n  type ViewSyncer,\n} from '../services/view-syncer/view-syncer.ts';\nimport type {ConnectParams} from './connect-params.ts';\nimport type {HandlerResult, MessageHandler} from './connection.ts';\n\nconst tracer = trace.getTracer('syncer-ws-server', version);\n\nexport class SyncerWsMessageHandler implements MessageHandler {\n  readonly #viewSyncer: ViewSyncer;\n  readonly #mutagen: Mutagen;\n  readonly #mutationLock: Lock;\n  readonly #lc: LogContext;\n  readonly #clientGroupID: string;\n  readonly #syncContext: SyncContext;\n  readonly #pusher: Pusher | undefined;\n\n  constructor(\n    lc: LogContext,\n    connectParams: ConnectParams,\n    viewSyncer: ViewSyncer,\n    mutagen: Mutagen,\n    pusher: Pusher | undefined,\n  ) {\n    const {\n      clientGroupID,\n      clientID,\n      profileID,\n      wsID,\n      baseCookie,\n      protocolVersion,\n      httpCookie,\n      origin,\n      userID,\n    } = connectParams;\n    this.#viewSyncer = viewSyncer;\n    this.#mutagen = mutagen;\n    this.#mutationLock = new Lock();\n    this.#lc = lc\n      .withContext('connection')\n      .withContext('clientID', clientID)\n      .withContext('clientGroupID', clientGroupID)\n      .withContext('wsID', wsID);\n    this.#clientGroupID = clientGroupID;\n    this.#pusher = pusher;\n    this.#syncContext = {\n      clientID,\n      profileID,\n      wsID,\n      baseCookie,\n      protocolVersion,\n      httpCookie,\n      origin,\n      userID,\n    };\n  }\n\n  async handleMessage(msg: Upstream): Promise<HandlerResult[]> {\n    const lc = this.#lc;\n    const msgType = msg[0];\n    const viewSyncer = this.#viewSyncer;\n    switch (msgType) {\n      case 'ping':\n        lc.error?.('Ping is not supported at this layer by Zero');\n        break;\n      case 'pull':\n        lc.error?.('Pull is not supported by Zero');\n        break;\n      case 'push': {\n        return startAsyncSpan<HandlerResult[]>(\n          tracer,\n          'connection.push',\n          async () => {\n            const {clientGroupID, mutations, auth: pushAuth} = msg[1];\n            if (clientGroupID !== this.#clientGroupID) {\n              return [\n                {\n                  type: 'fatal',\n                  error: {\n                    kind: ErrorKind.InvalidPush,\n                    message:\n                      `clientGroupID in mutation \"${clientGroupID}\" does not match ` +\n                      `clientGroupID of connection \"${this.#clientGroupID}`,\n                    origin: ErrorOrigin.ZeroCache,\n                  },\n                } satisfies HandlerResult,\n              ];\n            }\n\n            // for backwards compatibility, if the push contains auth, we update it\n            if (pushAuth) {\n              await viewSyncer.updateAuth(this.#syncContext, [\n                'updateAuth',\n                {auth: pushAuth},\n              ]);\n            }\n\n            if (mutations.length === 0) {\n              return [\n                {\n                  type: 'ok',\n                },\n              ];\n            }\n\n            // The client only ever sends 1 mutation per push.\n            // #pusher will throw if it sees a CRUD mutation.\n            // #mutagen will throw if it see a custom mutation.\n            if (mutations[0].type === 'custom') {\n              assert(\n                this.#pusher,\n                'A ZERO_MUTATE_URL must be set in order to process custom mutations.',\n              );\n              return [\n                this.#pusher.enqueuePush(\n                  this.#syncContext.clientID,\n                  msg[1],\n                  viewSyncer.auth?.raw,\n                  this.#syncContext.httpCookie,\n                  this.#syncContext.origin,\n                ),\n              ];\n            }\n\n            const auth = viewSyncer.auth;\n            assert(\n              auth?.type !== 'opaque',\n              'Only JWT auth is supported for CRUD mutations',\n            );\n\n            // Hold a connection-level lock while processing mutations so that:\n            // 1. Mutations are processed in the order in which they are received and\n            // 2. A single view syncer connection cannot hog multiple upstream connections.\n            const ret = await this.#mutationLock.withLock(async () => {\n              const errors: ErrorBody[] = [];\n              for (const mutation of mutations) {\n                const maybeError = await this.#mutagen.processMutation(\n                  mutation,\n                  auth?.decoded,\n                  this.#pusher !== undefined,\n                );\n                if (maybeError !== undefined) {\n                  errors.push({\n                    kind: maybeError[0],\n                    message: maybeError[1],\n                    origin: ErrorOrigin.ZeroCache,\n                  });\n                }\n              }\n              if (errors.length > 0) {\n                return {type: 'transient', errors} satisfies HandlerResult;\n              }\n              return {type: 'ok'} satisfies HandlerResult;\n            });\n            return [ret];\n          },\n        );\n      }\n      case 'changeDesiredQueries':\n        await startAsyncSpan(tracer, 'connection.changeDesiredQueries', () =>\n          viewSyncer.changeDesiredQueries(this.#syncContext, msg),\n        );\n        break;\n      case 'updateAuth':\n        await startAsyncSpan(tracer, 'connection.updateAuth', async () => {\n          await viewSyncer.updateAuth(this.#syncContext, msg);\n        });\n        break;\n      case 'deleteClients': {\n        const deletedClientIDs = await startAsyncSpan(\n          tracer,\n          'connection.deleteClients',\n          () => viewSyncer.deleteClients(this.#syncContext, msg),\n        );\n        if (this.#pusher && deletedClientIDs.length > 0) {\n          await this.#pusher.deleteClientMutations(deletedClientIDs);\n        }\n        break;\n      }\n      case 'initConnection': {\n        const ret: HandlerResult[] = [\n          {\n            type: 'stream',\n            source: 'viewSyncer',\n            stream: startSpan(tracer, 'connection.initConnection', () =>\n              viewSyncer.initConnection(this.#syncContext, msg),\n            ),\n          },\n        ];\n\n        // Given we support both CRUD and Custom mutators,\n        // we do not initialize the `pusher` unless the user has opted\n        // into custom mutations. We detect that by checking\n        // if the pushURL has been set.\n        if (this.#pusher) {\n          ret.push({\n            type: 'stream',\n            source: 'pusher',\n            stream: this.#pusher.initConnection(\n              this.#syncContext.clientID,\n              this.#syncContext.wsID,\n              msg[1].userPushURL,\n              msg[1].userPushHeaders,\n              () => viewSyncer.clearAuth(),\n            ),\n          });\n        }\n\n        return ret;\n      }\n      case 'closeConnection':\n        // This message is deprecated and no longer used.\n        break;\n\n      case 'inspect':\n        await startAsyncSpan(tracer, 'connection.inspect', () =>\n          viewSyncer.inspect(this.#syncContext, msg),\n        );\n        break;\n\n      case 'ackMutationResponses':\n        if (this.#pusher) {\n          await this.#pusher.ackMutationResponses(msg[1]);\n        }\n        break;\n\n      default:\n        unreachable(msgType);\n    }\n\n    return [{type: 'ok'}];\n  }\n}\n"],"names":["ErrorKind.InvalidPush","ErrorOrigin.ZeroCache"],"mappings":";;;;;;;;AAmBA,MAAM,SAAS,MAAM,UAAU,oBAAoB,OAAO;AAEnD,MAAM,uBAAiD;AAAA,EACnD;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAET,YACE,IACA,eACA,YACA,SACA,QACA;AACA,UAAM;AAAA,MACJ;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IAAA,IACE;AACJ,SAAK,cAAc;AACnB,SAAK,WAAW;AAChB,SAAK,gBAAgB,IAAI,KAAA;AACzB,SAAK,MAAM,GACR,YAAY,YAAY,EACxB,YAAY,YAAY,QAAQ,EAChC,YAAY,iBAAiB,aAAa,EAC1C,YAAY,QAAQ,IAAI;AAC3B,SAAK,iBAAiB;AACtB,SAAK,UAAU;AACf,SAAK,eAAe;AAAA,MAClB;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IAAA;AAAA,EAEJ;AAAA,EAEA,MAAM,cAAc,KAAyC;AAC3D,UAAM,KAAK,KAAK;AAChB,UAAM,UAAU,IAAI,CAAC;AACrB,UAAM,aAAa,KAAK;AACxB,YAAQ,SAAA;AAAA,MACN,KAAK;AACH,WAAG,QAAQ,6CAA6C;AACxD;AAAA,MACF,KAAK;AACH,WAAG,QAAQ,+BAA+B;AAC1C;AAAA,MACF,KAAK,QAAQ;AACX,eAAO;AAAA,UACL;AAAA,UACA;AAAA,UACA,YAAY;AACV,kBAAM,EAAC,eAAe,WAAW,MAAM,SAAA,IAAY,IAAI,CAAC;AACxD,gBAAI,kBAAkB,KAAK,gBAAgB;AACzC,qBAAO;AAAA,gBACL;AAAA,kBACE,MAAM;AAAA,kBACN,OAAO;AAAA,oBACL,MAAMA;AAAAA,oBACN,SACE,8BAA8B,aAAa,iDACX,KAAK,cAAc;AAAA,oBACrD,QAAQC;AAAAA,kBAAY;AAAA,gBACtB;AAAA,cACF;AAAA,YAEJ;AAGA,gBAAI,UAAU;AACZ,oBAAM,WAAW,WAAW,KAAK,cAAc;AAAA,gBAC7C;AAAA,gBACA,EAAC,MAAM,SAAA;AAAA,cAAQ,CAChB;AAAA,YACH;AAEA,gBAAI,UAAU,WAAW,GAAG;AAC1B,qBAAO;AAAA,gBACL;AAAA,kBACE,MAAM;AAAA,gBAAA;AAAA,cACR;AAAA,YAEJ;AAKA,gBAAI,UAAU,CAAC,EAAE,SAAS,UAAU;AAClC;AAAA,gBACE,KAAK;AAAA,gBACL;AAAA,cAAA;AAEF,qBAAO;AAAA,gBACL,KAAK,QAAQ;AAAA,kBACX,KAAK,aAAa;AAAA,kBAClB,IAAI,CAAC;AAAA,kBACL,WAAW,MAAM;AAAA,kBACjB,KAAK,aAAa;AAAA,kBAClB,KAAK,aAAa;AAAA,gBAAA;AAAA,cACpB;AAAA,YAEJ;AAEA,kBAAM,OAAO,WAAW;AACxB;AAAA,cACE,MAAM,SAAS;AAAA,cACf;AAAA,YAAA;AAMF,kBAAM,MAAM,MAAM,KAAK,cAAc,SAAS,YAAY;AACxD,oBAAM,SAAsB,CAAA;AAC5B,yBAAW,YAAY,WAAW;AAChC,sBAAM,aAAa,MAAM,KAAK,SAAS;AAAA,kBACrC;AAAA,kBACA,MAAM;AAAA,kBACN,KAAK,YAAY;AAAA,gBAAA;AAEnB,oBAAI,eAAe,QAAW;AAC5B,yBAAO,KAAK;AAAA,oBACV,MAAM,WAAW,CAAC;AAAA,oBAClB,SAAS,WAAW,CAAC;AAAA,oBACrB,QAAQA;AAAAA,kBAAY,CACrB;AAAA,gBACH;AAAA,cACF;AACA,kBAAI,OAAO,SAAS,GAAG;AACrB,uBAAO,EAAC,MAAM,aAAa,OAAA;AAAA,cAC7B;AACA,qBAAO,EAAC,MAAM,KAAA;AAAA,YAChB,CAAC;AACD,mBAAO,CAAC,GAAG;AAAA,UACb;AAAA,QAAA;AAAA,MAEJ;AAAA,MACA,KAAK;AACH,cAAM;AAAA,UAAe;AAAA,UAAQ;AAAA,UAAmC,MAC9D,WAAW,qBAAqB,KAAK,cAAc,GAAG;AAAA,QAAA;AAExD;AAAA,MACF,KAAK;AACH,cAAM,eAAe,QAAQ,yBAAyB,YAAY;AAChE,gBAAM,WAAW,WAAW,KAAK,cAAc,GAAG;AAAA,QACpD,CAAC;AACD;AAAA,MACF,KAAK,iBAAiB;AACpB,cAAM,mBAAmB,MAAM;AAAA,UAC7B;AAAA,UACA;AAAA,UACA,MAAM,WAAW,cAAc,KAAK,cAAc,GAAG;AAAA,QAAA;AAEvD,YAAI,KAAK,WAAW,iBAAiB,SAAS,GAAG;AAC/C,gBAAM,KAAK,QAAQ,sBAAsB,gBAAgB;AAAA,QAC3D;AACA;AAAA,MACF;AAAA,MACA,KAAK,kBAAkB;AACrB,cAAM,MAAuB;AAAA,UAC3B;AAAA,YACE,MAAM;AAAA,YACN,QAAQ;AAAA,YACR,QAAQ;AAAA,cAAU;AAAA,cAAQ;AAAA,cAA6B,MACrD,WAAW,eAAe,KAAK,cAAc,GAAG;AAAA,YAAA;AAAA,UAClD;AAAA,QACF;AAOF,YAAI,KAAK,SAAS;AAChB,cAAI,KAAK;AAAA,YACP,MAAM;AAAA,YACN,QAAQ;AAAA,YACR,QAAQ,KAAK,QAAQ;AAAA,cACnB,KAAK,aAAa;AAAA,cAClB,KAAK,aAAa;AAAA,cAClB,IAAI,CAAC,EAAE;AAAA,cACP,IAAI,CAAC,EAAE;AAAA,cACP,MAAM,WAAW,UAAA;AAAA,YAAU;AAAA,UAC7B,CACD;AAAA,QACH;AAEA,eAAO;AAAA,MACT;AAAA,MACA,KAAK;AAEH;AAAA,MAEF,KAAK;AACH,cAAM;AAAA,UAAe;AAAA,UAAQ;AAAA,UAAsB,MACjD,WAAW,QAAQ,KAAK,cAAc,GAAG;AAAA,QAAA;AAE3C;AAAA,MAEF,KAAK;AACH,YAAI,KAAK,SAAS;AAChB,gBAAM,KAAK,QAAQ,qBAAqB,IAAI,CAAC,CAAC;AAAA,QAChD;AACA;AAAA,MAEF;AACE,oBAAmB;AAAA,IAAA;AAGvB,WAAO,CAAC,EAAC,MAAM,MAAK;AAAA,EACtB;AACF;"}
+{"version":3,"file":"syncer-ws-message-handler.js","names":["#viewSyncer","#mutagen","#mutationLock","#lc","#clientGroupID","#syncContext","#pusher"],"sources":["../../../../../zero-cache/src/workers/syncer-ws-message-handler.ts"],"sourcesContent":["import {trace} from '@opentelemetry/api';\nimport {Lock} from '@rocicorp/lock';\nimport type {LogContext} from '@rocicorp/logger';\nimport {startAsyncSpan, startSpan} from '../../../otel/src/span.ts';\nimport {version} from '../../../otel/src/version.ts';\nimport {assert, unreachable} from '../../../shared/src/asserts.ts';\nimport {ErrorKind} from '../../../zero-protocol/src/error-kind.ts';\nimport {ErrorOrigin} from '../../../zero-protocol/src/error-origin.ts';\nimport type {ErrorBody} from '../../../zero-protocol/src/error.ts';\nimport type {Upstream} from '../../../zero-protocol/src/up.ts';\nimport type {Mutagen} from '../services/mutagen/mutagen.ts';\nimport type {Pusher} from '../services/mutagen/pusher.ts';\nimport {\n  type SyncContext,\n  type ViewSyncer,\n} from '../services/view-syncer/view-syncer.ts';\nimport type {ConnectParams} from './connect-params.ts';\nimport type {HandlerResult, MessageHandler} from './connection.ts';\n\nconst tracer = trace.getTracer('syncer-ws-server', version);\n\nexport class SyncerWsMessageHandler implements MessageHandler {\n  readonly #viewSyncer: ViewSyncer;\n  readonly #mutagen: Mutagen;\n  readonly #mutationLock: Lock;\n  readonly #lc: LogContext;\n  readonly #clientGroupID: string;\n  readonly #syncContext: SyncContext;\n  readonly #pusher: Pusher | undefined;\n\n  constructor(\n    lc: LogContext,\n    connectParams: ConnectParams,\n    viewSyncer: ViewSyncer,\n    mutagen: Mutagen,\n    pusher: Pusher | undefined,\n  ) {\n    const {\n      clientGroupID,\n      clientID,\n      profileID,\n      wsID,\n      baseCookie,\n      protocolVersion,\n      httpCookie,\n      origin,\n      userID,\n    } = connectParams;\n    this.#viewSyncer = viewSyncer;\n    this.#mutagen = mutagen;\n    this.#mutationLock = new Lock();\n    this.#lc = lc\n      .withContext('connection')\n      .withContext('clientID', clientID)\n      .withContext('clientGroupID', clientGroupID)\n      .withContext('wsID', wsID);\n    this.#clientGroupID = clientGroupID;\n    this.#pusher = pusher;\n    this.#syncContext = {\n      clientID,\n      profileID,\n      wsID,\n      baseCookie,\n      protocolVersion,\n      httpCookie,\n      origin,\n      userID,\n    };\n  }\n\n  async handleMessage(msg: Upstream): Promise<HandlerResult[]> {\n    const lc = this.#lc;\n    const msgType = msg[0];\n    const viewSyncer = this.#viewSyncer;\n    switch (msgType) {\n      case 'ping':\n        lc.error?.('Ping is not supported at this layer by Zero');\n        break;\n      case 'pull':\n        lc.error?.('Pull is not supported by Zero');\n        break;\n      case 'push': {\n        return startAsyncSpan<HandlerResult[]>(\n          tracer,\n          'connection.push',\n          async () => {\n            const {clientGroupID, mutations, auth: pushAuth} = msg[1];\n            if (clientGroupID !== this.#clientGroupID) {\n              return [\n                {\n                  type: 'fatal',\n                  error: {\n                    kind: ErrorKind.InvalidPush,\n                    message:\n                      `clientGroupID in mutation \"${clientGroupID}\" does not match ` +\n                      `clientGroupID of connection \"${this.#clientGroupID}`,\n                    origin: ErrorOrigin.ZeroCache,\n                  },\n                } satisfies HandlerResult,\n              ];\n            }\n\n            // for backwards compatibility, if the push contains auth, we update it\n            if (pushAuth) {\n              await viewSyncer.updateAuth(this.#syncContext, [\n                'updateAuth',\n                {auth: pushAuth},\n              ]);\n            }\n\n            if (mutations.length === 0) {\n              return [\n                {\n                  type: 'ok',\n                },\n              ];\n            }\n\n            // The client only ever sends 1 mutation per push.\n            // #pusher will throw if it sees a CRUD mutation.\n            // #mutagen will throw if it see a custom mutation.\n            if (mutations[0].type === 'custom') {\n              assert(\n                this.#pusher,\n                'A ZERO_MUTATE_URL must be set in order to process custom mutations.',\n              );\n              return [\n                this.#pusher.enqueuePush(\n                  this.#syncContext.clientID,\n                  msg[1],\n                  viewSyncer.auth?.raw,\n                  this.#syncContext.httpCookie,\n                  this.#syncContext.origin,\n                ),\n              ];\n            }\n\n            const auth = viewSyncer.auth;\n            assert(\n              auth?.type !== 'opaque',\n              'Only JWT auth is supported for CRUD mutations',\n            );\n\n            // Hold a connection-level lock while processing mutations so that:\n            // 1. Mutations are processed in the order in which they are received and\n            // 2. A single view syncer connection cannot hog multiple upstream connections.\n            const ret = await this.#mutationLock.withLock(async () => {\n              const errors: ErrorBody[] = [];\n              for (const mutation of mutations) {\n                const maybeError = await this.#mutagen.processMutation(\n                  mutation,\n                  auth?.decoded,\n                  this.#pusher !== undefined,\n                );\n                if (maybeError !== undefined) {\n                  errors.push({\n                    kind: maybeError[0],\n                    message: maybeError[1],\n                    origin: ErrorOrigin.ZeroCache,\n                  });\n                }\n              }\n              if (errors.length > 0) {\n                return {type: 'transient', errors} satisfies HandlerResult;\n              }\n              return {type: 'ok'} satisfies HandlerResult;\n            });\n            return [ret];\n          },\n        );\n      }\n      case 'changeDesiredQueries':\n        await startAsyncSpan(tracer, 'connection.changeDesiredQueries', () =>\n          viewSyncer.changeDesiredQueries(this.#syncContext, msg),\n        );\n        break;\n      case 'updateAuth':\n        await startAsyncSpan(tracer, 'connection.updateAuth', async () => {\n          await viewSyncer.updateAuth(this.#syncContext, msg);\n        });\n        break;\n      case 'deleteClients': {\n        const deletedClientIDs = await startAsyncSpan(\n          tracer,\n          'connection.deleteClients',\n          () => viewSyncer.deleteClients(this.#syncContext, msg),\n        );\n        if (this.#pusher && deletedClientIDs.length > 0) {\n          await this.#pusher.deleteClientMutations(deletedClientIDs);\n        }\n        break;\n      }\n      case 'initConnection': {\n        const ret: HandlerResult[] = [\n          {\n            type: 'stream',\n            source: 'viewSyncer',\n            stream: startSpan(tracer, 'connection.initConnection', () =>\n              viewSyncer.initConnection(this.#syncContext, msg),\n            ),\n          },\n        ];\n\n        // Given we support both CRUD and Custom mutators,\n        // we do not initialize the `pusher` unless the user has opted\n        // into custom mutations. We detect that by checking\n        // if the pushURL has been set.\n        if (this.#pusher) {\n          ret.push({\n            type: 'stream',\n            source: 'pusher',\n            stream: this.#pusher.initConnection(\n              this.#syncContext.clientID,\n              this.#syncContext.wsID,\n              msg[1].userPushURL,\n              msg[1].userPushHeaders,\n              () => viewSyncer.clearAuth(),\n            ),\n          });\n        }\n\n        return ret;\n      }\n      case 'closeConnection':\n        // This message is deprecated and no longer used.\n        break;\n\n      case 'inspect':\n        await startAsyncSpan(tracer, 'connection.inspect', () =>\n          viewSyncer.inspect(this.#syncContext, msg),\n        );\n        break;\n\n      case 'ackMutationResponses':\n        if (this.#pusher) {\n          await this.#pusher.ackMutationResponses(msg[1]);\n        }\n        break;\n\n      default:\n        unreachable(msgType);\n    }\n\n    return [{type: 'ok'}];\n  }\n}\n"],"mappings":";;;;;;;;;AAmBA,IAAM,SAAS,MAAM,UAAU,oBAAoB,QAAQ;AAE3D,IAAa,yBAAb,MAA8D;CAC5D;CACA;CACA;CACA;CACA;CACA;CACA;CAEA,YACE,IACA,eACA,YACA,SACA,QACA;EACA,MAAM,EACJ,eACA,UACA,WACA,MACA,YACA,iBACA,YACA,QACA,WACE;AACJ,QAAA,aAAmB;AACnB,QAAA,UAAgB;AAChB,QAAA,eAAqB,IAAI,MAAM;AAC/B,QAAA,KAAW,GACR,YAAY,aAAa,CACzB,YAAY,YAAY,SAAS,CACjC,YAAY,iBAAiB,cAAc,CAC3C,YAAY,QAAQ,KAAK;AAC5B,QAAA,gBAAsB;AACtB,QAAA,SAAe;AACf,QAAA,cAAoB;GAClB;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACD;;CAGH,MAAM,cAAc,KAAyC;EAC3D,MAAM,KAAK,MAAA;EACX,MAAM,UAAU,IAAI;EACpB,MAAM,aAAa,MAAA;AACnB,UAAQ,SAAR;GACE,KAAK;AACH,OAAG,QAAQ,8CAA8C;AACzD;GACF,KAAK;AACH,OAAG,QAAQ,gCAAgC;AAC3C;GACF,KAAK,OACH,QAAO,eACL,QACA,mBACA,YAAY;IACV,MAAM,EAAC,eAAe,WAAW,MAAM,aAAY,IAAI;AACvD,QAAI,kBAAkB,MAAA,cACpB,QAAO,CACL;KACE,MAAM;KACN,OAAO;MACL,MAAM;MACN,SACE,8BAA8B,cAAc,gDACZ,MAAA;MAClC,QAAQ;MACT;KACF,CACF;AAIH,QAAI,SACF,OAAM,WAAW,WAAW,MAAA,aAAmB,CAC7C,cACA,EAAC,MAAM,UAAS,CACjB,CAAC;AAGJ,QAAI,UAAU,WAAW,EACvB,QAAO,CACL,EACE,MAAM,MACP,CACF;AAMH,QAAI,UAAU,GAAG,SAAS,UAAU;AAClC,YACE,MAAA,QACA,sEACD;AACD,YAAO,CACL,MAAA,OAAa,YACX,MAAA,YAAkB,UAClB,IAAI,IACJ,WAAW,MAAM,KACjB,MAAA,YAAkB,YAClB,MAAA,YAAkB,OACnB,CACF;;IAGH,MAAM,OAAO,WAAW;AACxB,WACE,MAAM,SAAS,UACf,gDACD;AA0BD,WAAO,CArBK,MAAM,MAAA,aAAmB,SAAS,YAAY;KACxD,MAAM,SAAsB,EAAE;AAC9B,UAAK,MAAM,YAAY,WAAW;MAChC,MAAM,aAAa,MAAM,MAAA,QAAc,gBACrC,UACA,MAAM,SACN,MAAA,WAAiB,KAAA,EAClB;AACD,UAAI,eAAe,KAAA,EACjB,QAAO,KAAK;OACV,MAAM,WAAW;OACjB,SAAS,WAAW;OACpB,QAAQ;OACT,CAAC;;AAGN,SAAI,OAAO,SAAS,EAClB,QAAO;MAAC,MAAM;MAAa;MAAO;AAEpC,YAAO,EAAC,MAAM,MAAK;MACnB,CACU;KAEf;GAEH,KAAK;AACH,UAAM,eAAe,QAAQ,yCAC3B,WAAW,qBAAqB,MAAA,aAAmB,IAAI,CACxD;AACD;GACF,KAAK;AACH,UAAM,eAAe,QAAQ,yBAAyB,YAAY;AAChE,WAAM,WAAW,WAAW,MAAA,aAAmB,IAAI;MACnD;AACF;GACF,KAAK,iBAAiB;IACpB,MAAM,mBAAmB,MAAM,eAC7B,QACA,kCACM,WAAW,cAAc,MAAA,aAAmB,IAAI,CACvD;AACD,QAAI,MAAA,UAAgB,iBAAiB,SAAS,EAC5C,OAAM,MAAA,OAAa,sBAAsB,iBAAiB;AAE5D;;GAEF,KAAK,kBAAkB;IACrB,MAAM,MAAuB,CAC3B;KACE,MAAM;KACN,QAAQ;KACR,QAAQ,UAAU,QAAQ,mCACxB,WAAW,eAAe,MAAA,aAAmB,IAAI,CAClD;KACF,CACF;AAMD,QAAI,MAAA,OACF,KAAI,KAAK;KACP,MAAM;KACN,QAAQ;KACR,QAAQ,MAAA,OAAa,eACnB,MAAA,YAAkB,UAClB,MAAA,YAAkB,MAClB,IAAI,GAAG,aACP,IAAI,GAAG,uBACD,WAAW,WAAW,CAC7B;KACF,CAAC;AAGJ,WAAO;;GAET,KAAK,kBAEH;GAEF,KAAK;AACH,UAAM,eAAe,QAAQ,4BAC3B,WAAW,QAAQ,MAAA,aAAmB,IAAI,CAC3C;AACD;GAEF,KAAK;AACH,QAAI,MAAA,OACF,OAAM,MAAA,OAAa,qBAAqB,IAAI,GAAG;AAEjD;GAEF,QACE,aAAY,QAAQ;;AAGxB,SAAO,CAAC,EAAC,MAAM,MAAK,CAAC"}