@tanstack/react-query-persist-client 5.0.0-alpha.7 → 5.0.0-alpha.70

package/build/lib/PersistQueryClientProvider.cjs

@@ -0,0 +1,75 @@
+'use strict';
+
+var _rollupPluginBabelHelpers = require('./_virtual/_rollupPluginBabelHelpers.cjs');
+require('client-only');
+var React = require('react');
+var queryPersistClientCore = require('@tanstack/query-persist-client-core');
+var reactQuery = require('@tanstack/react-query');
+
+function _interopNamespaceDefault(e) {
+  var n = Object.create(null);
+  if (e) {
+    Object.keys(e).forEach(function (k) {
+      if (k !== 'default') {
+        var d = Object.getOwnPropertyDescriptor(e, k);
+        Object.defineProperty(n, k, d.get ? d : {
+          enumerable: true,
+          get: function () { return e[k]; }
+        });
+      }
+    });
+  }
+  n.default = e;
+  return Object.freeze(n);
+}
+
+var React__namespace = /*#__PURE__*/_interopNamespaceDefault(React);
+
+const PersistQueryClientProvider = ({
+  client,
+  children,
+  persistOptions,
+  onSuccess,
+  ...props
+}) => {
+  const [isRestoring, setIsRestoring] = React__namespace.useState(true);
+  const refs = React__namespace.useRef({
+    persistOptions,
+    onSuccess
+  });
+  React__namespace.useEffect(() => {
+    refs.current = {
+      persistOptions,
+      onSuccess
+    };
+  });
+  React__namespace.useEffect(() => {
+    let isStale = false;
+    setIsRestoring(true);
+    const [unsubscribe, promise] = queryPersistClientCore.persistQueryClient({
+      ...refs.current.persistOptions,
+      queryClient: client
+    });
+    promise.then(async () => {
+      if (!isStale) {
+        try {
+          await refs.current.onSuccess?.();
+        } finally {
+          setIsRestoring(false);
+        }
+      }
+    });
+    return () => {
+      isStale = true;
+      unsubscribe();
+    };
+  }, [client]);
+  return /*#__PURE__*/React__namespace.createElement(reactQuery.QueryClientProvider, _rollupPluginBabelHelpers.extends({
+    client: client
+  }, props), /*#__PURE__*/React__namespace.createElement(reactQuery.IsRestoringProvider, {
+    value: isRestoring
+  }, children));
+};
+
+exports.PersistQueryClientProvider = PersistQueryClientProvider;
+//# sourceMappingURL=PersistQueryClientProvider.cjs.map

package/build/lib/PersistQueryClientProvider.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"PersistQueryClientProvider.cjs","sources":["../../src/PersistQueryClientProvider.tsx"],"sourcesContent":["import 'client-only'\nimport * as React from 'react'\n\nimport type { PersistQueryClientOptions } from '@tanstack/query-persist-client-core'\nimport { persistQueryClient } from '@tanstack/query-persist-client-core'\nimport type { QueryClientProviderProps } from '@tanstack/react-query'\nimport { QueryClientProvider, IsRestoringProvider } from '@tanstack/react-query'\n\nexport type PersistQueryClientProviderProps = QueryClientProviderProps & {\n  persistOptions: Omit<PersistQueryClientOptions, 'queryClient'>\n  onSuccess?: () => Promise<unknown> | unknown\n}\n\nexport const PersistQueryClientProvider = ({\n  client,\n  children,\n  persistOptions,\n  onSuccess,\n  ...props\n}: PersistQueryClientProviderProps): JSX.Element => {\n  const [isRestoring, setIsRestoring] = React.useState(true)\n  const refs = React.useRef({ persistOptions, onSuccess })\n\n  React.useEffect(() => {\n    refs.current = { persistOptions, onSuccess }\n  })\n\n  React.useEffect(() => {\n    let isStale = false\n    setIsRestoring(true)\n    const [unsubscribe, promise] = persistQueryClient({\n      ...refs.current.persistOptions,\n      queryClient: client,\n    })\n\n    promise.then(async () => {\n      if (!isStale) {\n        try {\n          await refs.current.onSuccess?.()\n        } finally {\n          setIsRestoring(false)\n        }\n      }\n    })\n\n    return () => {\n      isStale = true\n      unsubscribe()\n    }\n  }, [client])\n\n  return (\n    <QueryClientProvider client={client} {...props}>\n      <IsRestoringProvider value={isRestoring}>{children}</IsRestoringProvider>\n    </QueryClientProvider>\n  )\n}\n"],"names":["PersistQueryClientProvider","client","children","persistOptions","onSuccess","props","isRestoring","setIsRestoring","React","useState","refs","useRef","useEffect","current","isStale","unsubscribe","promise","persistQueryClient","queryClient","then","createElement","QueryClientProvider","_extends","IsRestoringProvider","value"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;AAaO,MAAMA,0BAA0B,GAAGA,CAAC;EACzCC,MAAM;EACNC,QAAQ;EACRC,cAAc;EACdC,SAAS;EACT,GAAGC,KAAAA;AAC4B,CAAC,KAAkB;EAClD,MAAM,CAACC,WAAW,EAAEC,cAAc,CAAC,GAAGC,gBAAK,CAACC,QAAQ,CAAC,IAAI,CAAC,CAAA;AAC1D,EAAA,MAAMC,IAAI,GAAGF,gBAAK,CAACG,MAAM,CAAC;IAAER,cAAc;AAAEC,IAAAA,SAAAA;AAAU,GAAC,CAAC,CAAA;EAExDI,gBAAK,CAACI,SAAS,CAAC,MAAM;IACpBF,IAAI,CAACG,OAAO,GAAG;MAAEV,cAAc;AAAEC,MAAAA,SAAAA;KAAW,CAAA;AAC9C,GAAC,CAAC,CAAA;EAEFI,gBAAK,CAACI,SAAS,CAAC,MAAM;IACpB,IAAIE,OAAO,GAAG,KAAK,CAAA;IACnBP,cAAc,CAAC,IAAI,CAAC,CAAA;AACpB,IAAA,MAAM,CAACQ,WAAW,EAAEC,OAAO,CAAC,GAAGC,yCAAkB,CAAC;AAChD,MAAA,GAAGP,IAAI,CAACG,OAAO,CAACV,cAAc;AAC9Be,MAAAA,WAAW,EAAEjB,MAAAA;AACf,KAAC,CAAC,CAAA;IAEFe,OAAO,CAACG,IAAI,CAAC,YAAY;MACvB,IAAI,CAACL,OAAO,EAAE;QACZ,IAAI;AACF,UAAA,MAAMJ,IAAI,CAACG,OAAO,CAACT,SAAS,IAAI,CAAA;AAClC,SAAC,SAAS;UACRG,cAAc,CAAC,KAAK,CAAC,CAAA;AACvB,SAAA;AACF,OAAA;AACF,KAAC,CAAC,CAAA;AAEF,IAAA,OAAO,MAAM;AACXO,MAAAA,OAAO,GAAG,IAAI,CAAA;AACdC,MAAAA,WAAW,EAAE,CAAA;KACd,CAAA;AACH,GAAC,EAAE,CAACd,MAAM,CAAC,CAAC,CAAA;AAEZ,EAAA,oBACEO,gBAAA,CAAAY,aAAA,CAACC,8BAAmB,EAAAC,iCAAA,CAAA;AAACrB,IAAAA,MAAM,EAAEA,MAAAA;AAAO,GAAA,EAAKI,KAAK,CAC5CG,eAAAA,gBAAA,CAAAY,aAAA,CAACG,8BAAmB,EAAA;AAACC,IAAAA,KAAK,EAAElB,WAAAA;GAAcJ,EAAAA,QAA8B,CACrD,CAAC,CAAA;AAE1B;;;;"}

package/build/lib/PersistQueryClientProvider.d.ts

@@ -1,7 +1,9 @@
+import 'client-only';
 import type { PersistQueryClientOptions } from '@tanstack/query-persist-client-core';
 import type { QueryClientProviderProps } from '@tanstack/react-query';
-export declare type PersistQueryClientProviderProps = QueryClientProviderProps & {
+export type PersistQueryClientProviderProps = QueryClientProviderProps & {
     persistOptions: Omit<PersistQueryClientOptions, 'queryClient'>;
-    onSuccess?: () => void;
+    onSuccess?: () => Promise<unknown> | unknown;
 };
 export declare const PersistQueryClientProvider: ({ client, children, persistOptions, onSuccess, ...props }: PersistQueryClientProviderProps) => JSX.Element;
+//# sourceMappingURL=PersistQueryClientProvider.d.ts.map

package/build/lib/PersistQueryClientProvider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"PersistQueryClientProvider.d.ts","sourceRoot":"","sources":["../../src/PersistQueryClientProvider.tsx"],"names":[],"mappings":"AAAA,OAAO,aAAa,CAAA;AAGpB,OAAO,KAAK,EAAE,yBAAyB,EAAE,MAAM,qCAAqC,CAAA;AAEpF,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,uBAAuB,CAAA;AAGrE,MAAM,MAAM,+BAA+B,GAAG,wBAAwB,GAAG;IACvE,cAAc,EAAE,IAAI,CAAC,yBAAyB,EAAE,aAAa,CAAC,CAAA;IAC9D,SAAS,CAAC,EAAE,MAAM,OAAO,CAAC,OAAO,CAAC,GAAG,OAAO,CAAA;CAC7C,CAAA;AAED,eAAO,MAAM,0BAA0B,8DAMpC,+BAA+B,KAAG,WAqCpC,CAAA"}

package/build/lib/PersistQueryClientProvider.js

@@ -1,28 +1,8 @@
-'use strict';
-
-var _rollupPluginBabelHelpers = require('./_virtual/_rollupPluginBabelHelpers.js');
-var React = require('react');
-var queryPersistClientCore = require('@tanstack/query-persist-client-core');
-var reactQuery = require('@tanstack/react-query');
-
-function _interopNamespaceDefault(e) {
-  var n = Object.create(null);
-  if (e) {
-    Object.keys(e).forEach(function (k) {
-      if (k !== 'default') {
-        var d = Object.getOwnPropertyDescriptor(e, k);
-        Object.defineProperty(n, k, d.get ? d : {
-          enumerable: true,
-          get: function () { return e[k]; }
-        });
-      }
-    });
-  }
-  n.default = e;
-  return Object.freeze(n);
-}
-
-var React__namespace = /*#__PURE__*/_interopNamespaceDefault(React);
+import { extends as _extends } from './_virtual/_rollupPluginBabelHelpers.js';
+import 'client-only';
+import * as React from 'react';
+import { persistQueryClient } from '@tanstack/query-persist-client-core';
+import { QueryClientProvider, IsRestoringProvider } from '@tanstack/react-query';
 
 const PersistQueryClientProvider = ({
   client,
@@ -31,28 +11,31 @@ const PersistQueryClientProvider = ({
   onSuccess,
   ...props
 }) => {
-  const [isRestoring, setIsRestoring] = React__namespace.useState(true);
-  const refs = React__namespace.useRef({
+  const [isRestoring, setIsRestoring] = React.useState(true);
+  const refs = React.useRef({
     persistOptions,
     onSuccess
   });
-  React__namespace.useEffect(() => {
+  React.useEffect(() => {
     refs.current = {
       persistOptions,
       onSuccess
     };
   });
-  React__namespace.useEffect(() => {
+  React.useEffect(() => {
     let isStale = false;
     setIsRestoring(true);
-    const [unsubscribe, promise] = queryPersistClientCore.persistQueryClient({
+    const [unsubscribe, promise] = persistQueryClient({
       ...refs.current.persistOptions,
       queryClient: client
     });
-    promise.then(() => {
+    promise.then(async () => {
       if (!isStale) {
-        refs.current.onSuccess == null ? void 0 : refs.current.onSuccess();
-        setIsRestoring(false);
+        try {
+          await refs.current.onSuccess?.();
+        } finally {
+          setIsRestoring(false);
+        }
       }
     });
     return () => {
@@ -60,12 +43,12 @@ const PersistQueryClientProvider = ({
       unsubscribe();
     };
   }, [client]);
-  return /*#__PURE__*/React__namespace.createElement(reactQuery.QueryClientProvider, _rollupPluginBabelHelpers.extends({
+  return /*#__PURE__*/React.createElement(QueryClientProvider, _extends({
     client: client
-  }, props), /*#__PURE__*/React__namespace.createElement(reactQuery.IsRestoringProvider, {
+  }, props), /*#__PURE__*/React.createElement(IsRestoringProvider, {
     value: isRestoring
   }, children));
 };
 
-exports.PersistQueryClientProvider = PersistQueryClientProvider;
+export { PersistQueryClientProvider };
 //# sourceMappingURL=PersistQueryClientProvider.js.map

package/build/lib/PersistQueryClientProvider.js.map

@@ -1 +1 @@
-{"version":3,"file":"PersistQueryClientProvider.js","sources":["../../src/PersistQueryClientProvider.tsx"],"sourcesContent":["'use client'\nimport * as React from 'react'\n\nimport type { PersistQueryClientOptions } from '@tanstack/query-persist-client-core'\nimport { persistQueryClient } from '@tanstack/query-persist-client-core'\nimport type { QueryClientProviderProps } from '@tanstack/react-query'\nimport { QueryClientProvider, IsRestoringProvider } from '@tanstack/react-query'\n\nexport type PersistQueryClientProviderProps = QueryClientProviderProps & {\n  persistOptions: Omit<PersistQueryClientOptions, 'queryClient'>\n  onSuccess?: () => void\n}\n\nexport const PersistQueryClientProvider = ({\n  client,\n  children,\n  persistOptions,\n  onSuccess,\n  ...props\n}: PersistQueryClientProviderProps): JSX.Element => {\n  const [isRestoring, setIsRestoring] = React.useState(true)\n  const refs = React.useRef({ persistOptions, onSuccess })\n\n  React.useEffect(() => {\n    refs.current = { persistOptions, onSuccess }\n  })\n\n  React.useEffect(() => {\n    let isStale = false\n    setIsRestoring(true)\n    const [unsubscribe, promise] = persistQueryClient({\n      ...refs.current.persistOptions,\n      queryClient: client,\n    })\n\n    promise.then(() => {\n      if (!isStale) {\n        refs.current.onSuccess?.()\n        setIsRestoring(false)\n      }\n    })\n\n    return () => {\n      isStale = true\n      unsubscribe()\n    }\n  }, [client])\n\n  return (\n    <QueryClientProvider client={client} {...props}>\n      <IsRestoringProvider value={isRestoring}>{children}</IsRestoringProvider>\n    </QueryClientProvider>\n  )\n}\n"],"names":["PersistQueryClientProvider","client","children","persistOptions","onSuccess","props","isRestoring","setIsRestoring","React","useState","refs","useRef","useEffect","current","isStale","unsubscribe","promise","persistQueryClient","queryClient","then","QueryClientProvider","_extends","IsRestoringProvider"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAaO,MAAMA,0BAA0B,GAAG,CAAC;EACzCC,MAAM;EACNC,QAAQ;EACRC,cAAc;EACdC,SAAS;EACT,GAAGC,KAAAA;AAC4B,CAAC,KAAkB;EAClD,MAAM,CAACC,WAAW,EAAEC,cAAc,CAAC,GAAGC,gBAAK,CAACC,QAAQ,CAAC,IAAI,CAAC,CAAA;AAC1D,EAAA,MAAMC,IAAI,GAAGF,gBAAK,CAACG,MAAM,CAAC;IAAER,cAAc;AAAEC,IAAAA,SAAAA;AAAU,GAAC,CAAC,CAAA;EAExDI,gBAAK,CAACI,SAAS,CAAC,MAAM;IACpBF,IAAI,CAACG,OAAO,GAAG;MAAEV,cAAc;AAAEC,MAAAA,SAAAA;KAAW,CAAA;AAC9C,GAAC,CAAC,CAAA;EAEFI,gBAAK,CAACI,SAAS,CAAC,MAAM;IACpB,IAAIE,OAAO,GAAG,KAAK,CAAA;IACnBP,cAAc,CAAC,IAAI,CAAC,CAAA;AACpB,IAAA,MAAM,CAACQ,WAAW,EAAEC,OAAO,CAAC,GAAGC,yCAAkB,CAAC;AAChD,MAAA,GAAGP,IAAI,CAACG,OAAO,CAACV,cAAc;AAC9Be,MAAAA,WAAW,EAAEjB,MAAAA;AACf,KAAC,CAAC,CAAA;IAEFe,OAAO,CAACG,IAAI,CAAC,MAAM;MACjB,IAAI,CAACL,OAAO,EAAE;QACZJ,IAAI,CAACG,OAAO,CAACT,SAAS,IAAA,IAAA,GAAA,KAAA,CAAA,GAAtBM,IAAI,CAACG,OAAO,CAACT,SAAS,EAAI,CAAA;QAC1BG,cAAc,CAAC,KAAK,CAAC,CAAA;AACvB,OAAA;AACF,KAAC,CAAC,CAAA;AAEF,IAAA,OAAO,MAAM;AACXO,MAAAA,OAAO,GAAG,IAAI,CAAA;AACdC,MAAAA,WAAW,EAAE,CAAA;KACd,CAAA;AACH,GAAC,EAAE,CAACd,MAAM,CAAC,CAAC,CAAA;AAEZ,EAAA,oBACEO,+BAACY,8BAAmB,EAAAC,iCAAA,CAAA;AAAC,IAAA,MAAM,EAAEpB,MAAAA;GAAYI,EAAAA,KAAK,CAC5C,eAAAG,gBAAA,CAAA,aAAA,CAACc,8BAAmB,EAAA;AAAC,IAAA,KAAK,EAAEhB,WAAAA;GAAcJ,EAAAA,QAAQ,CAAuB,CACrD,CAAA;AAE1B;;;;"}
+{"version":3,"file":"PersistQueryClientProvider.js","sources":["../../src/PersistQueryClientProvider.tsx"],"sourcesContent":["import 'client-only'\nimport * as React from 'react'\n\nimport type { PersistQueryClientOptions } from '@tanstack/query-persist-client-core'\nimport { persistQueryClient } from '@tanstack/query-persist-client-core'\nimport type { QueryClientProviderProps } from '@tanstack/react-query'\nimport { QueryClientProvider, IsRestoringProvider } from '@tanstack/react-query'\n\nexport type PersistQueryClientProviderProps = QueryClientProviderProps & {\n  persistOptions: Omit<PersistQueryClientOptions, 'queryClient'>\n  onSuccess?: () => Promise<unknown> | unknown\n}\n\nexport const PersistQueryClientProvider = ({\n  client,\n  children,\n  persistOptions,\n  onSuccess,\n  ...props\n}: PersistQueryClientProviderProps): JSX.Element => {\n  const [isRestoring, setIsRestoring] = React.useState(true)\n  const refs = React.useRef({ persistOptions, onSuccess })\n\n  React.useEffect(() => {\n    refs.current = { persistOptions, onSuccess }\n  })\n\n  React.useEffect(() => {\n    let isStale = false\n    setIsRestoring(true)\n    const [unsubscribe, promise] = persistQueryClient({\n      ...refs.current.persistOptions,\n      queryClient: client,\n    })\n\n    promise.then(async () => {\n      if (!isStale) {\n        try {\n          await refs.current.onSuccess?.()\n        } finally {\n          setIsRestoring(false)\n        }\n      }\n    })\n\n    return () => {\n      isStale = true\n      unsubscribe()\n    }\n  }, [client])\n\n  return (\n    <QueryClientProvider client={client} {...props}>\n      <IsRestoringProvider value={isRestoring}>{children}</IsRestoringProvider>\n    </QueryClientProvider>\n  )\n}\n"],"names":["PersistQueryClientProvider","client","children","persistOptions","onSuccess","props","isRestoring","setIsRestoring","React","useState","refs","useRef","useEffect","current","isStale","unsubscribe","promise","persistQueryClient","queryClient","then","createElement","QueryClientProvider","_extends","IsRestoringProvider","value"],"mappings":";;;;;;AAaO,MAAMA,0BAA0B,GAAGA,CAAC;EACzCC,MAAM;EACNC,QAAQ;EACRC,cAAc;EACdC,SAAS;EACT,GAAGC,KAAAA;AAC4B,CAAC,KAAkB;EAClD,MAAM,CAACC,WAAW,EAAEC,cAAc,CAAC,GAAGC,KAAK,CAACC,QAAQ,CAAC,IAAI,CAAC,CAAA;AAC1D,EAAA,MAAMC,IAAI,GAAGF,KAAK,CAACG,MAAM,CAAC;IAAER,cAAc;AAAEC,IAAAA,SAAAA;AAAU,GAAC,CAAC,CAAA;EAExDI,KAAK,CAACI,SAAS,CAAC,MAAM;IACpBF,IAAI,CAACG,OAAO,GAAG;MAAEV,cAAc;AAAEC,MAAAA,SAAAA;KAAW,CAAA;AAC9C,GAAC,CAAC,CAAA;EAEFI,KAAK,CAACI,SAAS,CAAC,MAAM;IACpB,IAAIE,OAAO,GAAG,KAAK,CAAA;IACnBP,cAAc,CAAC,IAAI,CAAC,CAAA;AACpB,IAAA,MAAM,CAACQ,WAAW,EAAEC,OAAO,CAAC,GAAGC,kBAAkB,CAAC;AAChD,MAAA,GAAGP,IAAI,CAACG,OAAO,CAACV,cAAc;AAC9Be,MAAAA,WAAW,EAAEjB,MAAAA;AACf,KAAC,CAAC,CAAA;IAEFe,OAAO,CAACG,IAAI,CAAC,YAAY;MACvB,IAAI,CAACL,OAAO,EAAE;QACZ,IAAI;AACF,UAAA,MAAMJ,IAAI,CAACG,OAAO,CAACT,SAAS,IAAI,CAAA;AAClC,SAAC,SAAS;UACRG,cAAc,CAAC,KAAK,CAAC,CAAA;AACvB,SAAA;AACF,OAAA;AACF,KAAC,CAAC,CAAA;AAEF,IAAA,OAAO,MAAM;AACXO,MAAAA,OAAO,GAAG,IAAI,CAAA;AACdC,MAAAA,WAAW,EAAE,CAAA;KACd,CAAA;AACH,GAAC,EAAE,CAACd,MAAM,CAAC,CAAC,CAAA;AAEZ,EAAA,oBACEO,KAAA,CAAAY,aAAA,CAACC,mBAAmB,EAAAC,QAAA,CAAA;AAACrB,IAAAA,MAAM,EAAEA,MAAAA;AAAO,GAAA,EAAKI,KAAK,CAC5CG,eAAAA,KAAA,CAAAY,aAAA,CAACG,mBAAmB,EAAA;AAACC,IAAAA,KAAK,EAAElB,WAAAA;GAAcJ,EAAAA,QAA8B,CACrD,CAAC,CAAA;AAE1B;;;;"}

package/build/lib/PersistQueryClientProvider.legacy.cjs

@@ -0,0 +1,75 @@
+'use strict';
+
+var _rollupPluginBabelHelpers = require('./_virtual/_rollupPluginBabelHelpers.legacy.cjs');
+require('client-only');
+var React = require('react');
+var queryPersistClientCore = require('@tanstack/query-persist-client-core');
+var reactQuery = require('@tanstack/react-query');
+
+function _interopNamespaceDefault(e) {
+  var n = Object.create(null);
+  if (e) {
+    Object.keys(e).forEach(function (k) {
+      if (k !== 'default') {
+        var d = Object.getOwnPropertyDescriptor(e, k);
+        Object.defineProperty(n, k, d.get ? d : {
+          enumerable: true,
+          get: function () { return e[k]; }
+        });
+      }
+    });
+  }
+  n.default = e;
+  return Object.freeze(n);
+}
+
+var React__namespace = /*#__PURE__*/_interopNamespaceDefault(React);
+
+const PersistQueryClientProvider = ({
+  client,
+  children,
+  persistOptions,
+  onSuccess,
+  ...props
+}) => {
+  const [isRestoring, setIsRestoring] = React__namespace.useState(true);
+  const refs = React__namespace.useRef({
+    persistOptions,
+    onSuccess
+  });
+  React__namespace.useEffect(() => {
+    refs.current = {
+      persistOptions,
+      onSuccess
+    };
+  });
+  React__namespace.useEffect(() => {
+    let isStale = false;
+    setIsRestoring(true);
+    const [unsubscribe, promise] = queryPersistClientCore.persistQueryClient({
+      ...refs.current.persistOptions,
+      queryClient: client
+    });
+    promise.then(async () => {
+      if (!isStale) {
+        try {
+          await (refs.current.onSuccess == null ? void 0 : refs.current.onSuccess());
+        } finally {
+          setIsRestoring(false);
+        }
+      }
+    });
+    return () => {
+      isStale = true;
+      unsubscribe();
+    };
+  }, [client]);
+  return /*#__PURE__*/React__namespace.createElement(reactQuery.QueryClientProvider, _rollupPluginBabelHelpers.extends({
+    client: client
+  }, props), /*#__PURE__*/React__namespace.createElement(reactQuery.IsRestoringProvider, {
+    value: isRestoring
+  }, children));
+};
+
+exports.PersistQueryClientProvider = PersistQueryClientProvider;
+//# sourceMappingURL=PersistQueryClientProvider.legacy.cjs.map

package/build/lib/PersistQueryClientProvider.legacy.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"PersistQueryClientProvider.legacy.cjs","sources":["../../src/PersistQueryClientProvider.tsx"],"sourcesContent":["import 'client-only'\nimport * as React from 'react'\n\nimport type { PersistQueryClientOptions } from '@tanstack/query-persist-client-core'\nimport { persistQueryClient } from '@tanstack/query-persist-client-core'\nimport type { QueryClientProviderProps } from '@tanstack/react-query'\nimport { QueryClientProvider, IsRestoringProvider } from '@tanstack/react-query'\n\nexport type PersistQueryClientProviderProps = QueryClientProviderProps & {\n  persistOptions: Omit<PersistQueryClientOptions, 'queryClient'>\n  onSuccess?: () => Promise<unknown> | unknown\n}\n\nexport const PersistQueryClientProvider = ({\n  client,\n  children,\n  persistOptions,\n  onSuccess,\n  ...props\n}: PersistQueryClientProviderProps): JSX.Element => {\n  const [isRestoring, setIsRestoring] = React.useState(true)\n  const refs = React.useRef({ persistOptions, onSuccess })\n\n  React.useEffect(() => {\n    refs.current = { persistOptions, onSuccess }\n  })\n\n  React.useEffect(() => {\n    let isStale = false\n    setIsRestoring(true)\n    const [unsubscribe, promise] = persistQueryClient({\n      ...refs.current.persistOptions,\n      queryClient: client,\n    })\n\n    promise.then(async () => {\n      if (!isStale) {\n        try {\n          await refs.current.onSuccess?.()\n        } finally {\n          setIsRestoring(false)\n        }\n      }\n    })\n\n    return () => {\n      isStale = true\n      unsubscribe()\n    }\n  }, [client])\n\n  return (\n    <QueryClientProvider client={client} {...props}>\n      <IsRestoringProvider value={isRestoring}>{children}</IsRestoringProvider>\n    </QueryClientProvider>\n  )\n}\n"],"names":["PersistQueryClientProvider","client","children","persistOptions","onSuccess","props","isRestoring","setIsRestoring","React","useState","refs","useRef","useEffect","current","isStale","unsubscribe","promise","persistQueryClient","queryClient","then","createElement","QueryClientProvider","_extends","IsRestoringProvider","value"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;AAaO,MAAMA,0BAA0B,GAAGA,CAAC;EACzCC,MAAM;EACNC,QAAQ;EACRC,cAAc;EACdC,SAAS;EACT,GAAGC,KAAAA;AAC4B,CAAC,KAAkB;EAClD,MAAM,CAACC,WAAW,EAAEC,cAAc,CAAC,GAAGC,gBAAK,CAACC,QAAQ,CAAC,IAAI,CAAC,CAAA;AAC1D,EAAA,MAAMC,IAAI,GAAGF,gBAAK,CAACG,MAAM,CAAC;IAAER,cAAc;AAAEC,IAAAA,SAAAA;AAAU,GAAC,CAAC,CAAA;EAExDI,gBAAK,CAACI,SAAS,CAAC,MAAM;IACpBF,IAAI,CAACG,OAAO,GAAG;MAAEV,cAAc;AAAEC,MAAAA,SAAAA;KAAW,CAAA;AAC9C,GAAC,CAAC,CAAA;EAEFI,gBAAK,CAACI,SAAS,CAAC,MAAM;IACpB,IAAIE,OAAO,GAAG,KAAK,CAAA;IACnBP,cAAc,CAAC,IAAI,CAAC,CAAA;AACpB,IAAA,MAAM,CAACQ,WAAW,EAAEC,OAAO,CAAC,GAAGC,yCAAkB,CAAC;AAChD,MAAA,GAAGP,IAAI,CAACG,OAAO,CAACV,cAAc;AAC9Be,MAAAA,WAAW,EAAEjB,MAAAA;AACf,KAAC,CAAC,CAAA;IAEFe,OAAO,CAACG,IAAI,CAAC,YAAY;MACvB,IAAI,CAACL,OAAO,EAAE;QACZ,IAAI;AACF,UAAA,OAAMJ,IAAI,CAACG,OAAO,CAACT,SAAS,IAAA,IAAA,GAAA,KAAA,CAAA,GAAtBM,IAAI,CAACG,OAAO,CAACT,SAAS,EAAI,CAAA,CAAA;AAClC,SAAC,SAAS;UACRG,cAAc,CAAC,KAAK,CAAC,CAAA;AACvB,SAAA;AACF,OAAA;AACF,KAAC,CAAC,CAAA;AAEF,IAAA,OAAO,MAAM;AACXO,MAAAA,OAAO,GAAG,IAAI,CAAA;AACdC,MAAAA,WAAW,EAAE,CAAA;KACd,CAAA;AACH,GAAC,EAAE,CAACd,MAAM,CAAC,CAAC,CAAA;AAEZ,EAAA,oBACEO,gBAAA,CAAAY,aAAA,CAACC,8BAAmB,EAAAC,iCAAA,CAAA;AAACrB,IAAAA,MAAM,EAAEA,MAAAA;AAAO,GAAA,EAAKI,KAAK,CAC5CG,eAAAA,gBAAA,CAAAY,aAAA,CAACG,8BAAmB,EAAA;AAACC,IAAAA,KAAK,EAAElB,WAAAA;GAAcJ,EAAAA,QAA8B,CACrD,CAAC,CAAA;AAE1B;;;;"}

package/build/lib/{PersistQueryClientProvider.esm.js → PersistQueryClientProvider.legacy.js}

@@ -1,4 +1,5 @@
-import { extends as _extends } from './_virtual/_rollupPluginBabelHelpers.esm.js';
+import { extends as _extends } from './_virtual/_rollupPluginBabelHelpers.legacy.js';
+import 'client-only';
 import * as React from 'react';
 import { persistQueryClient } from '@tanstack/query-persist-client-core';
 import { QueryClientProvider, IsRestoringProvider } from '@tanstack/react-query';
@@ -28,10 +29,13 @@ const PersistQueryClientProvider = ({
       ...refs.current.persistOptions,
       queryClient: client
     });
-    promise.then(() => {
+    promise.then(async () => {
       if (!isStale) {
-        refs.current.onSuccess == null ? void 0 : refs.current.onSuccess();
-        setIsRestoring(false);
+        try {
+          await (refs.current.onSuccess == null ? void 0 : refs.current.onSuccess());
+        } finally {
+          setIsRestoring(false);
+        }
       }
     });
     return () => {
@@ -47,4 +51,4 @@ const PersistQueryClientProvider = ({
 };
 
 export { PersistQueryClientProvider };
-//# sourceMappingURL=PersistQueryClientProvider.esm.js.map
+//# sourceMappingURL=PersistQueryClientProvider.legacy.js.map

package/build/lib/PersistQueryClientProvider.legacy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"PersistQueryClientProvider.legacy.js","sources":["../../src/PersistQueryClientProvider.tsx"],"sourcesContent":["import 'client-only'\nimport * as React from 'react'\n\nimport type { PersistQueryClientOptions } from '@tanstack/query-persist-client-core'\nimport { persistQueryClient } from '@tanstack/query-persist-client-core'\nimport type { QueryClientProviderProps } from '@tanstack/react-query'\nimport { QueryClientProvider, IsRestoringProvider } from '@tanstack/react-query'\n\nexport type PersistQueryClientProviderProps = QueryClientProviderProps & {\n  persistOptions: Omit<PersistQueryClientOptions, 'queryClient'>\n  onSuccess?: () => Promise<unknown> | unknown\n}\n\nexport const PersistQueryClientProvider = ({\n  client,\n  children,\n  persistOptions,\n  onSuccess,\n  ...props\n}: PersistQueryClientProviderProps): JSX.Element => {\n  const [isRestoring, setIsRestoring] = React.useState(true)\n  const refs = React.useRef({ persistOptions, onSuccess })\n\n  React.useEffect(() => {\n    refs.current = { persistOptions, onSuccess }\n  })\n\n  React.useEffect(() => {\n    let isStale = false\n    setIsRestoring(true)\n    const [unsubscribe, promise] = persistQueryClient({\n      ...refs.current.persistOptions,\n      queryClient: client,\n    })\n\n    promise.then(async () => {\n      if (!isStale) {\n        try {\n          await refs.current.onSuccess?.()\n        } finally {\n          setIsRestoring(false)\n        }\n      }\n    })\n\n    return () => {\n      isStale = true\n      unsubscribe()\n    }\n  }, [client])\n\n  return (\n    <QueryClientProvider client={client} {...props}>\n      <IsRestoringProvider value={isRestoring}>{children}</IsRestoringProvider>\n    </QueryClientProvider>\n  )\n}\n"],"names":["PersistQueryClientProvider","client","children","persistOptions","onSuccess","props","isRestoring","setIsRestoring","React","useState","refs","useRef","useEffect","current","isStale","unsubscribe","promise","persistQueryClient","queryClient","then","createElement","QueryClientProvider","_extends","IsRestoringProvider","value"],"mappings":";;;;;;AAaO,MAAMA,0BAA0B,GAAGA,CAAC;EACzCC,MAAM;EACNC,QAAQ;EACRC,cAAc;EACdC,SAAS;EACT,GAAGC,KAAAA;AAC4B,CAAC,KAAkB;EAClD,MAAM,CAACC,WAAW,EAAEC,cAAc,CAAC,GAAGC,KAAK,CAACC,QAAQ,CAAC,IAAI,CAAC,CAAA;AAC1D,EAAA,MAAMC,IAAI,GAAGF,KAAK,CAACG,MAAM,CAAC;IAAER,cAAc;AAAEC,IAAAA,SAAAA;AAAU,GAAC,CAAC,CAAA;EAExDI,KAAK,CAACI,SAAS,CAAC,MAAM;IACpBF,IAAI,CAACG,OAAO,GAAG;MAAEV,cAAc;AAAEC,MAAAA,SAAAA;KAAW,CAAA;AAC9C,GAAC,CAAC,CAAA;EAEFI,KAAK,CAACI,SAAS,CAAC,MAAM;IACpB,IAAIE,OAAO,GAAG,KAAK,CAAA;IACnBP,cAAc,CAAC,IAAI,CAAC,CAAA;AACpB,IAAA,MAAM,CAACQ,WAAW,EAAEC,OAAO,CAAC,GAAGC,kBAAkB,CAAC;AAChD,MAAA,GAAGP,IAAI,CAACG,OAAO,CAACV,cAAc;AAC9Be,MAAAA,WAAW,EAAEjB,MAAAA;AACf,KAAC,CAAC,CAAA;IAEFe,OAAO,CAACG,IAAI,CAAC,YAAY;MACvB,IAAI,CAACL,OAAO,EAAE;QACZ,IAAI;AACF,UAAA,OAAMJ,IAAI,CAACG,OAAO,CAACT,SAAS,IAAA,IAAA,GAAA,KAAA,CAAA,GAAtBM,IAAI,CAACG,OAAO,CAACT,SAAS,EAAI,CAAA,CAAA;AAClC,SAAC,SAAS;UACRG,cAAc,CAAC,KAAK,CAAC,CAAA;AACvB,SAAA;AACF,OAAA;AACF,KAAC,CAAC,CAAA;AAEF,IAAA,OAAO,MAAM;AACXO,MAAAA,OAAO,GAAG,IAAI,CAAA;AACdC,MAAAA,WAAW,EAAE,CAAA;KACd,CAAA;AACH,GAAC,EAAE,CAACd,MAAM,CAAC,CAAC,CAAA;AAEZ,EAAA,oBACEO,KAAA,CAAAY,aAAA,CAACC,mBAAmB,EAAAC,QAAA,CAAA;AAACrB,IAAAA,MAAM,EAAEA,MAAAA;AAAO,GAAA,EAAKI,KAAK,CAC5CG,eAAAA,KAAA,CAAAY,aAAA,CAACG,mBAAmB,EAAA;AAACC,IAAAA,KAAK,EAAElB,WAAAA;GAAcJ,EAAAA,QAA8B,CACrD,CAAC,CAAA;AAE1B;;;;"}

package/build/lib/__tests__/PersistQueryClientProvider.test.d.ts

@@ -1 +1,2 @@
 export {};
+//# sourceMappingURL=PersistQueryClientProvider.test.d.ts.map

package/build/lib/__tests__/PersistQueryClientProvider.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"PersistQueryClientProvider.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/PersistQueryClientProvider.test.tsx"],"names":[],"mappings":""}

package/build/lib/__tests__/utils.d.ts

@@ -1,7 +1,8 @@
 /// <reference types="node" />
-import type { QueryClientConfig } from '@tanstack/query-core';
-import { QueryClient } from '@tanstack/query-core';
+import type { QueryClientConfig } from '@tanstack/react-query';
+import { QueryClient } from '@tanstack/react-query';
 export declare function createQueryClient(config?: QueryClientConfig): QueryClient;
 export declare function queryKey(): Array<string>;
 export declare function sleep(timeout: number): Promise<void>;
 export declare function setActTimeout(fn: () => void, ms?: number): NodeJS.Timeout;
+//# sourceMappingURL=utils.d.ts.map

package/build/lib/__tests__/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/__tests__/utils.ts"],"names":[],"mappings":";AAEA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAA;AAC9D,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAA;AAEnD,wBAAgB,iBAAiB,CAAC,MAAM,CAAC,EAAE,iBAAiB,GAAG,WAAW,CAEzE;AAGD,wBAAgB,QAAQ,IAAI,KAAK,CAAC,MAAM,CAAC,CAGxC;AAED,wBAAgB,KAAK,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAIpD;AAED,wBAAgB,aAAa,CAAC,EAAE,EAAE,MAAM,IAAI,EAAE,EAAE,CAAC,EAAE,MAAM,kBAMxD"}

package/build/lib/_virtual/{_rollupPluginBabelHelpers.mjs → _rollupPluginBabelHelpers.cjs}

@@ -1,3 +1,5 @@
+'use strict';
+
 function _extends() {
   _extends = Object.assign ? Object.assign.bind() : function (target) {
     for (var i = 1; i < arguments.length; i++) {
@@ -13,5 +15,5 @@ function _extends() {
   return _extends.apply(this, arguments);
 }
 
-export { _extends as extends };
-//# sourceMappingURL=_rollupPluginBabelHelpers.mjs.map
+exports.extends = _extends;
+//# sourceMappingURL=_rollupPluginBabelHelpers.cjs.map

package/build/lib/_virtual/_rollupPluginBabelHelpers.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"_rollupPluginBabelHelpers.cjs","sources":[],"sourcesContent":[],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;"}

package/build/lib/_virtual/_rollupPluginBabelHelpers.js

@@ -1,5 +1,3 @@
-'use strict';
-
 function _extends() {
   _extends = Object.assign ? Object.assign.bind() : function (target) {
     for (var i = 1; i < arguments.length; i++) {
@@ -15,5 +13,5 @@ function _extends() {
   return _extends.apply(this, arguments);
 }
 
-exports.extends = _extends;
+export { _extends as extends };
 //# sourceMappingURL=_rollupPluginBabelHelpers.js.map

package/build/lib/_virtual/_rollupPluginBabelHelpers.js.map

@@ -1 +1 @@
-{"version":3,"file":"_rollupPluginBabelHelpers.js","sources":[],"sourcesContent":[],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;"}
+{"version":3,"file":"_rollupPluginBabelHelpers.js","sources":[],"sourcesContent":[],"names":[],"mappings":";;;;;;;;;;;;;;;;;"}

package/build/lib/_virtual/_rollupPluginBabelHelpers.legacy.cjs

@@ -0,0 +1,19 @@
+'use strict';
+
+function _extends() {
+  _extends = Object.assign ? Object.assign.bind() : function (target) {
+    for (var i = 1; i < arguments.length; i++) {
+      var source = arguments[i];
+      for (var key in source) {
+        if (Object.prototype.hasOwnProperty.call(source, key)) {
+          target[key] = source[key];
+        }
+      }
+    }
+    return target;
+  };
+  return _extends.apply(this, arguments);
+}
+
+exports.extends = _extends;
+//# sourceMappingURL=_rollupPluginBabelHelpers.legacy.cjs.map

package/build/lib/_virtual/_rollupPluginBabelHelpers.legacy.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"_rollupPluginBabelHelpers.legacy.cjs","sources":[],"sourcesContent":[],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;"}

package/build/lib/_virtual/{_rollupPluginBabelHelpers.esm.js → _rollupPluginBabelHelpers.legacy.js}

@@ -14,4 +14,4 @@ function _extends() {
 }
 
 export { _extends as extends };
-//# sourceMappingURL=_rollupPluginBabelHelpers.esm.js.map
+//# sourceMappingURL=_rollupPluginBabelHelpers.legacy.js.map

package/build/lib/_virtual/_rollupPluginBabelHelpers.legacy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"_rollupPluginBabelHelpers.legacy.js","sources":[],"sourcesContent":[],"names":[],"mappings":";;;;;;;;;;;;;;;;;"}

package/build/lib/index.cjs

@@ -0,0 +1,15 @@
+'use strict';
+
+var queryPersistClientCore = require('@tanstack/query-persist-client-core');
+var PersistQueryClientProvider = require('./PersistQueryClientProvider.cjs');
+
+
+
+exports.PersistQueryClientProvider = PersistQueryClientProvider.PersistQueryClientProvider;
+Object.keys(queryPersistClientCore).forEach(function (k) {
+	if (k !== 'default' && !exports.hasOwnProperty(k)) Object.defineProperty(exports, k, {
+		enumerable: true,
+		get: function () { return queryPersistClientCore[k]; }
+	});
+});
+//# sourceMappingURL=index.cjs.map

package/build/lib/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.cjs","sources":[],"sourcesContent":[],"names":[],"mappings":";;;;;;;;;;;;;"}

package/build/lib/index.d.ts

@@ -1,2 +1,3 @@
 export * from '@tanstack/query-persist-client-core';
 export * from './PersistQueryClientProvider';
+//# sourceMappingURL=index.d.ts.map

package/build/lib/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AACA,cAAc,qCAAqC,CAAA;AAEnD,cAAc,8BAA8B,CAAA"}

package/build/lib/index.js

@@ -1,15 +1,3 @@
-'use strict';
-
-var queryPersistClientCore = require('@tanstack/query-persist-client-core');
-var PersistQueryClientProvider = require('./PersistQueryClientProvider.js');
-
-
-
-exports.PersistQueryClientProvider = PersistQueryClientProvider.PersistQueryClientProvider;
-Object.keys(queryPersistClientCore).forEach(function (k) {
-	if (k !== 'default' && !exports.hasOwnProperty(k)) Object.defineProperty(exports, k, {
-		enumerable: true,
-		get: function () { return queryPersistClientCore[k]; }
-	});
-});
+export * from '@tanstack/query-persist-client-core';
+export { PersistQueryClientProvider } from './PersistQueryClientProvider.js';
 //# sourceMappingURL=index.js.map

package/build/lib/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sources":[],"sourcesContent":[],"names":[],"mappings":";;;;;;;;;;;;;"}
+{"version":3,"file":"index.js","sources":[],"sourcesContent":[],"names":[],"mappings":";"}

package/build/lib/index.legacy.cjs

@@ -0,0 +1,15 @@
+'use strict';
+
+var queryPersistClientCore = require('@tanstack/query-persist-client-core');
+var PersistQueryClientProvider = require('./PersistQueryClientProvider.legacy.cjs');
+
+
+
+exports.PersistQueryClientProvider = PersistQueryClientProvider.PersistQueryClientProvider;
+Object.keys(queryPersistClientCore).forEach(function (k) {
+	if (k !== 'default' && !exports.hasOwnProperty(k)) Object.defineProperty(exports, k, {
+		enumerable: true,
+		get: function () { return queryPersistClientCore[k]; }
+	});
+});
+//# sourceMappingURL=index.legacy.cjs.map

package/build/lib/index.legacy.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.legacy.cjs","sources":[],"sourcesContent":[],"names":[],"mappings":";;;;;;;;;;;;;"}

package/build/lib/{index.esm.js → index.legacy.js}

@@ -1,3 +1,3 @@
 export * from '@tanstack/query-persist-client-core';
-export { PersistQueryClientProvider } from './PersistQueryClientProvider.esm.js';
-//# sourceMappingURL=index.esm.js.map
+export { PersistQueryClientProvider } from './PersistQueryClientProvider.legacy.js';
+//# sourceMappingURL=index.legacy.js.map

package/build/lib/index.legacy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.legacy.js","sources":[],"sourcesContent":[],"names":[],"mappings":";"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tanstack/react-query-persist-client",
-  "version": "5.0.0-alpha.7",
+  "version": "5.0.0-alpha.70",
   "description": "React bindings to work with persisters in TanStack/react-query",
   "author": "tannerlinsley",
   "license": "MIT",
@@ -10,42 +10,47 @@
     "type": "github",
     "url": "https://github.com/sponsors/tannerlinsley"
   },
+  "type": "module",
   "types": "build/lib/index.d.ts",
-  "main": "build/lib/index.js",
-  "module": "build/lib/index.esm.js",
+  "main": "build/lib/index.legacy.cjs",
+  "module": "build/lib/index.legacy.js",
   "exports": {
     ".": {
       "types": "./build/lib/index.d.ts",
-      "import": "./build/lib/index.mjs",
-      "default": "./build/lib/index.js"
+      "import": "./build/lib/index.js",
+      "require": "./build/lib/index.cjs",
+      "default": "./build/lib/index.cjs"
     },
     "./package.json": "./package.json"
   },
   "sideEffects": false,
   "files": [
     "build/lib/*",
-    "build/umd/*",
     "src"
   ],
+  "dependencies": {
+    "client-only": "0.0.1",
+    "@tanstack/query-persist-client-core": "5.0.0-alpha.70"
+  },
   "devDependencies": {
-    "@types/react": "^18.0.14",
-    "@types/react-dom": "^18.0.5",
+    "@types/react": "^18.2.4",
+    "@types/react-dom": "^18.2.4",
     "react": "^18.2.0",
     "react-dom": "^18.2.0",
-    "@tanstack/react-query": "5.0.0-alpha.7"
-  },
-  "dependencies": {
-    "@tanstack/query-persist-client-core": "5.0.0-alpha.7"
+    "@tanstack/react-query": "5.0.0-alpha.70"
   },
   "peerDependencies": {
-    "@tanstack/react-query": "5.0.0-alpha.7"
+    "@tanstack/react-query": "5.0.0-alpha.70"
   },
   "scripts": {
-    "clean": "rimraf ./build",
+    "clean": "rimraf ./build && rimraf ./coverage",
     "test:eslint": "eslint --ext .ts,.tsx ./src",
-    "test:types": "tsc",
+    "test:types": "tsc --noEmit",
     "test:lib": "vitest run --coverage",
     "test:lib:dev": "pnpm run test:lib --watch",
-    "build:types": "tsc --build"
+    "test:build": "publint --strict",
+    "build": "pnpm build:rollup && pnpm build:types",
+    "build:rollup": "rollup --config rollup.config.js",
+    "build:types": "tsc --emitDeclarationOnly"
   }
 }

package/src/PersistQueryClientProvider.tsx

@@ -1,4 +1,4 @@
-'use client'
+import 'client-only'
 import * as React from 'react'
 
 import type { PersistQueryClientOptions } from '@tanstack/query-persist-client-core'
@@ -8,7 +8,7 @@ import { QueryClientProvider, IsRestoringProvider } from '@tanstack/react-query'
 
 export type PersistQueryClientProviderProps = QueryClientProviderProps & {
   persistOptions: Omit<PersistQueryClientOptions, 'queryClient'>
-  onSuccess?: () => void
+  onSuccess?: () => Promise<unknown> | unknown
 }
 
 export const PersistQueryClientProvider = ({
@@ -33,10 +33,13 @@ export const PersistQueryClientProvider = ({
       queryClient: client,
     })
 
-    promise.then(() => {
+    promise.then(async () => {
       if (!isStale) {
-        refs.current.onSuccess?.()
-        setIsRestoring(false)
+        try {
+          await refs.current.onSuccess?.()
+        } finally {
+          setIsRestoring(false)
+        }
       }
     })