[Chat] Add chat invite as message embed type (#10728)

src/state/queries/join-links.ts

@@ -1,4 +1,9 @@
-import {AtpAgent} from '@atproto/api'
+import {useCallback} from 'react'
+import {
+  AtpAgent,
+  type ChatBskyGroupDefs,
+  type ChatBskyGroupGetJoinLinkPreviews,
+} from '@atproto/api'
 import {useQuery, useQueryClient} from '@tanstack/react-query'
 
 import {CHAT_SERVICE, DM_SERVICE_HEADERS} from '#/lib/constants'
@@ -17,14 +22,39 @@ export const createJoinLinkPreviewQueryKey = (args: {
     persistedVersion: 1,
   })
 
+async function fetchJoinLinkPreviews({
+  agent,
+  codes,
+  hasSession,
+}: {
+  agent: AtpAgent
+  codes: string[]
+  hasSession: boolean
+}) {
+  const previewAgent = new AtpAgent({service: CHAT_SERVICE})
+  const res = hasSession
+    ? await agent.chat.bsky.group.getJoinLinkPreviews(
+        {codes},
+        {headers: DM_SERVICE_HEADERS},
+      )
+    : await previewAgent.chat.bsky.group.getJoinLinkPreviews({codes})
+  return res.data
+}
+
 export function useJoinLinkPreviewsQuery({
   codes,
   hasSession,
   staleTime = STALE.MINUTES.ONE,
+  initialData,
 }: {
   codes?: string[]
   hasSession: boolean
   staleTime?: number
+  /**
+   * Seed the query with an already-known preview (e.g. a DM message embed
+   * already carries the resolved view), avoiding a duplicate fetch.
+   */
+  initialData?: ChatBskyGroupGetJoinLinkPreviews.OutputSchema
 }) {
   const agent = useAgent()
 
@@ -33,14 +63,7 @@ export function useJoinLinkPreviewsQuery({
     queryFn: async () => {
       if (!codes) throw new Error('No invite code')
       try {
-        const previewAgent = new AtpAgent({service: CHAT_SERVICE})
-        const res = hasSession
-          ? await agent.chat.bsky.group.getJoinLinkPreviews(
-              {codes},
-              {headers: DM_SERVICE_HEADERS},
-            )
-          : await previewAgent.chat.bsky.group.getJoinLinkPreviews({codes})
-        return res.data
+        return await fetchJoinLinkPreviews({agent, codes, hasSession})
       } catch (error) {
         logger.error('Failed to fetch join link preview', {safeMessage: error})
         throw error
@@ -48,6 +71,7 @@ export function useJoinLinkPreviewsQuery({
     },
     enabled: codes != null && codes.length > 0,
     staleTime,
+    initialData,
   })
 }
 
@@ -58,17 +82,42 @@ export function usePrefetchJoinLinkPreviews() {
   return ({codes, hasSession}: {codes: string[]; hasSession: boolean}) => {
     return queryClient.prefetchQuery({
       queryKey: createJoinLinkPreviewQueryKey({codes, hasSession}),
-      queryFn: async () => {
-        const previewAgent = new AtpAgent({service: CHAT_SERVICE})
-        const res = hasSession
-          ? await agent.chat.bsky.group.getJoinLinkPreviews(
-              {codes},
-              {headers: DM_SERVICE_HEADERS},
-            )
-          : await previewAgent.chat.bsky.group.getJoinLinkPreviews({codes})
-        return res.data
-      },
+      queryFn: () => fetchJoinLinkPreviews({agent, codes, hasSession}),
       staleTime: STALE.MINUTES.ONE,
     })
   }
 }
+
+/**
+ * Imperatively fetch (or read from cache) a single join link preview by code.
+ * Used when sending a DM invite embed so we can build an optimistic view.
+ * Returns undefined if the preview can't be resolved.
+ */
+export function useGetJoinLinkPreview() {
+  const agent = useAgent()
+  const queryClient = useQueryClient()
+
+  return useCallback(
+    async ({
+      code,
+      hasSession,
+    }: {
+      code: string
+      hasSession: boolean
+    }): Promise<ChatBskyGroupDefs.JoinLinkPreviewView | undefined> => {
+      try {
+        const data = await queryClient.fetchQuery({
+          queryKey: createJoinLinkPreviewQueryKey({codes: [code], hasSession}),
+          queryFn: () =>
+            fetchJoinLinkPreviews({agent, codes: [code], hasSession}),
+          staleTime: STALE.MINUTES.ONE,
+        })
+        return data.joinLinkPreviews[0]
+      } catch (error) {
+        logger.error('Failed to fetch join link preview', {safeMessage: error})
+        return undefined
+      }
+    },
+    [agent, queryClient],
+  )
+}