Adjust rte to async translate function

vivid-planet · Dec 5, 2023 · 216df08 · 216df08
1 parent c14f915
commit 216df08
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/packages/admin/admin-rte/src/core/translation/ToolbarButton.tsx b/packages/admin/admin-rte/src/core/translation/ToolbarButton.tsx
@@ -1,4 +1,4 @@
-import { useTranslationConfig } from "@comet/admin";
+import { useTranslationConfig } from "@Comet/admin";
 import TranslateIcon from "@mui/icons-material/Translate";
 import Tooltip from "@mui/material/Tooltip";
 import { convertToRaw } from "draft-js";
@@ -13,17 +13,18 @@ import { translateAndTransformXmlToState } from "./xml/translateAndTransformToSt
 function ToolbarButton({ editorState, setEditorState }: IControlProps): React.ReactElement {
     const { translate } = useTranslationConfig();
 
-    function handleClick(event: React.MouseEvent) {
+    async function handleClick(event: React.MouseEvent) {
         event.preventDefault();
 
         const contentState = editorState.getCurrentContent();
 
         const xml = transformStateToXml(contentState);
 
         if (translate) {
-            const translation = xml.map((item) => ({ original: item, replaceWith: translate(item) }));
+            const translationPromises = xml.map(async (item) => ({ original: item, replaceWith: await translate(item) }));
+            const translations = await Promise.all(translationPromises);
 
-            const translatedState = translateAndTransformXmlToState(contentState, convertToRaw(contentState), translation);
+            const translatedState = translateAndTransformXmlToState(contentState, convertToRaw(contentState), translations);
 
             setEditorState(translatedState);
         }