convert useLoadDraft() hook to regular async function

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

src/view/com/composer/Composer.tsx

@@ -131,7 +131,7 @@ import {useAnalytics} from '#/analytics'
 import {IS_ANDROID, IS_IOS, IS_NATIVE, IS_WEB} from '#/env'
 import {BottomSheetPortalProvider} from '../../../../modules/bottom-sheet'
 import {draftToComposerPosts} from './drafts/state/api'
-import {useLoadDraft, useSaveDraftMutation} from './drafts/state/queries'
+import {loadDraft, useSaveDraftMutation} from './drafts/state/queries'
 import {type DraftSummary} from './drafts/state/schema'
 import {PostLanguageSelect} from './select-language/PostLanguageSelect'
 import {
@@ -193,7 +193,6 @@ export const ComposePost = ({
   const discardPromptControl = Prompt.usePromptControl()
   const {mutateAsync: saveDraft, isPending: _isSavingDraft} =
     useSaveDraftMutation()
-  const loadDraft = useLoadDraft()
   const {closeAllDialogs} = useDialogStateControlContext()
   const {closeAllModals} = useModalControls()
   const {data: preferences} = usePreferencesQuery()
@@ -343,7 +342,7 @@ export const ComposePost = ({
         loadedMedia,
       })
     },
-    [loadDraft, composerDispatch],
+    [composerDispatch],
   )
 
   const [publishOnUpload, setPublishOnUpload] = useState(false)

src/view/com/composer/drafts/state/queries.ts

@@ -1,4 +1,3 @@
-import {useCallback} from 'react'
 import {AppBskyDraftCreateDraft, type AppBskyDraftDefs} from '@atproto/api'
 import {
   useInfiniteQuery,
@@ -40,16 +39,12 @@ export function useDraftsQuery() {
 }
 
 /**
- * Hook to load a draft's local media for editing.
+ * Load a draft's local media for editing.
  * Takes the full Draft object (from DraftSummary) to avoid re-fetching.
  */
-export function useLoadDraft() {
+export async function loadDraft(draft: AppBskyDraftDefs.Draft): Promise<{
-  return useCallback(
-    async (
-      draft: AppBskyDraftDefs.Draft,
-    ): Promise<{
   loadedMedia: Map<string, string>
-    }> => {
+}> {
   // Load local media files
   const loadedMedia = new Map<string, string>()
   for (const post of draft.posts) {
@@ -84,9 +79,6 @@ export function useLoadDraft() {
   }
 
   return {loadedMedia}
-    },
-    [],
-  )
 }
 
 /**
@@ -158,12 +150,15 @@ export function useDeleteDraftMutation() {
   return useMutation({
     mutationFn: async ({
       draftId,
-      draft,
     }: {
       draftId: string
       draft: AppBskyDraftDefs.Draft
     }) => {
-      // Delete local media files
+      // Delete from server first - if this fails, we keep local media for retry
+      await agent.app.bsky.draft.deleteDraft({id: draftId})
+    },
+    onSuccess: async (_, {draft}) => {
+      // Only delete local media after server deletion succeeds
       for (const post of draft.posts) {
         if (post.embedImages) {
           for (const img of post.embedImages) {
@@ -176,11 +171,6 @@ export function useDeleteDraftMutation() {
           }
         }
       }
-
-      // Delete from server
-      await agent.app.bsky.draft.deleteDraft({id: draftId})
-    },
-    onSuccess: () => {
       queryClient.invalidateQueries({queryKey: DRAFTS_QUERY_KEY})
     },
   })