@daveyplate/better-auth-tanstack/src/lib/prefetch-session.ts

Tanstack hooks for better-auth

import type { AnyUseQueryOptions, QueryClient } from "@tanstack/react-query"

import type { AuthClient } from "../types/auth-client"
import type { AuthQueryOptions } from "./auth-query-provider"

export async function prefetchSession<TAuthClient extends AuthClient>(
    authClient: TAuthClient,
    queryClient: QueryClient,
    queryOptions?: AuthQueryOptions,
    options?: Partial<AnyUseQueryOptions>
) {
    const { error, data } = await authClient.getSession()

    const mergedOptions = {
        ...queryOptions?.queryOptions,
        ...queryOptions?.sessionQueryOptions,
        ...options
    }

    await queryClient.prefetchQuery({
        ...mergedOptions,
        queryKey: queryOptions?.sessionKey,
        queryFn: () => data as SessionData
    })

    type SessionData = TAuthClient["$Infer"]["Session"] | undefined
    type User = TAuthClient["$Infer"]["Session"]["user"] | undefined
    type Session = TAuthClient["$Infer"]["Session"]["session"] | undefined

    return {
        error,
        data: data as SessionData,
        session: data?.session as Session,
        user: data?.user as User
    }
}