From 6b8bf8161ea2796c44f69c2a000d558ccc52e101 Mon Sep 17 00:00:00 2001
From: Jan Prochazka <jan.prochazka@gmail.com>
Date: Thu, 3 Jun 2021 20:58:38 +0200
Subject: [PATCH] quick export WIP

---
 packages/types/extensions.d.ts                |   7 +
 .../src/appobj/DatabaseObjectAppObject.svelte | 203 ++++++++++--------
 .../web/src/plugins/PluginsProvider.svelte    |  10 +-
 .../web/src/utility/exportElectronFile.ts     |  29 +++
 .../dbgate-plugin-csv/src/frontend/index.js   |  25 +++
 .../dbgate-plugin-excel/src/frontend/index.js |  13 ++
 6 files changed, 197 insertions(+), 90 deletions(-)
 create mode 100644 packages/web/src/utility/exportElectronFile.ts

diff --git a/packages/types/extensions.d.ts b/packages/types/extensions.d.ts
index 142b6658a..fca442fe9 100644
--- a/packages/types/extensions.d.ts
+++ b/packages/types/extensions.d.ts
@@ -33,9 +33,16 @@ export interface PluginDefinition {
   content: any;
 }
 
+export interface QuickExportDefinition {
+  label: string;
+  createWriter: (fileName: string) => { functionName: string; props: any };
+  extension: string;
+}
+
 export interface ExtensionsDirectory {
   plugins: PluginDefinition[];
   fileFormats: FileFormatDefinition[];
+  quickExports: QuickExportDefinition[];
   drivers: EngineDriver[];
   themes: ThemeDefinition[];
 }
diff --git a/packages/web/src/appobj/DatabaseObjectAppObject.svelte b/packages/web/src/appobj/DatabaseObjectAppObject.svelte
index f125e77cc..edfc2de46 100644
--- a/packages/web/src/appobj/DatabaseObjectAppObject.svelte
+++ b/packages/web/src/appobj/DatabaseObjectAppObject.svelte
@@ -1,6 +1,7 @@
 <script lang="ts" context="module">
   export const extractKey = ({ schemaName, pureName }) => (schemaName ? `${schemaName}.${pureName}` : pureName);
   export const createMatcher = ({ pureName }) => filter => filterName(filter, pureName);
+  const electron = getElectron();
 
   const icons = {
     tables: 'img table',
@@ -50,6 +51,10 @@
         label: 'Export',
         isExport: true,
       },
+      electron && {
+        label: 'Quick export',
+        isQuickExport: true,
+      },
       {
         label: 'Open in free table editor',
         isOpenFreeTable: true,
@@ -112,6 +117,10 @@
         label: 'Export',
         isExport: true,
       },
+      electron && {
+        label: 'Quick export',
+        isQuickExport: true,
+      },
       {
         label: 'Open in free table editor',
         isOpenFreeTable: true,
@@ -169,6 +178,10 @@
         label: 'Export',
         isExport: true,
       },
+      electron && {
+        label: 'Quick export',
+        isQuickExport: true,
+      },
       {
         label: 'Open in free table editor',
         isOpenFreeTable: true,
@@ -265,6 +278,10 @@
         label: 'Export',
         isExport: true,
       },
+      electron && {
+        label: 'Quick export',
+        isQuickExport: true,
+      },
       {
         divider: true,
       },
@@ -311,6 +328,7 @@
       { forceNewTab }
     );
   }
+
 </script>
 
 <script lang="ts">
@@ -326,7 +344,9 @@
   import { findEngineDriver } from 'dbgate-tools';
   import uuidv1 from 'uuid/v1';
   import SqlGeneratorModal from '../modals/SqlGeneratorModal.svelte';
-import getConnectionLabel from '../utility/getConnectionLabel';
+  import getConnectionLabel from '../utility/getConnectionLabel';
+  import getElectron from '../utility/getElectron';
+  import { exportElectronFile } from '../utility/exportElectronFile';
 
   export let data;
 
@@ -370,100 +390,105 @@ import getConnectionLabel from '../utility/getConnectionLabel';
 
   function createMenu() {
     const { objectTypeField } = data;
-    return menus[objectTypeField].map(menu => {
-      if (menu.divider) return menu;
-      return {
-        text: menu.label,
-        onClick: async () => {
-          if (menu.isExport) {
-            showModal(ImportExportModal, {
-              initialValues: {
-                sourceStorageType: 'database',
-                sourceConnectionId: data.conid,
-                sourceDatabaseName: data.database,
-                sourceSchemaName: data.schemaName,
-                sourceList: [data.pureName],
-              },
-            });
-          } else if (menu.isOpenFreeTable) {
-            const coninfo = await getConnectionInfo(data);
-            openNewTab({
-              title: data.pureName,
-              icon: 'img free-table',
-              tabComponent: 'FreeTableTab',
-              props: {
-                initialArgs: {
-                  functionName: 'tableReader',
-                  props: {
-                    connection: {
-                      ...coninfo,
-                      database: data.database,
+    return menus[objectTypeField]
+      .filter(x => x)
+      .map(menu => {
+        if (menu.divider) return menu;
+        return {
+          text: menu.label,
+          onClick: async () => {
+            if (menu.isExport) {
+              showModal(ImportExportModal, {
+                initialValues: {
+                  sourceStorageType: 'database',
+                  sourceConnectionId: data.conid,
+                  sourceDatabaseName: data.database,
+                  sourceSchemaName: data.schemaName,
+                  sourceList: [data.pureName],
+                },
+              });
+            } else if (menu.isQuickExport) {
+              exportElectronFile(data);
+            } else if (menu.isOpenFreeTable) {
+              const coninfo = await getConnectionInfo(data);
+              openNewTab({
+                title: data.pureName,
+                icon: 'img free-table',
+                tabComponent: 'FreeTableTab',
+                props: {
+                  initialArgs: {
+                    functionName: 'tableReader',
+                    props: {
+                      connection: {
+                        ...coninfo,
+                        database: data.database,
+                      },
+                      schemaName: data.schemaName,
+                      pureName: data.pureName,
                     },
-                    schemaName: data.schemaName,
-                    pureName: data.pureName,
                   },
                 },
-              },
-            });
-          } else if (menu.isActiveChart) {
-            const driver = await getDriver();
-            const dmp = driver.createDumper();
-            dmp.put('^select * from %f', data);
-            openNewTab(
-              {
-                title: data.pureName,
-                icon: 'img chart',
-                tabComponent: 'ChartTab',
-                props: {
-                  conid: data.conid,
-                  database: data.database,
+              });
+            } else if (menu.isActiveChart) {
+              const driver = await getDriver();
+              const dmp = driver.createDumper();
+              dmp.put('^select * from %f', data);
+              openNewTab(
+                {
+                  title: data.pureName,
+                  icon: 'img chart',
+                  tabComponent: 'ChartTab',
+                  props: {
+                    conid: data.conid,
+                    database: data.database,
+                  },
                 },
-              },
-              {
-                editor: {
-                  config: { chartType: 'bar' },
-                  sql: dmp.s,
+                {
+                  editor: {
+                    config: { chartType: 'bar' },
+                    sql: dmp.s,
+                  },
+                }
+              );
+            } else if (menu.isQueryDesigner) {
+              openNewTab(
+                {
+                  title: 'Query #',
+                  icon: 'img query-design',
+                  tabComponent: 'QueryDesignTab',
+                  props: {
+                    conid: data.conid,
+                    database: data.database,
+                  },
                 },
-              }
-            );
-          } else if (menu.isQueryDesigner) {
-            openNewTab(
-              {
-                title: 'Query #',
-                icon: 'img query-design',
-                tabComponent: 'QueryDesignTab',
-                props: {
-                  conid: data.conid,
-                  database: data.database,
-                },
-              },
-              {
-                editor: {
-                  tables: [
-                    {
-                      ...data,
-                      designerId: uuidv1(),
-                      left: 50,
-                      top: 50,
-                    },
-                  ],
-                },
-              }
-            );
-          } else if (menu.sqlGeneratorProps) {
-            showModal(SqlGeneratorModal, {
-              initialObjects: [data],
-              initialConfig: menu.sqlGeneratorProps,
-              conid: data.conid,
-              database: data.database,
-            });
-          } else {
-            openDatabaseObjectDetail(menu.tab, menu.scriptTemplate, data, menu.forceNewTab, menu.initialData);
-          }
-        },
-      };
-    });
+                {
+                  editor: {
+                    tables: [
+                      {
+                        ...data,
+                        designerId: uuidv1(),
+                        left: 50,
+                        top: 50,
+                      },
+                    ],
+                  },
+                }
+              );
+            } else if (menu.sqlGeneratorProps) {
+              showModal(SqlGeneratorModal, {
+                initialObjects: [data],
+                initialConfig: menu.sqlGeneratorProps,
+                conid: data.conid,
+                database: data.database,
+              });
+            } else {
+              openDatabaseObjectDetail(menu.tab, menu.scriptTemplate, data, menu.forceNewTab, menu.initialData);
+            }
+          },
+        };
+      });
   }
+
 </script>
 
 <AppObjectCore
diff --git a/packages/web/src/plugins/PluginsProvider.svelte b/packages/web/src/plugins/PluginsProvider.svelte
index a6ebe00a1..d0d198290 100644
--- a/packages/web/src/plugins/PluginsProvider.svelte
+++ b/packages/web/src/plugins/PluginsProvider.svelte
@@ -40,18 +40,26 @@
   function buildDrivers(plugins) {
     const res = [];
     for (const { content } of plugins) {
-      // if (content.driver) res.push(content.driver);
       if (content.drivers) res.push(...content.drivers);
     }
     return res;
   }
 
+  function buildQuickExports(plugins) {
+    const res = [];
+    for (const { content } of plugins) {
+      if (content.quickExports) res.push(...content.quickExports);
+    }
+    return res;
+  }
+
   export function buildExtensions(plugins) {
     const extensions = {
       plugins,
       fileFormats: buildFileFormats(plugins),
       themes: buildThemes(plugins),
       drivers: buildDrivers(plugins),
+      quickExports: buildQuickExports(plugins),
     };
     return extensions;
   }
diff --git a/packages/web/src/utility/exportElectronFile.ts b/packages/web/src/utility/exportElectronFile.ts
new file mode 100644
index 000000000..20376e0af
--- /dev/null
+++ b/packages/web/src/utility/exportElectronFile.ts
@@ -0,0 +1,29 @@
+import { showModal } from '../modals/modalTools';
+import { get } from 'svelte/store';
+import newQuery from '../query/newQuery';
+import ImportExportModal from '../modals/ImportExportModal.svelte';
+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() {
+  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);
+    });
+}
diff --git a/plugins/dbgate-plugin-csv/src/frontend/index.js b/plugins/dbgate-plugin-csv/src/frontend/index.js
index 33b0f6a17..94a199e3d 100644
--- a/plugins/dbgate-plugin-csv/src/frontend/index.js
+++ b/plugins/dbgate-plugin-csv/src/frontend/index.js
@@ -43,4 +43,29 @@ const fileFormat = {
 
 export default {
   fileFormats: [fileFormat],
+
+  quickExports: [
+    {
+      label: 'CSV file',
+      extension: 'csv',
+      createWriter: (fileName) => ({
+        functionName: 'writer',
+        props: {
+          fileName,
+          delimiter: ',',
+        },
+      }),
+    },
+    {
+      label: 'CSV file (semicolor separated)',
+      extension: 'csv',
+      createWriter: (fileName) => ({
+        functionName: 'writer',
+        props: {
+          fileName,
+          delimiter: ';',
+        },
+      }),
+    },
+  ],
 };
diff --git a/plugins/dbgate-plugin-excel/src/frontend/index.js b/plugins/dbgate-plugin-excel/src/frontend/index.js
index 94d515376..f79e18307 100644
--- a/plugins/dbgate-plugin-excel/src/frontend/index.js
+++ b/plugins/dbgate-plugin-excel/src/frontend/index.js
@@ -64,5 +64,18 @@ const fileFormat = {
 
 export default {
   fileFormats: [fileFormat],
+  quickExports: [
+    {
+      label: 'MS Excel',
+      extension: 'xlsx',
+      createWriter: (fileName) => ({
+        functionName: 'writer',
+        props: {
+          fileName,
+          sheetName: 'data',
+        },
+      }),
+    },
+  ],
   initialize,
 };