@electric-sql/client 1.0.2 → 1.0.4

package/dist/index.mjs

@@ -284,6 +284,7 @@ var WHERE_QUERY_PARAM = `where`;
 var REPLICA_PARAM = `replica`;
 var WHERE_PARAMS_PARAM = `params`;
 var FORCE_DISCONNECT_AND_REFRESH = `force-disconnect-and-refresh`;
+var PAUSE_STREAM = `pause-stream`;
 
 // src/fetch.ts
 var HTTP_RETRY_STATUS_CODES = [429];
@@ -329,12 +330,15 @@ function createFetchWithBackoff(fetchClient, backoffOptions = BackoffDefaults) {
     }
   });
 }
+var NO_BODY_STATUS_CODES = [201, 204, 205];
 function createFetchWithConsumedMessages(fetchClient) {
   return (...args) => __async(this, null, function* () {
     const url = args[0];
     const res = yield fetchClient(...args);
     try {
-      if (res.body === null) return res;
+      if (res.status < 200 || NO_BODY_STATUS_CODES.includes(res.status)) {
+        return res;
+      }
       const text = yield res.text();
       return new Response(text, res);
     } catch (err) {
@@ -549,7 +553,7 @@ function resolveHeaders(headers) {
     return Object.fromEntries(resolvedEntries);
   });
 }
-var _error, _fetchClient2, _messageParser, _subscribers, _started, _lastOffset, _liveCacheBuster, _lastSyncedAt, _isUpToDate, _connected, _shapeHandle, _schema, _onError, _requestAbortController, _isRefreshing, _tickPromise, _tickPromiseResolver, _tickPromiseRejecter, _ShapeStream_instances, start_fn, nextTick_fn, publish_fn, sendErrorToSubscribers_fn, reset_fn;
+var _error, _fetchClient2, _messageParser, _subscribers, _started, _state, _lastOffset, _liveCacheBuster, _lastSyncedAt, _isUpToDate, _connected, _shapeHandle, _schema, _onError, _requestAbortController, _isRefreshing, _tickPromise, _tickPromiseResolver, _tickPromiseRejecter, _ShapeStream_instances, start_fn, requestShape_fn, pause_fn, resume_fn, nextTick_fn, publish_fn, sendErrorToSubscribers_fn, subscribeToVisibilityChanges_fn, reset_fn;
 var ShapeStream = class {
   constructor(options) {
     __privateAdd(this, _ShapeStream_instances);
@@ -558,6 +562,7 @@ var ShapeStream = class {
     __privateAdd(this, _messageParser);
     __privateAdd(this, _subscribers, /* @__PURE__ */ new Map());
     __privateAdd(this, _started, false);
+    __privateAdd(this, _state, `active`);
     __privateAdd(this, _lastOffset);
     __privateAdd(this, _liveCacheBuster);
     // Seconds since our Electric Epoch 😎
@@ -594,6 +599,7 @@ var ShapeStream = class {
         createFetchWithChunkBuffer(fetchWithBackoffClient)
       )
     ));
+    __privateMethod(this, _ShapeStream_instances, subscribeToVisibilityChanges_fn).call(this);
   }
   get shapeHandle() {
     return __privateGet(this, _shapeHandle);
@@ -639,6 +645,9 @@ var ShapeStream = class {
   hasStarted() {
     return __privateGet(this, _started);
   }
+  isPaused() {
+    return __privateGet(this, _state) === `paused`;
+  }
   /**
    * Refreshes the shape stream.
    * This preemptively aborts any ongoing long poll and reconnects without
@@ -662,6 +671,7 @@ _fetchClient2 = new WeakMap();
 _messageParser = new WeakMap();
 _subscribers = new WeakMap();
 _started = new WeakMap();
+_state = new WeakMap();
 _lastOffset = new WeakMap();
 _liveCacheBuster = new WeakMap();
 _lastSyncedAt = new WeakMap();
@@ -678,128 +688,10 @@ _tickPromiseRejecter = new WeakMap();
 _ShapeStream_instances = new WeakSet();
 start_fn = function() {
   return __async(this, null, function* () {
-    var _a, _b, _c, _d, _e;
-    if (__privateGet(this, _started)) throw new Error(`Cannot start stream twice`);
+    var _a;
     __privateSet(this, _started, true);
     try {
-      while (!((_a = this.options.signal) == null ? void 0 : _a.aborted) && !__privateGet(this, _isUpToDate) || this.options.subscribe) {
-        const { url, signal } = this.options;
-        const [requestHeaders, params] = yield Promise.all([
-          resolveHeaders(this.options.headers),
-          this.options.params ? toInternalParams(convertWhereParamsToObj(this.options.params)) : void 0
-        ]);
-        if (params) {
-          validateParams(params);
-        }
-        const fetchUrl = new URL(url);
-        if (params) {
-          if (params.table)
-            setQueryParam(fetchUrl, TABLE_QUERY_PARAM, params.table);
-          if (params.where)
-            setQueryParam(fetchUrl, WHERE_QUERY_PARAM, params.where);
-          if (params.columns)
-            setQueryParam(fetchUrl, COLUMNS_QUERY_PARAM, params.columns);
-          if (params.replica)
-            setQueryParam(fetchUrl, REPLICA_PARAM, params.replica);
-          if (params.params)
-            setQueryParam(fetchUrl, WHERE_PARAMS_PARAM, params.params);
-          const customParams = __spreadValues({}, params);
-          delete customParams.table;
-          delete customParams.where;
-          delete customParams.columns;
-          delete customParams.replica;
-          delete customParams.params;
-          for (const [key, value] of Object.entries(customParams)) {
-            setQueryParam(fetchUrl, key, value);
-          }
-        }
-        fetchUrl.searchParams.set(OFFSET_QUERY_PARAM, __privateGet(this, _lastOffset));
-        if (__privateGet(this, _isUpToDate)) {
-          if (!__privateGet(this, _isRefreshing)) {
-            fetchUrl.searchParams.set(LIVE_QUERY_PARAM, `true`);
-          }
-          fetchUrl.searchParams.set(
-            LIVE_CACHE_BUSTER_QUERY_PARAM,
-            __privateGet(this, _liveCacheBuster)
-          );
-        }
-        if (__privateGet(this, _shapeHandle)) {
-          fetchUrl.searchParams.set(
-            SHAPE_HANDLE_QUERY_PARAM,
-            __privateGet(this, _shapeHandle)
-          );
-        }
-        fetchUrl.searchParams.sort();
-        __privateSet(this, _requestAbortController, new AbortController());
-        let abortListener;
-        if (signal) {
-          abortListener = () => {
-            var _a2;
-            (_a2 = __privateGet(this, _requestAbortController)) == null ? void 0 : _a2.abort(signal.reason);
-          };
-          signal.addEventListener(`abort`, abortListener, { once: true });
-          if (signal.aborted) {
-            (_b = __privateGet(this, _requestAbortController)) == null ? void 0 : _b.abort(signal.reason);
-          }
-        }
-        let response;
-        try {
-          response = yield __privateGet(this, _fetchClient2).call(this, fetchUrl.toString(), {
-            signal: __privateGet(this, _requestAbortController).signal,
-            headers: requestHeaders
-          });
-          __privateSet(this, _connected, true);
-        } catch (e) {
-          if ((e instanceof FetchError || e instanceof FetchBackoffAbortError) && __privateGet(this, _requestAbortController).signal.aborted && __privateGet(this, _requestAbortController).signal.reason === FORCE_DISCONNECT_AND_REFRESH) {
-            continue;
-          }
-          if (e instanceof FetchBackoffAbortError) break;
-          if (!(e instanceof FetchError)) throw e;
-          if (e.status == 409) {
-            const newShapeHandle = e.headers[SHAPE_HANDLE_HEADER];
-            __privateMethod(this, _ShapeStream_instances, reset_fn).call(this, newShapeHandle);
-            yield __privateMethod(this, _ShapeStream_instances, publish_fn).call(this, e.json);
-            continue;
-          } else {
-            __privateMethod(this, _ShapeStream_instances, sendErrorToSubscribers_fn).call(this, e);
-            throw e;
-          }
-        } finally {
-          if (abortListener && signal) {
-            signal.removeEventListener(`abort`, abortListener);
-          }
-          __privateSet(this, _requestAbortController, void 0);
-        }
-        const { headers } = response;
-        const shapeHandle = headers.get(SHAPE_HANDLE_HEADER);
-        if (shapeHandle) {
-          __privateSet(this, _shapeHandle, shapeHandle);
-        }
-        const lastOffset = headers.get(CHUNK_LAST_OFFSET_HEADER);
-        if (lastOffset) {
-          __privateSet(this, _lastOffset, lastOffset);
-        }
-        const liveCacheBuster = headers.get(LIVE_CACHE_BUSTER_HEADER);
-        if (liveCacheBuster) {
-          __privateSet(this, _liveCacheBuster, liveCacheBuster);
-        }
-        const getSchema = () => {
-          const schemaHeader = headers.get(SHAPE_SCHEMA_HEADER);
-          return schemaHeader ? JSON.parse(schemaHeader) : {};
-        };
-        __privateSet(this, _schema, (_c = __privateGet(this, _schema)) != null ? _c : getSchema());
-        const messages = yield response.text();
-        const batch = __privateGet(this, _messageParser).parse(messages, __privateGet(this, _schema));
-        if (batch.length > 0) {
-          const lastMessage = batch[batch.length - 1];
-          if (isUpToDateMessage(lastMessage)) {
-            __privateSet(this, _lastSyncedAt, Date.now());
-            __privateSet(this, _isUpToDate, true);
-          }
-          yield __privateMethod(this, _ShapeStream_instances, publish_fn).call(this, batch);
-        }
-        (_d = __privateGet(this, _tickPromiseResolver)) == null ? void 0 : _d.call(this);
-      }
+      yield __privateMethod(this, _ShapeStream_instances, requestShape_fn).call(this);
     } catch (err) {
       __privateSet(this, _error, err);
       if (__privateGet(this, _onError)) {
@@ -820,10 +712,155 @@ start_fn = function() {
       throw err;
     } finally {
       __privateSet(this, _connected, false);
-      (_e = __privateGet(this, _tickPromiseRejecter)) == null ? void 0 : _e.call(this);
+      (_a = __privateGet(this, _tickPromiseRejecter)) == null ? void 0 : _a.call(this);
     }
   });
 };
+requestShape_fn = function() {
+  return __async(this, null, function* () {
+    var _a, _b, _c, _d;
+    if (__privateGet(this, _state) === `pause-requested`) {
+      __privateSet(this, _state, `paused`);
+      return;
+    }
+    if (!this.options.subscribe && (((_a = this.options.signal) == null ? void 0 : _a.aborted) || __privateGet(this, _isUpToDate))) {
+      return;
+    }
+    const resumingFromPause = __privateGet(this, _state) === `paused`;
+    __privateSet(this, _state, `active`);
+    const { url, signal } = this.options;
+    const [requestHeaders, params] = yield Promise.all([
+      resolveHeaders(this.options.headers),
+      this.options.params ? toInternalParams(convertWhereParamsToObj(this.options.params)) : void 0
+    ]);
+    if (params) {
+      validateParams(params);
+    }
+    const fetchUrl = new URL(url);
+    if (params) {
+      if (params.table) setQueryParam(fetchUrl, TABLE_QUERY_PARAM, params.table);
+      if (params.where) setQueryParam(fetchUrl, WHERE_QUERY_PARAM, params.where);
+      if (params.columns)
+        setQueryParam(fetchUrl, COLUMNS_QUERY_PARAM, params.columns);
+      if (params.replica) setQueryParam(fetchUrl, REPLICA_PARAM, params.replica);
+      if (params.params)
+        setQueryParam(fetchUrl, WHERE_PARAMS_PARAM, params.params);
+      const customParams = __spreadValues({}, params);
+      delete customParams.table;
+      delete customParams.where;
+      delete customParams.columns;
+      delete customParams.replica;
+      delete customParams.params;
+      for (const [key, value] of Object.entries(customParams)) {
+        setQueryParam(fetchUrl, key, value);
+      }
+    }
+    fetchUrl.searchParams.set(OFFSET_QUERY_PARAM, __privateGet(this, _lastOffset));
+    if (__privateGet(this, _isUpToDate)) {
+      if (!__privateGet(this, _isRefreshing) && !resumingFromPause) {
+        fetchUrl.searchParams.set(LIVE_QUERY_PARAM, `true`);
+      }
+      fetchUrl.searchParams.set(
+        LIVE_CACHE_BUSTER_QUERY_PARAM,
+        __privateGet(this, _liveCacheBuster)
+      );
+    }
+    if (__privateGet(this, _shapeHandle)) {
+      fetchUrl.searchParams.set(SHAPE_HANDLE_QUERY_PARAM, __privateGet(this, _shapeHandle));
+    }
+    fetchUrl.searchParams.sort();
+    __privateSet(this, _requestAbortController, new AbortController());
+    let abortListener;
+    if (signal) {
+      abortListener = () => {
+        var _a2;
+        (_a2 = __privateGet(this, _requestAbortController)) == null ? void 0 : _a2.abort(signal.reason);
+      };
+      signal.addEventListener(`abort`, abortListener, { once: true });
+      if (signal.aborted) {
+        (_b = __privateGet(this, _requestAbortController)) == null ? void 0 : _b.abort(signal.reason);
+      }
+    }
+    let response;
+    try {
+      response = yield __privateGet(this, _fetchClient2).call(this, fetchUrl.toString(), {
+        signal: __privateGet(this, _requestAbortController).signal,
+        headers: requestHeaders
+      });
+      __privateSet(this, _connected, true);
+    } catch (e) {
+      if ((e instanceof FetchError || e instanceof FetchBackoffAbortError) && __privateGet(this, _requestAbortController).signal.aborted && __privateGet(this, _requestAbortController).signal.reason === FORCE_DISCONNECT_AND_REFRESH) {
+        return __privateMethod(this, _ShapeStream_instances, requestShape_fn).call(this);
+      }
+      if (e instanceof FetchBackoffAbortError) {
+        if (__privateGet(this, _requestAbortController).signal.aborted && __privateGet(this, _requestAbortController).signal.reason === PAUSE_STREAM) {
+          __privateSet(this, _state, `paused`);
+        }
+        return;
+      }
+      if (!(e instanceof FetchError)) throw e;
+      if (e.status == 409) {
+        const newShapeHandle = e.headers[SHAPE_HANDLE_HEADER];
+        __privateMethod(this, _ShapeStream_instances, reset_fn).call(this, newShapeHandle);
+        yield __privateMethod(this, _ShapeStream_instances, publish_fn).call(this, e.json);
+        return __privateMethod(this, _ShapeStream_instances, requestShape_fn).call(this);
+      } else {
+        __privateMethod(this, _ShapeStream_instances, sendErrorToSubscribers_fn).call(this, e);
+        throw e;
+      }
+    } finally {
+      if (abortListener && signal) {
+        signal.removeEventListener(`abort`, abortListener);
+      }
+      __privateSet(this, _requestAbortController, void 0);
+    }
+    const { headers, status } = response;
+    const shapeHandle = headers.get(SHAPE_HANDLE_HEADER);
+    if (shapeHandle) {
+      __privateSet(this, _shapeHandle, shapeHandle);
+    }
+    const lastOffset = headers.get(CHUNK_LAST_OFFSET_HEADER);
+    if (lastOffset) {
+      __privateSet(this, _lastOffset, lastOffset);
+    }
+    const liveCacheBuster = headers.get(LIVE_CACHE_BUSTER_HEADER);
+    if (liveCacheBuster) {
+      __privateSet(this, _liveCacheBuster, liveCacheBuster);
+    }
+    const getSchema = () => {
+      const schemaHeader = headers.get(SHAPE_SCHEMA_HEADER);
+      return schemaHeader ? JSON.parse(schemaHeader) : {};
+    };
+    __privateSet(this, _schema, (_c = __privateGet(this, _schema)) != null ? _c : getSchema());
+    if (status === 204) {
+      __privateSet(this, _lastSyncedAt, Date.now());
+    }
+    const messages = (yield response.text()) || `[]`;
+    const batch = __privateGet(this, _messageParser).parse(messages, __privateGet(this, _schema));
+    if (batch.length > 0) {
+      const lastMessage = batch[batch.length - 1];
+      if (isUpToDateMessage(lastMessage)) {
+        __privateSet(this, _lastSyncedAt, Date.now());
+        __privateSet(this, _isUpToDate, true);
+      }
+      yield __privateMethod(this, _ShapeStream_instances, publish_fn).call(this, batch);
+    }
+    (_d = __privateGet(this, _tickPromiseResolver)) == null ? void 0 : _d.call(this);
+    return __privateMethod(this, _ShapeStream_instances, requestShape_fn).call(this);
+  });
+};
+pause_fn = function() {
+  var _a;
+  if (__privateGet(this, _started) && __privateGet(this, _state) === `active`) {
+    __privateSet(this, _state, `pause-requested`);
+    (_a = __privateGet(this, _requestAbortController)) == null ? void 0 : _a.abort(PAUSE_STREAM);
+  }
+};
+resume_fn = function() {
+  if (__privateGet(this, _started) && __privateGet(this, _state) === `paused`) {
+    __privateMethod(this, _ShapeStream_instances, start_fn).call(this);
+  }
+};
 nextTick_fn = function() {
   return __async(this, null, function* () {
     if (__privateGet(this, _tickPromise)) {
@@ -861,6 +898,18 @@ sendErrorToSubscribers_fn = function(error) {
     errorFn == null ? void 0 : errorFn(error);
   });
 };
+subscribeToVisibilityChanges_fn = function() {
+  if (typeof document === `object` && typeof document.hidden === `boolean` && typeof document.addEventListener === `function`) {
+    const visibilityHandler = () => {
+      if (document.hidden) {
+        __privateMethod(this, _ShapeStream_instances, pause_fn).call(this);
+      } else {
+        __privateMethod(this, _ShapeStream_instances, resume_fn).call(this);
+      }
+    };
+    document.addEventListener(`visibilitychange`, visibilityHandler);
+  }
+};
 /**
  * Resets the state of the stream, optionally with a provided
  * shape handle