From c0d664d399f8f5dab81fcf26046ac81d24eb4e10 Mon Sep 17 00:00:00 2001
From: Jan Prochazka <jan.prochazka@sprinx.cz>
Date: Fri, 16 Aug 2024 12:52:37 +0200
Subject: [PATCH] generic drop collection

---
 .../web/src/appobj/DatabaseObjectAppObject.svelte     | 11 +++++++----
 plugins/dbgate-plugin-mongo/src/backend/driver.js     |  2 ++
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/packages/web/src/appobj/DatabaseObjectAppObject.svelte b/packages/web/src/appobj/DatabaseObjectAppObject.svelte
index b3ba1b1d5..962ae9b32 100644
--- a/packages/web/src/appobj/DatabaseObjectAppObject.svelte
+++ b/packages/web/src/appobj/DatabaseObjectAppObject.svelte
@@ -1,5 +1,5 @@
 <script lang="ts" context="module">
-  import {copyTextToClipboard} from "../utility/clipboard";
+  import { copyTextToClipboard } from '../utility/clipboard';
 
   export const extractKey = ({ schemaName, pureName }) => (schemaName ? `${schemaName}.${pureName}` : pureName);
   export const createMatcher =
@@ -85,7 +85,7 @@
       {
         label: 'Copy table name',
         isCopyTableName: true,
-        requiresWriteAccess: false
+        requiresWriteAccess: false,
       },
       {
         label: 'Create table backup',
@@ -515,7 +515,10 @@
         message: `Really drop collection ${data.pureName}?`,
         onConfirm: async () => {
           const dbid = _.pick(data, ['conid', 'database']);
-          saveScriptToDatabase(dbid, `db.dropCollection('${data.pureName}')`);
+          runOperationOnDatabase(dbid, {
+            type: 'dropCollection',
+            collection: data.pureName,
+          });
         },
       });
     } else if (menu.isCopyTableName) {
@@ -786,7 +789,7 @@
   import SqlGeneratorModal from '../modals/SqlGeneratorModal.svelte';
   import { exportQuickExportFile } from '../utility/exportFileTools';
   import createQuickExportMenu from '../utility/createQuickExportMenu';
-  import ConfirmSqlModal, { saveScriptToDatabase } from '../modals/ConfirmSqlModal.svelte';
+  import ConfirmSqlModal, { runOperationOnDatabase, saveScriptToDatabase } from '../modals/ConfirmSqlModal.svelte';
   import { alterDatabaseDialog, renameDatabaseObjectDialog } from '../utility/alterDatabaseTools';
   import ConfirmModal from '../modals/ConfirmModal.svelte';
   import { apiCall } from '../utility/api';
diff --git a/plugins/dbgate-plugin-mongo/src/backend/driver.js b/plugins/dbgate-plugin-mongo/src/backend/driver.js
index 9466998a7..64762e051 100644
--- a/plugins/dbgate-plugin-mongo/src/backend/driver.js
+++ b/plugins/dbgate-plugin-mongo/src/backend/driver.js
@@ -105,6 +105,8 @@ const driver = {
     switch (type) {
       case 'createCollection':
         await this.script(pool, `db.createCollection('${operation.collection}')`);
+      case 'dropOperation':
+        await this.script(pool, `db.dropCollection('${operation.collection}')`);
       default:
         throw new Error(`Operation type ${type} not supported`);
     }