fix: wait entry content for translation

hyoban · hyoban · commit 1864548140f9 · 2025-06-11T22:49:17.000+08:00
diff --git a/packages/internal/store/src/translation/hooks.ts b/packages/internal/store/src/translation/hooks.ts
@@ -4,6 +4,7 @@ import { useQueries } from "@tanstack/react-query"
 import { useCallback } from "react"
 
 import { useEntryList } from "../entry/hooks"
+import type { EntryModel } from "../entry/types"
 import { translationSyncService, useTranslationStore } from "./store"
 
 export const usePrefetchEntryTranslation = ({
@@ -21,23 +22,29 @@ export const usePrefetchEntryTranslation = ({
   language: SupportedActionLanguage
   checkLanguage: (params: { content: string; language: SupportedActionLanguage }) => boolean
 }) => {
-  const entryList =
-    useEntryList(entryIds)
-      ?.filter((entry) => entry !== null && (translation || !!entry?.settings?.translation))
-      .map((entry) => entry!.id) || []
+  const entryList = (useEntryList(entryIds)?.filter(
+    (entry) => entry !== null && (translation || !!entry?.settings?.translation),
+  ) || []) as EntryModel[]
 
   return useQueries({
-    queries: entryList.map((entryId) => ({
-      queryKey: ["translation", entryId, language, withContent, target],
-      queryFn: () =>
-        translationSyncService.generateTranslation({
-          entryId,
-          language,
-          withContent,
-          target,
-          checkLanguage,
-        }),
-    })),
+    queries: entryList.map((entry) => {
+      const entryId = entry.id
+      const targetContent =
+        target === "readabilityContent" ? entry.readabilityContent : entry.content
+      const finalWithContent = withContent && !!targetContent
+
+      return {
+        queryKey: ["translation", entryId, language, finalWithContent, target],
+        queryFn: () =>
+          translationSyncService.generateTranslation({
+            entryId,
+            language,
+            withContent: finalWithContent,
+            target,
+            checkLanguage,
+          }),
+      }
+    }),
   })
 }
 
