import {
PersistedClient,
Persister,
} from "@tanstack/react-query-persist-client";
export const TANSTACK_QUERY_CACHE_KEY = "tanstack-query-cache-key";
// Declare chrome namespace for TypeScript
declare const chrome: {
storage: {
local: {
set: (items: Record<string, string>) => Promise<void>;
get: (keys: string | string[]) => Promise<Record<string, string>>;
remove: (keys: string | string[]) => Promise<void>;
};
};
};
/**
* Creates an AsyncStorage-like interface for Chrome's extension storage.
*
* @param storage The Chrome storage area to use (e.g., `chrome.storage.local`).
* @returns An object that mimics the AsyncStorage interface.
*/
export const createChromeStorage = (
storage: typeof chrome.storage.local = globalThis.chrome?.storage?.local,
): Persister => {
// Check if we are in a Chrome extension environment
if (typeof chrome === "undefined" || !chrome.storage) {
// Return a noop persister for non-extension environments
return {
persistClient: async () => {
return;
},
restoreClient: async () => undefined,
removeClient: async () => {
return;
},
};
}
return {
persistClient: async (client: PersistedClient) => {
await storage.set({ [TANSTACK_QUERY_CACHE_KEY]: JSON.stringify(client) });
},
restoreClient: async () => {
const result = await storage.get(TANSTACK_QUERY_CACHE_KEY);
return result[TANSTACK_QUERY_CACHE_KEY]
? JSON.parse(result[TANSTACK_QUERY_CACHE_KEY])
: undefined;
},
removeClient: async () => {
await storage.remove(TANSTACK_QUERY_CACHE_KEY);
},
};
};
