@rocicorp/zero 1.11.0-canary.8 → 1.11.0-canary.9

package/out/zero-client/src/client/zero.js

@@ -20,7 +20,7 @@ import { isMutatorRegistry, iterateMutators } from "../../../zql/src/mutate/muta
 import { createRunnableBuilder } from "../../../zql/src/query/create-builder.js";
 import "../../../zql/src/query/query.js";
 import { addContextToQuery } from "../../../zql/src/query/query-registry.js";
-import { Closed, Connected, Connecting, Disconnected, Error as Error$1, NeedsAuth } from "./connection-status-enum.js";
+import { Closed, Connected, Connecting, Disconnected, Error as Error$1, Initializing, NeedsAuth } from "./connection-status-enum.js";
 import { getDocumentVisibilityWatcher } from "../../../shared/src/document-visible.js";
 import { ReplicacheImpl } from "../../../replicache/src/replicache-impl.js";
 import "../../../replicache/src/impl.js";
@@ -38,7 +38,7 @@ import { isClientMetric } from "../../../zql/src/query/metrics-delegate.js";
 import { nanoid } from "../util/nanoid.js";
 import { ActiveClientsManager } from "./active-clients-manager.js";
 import { AbruptClose, CleanClose, ClientClosed, ConnectTimeout, Hidden, Internal, InvalidMessage as InvalidMessage$1, NoSocketOrigin, PingTimeout, PullTimeout, UnexpectedBaseCookie } from "./client-error-kind-enum.js";
-import { ClientError, NO_STATUS_TRANSITION, getBackoffParams, getErrorConnectionTransition, isAuthError, isClientError, isServerError, isZeroError } from "./error.js";
+import { ClientError, NO_STATUS_TRANSITION, getBackoffParams, getErrorConnectionTransition, isClientError, isServerError, isZeroError } from "./error.js";
 import { ConnectionManager, throwIfConnectionError } from "./connection-manager.js";
 import { ConnectionImpl } from "./connection.js";
 import { IVMSourceBranch } from "./ivm-branch.js";
@@ -122,7 +122,7 @@ var serverAheadReloadReason = `Server reported that client is ahead of server. T
 function onClientStateNotFoundServerReason(serverErrMsg) {
 	return `Server could not find state needed to synchronize this client. ${serverErrMsg}`;
 }
-var ON_CLIENT_STATE_NOT_FOUND_REASON_CLIENT = "The local persistent state needed to synchronize this client has been garbage collected.";
+var ON_CLIENT_STATE_NOT_FOUND_REASON_CLIENT = "The local persistent state needed to synchronize this client has been garbage collected or was found to be corrupt.";
 var CLOSE_CODE_NORMAL = 1e3;
 var Zero = class Zero {
 	version = version;
@@ -307,7 +307,6 @@ var Zero = class Zero {
 		}, (ast, customQueryID, ttl, gotCallback) => this.#queryManager.addCustom(ast, customQueryID, ttl, gotCallback), (ast, ttl) => {
 			if (enableLegacyQueries) this.#queryManager.updateLegacy(ast, ttl);
 		}, (customQueryID, ttl) => this.#queryManager.updateCustom(customQueryID, ttl), () => this.#queryManager.flushBatch(), batchViewUpdates, this.#addMetric, assertValidRunOptions);
-		this.#zeroContext.deferPipelines();
 		this.query = createRunnableBuilder(this.#zeroContext, schema);
 		const rep = new ReplicacheImpl(replicacheOptions, {
 			enableClientGroupForking: false,
@@ -378,15 +377,16 @@ var Zero = class Zero {
 		});
 		this.#metrics.tags.push(`version:${this.version}`);
 		this.#pokeHandler = new PokeHandler(async (poke) => {
+			const socket = this.#socket;
 			await this.#rep.poke(poke);
-			this.#queryManager.markGotQueriesAuthoritative();
+			if (socket !== void 0 && this.#socket === socket) this.#queryManager.markGotQueriesAuthoritative();
 		}, (e) => this.#onPokeError(e), rep.clientID, schema, this.#lc, this.#mutationTracker);
 		this.#visibilityWatcher = getDocumentVisibilityWatcher(getBrowserGlobal("document"), hiddenTabDisconnectDelay, this.#closeAbortController.signal);
 		this.#runLoop();
 		this.#expose();
 	}
 	#enableRefresh() {
-		return this.#forceEnableRefresh || !this.#connectionManager.is("connected") && !this.#connectionManager.is("connecting");
+		return this.#forceEnableRefresh || !this.#connectionManager.is("connected") && !this.#connectionManager.is("connecting") && !this.#connectionManager.is("initializing");
 	}
 	#expose() {
 		const g = globalThis;
@@ -437,8 +437,11 @@ var Zero = class Zero {
 	* This function is useful when you want to populate the cache ahead of time,
 	* for example after login, to avoid a flash of loading screen on the next page.
 	*
-	* Returns an object with two properties:
-	* - `complete`: a Promise that resolves when the data is loaded
+	* Returns an object with three properties:
+	* - `complete`: a Promise that resolves when the server has confirmed the
+	*   data on this connection
+	* - `cached`: a Promise that resolves as soon as the store holds a
+	*   server-confirmed result, from this connection or a previous one
 	* - `cleanup`: a function that can be called to cancel the preload
 	*
 	* @example
@@ -455,7 +458,9 @@ var Zero = class Zero {
 	* Executes a query once and returns the results.
 	*
 	* By default, runs immediately with whatever data is available locally.
-	* Use `{type: 'complete'}` to wait for fresh results from the server.
+	* Use `{type: 'complete'}` to wait for fresh results from the server, or
+	* `{type: 'cached'}` to accept a result the server confirmed on a previous
+	* connection when the store holds one.
 	*
 	* @param query - The query to execute
 	* @param runOptions - Options controlling query execution
@@ -1011,6 +1016,7 @@ var Zero = class Zero {
 			case Closed:
 				lc.debug?.("disconnect() called while closed");
 				return;
+			case Initializing:
 			case Disconnected:
 			case Connecting:
 			case NeedsAuth:
@@ -1154,6 +1160,26 @@ var Zero = class Zero {
 		};
 		const { auth } = this.#options;
 		this.#setAuth(auth);
+		const { signal: closeSignal } = this.#closeAbortController;
+		const closed = resolver();
+		const onClose = () => closed.resolve();
+		closeSignal.addEventListener("abort", onClose, { once: true });
+		try {
+			if ((await promiseRace({
+				initialized: Promise.all([
+					this.#rep.cookie,
+					this.clientGroupID,
+					this.#activeClientsManager
+				]),
+				closed: closed.promise
+			})).key === "closed") {
+				this.#lc.debug?.("Closed while initializing, not connecting");
+				return;
+			}
+		} catch {} finally {
+			closeSignal.removeEventListener("abort", onClose);
+		}
+		this.#connectionManager.initialized();
 		let backoffMs;
 		let additionalConnectParams;
 		while (this.#connectionManager.shouldContinueRunLoop()) {
@@ -1267,12 +1293,24 @@ var Zero = class Zero {
 						})).key === "connectRequest") this.#connectionManager.resumeFromConnectRequest();
 						break;
 					case Closed: break;
+					case Initializing: unreachable();
 					default: unreachable(currentState);
 				}
 			} catch (ex) {
 				const isClientClosedError = isClientError(ex) && ex.kind === "ClientClosed";
+				const transition = getErrorConnectionTransition(ex);
 				if (!this.#connectionManager.is("connected") && !isClientClosedError) {
-					const level = isAuthError(ex) ? "warn" : "error";
+					let level;
+					switch (transition.status) {
+						case Error$1:
+							level = "error";
+							break;
+						case Disconnected:
+						case Closed:
+							level = "info";
+							break;
+						default: level = "warn";
+					}
 					const kind = isServerError(ex) ? ex.kind : "Unknown Error";
 					lc[level]?.("Failed to connect", ex, ...getErrorCauses(ex), kind, {
 						lmid: this.#lastMutationIDReceived,
@@ -1280,7 +1318,6 @@ var Zero = class Zero {
 					});
 				}
 				lc.debug?.("Got an exception in the run loop", "state:", this.#connectionManager.state, "exception:", ex);
-				const transition = getErrorConnectionTransition(ex);
 				let sleepMs = void 0;
 				switch (transition.status) {
 					case NO_STATUS_TRANSITION: {