@electric-sql/client 1.0.2 → 1.0.4

package/dist/index.legacy-esm.js

@@ -262,6 +262,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];
@@ -307,12 +308,15 @@ function createFetchWithBackoff(fetchClient, backoffOptions = BackoffDefaults) {
     }
   };
 }
+var NO_BODY_STATUS_CODES = [201, 204, 205];
 function createFetchWithConsumedMessages(fetchClient) {
   return async (...args) => {
     const url = args[0];
     const res = await fetchClient(...args);
     try {
-      if (res.body === null) return res;
+      if (res.status < 200 || NO_BODY_STATUS_CODES.includes(res.status)) {
+        return res;
+      }
       const text = await res.text();
       return new Response(text, res);
     } catch (err) {
@@ -519,7 +523,7 @@ async 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);
@@ -528,6 +532,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 😎
@@ -564,6 +569,7 @@ var ShapeStream = class {
         createFetchWithChunkBuffer(fetchWithBackoffClient)
       )
     ));
+    __privateMethod(this, _ShapeStream_instances, subscribeToVisibilityChanges_fn).call(this);
   }
   get shapeHandle() {
     return __privateGet(this, _shapeHandle);
@@ -609,6 +615,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
@@ -630,6 +639,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();
@@ -645,128 +655,10 @@ _tickPromiseResolver = new WeakMap();
 _tickPromiseRejecter = new WeakMap();
 _ShapeStream_instances = new WeakSet();
 start_fn = async 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] = await 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 = await __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);
-          await __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 = await 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);
-        }
-        await __privateMethod(this, _ShapeStream_instances, publish_fn).call(this, batch);
-      }
-      (_d = __privateGet(this, _tickPromiseResolver)) == null ? void 0 : _d.call(this);
-    }
+    await __privateMethod(this, _ShapeStream_instances, requestShape_fn).call(this);
   } catch (err) {
     __privateSet(this, _error, err);
     if (__privateGet(this, _onError)) {
@@ -787,7 +679,150 @@ start_fn = async 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 = async 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] = await 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 = await __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);
+      await __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 = await 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);
+    }
+    await __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 = async function() {
@@ -823,6 +858,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