From 817286d32620df7978f21d1064544f66abd6bb9a Mon Sep 17 00:00:00 2001
From: Jan Prochazka <jan.prochazka@gmail.com>
Date: Sat, 10 Apr 2021 17:21:29 +0200
Subject: [PATCH] export mongo grid, generate query

---
 .../datagrid/CollectionDataGridCore.svelte    | 62 +++++++++++++++++++
 .../web/src/impexp/SourceTargetConfig.svelte  |  2 +-
 2 files changed, 63 insertions(+), 1 deletion(-)

diff --git a/packages/web/src/datagrid/CollectionDataGridCore.svelte b/packages/web/src/datagrid/CollectionDataGridCore.svelte
index 57455b19..6690d58c 100644
--- a/packages/web/src/datagrid/CollectionDataGridCore.svelte
+++ b/packages/web/src/datagrid/CollectionDataGridCore.svelte
@@ -1,4 +1,23 @@
 <script context="module" lang="ts">
+  const getCurrentEditor = () => getActiveComponent('CollectionDataGridCore');
+
+  registerCommand({
+    id: 'collectionDataGrid.openQuery',
+    category: 'Data grid',
+    name: 'Open query',
+    testEnabled: () => getCurrentEditor() != null,
+    onClick: () => getCurrentEditor().openQuery(),
+  });
+
+  registerCommand({
+    id: 'collectionDataGrid.export',
+    category: 'Data grid',
+    name: 'Export',
+    keyText: 'Ctrl+E',
+    testEnabled: () => getCurrentEditor() != null,
+    onClick: () => getCurrentEditor().exportGrid(),
+  });
+
   function buildGridMongoCondition(props) {
     const filters = props?.display?.config?.filters;
 
@@ -100,6 +119,7 @@
   import { parseFilter } from 'dbgate-filterparser';
   import { scriptToSql } from 'dbgate-sqltree';
   import _ from 'lodash';
+  import registerCommand from '../commands/registerCommand';
   import ErrorInfo from '../elements/ErrorInfo.svelte';
   import ConfirmNoSqlModal from '../modals/ConfirmNoSqlModal.svelte';
   import ErrorMessageModal from '../modals/ErrorMessageModal.svelte';
@@ -107,6 +127,8 @@
   import { showModal } from '../modals/modalTools';
 
   import axiosInstance from '../utility/axiosInstance';
+  import { registerMenu } from '../utility/contextMenu';
+  import createActivator, { getActiveComponent } from '../utility/createActivator';
   import openNewTab from '../utility/openNewTab';
   import ChangeSetGrider from './ChangeSetGrider';
 
@@ -129,6 +151,8 @@
 
   export let loadedRows = [];
 
+  export const activator = createActivator('CollectionDataGridCore', false);
+
   // $: console.log('loadedRows BIND', loadedRows);
   $: grider = new ChangeSetGrider(
     loadedRows,
@@ -141,6 +165,44 @@
   );
   // $: console.log('GRIDER', grider);
   // $: if (onChangeGrider) onChangeGrider(grider);
+
+  function getExportQuery() {
+    return `db.collection('${pureName}')
+      .find(${JSON.stringify(buildGridMongoCondition($$props) || {})})
+      .sort(${JSON.stringify(buildMongoSort($$props) || {})})`;
+  }
+
+  export function exportGrid() {
+    const initialValues: any = {};
+    initialValues.sourceStorageType = 'query';
+    initialValues.sourceConnectionId = conid;
+    initialValues.sourceDatabaseName = database;
+    initialValues.sourceSql = getExportQuery();
+    initialValues.sourceList = [pureName];
+    showModal(ImportExportModal, { initialValues });
+  }
+
+  export function openQuery() {
+    openNewTab(
+      {
+        title: 'Query #',
+        icon: 'img sql-file',
+        tabComponent: 'QueryTab',
+        props: {
+          conid,
+          database,
+        },
+      },
+      {
+        editor: getExportQuery(),
+      }
+    );
+  }
+
+  registerMenu(
+    { command: 'collectionDataGrid.openQuery', tag: 'export' },
+    { command: 'collectionDataGrid.export', tag: 'export' }
+  );
 </script>
 
 <LoadingDataGridCore
diff --git a/packages/web/src/impexp/SourceTargetConfig.svelte b/packages/web/src/impexp/SourceTargetConfig.svelte
index 547864b2..74a4948b 100644
--- a/packages/web/src/impexp/SourceTargetConfig.svelte
+++ b/packages/web/src/impexp/SourceTargetConfig.svelte
@@ -42,7 +42,7 @@
             label: `${format.name} files(s)`,
             directions: getFileFormatDirections(format),
           })),
-          { value: 'query', label: 'SQL Query', directions: ['source'] },
+          { value: 'query', label: 'Query', directions: ['source'] },
           { value: 'archive', label: 'Archive', directions: ['source', 'target'] },
         ];