From 27311afb3150b424d112b6a5d6647291ce6c200b Mon Sep 17 00:00:00 2001
From: Jan Prochazka <jan.prochazka@gmail.com>
Date: Sun, 6 Jun 2021 10:25:52 +0200
Subject: [PATCH] quick exports - basic skeleton working

---
 .../src/appobj/DatabaseObjectAppObject.svelte | 65 +++++++++++--------
 packages/web/src/modals/DropDownMenu.svelte   | 32 +++++++--
 .../web/src/utility/exportElectronFile.ts     | 53 ++++++++-------
 .../dbgate-plugin-csv/src/frontend/index.js   |  4 +-
 .../dbgate-plugin-excel/src/frontend/index.js |  6 +-
 5 files changed, 96 insertions(+), 64 deletions(-)

diff --git a/packages/web/src/appobj/DatabaseObjectAppObject.svelte b/packages/web/src/appobj/DatabaseObjectAppObject.svelte
index ac0196715..9578e9db2 100644
--- a/packages/web/src/appobj/DatabaseObjectAppObject.svelte
+++ b/packages/web/src/appobj/DatabaseObjectAppObject.svelte
@@ -47,15 +47,15 @@
       {
         divider: true,
       },
+      electron && {
+        label: 'Quick export',
+        isQuickExport: true,
+        functionName: 'tableReader',
+      },
       {
         label: 'Export',
         isExport: true,
       },
-      // electron &&
-      {
-        label: 'Quick export',
-        isQuickExport: true,
-      },
       {
         label: 'Open in free table editor',
         isOpenFreeTable: true,
@@ -114,14 +114,14 @@
       {
         divider: true,
       },
-      {
-        label: 'Export',
-        isExport: true,
-      },
       electron && {
         label: 'Quick export',
         isQuickExport: true,
       },
+      {
+        label: 'Export',
+        isExport: true,
+      },
       {
         label: 'Open in free table editor',
         isOpenFreeTable: true,
@@ -175,14 +175,14 @@
       {
         divider: true,
       },
-      {
-        label: 'Export',
-        isExport: true,
-      },
       electron && {
         label: 'Quick export',
         isQuickExport: true,
       },
+      {
+        label: 'Export',
+        isExport: true,
+      },
       {
         label: 'Open in free table editor',
         isOpenFreeTable: true,
@@ -275,14 +275,14 @@
           },
         },
       },
-      {
-        label: 'Export',
-        isExport: true,
-      },
       electron && {
         label: 'Quick export',
         isQuickExport: true,
       },
+      {
+        label: 'Export',
+        isExport: true,
+      },
       {
         divider: true,
       },
@@ -399,16 +399,26 @@
         if (menu.isQuickExport) {
           return {
             text: menu.label,
-            submenu: [
-              {
-                text: 'CSV file',
-                isQuickExport: true,
+            submenu: $extensions.quickExports.map(fmt => ({
+              text: fmt.label,
+              onClick: async () => {
+                const coninfo = await getConnectionInfo(data);
+                exportElectronFile(
+                  data.pureName,
+                  {
+                    functionName: menu.functionName,
+                    props: {
+                      connection: {
+                        ..._.omit(coninfo, ['_id', 'displayName']),
+                        ..._.pick(data, ['database']),
+                      },
+                      ..._.pick(data, ['pureName', 'schemaName']),
+                    },
+                  },
+                  fmt
+                );
               },
-              {
-                text: 'Excel',
-                isQuickExport: true,
-              },
-            ],
+            })),
           };
         }
 
@@ -425,8 +435,6 @@
                   sourceList: [data.pureName],
                 },
               });
-            } else if (menu.isQuickExport) {
-              exportElectronFile(data);
             } else if (menu.isOpenFreeTable) {
               const coninfo = await getConnectionInfo(data);
               openNewTab({
@@ -511,6 +519,7 @@
 
 <AppObjectCore
   {...$$restProps}
+  module={$$props.module}
   {data}
   title={data.schemaName ? `${data.schemaName}.${data.pureName}` : data.pureName}
   icon={icons[data.objectTypeField]}
diff --git a/packages/web/src/modals/DropDownMenu.svelte b/packages/web/src/modals/DropDownMenu.svelte
index 043b57ec5..7961b2e22 100644
--- a/packages/web/src/modals/DropDownMenu.svelte
+++ b/packages/web/src/modals/DropDownMenu.svelte
@@ -68,6 +68,7 @@
   export let items;
   export let top;
   export let left;
+  export let onCloseParent;
 
   let element;
 
@@ -90,6 +91,7 @@
       return;
     }
     dispatch('close');
+    if (onCloseParent) onCloseParent();
     if (item.onClick) item.onClick();
   }
 
@@ -106,14 +108,23 @@
   $: compacted = _.compact(extracted.map(x => mapItem(x, $commandsCustomized)));
   $: filtered = compacted.filter(x => !x.disabled || !x.hideDisabled);
 
+  const handleClickOutside = event => {
+    // if (element && !element.contains(event.target) && !event.defaultPrevented) {
+    if (event.target.closest('ul.dropDownMenuMarker')) return;
+
+    dispatch('close');
+  };
+
+  onMount(() => {
+    document.addEventListener('mousedown', handleClickOutside, true);
+    return () => {
+      document.removeEventListener('mousedown', handleClickOutside, true);
+    };
+  });
+
 </script>
 
-<ul
-  style={`left: ${left}px; top: ${top}px`}
-  use:clickOutside
-  on:clickOutside={() => dispatch('close')}
-  bind:this={element}
->
+<ul class="dropDownMenuMarker" style={`left: ${left}px; top: ${top}px`} bind:this={element}>
   {#each filtered as item}
     {#if item.divider}
       <li class="divider" />
@@ -141,7 +152,14 @@
   {/each}
 </ul>
 {#if submenuItem?.submenu}
-  <svelte:self items={submenuItem?.submenu} {...submenuOffset} />
+  <svelte:self
+    items={submenuItem?.submenu}
+    {...submenuOffset}
+    onCloseParent={() => {
+      if (onCloseParent) onCloseParent();
+      dispatch('close');
+    }}
+  />
 {/if}
 
 <style>
diff --git a/packages/web/src/utility/exportElectronFile.ts b/packages/web/src/utility/exportElectronFile.ts
index 20376e0af..86aaea777 100644
--- a/packages/web/src/utility/exportElectronFile.ts
+++ b/packages/web/src/utility/exportElectronFile.ts
@@ -1,29 +1,34 @@
-import { showModal } from '../modals/modalTools';
-import { get } from 'svelte/store';
-import newQuery from '../query/newQuery';
-import ImportExportModal from '../modals/ImportExportModal.svelte';
+import ScriptWriter from '../impexp/ScriptWriter';
 import getElectron from './getElectron';
-import { currentDatabase, extensions } from '../stores';
-import { getUploadListener } from './uploadFiles';
 import axiosInstance from '../utility/axiosInstance';
-import { getDatabaseFileLabel } from './getConnectionLabel';
 
-function getFileFormatFilters(extensions) {
-  return extensions.quickExports.map(x => ({ name: x.label, extensions: [x.extension] }));
-}
-
-export async function exportElectronFile() {
+export async function exportElectronFile(dataName, reader, format) {
   const electron = getElectron();
-  const ext = get(extensions);
-  const filters = getFileFormatFilters(ext);
-  console.log('FLT', filters);
-  electron.remote.dialog
-    .showSaveDialog(electron.remote.getCurrentWindow(), {
-      filters,
-    })
-    .then(filePaths => {
-      console.log('filePaths ASYNC2', filePaths);
-      const filePath = filePaths && filePaths[0];
-      console.log('filePath', filePath);
-    });
+  const filters = [{ name: format.label, extensions: [format.extension] }];
+
+  const filePath = electron.remote.dialog.showSaveDialogSync(electron.remote.getCurrentWindow(), {
+    filters,
+    defaultPath: `${dataName}.${format.extension}`,
+    properties: ['showOverwriteConfirmation'],
+  });
+  if (!filePath) return;
+
+  const script = new ScriptWriter();
+
+  const sourceVar = script.allocVariable();
+  script.assign(sourceVar, reader.functionName, reader.props);
+
+  console.log('format.createWriter(filePath, dataName)', format.createWriter(filePath, dataName));
+
+  const targetVar = script.allocVariable();
+  const writer = format.createWriter(filePath, dataName);
+  script.assign(targetVar, writer.functionName, writer.props);
+
+  script.copyStream(sourceVar, targetVar);
+  script.put();
+
+  console.log('script.getScript()', script.getScript());
+
+  const resp = await axiosInstance.post('runners/start', { script: script.getScript() });
+  const runid = resp.data.runid;
 }
diff --git a/plugins/dbgate-plugin-csv/src/frontend/index.js b/plugins/dbgate-plugin-csv/src/frontend/index.js
index 94a199e3d..f9e871e5d 100644
--- a/plugins/dbgate-plugin-csv/src/frontend/index.js
+++ b/plugins/dbgate-plugin-csv/src/frontend/index.js
@@ -49,7 +49,7 @@ export default {
       label: 'CSV file',
       extension: 'csv',
       createWriter: (fileName) => ({
-        functionName: 'writer',
+        functionName: 'writer@dbgate-plugin-csv',
         props: {
           fileName,
           delimiter: ',',
@@ -60,7 +60,7 @@ export default {
       label: 'CSV file (semicolor separated)',
       extension: 'csv',
       createWriter: (fileName) => ({
-        functionName: 'writer',
+        functionName: 'writer@dbgate-plugin-csv',
         props: {
           fileName,
           delimiter: ';',
diff --git a/plugins/dbgate-plugin-excel/src/frontend/index.js b/plugins/dbgate-plugin-excel/src/frontend/index.js
index f79e18307..25ceab8b0 100644
--- a/plugins/dbgate-plugin-excel/src/frontend/index.js
+++ b/plugins/dbgate-plugin-excel/src/frontend/index.js
@@ -68,11 +68,11 @@ export default {
     {
       label: 'MS Excel',
       extension: 'xlsx',
-      createWriter: (fileName) => ({
-        functionName: 'writer',
+      createWriter: (fileName, dataName) => ({
+        functionName: 'writer@dbgate-plugin-excel',
         props: {
           fileName,
-          sheetName: 'data',
+          sheetName: dataName,
         },
       }),
     },