From b35222f86ba2961fa87bf90d976b4b27ba747d92 Mon Sep 17 00:00:00 2001
From: Romein van Buren <romein@vburen.nl>
Date: Mon, 23 Jan 2023 13:35:31 +0100
Subject: [PATCH] Truncate collections

---
 .../src/organisms/connection/hosttree.svelte  | 18 +++++++---
 frontend/wailsjs/go/app/App.d.ts              |  2 ++
 frontend/wailsjs/go/app/App.js                |  4 +++
 internal/app/collection.go                    | 35 +++++++++++++++++++
 4 files changed, 54 insertions(+), 5 deletions(-)

diff --git a/frontend/src/organisms/connection/hosttree.svelte b/frontend/src/organisms/connection/hosttree.svelte
index 89d1c13..73c9d76 100644
--- a/frontend/src/organisms/connection/hosttree.svelte
+++ b/frontend/src/organisms/connection/hosttree.svelte
@@ -1,7 +1,7 @@
 <script>
   import { busy, connections } from '../../stores';
   import { createEventDispatcher } from 'svelte';
-  import { DropCollection, DropDatabase, OpenCollection, OpenConnection, OpenDatabase } from '../../../wailsjs/go/app/App';
+  import { DropCollection, DropDatabase, OpenCollection, OpenConnection, OpenDatabase, TruncateCollection } from '../../../wailsjs/go/app/App';
   import Grid from '../../components/grid.svelte';
   import { WindowSetTitle } from '../../../wailsjs/runtime/runtime';
 
@@ -69,6 +69,13 @@
     busy.end();
   }
 
+  async function truncateCollection(dbKey, collKey) {
+    busy.start();
+    await TruncateCollection(activeHostKey, dbKey, collKey);
+    await reload();
+    busy.end();
+  }
+
   async function dropCollection(dbKey, collKey) {
     busy.start();
     await DropCollection(activeHostKey, dbKey, collKey);
@@ -94,21 +101,22 @@
         name: collKey,
         icon: 'list',
         menu: [
-          { label: `Rename ${collKey} collection…`, fn: () => dispatch('renameCollection', collKey) },
-          { label: `Drop ${collKey}…`, fn: () => dropCollection(dbKey, collKey) },
+          { label: 'Rename collection…', fn: () => dispatch('renameCollection', collKey) },
+          { label: 'Truncate collection…', fn: () => truncateCollection(dbKey, collKey) },
+          { label: 'Drop collection…', fn: () => dropCollection(dbKey, collKey) },
           { separator: true },
           { label: 'New collection…', fn: () => dispatch('newCollection') },
         ],
       })) || [],
       menu: [
-        { label: `Drop database ${dbKey}…`, fn: () => dropDatabase(dbKey) },
+        { label: 'Drop database…', fn: () => dropDatabase(dbKey) },
         { separator: true },
         { label: 'New database…', fn: () => dispatch('newDatabase') },
       ],
     })),
     menu: [
       { label: 'New database…', fn: () => dispatch('newDatabase') },
-        { separator: true },
+      { separator: true },
       { label: `Edit host ${hosts[hostKey].name}…`, fn: () => dispatch('editHost', hostKey) },
     ],
   }))}
diff --git a/frontend/wailsjs/go/app/App.d.ts b/frontend/wailsjs/go/app/App.d.ts
index 373847c..903d449 100755
--- a/frontend/wailsjs/go/app/App.d.ts
+++ b/frontend/wailsjs/go/app/App.d.ts
@@ -41,6 +41,8 @@ export function RenameCollection(arg1:string,arg2:string,arg3:string,arg4:string
 
 export function Settings():Promise<app.Settings>;
 
+export function TruncateCollection(arg1:string,arg2:string,arg3:string):Promise<boolean>;
+
 export function UpdateHost(arg1:string,arg2:string):Promise<void>;
 
 export function UpdateItems(arg1:string,arg2:string,arg3:string,arg4:string):Promise<number>;
diff --git a/frontend/wailsjs/go/app/App.js b/frontend/wailsjs/go/app/App.js
index 5bb6c4f..f6f24c8 100755
--- a/frontend/wailsjs/go/app/App.js
+++ b/frontend/wailsjs/go/app/App.js
@@ -74,6 +74,10 @@ export function Settings() {
   return window['go']['app']['App']['Settings']();
 }
 
+export function TruncateCollection(arg1, arg2, arg3) {
+  return window['go']['app']['App']['TruncateCollection'](arg1, arg2, arg3);
+}
+
 export function UpdateHost(arg1, arg2) {
   return window['go']['app']['App']['UpdateHost'](arg1, arg2);
 }
diff --git a/internal/app/collection.go b/internal/app/collection.go
index 48e4abc..d85991a 100644
--- a/internal/app/collection.go
+++ b/internal/app/collection.go
@@ -57,6 +57,39 @@ func (a *App) RenameCollection(hostKey, dbKey, collKey, newCollKey string) bool
 	return true
 }
 
+func (a *App) TruncateCollection(hostKey, dbKey, collKey string) bool {
+	sure, _ := runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+		Title:         "Confirm",
+		Message:       "Are you sure you want to remove all items from " + collKey + "?",
+		Buttons:       []string{"Yes", "No"},
+		DefaultButton: "Yes",
+		CancelButton:  "No",
+	})
+	if sure != "Yes" {
+		return false
+	}
+
+	client, ctx, close, err := a.connectToHost(hostKey)
+	if err != nil {
+		fmt.Println(err.Error())
+		return false
+	}
+
+	_, err = client.Database(dbKey).Collection(collKey).DeleteMany(ctx, bson.D{})
+	if err != nil {
+		fmt.Println(err.Error())
+		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+			Type:    runtime.ErrorDialog,
+			Title:   "Could not truncate " + collKey,
+			Message: err.Error(),
+		})
+		return false
+	}
+
+	defer close()
+	return true
+}
+
 func (a *App) DropCollection(hostKey, dbKey, collKey string) bool {
 	sure, _ := runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
 		Title:         "Confirm",
@@ -74,6 +107,7 @@ func (a *App) DropCollection(hostKey, dbKey, collKey string) bool {
 		fmt.Println(err.Error())
 		return false
 	}
+
 	err = client.Database(dbKey).Collection(collKey).Drop(ctx)
 	if err != nil {
 		fmt.Println(err.Error())
@@ -84,6 +118,7 @@ func (a *App) DropCollection(hostKey, dbKey, collKey string) bool {
 		})
 		return false
 	}
+
 	defer close()
 	return true
 }