@better-auth-ui/core/src/plugins/organization/invitation-query.ts

Authentication components and data utilities for [Better Auth](https://better-auth.com), available for React and Solid.

import type { QueryClient, QueryOptions } from "@tanstack/query-core"
import type { InferData } from "../../lib/auth-client"
import { createAuthQueryFetchOptions } from "../../lib/auth-query-retry"
import type { OrganizationAuthClient } from "./organization-auth-client"
import { organizationQueryKeys } from "./organization-query-keys"

export type InvitationData<
  TAuthClient extends OrganizationAuthClient = OrganizationAuthClient
> = InferData<TAuthClient["organization"]["getInvitation"]>

export type InvitationParams<
  TAuthClient extends OrganizationAuthClient = OrganizationAuthClient
> = Parameters<TAuthClient["organization"]["getInvitation"]>[0]

export type InvitationOptions<
  TAuthClient extends OrganizationAuthClient = OrganizationAuthClient
> = Omit<QueryOptions<InvitationData<TAuthClient>>, "queryKey"> &
  InvitationParams<TAuthClient>

export function invitationOptions<TAuthClient extends OrganizationAuthClient>(
  authClient: TAuthClient,
  userId: string | undefined,
  params: InvitationParams<TAuthClient>
) {
  type TData = InvitationData<TAuthClient>
  const queryKey = organizationQueryKeys.invitations.detail(
    userId,
    params.query
  )

  return {
    queryKey,
    queryFn: ({ signal }: { signal: AbortSignal }) =>
      authClient.organization.getInvitation({
        ...params,
        fetchOptions: createAuthQueryFetchOptions(params.fetchOptions, signal)
      }) as Promise<TData>
  } satisfies QueryOptions<TData>
}

export const ensureInvitation = <TAuthClient extends OrganizationAuthClient>(
  queryClient: QueryClient,
  authClient: TAuthClient,
  userId: string | undefined,
  options: InvitationOptions<TAuthClient>
) => {
  const { query, fetchOptions, ...queryOptions } = options

  return queryClient.ensureQueryData({
    ...invitationOptions(authClient, userId, { query, fetchOptions }),
    ...queryOptions
  })
}

export const prefetchInvitation = <TAuthClient extends OrganizationAuthClient>(
  queryClient: QueryClient,
  authClient: TAuthClient,
  userId: string | undefined,
  options: InvitationOptions<TAuthClient>
) => {
  const { query, fetchOptions, ...queryOptions } = options

  return queryClient.prefetchQuery({
    ...invitationOptions(authClient, userId, { query, fetchOptions }),
    ...queryOptions
  })
}

export const fetchInvitation = <TAuthClient extends OrganizationAuthClient>(
  queryClient: QueryClient,
  authClient: TAuthClient,
  userId: string | undefined,
  options: InvitationOptions<TAuthClient>
) => {
  const { query, fetchOptions, ...queryOptions } = options

  return queryClient.fetchQuery({
    ...invitationOptions(authClient, userId, { query, fetchOptions }),
    ...queryOptions
  })
}