@rocicorp/zero 1.11.0-canary.2 → 1.11.0-canary.21

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

@@ -4,14 +4,11 @@ import { AbortError } from "../../../shared/src/abort-error.js";
 import { sleep, sleepWithAbort } from "../../../shared/src/sleep.js";
 import { parse } from "../../../shared/src/valita.js";
 import "../../../replicache/src/deleted-clients.js";
-import { localNavigator } from "../../../shared/src/navigator.js";
-import { getKVStoreProvider } from "../../../replicache/src/get-kv-store-provider.js";
-import { IDBDatabasesStore } from "../../../replicache/src/persist/idb-databases-store.js";
-import { dropDatabase } from "../../../replicache/src/persist/collect-idb-databases.js";
+import { dropDatabase, dropMatchingDatabases } from "../../../replicache/src/persist/collect-idb-databases.js";
 import { must } from "../../../shared/src/must.js";
 import { clientToServer } from "../../../zero-schema/src/name-mapper.js";
 import { h64 } from "../../../shared/src/hash.js";
-import { getErrorMessage } from "../../../shared/src/error.js";
+import { getErrorCauses, getErrorMessage } from "../../../shared/src/error.js";
 import { ClientNotFound, InvalidMessage } from "../../../zero-protocol/src/error-kind-enum.js";
 import { ProtocolError } from "../../../zero-protocol/src/error.js";
 import "../../../zero-protocol/src/client-schema.js";
@@ -20,10 +17,11 @@ 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";
+import { localNavigator } from "../../../shared/src/navigator.js";
 import { promiseRace } from "../../../shared/src/promise-race.js";
 import { emptyFunction } from "../../../shared/src/sentinels.js";
 import { Subscribable } from "../../../shared/src/subscribable.js";
@@ -38,7 +36,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 +120,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;
@@ -232,7 +230,7 @@ var Zero = class Zero {
 	* Constructs a new Zero client.
 	*/
 	constructor(options) {
-		const { userID, storageKey, onOnlineChange, onUpdateNeeded, onClientStateNotFound, hiddenTabDisconnectDelay = DEFAULT_DISCONNECT_HIDDEN_DELAY_MS, pingTimeoutMs = DEFAULT_PING_TIMEOUT_MS, disconnectTimeoutMs = DEFAULT_DISCONNECT_TIMEOUT_MS, schema, batchViewUpdates = (applyViewUpdates) => applyViewUpdates(), maxRecentQueries = 0, slowMaterializeThreshold = 5e3 } = options;
+		const { userID, storageKey, onOnlineChange, onUpdateNeeded, onClientStateNotFound, onStorageFailure, hiddenTabDisconnectDelay = DEFAULT_DISCONNECT_HIDDEN_DELAY_MS, pingTimeoutMs = DEFAULT_PING_TIMEOUT_MS, disconnectTimeoutMs = DEFAULT_DISCONNECT_TIMEOUT_MS, schema, batchViewUpdates = (applyViewUpdates) => applyViewUpdates(), maxRecentQueries = 0, slowMaterializeThreshold = 5e3 } = options;
 		if (userID === "") throw new ClientError({
 			kind: Internal,
 			message: "ZeroOptions.userID should not be empty. Omit it entirely for logged-out clients."
@@ -301,13 +299,13 @@ var Zero = class Zero {
 			licenseKey: "zero-client-static-key",
 			kvStore
 		};
+		this.#visibilityWatcher = getDocumentVisibilityWatcher(getBrowserGlobal("document"), hiddenTabDisconnectDelay, this.#closeAbortController.signal);
 		this.#zeroContext = new ZeroContext(lc, this.#ivmMain, (ast, ttl, gotCallback) => {
 			if (enableLegacyQueries) return this.#queryManager.addLegacy(ast, ttl, gotCallback);
 			return emptyFunction;
 		}, (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();
+		}, (customQueryID, ttl) => this.#queryManager.updateCustom(customQueryID, ttl), () => this.#queryManager.flushBatch(), batchViewUpdates, this.#addMetric, assertValidRunOptions, this.#visibilityWatcher);
 		this.query = createRunnableBuilder(this.#zeroContext, schema);
 		const rep = new ReplicacheImpl(replicacheOptions, {
 			enableClientGroupForking: false,
@@ -343,6 +341,7 @@ var Zero = class Zero {
 		this.#rep.onClientStateNotFound = () => {
 			this.#onClientStateNotFound(ClientNotFound, ON_CLIENT_STATE_NOT_FOUND_REASON_CLIENT);
 		};
+		this.#rep.onStorageFailure = onStorageFailure ?? null;
 		const mutatorProxy = new MutatorProxy(this.#lc, this.#connectionManager, this.#mutationTracker);
 		const { mutators } = options;
 		const registeredMutators = new Set(isMutatorRegistry(mutators) ? iterateMutators(mutators) : void 0);
@@ -378,15 +377,15 @@ 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 +436,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 +457,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
@@ -500,6 +504,16 @@ var Zero = class Zero {
 		return this.#rep.idbName;
 	}
 	/**
+	* The KV store backing this instance of Zero. Its `kind` is the storage
+	* currently in use: `'idb'`, `'mem'`, `'op-sqlite'`, `'expo-sqlite'`, or
+	* whatever a custom store reports. An IndexedDB store that fails to open
+	* falls back to memory, so `kind` can change from `'idb'` to `'mem'` after
+	* the first read or write.
+	*/
+	get kvStore() {
+		return this.#rep.kvStore;
+	}
+	/**
 	* The schema version of the data understood by this application.
 	* See [[ZeroOptions.schemaVersion]].
 	*/
@@ -747,7 +761,7 @@ var Zero = class Zero {
 			assert(this.#socket, "Socket is not set before onClose");
 			lc = addWebSocketIDFromSocketToLogContext(this.#socket, lc);
 			const { code, reason, wasClean } = e;
-			if (code <= 1001 || code === 1006) lc.info?.("Got socket close event", {
+			if (code <= 1001 || code === 1005 || code === 1006) lc.info?.("Got socket close event", {
 				code,
 				reason,
 				wasClean
@@ -793,7 +807,7 @@ var Zero = class Zero {
 		}
 		lc.info?.(`${kind}: ${message}}`);
 		const error = new ProtocolError(downMessage[1]);
-		lc.error?.(`${error.kind}:\n\n${error.errorBody.message}`, error);
+		lc[getBackoffParams(error) === void 0 ? "error" : "warn"]?.(`${error.kind}:\n\n${error.errorBody.message}`, error);
 		this.#disconnect(lc, error);
 		if (kind === "VersionNotSupported") this.#onUpdateNeeded({
 			type: kind,
@@ -809,6 +823,7 @@ var Zero = class Zero {
 			await this.#rep.disableClientGroup();
 			this.#onClientStateNotFound(kind, onClientStateNotFoundServerReason(message));
 		} else if (kind === "InvalidConnectionRequestLastMutationID" || kind === "InvalidConnectionRequestBaseCookie") {
+			await Promise.all([this.#rep.stopPersist(), this.#rep.stopRefresh()]);
 			await dropDatabase(this.#rep.idbName, { kvStore: this.#kvStore });
 			this.#onClientStateNotFound(kind, serverAheadReloadReason);
 		}
@@ -919,11 +934,11 @@ var Zero = class Zero {
 		const clientGroupID = await this.clientGroupID;
 		this.#addConnectEvent(lc, attempt, "initializing active clients");
 		const activeClientsManager = await this.#activeClientsManager;
-		if (signal.aborted) throw signal.reason;
+		if (signal.aborted) throw attempt.abortReason;
 		const [ws, initConnectionQueries, deletedClients] = await createSocket(this.#rep, this.#queryManager, this.#deleteClientsManager, socketOrigin, connectCookie, this.clientID, clientGroupID, this.#clientSchema, this.userID, fromReplicacheAuthToken(this.#rep.auth), this.#lastMutationIDReceived, wsid, this.#options.logLevel === "debug", lc, this.#options.mutateURL, this.#options.mutateHeaders, this.#options.queryURL ?? this.#options.getQueriesURL, this.#options.queryHeaders, additionalConnectParams, activeClientsManager, this.#options.maxHeaderLength, (event) => this.#addConnectEvent(lc, attempt, event));
 		if (signal.aborted) {
 			ws.close();
-			throw signal.reason;
+			throw attempt.abortReason;
 		}
 		if (this.closed) {
 			ws.close();
@@ -942,7 +957,7 @@ var Zero = class Zero {
 		attempt.clearTimeout?.();
 		attempt.clearTimeout = this.#armConnectTimeout(lc, attempt, "ack");
 		await attempt.connected.promise;
-		if (signal.aborted) throw signal.reason;
+		if (signal.aborted) throw attempt.abortReason;
 		this.#mutationTracker.onConnected(this.#lastMutationIDReceived);
 		this.#rep.push().catch(() => {});
 	}
@@ -980,7 +995,11 @@ var Zero = class Zero {
 		lc.debug?.("Connect event", event);
 	}
 	#disconnect(lc, reason, closeCode) {
-		this.#currentConnectAttempt?.controller.abort(reason);
+		const attempt = this.#currentConnectAttempt;
+		if (attempt) {
+			attempt.abortReason = reason;
+			attempt.controller.abort();
+		}
 		if (shouldReportConnectError(reason)) {
 			this.#connectErrorCount++;
 			this.#metrics.lastConnectError.set(getLastConnectErrorValue(reason));
@@ -1007,6 +1026,7 @@ var Zero = class Zero {
 			case Closed:
 				lc.debug?.("disconnect() called while closed");
 				return;
+			case Initializing:
 			case Disconnected:
 			case Connecting:
 			case NeedsAuth:
@@ -1150,6 +1170,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()) {
@@ -1185,7 +1225,7 @@ var Zero = class Zero {
 						this.#addConnectEvent(lc, attempt, "starting the connection");
 						attempt.clearTimeout = this.#armConnectTimeout(lc, attempt, "setup");
 						const canceled = resolver();
-						const abortHandler = () => canceled.reject(attempt.controller.signal.reason);
+						const abortHandler = () => canceled.reject(attempt.abortReason);
 						attempt.controller.signal.addEventListener("abort", abortHandler, { once: true });
 						try {
 							try {
@@ -1256,27 +1296,38 @@ var Zero = class Zero {
 						})).key === "connectRequest") this.#connectionManager.resumeFromConnectRequest();
 						break;
 					case Error$1:
-						lc.info?.(`Run loop paused in error state. Call zero.connection.connect() to resume.`, currentState.reason);
+						lc.info?.(`Run loop paused in error state. Call zero.connection.connect() to resume.`, currentState.reason, ...getErrorCauses(currentState.reason));
 						if ((await promiseRace({
 							connectRequest: this.#connectionManager.waitForConnectRequest(),
 							stateChange: this.#connectionManager.waitForStateChange()
 						})).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, kind, {
+					lc[level]?.("Failed to connect", ex, ...getErrorCauses(ex), kind, {
 						lmid: this.#lastMutationIDReceived,
 						baseCookie: this.#connectCookie
 					});
 				}
 				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: {
@@ -1458,25 +1509,11 @@ var Zero = class Zero {
 	}
 	async delete() {
 		await this.close();
-		const idbDatabasesStore = new IDBDatabasesStore(getKVStoreProvider(this.#lc, this.#kvStore).create);
-		try {
-			const databases = await idbDatabasesStore.getDatabases();
-			const dbNamesToDelete = Object.values(databases).filter((database) => database.replicacheName === this.#rep.name).map((database) => database.name);
-			const deleteResults = await Promise.allSettled(dbNamesToDelete.map(async (dbName) => {
-				await dropDatabase(dbName, { kvStore: this.#kvStore });
-				return dbName;
-			}));
-			const deleted = [];
-			const errors = [];
-			for (const result of deleteResults) if (result.status === "fulfilled") deleted.push(result.value);
-			else errors.push(result.reason);
-			return {
-				deleted,
-				errors
-			};
-		} finally {
-			await idbDatabasesStore.close();
-		}
+		const { dropped, errors } = await dropMatchingDatabases((database) => database.replicacheName === this.#rep.name, { kvStore: this.#kvStore });
+		return {
+			deleted: dropped,
+			errors
+		};
 	}
 	#addMetric = (metric, value, ...args) => {
 		assert(isClientMetric(metric), `Invalid metric: ${metric}`);