5304790ed5
Some checks are pending
E2E / Playwright e2e (push) Waiting to run
- Added SessionGuard component to manage session expiration and online status. - Updated AuthProvider to streamline session fetching and handling. - Introduced IdentityProvidersSection for managing OAuth, SAML, and LDAP identity providers. - Implemented identity provider guides for easier configuration. - Enhanced mail settings with infinite scroll option for improved user experience. - Updated global styles and layout components for better consistency across the application.
104 lines
2.7 KiB
TypeScript
104 lines
2.7 KiB
TypeScript
"use client"
|
|
|
|
import { useState } from "react"
|
|
import { QueryClient } from "@tanstack/react-query"
|
|
import {
|
|
PersistQueryClientProvider,
|
|
type PersistedClient,
|
|
} from "@tanstack/react-query-persist-client"
|
|
import { openDB, type IDBPDatabase } from "idb"
|
|
import type { Persister } from "@tanstack/react-query-persist-client"
|
|
import {
|
|
isPreviewThumbQueryKey,
|
|
revokePreviewBlobData,
|
|
} from "@/lib/api/preview-blob-url"
|
|
import { ApiRequestError } from "@/lib/api/client"
|
|
|
|
const DB_NAME = "ultimail-query-cache"
|
|
const STORE_NAME = "query-cache"
|
|
|
|
let dbPromise: Promise<IDBPDatabase> | null = null
|
|
|
|
function getDb() {
|
|
if (!dbPromise) {
|
|
dbPromise = openDB(DB_NAME, 1, {
|
|
upgrade(db) {
|
|
db.createObjectStore(STORE_NAME)
|
|
},
|
|
})
|
|
}
|
|
return dbPromise
|
|
}
|
|
|
|
const idbPersister: Persister = {
|
|
persistClient: async (client: PersistedClient) => {
|
|
const db = await getDb()
|
|
await db.put(STORE_NAME, client, "cache")
|
|
},
|
|
restoreClient: async (): Promise<PersistedClient | undefined> => {
|
|
const db = await getDb()
|
|
const restored = await db.get<PersistedClient>(STORE_NAME, "cache")
|
|
if (!restored?.clientState?.queries) return restored
|
|
restored.clientState.queries = restored.clientState.queries.filter(
|
|
(entry) => !isPreviewThumbQueryKey(entry.queryKey)
|
|
)
|
|
return restored
|
|
},
|
|
removeClient: async () => {
|
|
const db = await getDb()
|
|
await db.delete(STORE_NAME, "cache")
|
|
},
|
|
}
|
|
|
|
function attachPreviewBlobGc(queryClient: QueryClient) {
|
|
return queryClient.getQueryCache().subscribe((event) => {
|
|
if (event.type === "removed") {
|
|
revokePreviewBlobData(event.query.state.data)
|
|
}
|
|
})
|
|
}
|
|
|
|
function makeQueryClient() {
|
|
return new QueryClient({
|
|
defaultOptions: {
|
|
queries: {
|
|
gcTime: 1000 * 60 * 60 * 24,
|
|
staleTime: 1000 * 60 * 5,
|
|
networkMode: "offlineFirst",
|
|
retry: (failureCount, error) => {
|
|
if (error instanceof ApiRequestError && error.status === 401) {
|
|
return false
|
|
}
|
|
return failureCount < 3
|
|
},
|
|
},
|
|
mutations: {
|
|
networkMode: "offlineFirst",
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
export function QueryProvider({ children }: { children: React.ReactNode }) {
|
|
const [queryClient] = useState(() => {
|
|
const client = makeQueryClient()
|
|
attachPreviewBlobGc(client)
|
|
return client
|
|
})
|
|
|
|
return (
|
|
<PersistQueryClientProvider
|
|
client={queryClient}
|
|
persistOptions={{
|
|
persister: idbPersister,
|
|
dehydrateOptions: {
|
|
shouldDehydrateQuery: (query) =>
|
|
query.state.status === "success" && !isPreviewThumbQueryKey(query.queryKey),
|
|
},
|
|
}}
|
|
>
|
|
{children}
|
|
</PersistQueryClientProvider>
|
|
)
|
|
}
|