Add endpointClient to SeamQueryContext

Author: razor-x

src/lib/seam/SeamQueryProvider.tsx

@@ -1,5 +1,6 @@
 import type {
   SeamHttp,
+  SeamHttpEndpoints,
   SeamHttpOptionsWithClientSessionToken,
 } from '@seamapi/http/connect'
 import { QueryClient, QueryClientProvider } from '@tanstack/react-query'
@@ -15,6 +16,7 @@ import { useSeamClient } from './use-seam-client.js'
 
 export interface SeamQueryContext {
   client: SeamHttp | null
+  endpointClient: SeamHttpEndpoints | null
   clientOptions?: SeamQueryProviderClientOptions | undefined
   publishableKey?: string | undefined
   userIdentifierKey?: string | undefined
@@ -112,21 +114,24 @@ function Session({
 const createDefaultSeamQueryContextValue = (): SeamQueryContext => {
   try {
     if (globalThis.seam == null) {
-      return { client: null }
+      return { client: null, endpointClient: null }
     }
     return createSeamQueryContextValue(globalThis.seam)
   } catch (err) {
     // eslint-disable-next-line no-console
     console.warn(err)
-    return { client: null }
+    return { client: null, endpointClient: null }
   }
 }
 
 const createSeamQueryContextValue = (
   options: SeamQueryProviderProps
 ): SeamQueryContext => {
   if (isSeamQueryProviderPropsWithClient(options)) {
-    return options
+    return {
+      ...options,
+      endpointClient: null,
+    }
   }
 
   if (isSeamQueryProviderPropsWithClientSessionToken(options)) {
@@ -135,6 +140,7 @@ const createSeamQueryContextValue = (
       clientSessionToken,
       clientOptions,
       client: null,
+      endpointClient: null,
     }
   }
 
@@ -145,10 +151,11 @@ const createSeamQueryContextValue = (
       userIdentifierKey,
       clientOptions,
       client: null,
+      endpointClient: null,
     }
   }
 
-  return { client: null }
+  return { client: null, endpointClient: null }
 }
 
 const defaultSeamQueryContextValue = createDefaultSeamQueryContextValue()

src/lib/seam/use-seam-client.ts

@@ -1,4 +1,4 @@
-import { SeamHttp } from '@seamapi/http/connect'
+import { SeamHttp, SeamHttpEndpoints } from '@seamapi/http/connect'
 import { useQuery } from '@tanstack/react-query'
 import { useEffect } from 'react'
 import { v4 as uuidv4 } from 'uuid'
@@ -7,6 +7,7 @@ import { useSeamQueryContext } from './SeamQueryProvider.js'
 
 export function useSeamClient(): {
   client: SeamHttp | null
+  endpointClient: SeamHttpEndpoints | null
   isPending: boolean
   isError: boolean
   error: unknown
@@ -22,7 +23,9 @@ export function useSeamClient(): {
     clientSessionToken != null ? '' : context.userIdentifierKey
   )
 
-  const { isPending, isError, error, data } = useQuery<SeamHttp>({
+  const { isPending, isError, error, data } = useQuery<
+    [SeamHttp, SeamHttpEndpoints]
+  >({
     queryKey: [
       'client',
       {
@@ -34,13 +37,19 @@ export function useSeamClient(): {
       },
     ],
     queryFn: async () => {
-      if (client != null) return client
+      if (client != null)
+        return [client, SeamHttpEndpoints.fromClient(client.client)]
 
       if (clientSessionToken != null) {
-        return SeamHttp.fromClientSessionToken(
+        const clientSessionTokenClient = SeamHttp.fromClientSessionToken(
           clientSessionToken,
           clientOptions
         )
+
+        return [
+          clientSessionTokenClient,
+          SeamHttpEndpoints.fromClient(clientSessionTokenClient.client),
+        ]
       }
 
       if (publishableKey == null) {
@@ -49,15 +58,25 @@ export function useSeamClient(): {
         )
       }
 
-      return await SeamHttp.fromPublishableKey(
+      const publishableKeyClient = await SeamHttp.fromPublishableKey(
         publishableKey,
         userIdentifierKey,
         clientOptions
       )
+      return [
+        publishableKeyClient,
+        SeamHttpEndpoints.fromClient(publishableKeyClient.client),
+      ]
     },
   })
 
-  return { client: data ?? null, isPending, isError, error }
+  return {
+    client: data?.[0] ?? null,
+    endpointClient: data?.[1] ?? null,
+    isPending,
+    isError,
+    error,
+  }
 }
 
 export class NullSeamClientError extends Error {