Extract to helper

buerokratt · Nov 19, 2024 · 965a794 · 965a794
1 parent d764fab
commit 965a794
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 34 deletions.
diff --git a/GUI/src/pages/Training/Intents/CommonIntents.tsx b/GUI/src/pages/Training/Intents/CommonIntents.tsx
@@ -32,6 +32,7 @@ import IntentExamplesTable from './IntentExamplesTable';
 import LoadingDialog from '../../../components/LoadingDialog';
 import ConnectServiceToIntentModal from 'pages/ConnectServiceToIntentModal';
 import withAuthorization, { ROLES } from 'hoc/with-authorization';
+import { saveCsv } from 'utils/save-csv';
 
 const CommonIntents: FC = () => {
   const { t } = useTranslation();
@@ -276,23 +277,7 @@ const CommonIntents: FC = () => {
     mutationFn: (intentModelData: { intentName: string }) =>
       downloadExamples(intentModelData),
     onSuccess: async (data) => {
-      // @ts-ignore
-      const blob = new Blob([data], { type: 'text/csv' });
-      const fileName = selectedIntent?.id + '.csv';
-
-      if (window.showSaveFilePicker) {
-        const handle = await window.showSaveFilePicker({ suggestedName: fileName });
-        const writable = await handle.createWritable();
-        await writable.write(blob);
-        writable.close();
-      } else {
-        const url = window.URL.createObjectURL(blob);
-        const a = document.createElement('a');
-        a.href = url;
-        a.download = fileName;
-        a.click();
-        window.URL.revokeObjectURL(url);
-      }
+      saveCsv(data, selectedIntent?.id || '');
 
       toast.open({
         type: 'success',

diff --git a/GUI/src/pages/Training/Intents/index.tsx b/GUI/src/pages/Training/Intents/index.tsx
@@ -33,6 +33,7 @@ import { Rule, RuleDTO } from '../../../types/rule';
 import { addStoryOrRule, deleteStoryOrRule } from '../../../services/stories';
 import IntentTabList from './IntentTabList';
 import useStore from "../../../store/store";
+import { saveCsv } from 'utils/save-intent-examples-csv';
 
 type Response = {
   name: string;
@@ -459,23 +460,7 @@ const Intents: FC = () => {
     mutationFn: (intentModelData: { intentName: string }) =>
       downloadExamples(intentModelData),
     onSuccess: async (data) => {
-      // @ts-ignore
-      const blob = new Blob([data], { type: 'text/csv' });
-      const fileName = selectedIntent?.id + '.csv';
-
-      if (window.showSaveFilePicker) {
-        const handle = await window.showSaveFilePicker({ suggestedName: fileName });
-        const writable = await handle.createWritable();
-        await writable.write(blob);
-        writable.close();
-      } else {
-        const url = window.URL.createObjectURL(blob);
-        const a = document.createElement('a');
-        a.href = url;
-        a.download = fileName;
-        a.click();
-        window.URL.revokeObjectURL(url);
-      }
+      saveCsv(data, selectedIntent?.id || '');
 
       toast.open({
         type: 'success',

diff --git a/GUI/src/utils/save-csv.ts b/GUI/src/utils/save-csv.ts
@@ -0,0 +1,19 @@
+export const saveCsv = async (data: unknown, name: string) => {
+  // @ts-ignore
+  const blob = new Blob([data], { type: 'text/csv' });
+  const fileName = name + '.csv';
+
+  if (window.showSaveFilePicker) {
+    const handle = await window.showSaveFilePicker({ suggestedName: fileName });
+    const writable = await handle.createWritable();
+    await writable.write(blob);
+    writable.close();
+  } else {
+    const url = window.URL.createObjectURL(blob);
+    const a = document.createElement('a');
+    a.href = url;
+    a.download = fileName;
+    a.click();
+    window.URL.revokeObjectURL(url);
+  }
+};