From 9662d469570406b05616dc4b51b9d704869a0a8d Mon Sep 17 00:00:00 2001
From: Romein van Buren <romein@vburen.nl>
Date: Sat, 14 Jan 2023 20:47:29 +0100
Subject: [PATCH] Split a bit of backend code

---
 frontend/src/app.svelte                       |   2 +-
 .../organisms/collection-detail/find.svelte   |   2 +-
 .../organisms/collection-detail/insert.svelte |   2 +-
 frontend/wailsjs/go/{main => app}/App.d.ts    |   8 +-
 frontend/wailsjs/go/app/App.js                |  31 ++
 frontend/wailsjs/go/main/App.js               |  31 --
 frontend/wailsjs/go/models.ts                 |   2 +-
 internal/app/app.go                           |  15 +
 internal/app/collection.go                    |  29 ++
 internal/app/collection_find.go               | 129 +++++++
 internal/app/collection_insert.go             |  49 +++
 internal/app/connection.go                    |  28 ++
 internal/app/database.go                      |  59 ++++
 internal/app/host.go                          |  58 ++++
 main.go                                       | 322 +-----------------
 15 files changed, 408 insertions(+), 359 deletions(-)
 rename frontend/wailsjs/go/{main => app}/App.d.ts (78%)
 create mode 100755 frontend/wailsjs/go/app/App.js
 delete mode 100755 frontend/wailsjs/go/main/App.js
 create mode 100644 internal/app/app.go
 create mode 100644 internal/app/collection.go
 create mode 100644 internal/app/collection_find.go
 create mode 100644 internal/app/collection_insert.go
 create mode 100644 internal/app/connection.go
 create mode 100644 internal/app/database.go
 create mode 100644 internal/app/host.go

diff --git a/frontend/src/app.svelte b/frontend/src/app.svelte
index 5e75190..39bc4b2 100644
--- a/frontend/src/app.svelte
+++ b/frontend/src/app.svelte
@@ -1,6 +1,6 @@
 <script>
   import { onMount } from 'svelte';
-  import { DropDatabase, Hosts, OpenCollection, OpenConnection, OpenDatabase } from '../wailsjs/go/main/App';
+  import { DropDatabase, Hosts, OpenCollection, OpenConnection, OpenDatabase } from '../wailsjs/go/app/App';
   import AddressBar from './organisms/addressbar/index.svelte';
   import Grid from './components/grid.svelte';
   import CollectionDetail from './organisms/collection-detail/index.svelte';
diff --git a/frontend/src/organisms/collection-detail/find.svelte b/frontend/src/organisms/collection-detail/find.svelte
index 848cfec..9e2cd03 100644
--- a/frontend/src/organisms/collection-detail/find.svelte
+++ b/frontend/src/organisms/collection-detail/find.svelte
@@ -1,5 +1,5 @@
 <script>
-  import { PerformFind } from '../../../wailsjs/go/main/App';
+  import { PerformFind } from '../../../wailsjs/go/app/App';
   import CodeExample from '../../components/code-example.svelte';
   import { onMount } from 'svelte';
   import { input } from '../../actions';
diff --git a/frontend/src/organisms/collection-detail/insert.svelte b/frontend/src/organisms/collection-detail/insert.svelte
index 4da6553..767488e 100644
--- a/frontend/src/organisms/collection-detail/insert.svelte
+++ b/frontend/src/organisms/collection-detail/insert.svelte
@@ -1,7 +1,7 @@
 <script>
   import { input } from '../../actions';
   import { createEventDispatcher } from 'svelte';
-  import { PerformInsert } from '../../../wailsjs/go/main/App';
+  import { PerformInsert } from '../../../wailsjs/go/app/App';
 
   export let collection;
 
diff --git a/frontend/wailsjs/go/main/App.d.ts b/frontend/wailsjs/go/app/App.d.ts
similarity index 78%
rename from frontend/wailsjs/go/main/App.d.ts
rename to frontend/wailsjs/go/app/App.d.ts
index 9240a72..cddb339 100755
--- a/frontend/wailsjs/go/main/App.d.ts
+++ b/frontend/wailsjs/go/app/App.d.ts
@@ -1,12 +1,12 @@
 // Cynhyrchwyd y ffeil hon yn awtomatig. PEIDIWCH Â MODIWL
 // This file is automatically generated. DO NOT EDIT
-import {map[string]main} from '../models';
+import {map[string]app} from '../models';
 import {primitive} from '../models';
-import {main} from '../models';
+import {app} from '../models';
 
 export function DropDatabase(arg1:string,arg2:string):Promise<boolean>;
 
-export function Hosts():Promise<map[string]main.Host>;
+export function Hosts():Promise<map[string]app.Host>;
 
 export function OpenCollection(arg1:string,arg2:string,arg3:string):Promise<primitive.M>;
 
@@ -14,6 +14,6 @@ export function OpenConnection(arg1:string):Promise<Array<string>>;
 
 export function OpenDatabase(arg1:string,arg2:string):Promise<Array<string>>;
 
-export function PerformFind(arg1:string,arg2:string,arg3:string,arg4:string):Promise<main.findResult>;
+export function PerformFind(arg1:string,arg2:string,arg3:string,arg4:string):Promise<app.findResult>;
 
 export function PerformInsert(arg1:string,arg2:string,arg3:string,arg4:string):Promise<any>;
diff --git a/frontend/wailsjs/go/app/App.js b/frontend/wailsjs/go/app/App.js
new file mode 100755
index 0000000..71cddef
--- /dev/null
+++ b/frontend/wailsjs/go/app/App.js
@@ -0,0 +1,31 @@
+// @ts-check
+// Cynhyrchwyd y ffeil hon yn awtomatig. PEIDIWCH Â MODIWL
+// This file is automatically generated. DO NOT EDIT
+
+export function DropDatabase(arg1, arg2) {
+  return window['go']['app']['App']['DropDatabase'](arg1, arg2);
+}
+
+export function Hosts() {
+  return window['go']['app']['App']['Hosts']();
+}
+
+export function OpenCollection(arg1, arg2, arg3) {
+  return window['go']['app']['App']['OpenCollection'](arg1, arg2, arg3);
+}
+
+export function OpenConnection(arg1) {
+  return window['go']['app']['App']['OpenConnection'](arg1);
+}
+
+export function OpenDatabase(arg1, arg2) {
+  return window['go']['app']['App']['OpenDatabase'](arg1, arg2);
+}
+
+export function PerformFind(arg1, arg2, arg3, arg4) {
+  return window['go']['app']['App']['PerformFind'](arg1, arg2, arg3, arg4);
+}
+
+export function PerformInsert(arg1, arg2, arg3, arg4) {
+  return window['go']['app']['App']['PerformInsert'](arg1, arg2, arg3, arg4);
+}
diff --git a/frontend/wailsjs/go/main/App.js b/frontend/wailsjs/go/main/App.js
deleted file mode 100755
index ff1b889..0000000
--- a/frontend/wailsjs/go/main/App.js
+++ /dev/null
@@ -1,31 +0,0 @@
-// @ts-check
-// Cynhyrchwyd y ffeil hon yn awtomatig. PEIDIWCH Â MODIWL
-// This file is automatically generated. DO NOT EDIT
-
-export function DropDatabase(arg1, arg2) {
-  return window['go']['main']['App']['DropDatabase'](arg1, arg2);
-}
-
-export function Hosts() {
-  return window['go']['main']['App']['Hosts']();
-}
-
-export function OpenCollection(arg1, arg2, arg3) {
-  return window['go']['main']['App']['OpenCollection'](arg1, arg2, arg3);
-}
-
-export function OpenConnection(arg1) {
-  return window['go']['main']['App']['OpenConnection'](arg1);
-}
-
-export function OpenDatabase(arg1, arg2) {
-  return window['go']['main']['App']['OpenDatabase'](arg1, arg2);
-}
-
-export function PerformFind(arg1, arg2, arg3, arg4) {
-  return window['go']['main']['App']['PerformFind'](arg1, arg2, arg3, arg4);
-}
-
-export function PerformInsert(arg1, arg2, arg3, arg4) {
-  return window['go']['main']['App']['PerformInsert'](arg1, arg2, arg3, arg4);
-}
diff --git a/frontend/wailsjs/go/models.ts b/frontend/wailsjs/go/models.ts
index 7a37ffd..a620a89 100755
--- a/frontend/wailsjs/go/models.ts
+++ b/frontend/wailsjs/go/models.ts
@@ -1,4 +1,4 @@
-export namespace main {
+export namespace app {
 	
 	export class findResult {
 	    total: number;
diff --git a/internal/app/app.go b/internal/app/app.go
new file mode 100644
index 0000000..c5da997
--- /dev/null
+++ b/internal/app/app.go
@@ -0,0 +1,15 @@
+package app
+
+import "context"
+
+type App struct {
+	ctx context.Context
+}
+
+func NewApp() *App {
+	return &App{}
+}
+
+func (a *App) Startup(ctx context.Context) {
+	a.ctx = ctx
+}
diff --git a/internal/app/collection.go b/internal/app/collection.go
new file mode 100644
index 0000000..392f963
--- /dev/null
+++ b/internal/app/collection.go
@@ -0,0 +1,29 @@
+package app
+
+import (
+	"fmt"
+
+	"github.com/wailsapp/wails/v2/pkg/runtime"
+	"go.mongodb.org/mongo-driver/bson"
+)
+
+func (a *App) OpenCollection(hostKey, dbKey, collKey string) (result bson.M) {
+	client, ctx, close, err := a.connectToHost(hostKey)
+	if err != nil {
+		fmt.Println(err.Error())
+		return nil
+	}
+	command := bson.M{"collStats": collKey}
+	err = client.Database(dbKey).RunCommand(ctx, command).Decode(&result)
+	if err != nil {
+		fmt.Println(err.Error())
+		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+			Type:    runtime.ErrorDialog,
+			Title:   "Could not retrieve collection list for " + dbKey,
+			Message: err.Error(),
+		})
+		return nil
+	}
+	defer close()
+	return result
+}
diff --git a/internal/app/collection_find.go b/internal/app/collection_find.go
new file mode 100644
index 0000000..5382aa9
--- /dev/null
+++ b/internal/app/collection_find.go
@@ -0,0 +1,129 @@
+package app
+
+import (
+	"encoding/json"
+	"fmt"
+
+	"github.com/wailsapp/wails/v2/pkg/runtime"
+
+	"go.mongodb.org/mongo-driver/bson"
+	mongoOptions "go.mongodb.org/mongo-driver/mongo/options"
+)
+
+type findResult struct {
+	Total   int64       `json:"total"`
+	Results interface{} `json:"results"`
+}
+
+func (a *App) PerformFind(hostKey, dbKey, collKey string, formJson string) findResult {
+	var out findResult
+	var form struct {
+		Fields string `json:"fields"`
+		Limit  int64  `json:"limit"`
+		Query  string `json:"query"`
+		Skip   int64  `json:"skip"`
+		Sort   string `json:"sort"`
+	}
+
+	err := json.Unmarshal([]byte(formJson), &form)
+	if err != nil {
+		fmt.Println(err.Error())
+		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+			Type:    runtime.ErrorDialog,
+			Title:   "Couldn't parse form",
+			Message: err.Error(),
+		})
+		return out
+	}
+
+	client, ctx, close, err := a.connectToHost(hostKey)
+	if err != nil {
+		fmt.Println(err.Error())
+		return out
+	}
+
+	defer close()
+	var query bson.M
+	var projection bson.M
+	var sort bson.M
+
+	err = json.Unmarshal([]byte(form.Query), &query)
+	if err != nil {
+		fmt.Println(err.Error())
+		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+			Type:    runtime.ErrorDialog,
+			Title:   "Invalid query",
+			Message: err.Error(),
+		})
+		return out
+	}
+
+	err = json.Unmarshal([]byte(form.Fields), &projection)
+	if err != nil {
+		fmt.Println(err.Error())
+		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+			Type:    runtime.ErrorDialog,
+			Title:   "Invalid projection",
+			Message: err.Error(),
+		})
+		return out
+	}
+
+	err = json.Unmarshal([]byte(form.Sort), &sort)
+	if err != nil {
+		fmt.Println(err.Error())
+		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+			Type:    runtime.ErrorDialog,
+			Title:   "Invalid sort",
+			Message: err.Error(),
+		})
+		return out
+	}
+
+	opt := mongoOptions.FindOptions{
+		Limit:      &form.Limit,
+		Projection: projection,
+		Skip:       &form.Skip,
+		Sort:       sort,
+	}
+
+	total, err := client.Database(dbKey).Collection(collKey).CountDocuments(ctx, query, nil)
+	if err != nil {
+		fmt.Println(err.Error())
+		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+			Type:    runtime.ErrorDialog,
+			Title:   "Encountered an error while counting documents",
+			Message: err.Error(),
+		})
+		return out
+	}
+
+	cur, err := client.Database(dbKey).Collection(collKey).Find(ctx, query, &opt)
+	if err != nil {
+		fmt.Println(err.Error())
+		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+			Type:    runtime.ErrorDialog,
+			Title:   "Encountered an error while performing query",
+			Message: err.Error(),
+		})
+		return out
+	}
+
+	defer cur.Close(ctx)
+	var results []bson.M
+	err = cur.All(ctx, &results)
+
+	if err != nil {
+		fmt.Println(err.Error())
+		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+			Type:    runtime.ErrorDialog,
+			Title:   "Encountered an error while performing query",
+			Message: err.Error(),
+		})
+		return out
+	}
+
+	out.Results = results
+	out.Total = total
+	return out
+}
diff --git a/internal/app/collection_insert.go b/internal/app/collection_insert.go
new file mode 100644
index 0000000..67a0464
--- /dev/null
+++ b/internal/app/collection_insert.go
@@ -0,0 +1,49 @@
+package app
+
+import (
+	"encoding/json"
+	"fmt"
+	"strings"
+
+	"github.com/wailsapp/wails/v2/pkg/runtime"
+)
+
+func (a *App) PerformInsert(hostKey, dbKey, collKey, jsonData string) interface{} {
+	var data []interface{}
+
+	jsonData = strings.TrimSpace(jsonData)
+	if strings.HasPrefix(jsonData, "{") {
+		jsonData = "[" + jsonData + "]"
+	}
+
+	err := json.Unmarshal([]byte(jsonData), &data)
+	if err != nil {
+		fmt.Println(err.Error())
+		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+			Type:    runtime.ErrorDialog,
+			Title:   "Couldn't parse JSON",
+			Message: err.Error(),
+		})
+		return nil
+	}
+
+	client, ctx, close, err := a.connectToHost(hostKey)
+	if err != nil {
+		fmt.Println(err.Error())
+		return nil
+	}
+
+	defer close()
+	res, err := client.Database(dbKey).Collection(collKey).InsertMany(ctx, data)
+	if err != nil {
+		fmt.Println(err.Error())
+		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+			Type:    runtime.ErrorDialog,
+			Title:   "Encountered an error while performing query",
+			Message: err.Error(),
+		})
+		return nil
+	}
+
+	return res.InsertedIDs
+}
diff --git a/internal/app/connection.go b/internal/app/connection.go
new file mode 100644
index 0000000..32ef687
--- /dev/null
+++ b/internal/app/connection.go
@@ -0,0 +1,28 @@
+package app
+
+import (
+	"fmt"
+
+	"github.com/wailsapp/wails/v2/pkg/runtime"
+	"go.mongodb.org/mongo-driver/bson"
+)
+
+func (a *App) OpenConnection(hostKey string) (databases []string) {
+	client, ctx, close, err := a.connectToHost(hostKey)
+	if err != nil {
+		fmt.Println(err.Error())
+		return nil
+	}
+	databases, err = client.ListDatabaseNames(ctx, bson.M{})
+	if err != nil {
+		fmt.Println(err.Error())
+		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+			Type:    runtime.ErrorDialog,
+			Title:   "Could not retrieve database list",
+			Message: err.Error(),
+		})
+		return nil
+	}
+	defer close()
+	return databases
+}
diff --git a/internal/app/database.go b/internal/app/database.go
new file mode 100644
index 0000000..963ed1a
--- /dev/null
+++ b/internal/app/database.go
@@ -0,0 +1,59 @@
+package app
+
+import (
+	"fmt"
+
+	"github.com/wailsapp/wails/v2/pkg/runtime"
+	"go.mongodb.org/mongo-driver/bson"
+)
+
+func (a *App) OpenDatabase(hostKey, dbKey string) (collections []string) {
+	client, ctx, close, err := a.connectToHost(hostKey)
+	if err != nil {
+		fmt.Println(err.Error())
+		return nil
+	}
+	collections, err = client.Database(dbKey).ListCollectionNames(ctx, bson.D{})
+	if err != nil {
+		fmt.Println(err.Error())
+		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+			Type:    runtime.ErrorDialog,
+			Title:   "Could not retrieve collection list for " + dbKey,
+			Message: err.Error(),
+		})
+		return nil
+	}
+	defer close()
+	return collections
+}
+
+func (a *App) DropDatabase(hostKey, dbKey string) bool {
+	sure, _ := runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+		Title:         "Confirm",
+		Message:       "Are you sure you want to drop " + dbKey + "?",
+		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).Drop(ctx)
+	if err != nil {
+		fmt.Println(err.Error())
+		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+			Type:    runtime.ErrorDialog,
+			Title:   "Could not drop " + dbKey,
+			Message: err.Error(),
+		})
+		return false
+	}
+	defer close()
+	return true
+}
diff --git a/internal/app/host.go b/internal/app/host.go
new file mode 100644
index 0000000..d5d2dff
--- /dev/null
+++ b/internal/app/host.go
@@ -0,0 +1,58 @@
+package app
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"time"
+
+	"github.com/wailsapp/wails/v2/pkg/runtime"
+	"go.mongodb.org/mongo-driver/mongo"
+	mongoOptions "go.mongodb.org/mongo-driver/mongo/options"
+)
+
+type Host struct {
+	Name string `json:"name"`
+	URI  string `json:"uri"`
+}
+
+var hosts = map[string]Host{
+	"localhost": {Name: "Localhost", URI: "mongodb://localhost:27017"},
+	"tig":       {Name: "cmdb.myinfra.nl"},
+	"vbt":       {Name: "vbtverhuurmakelaars.nl"},
+}
+
+func (a *App) Hosts() map[string]Host {
+	return hosts
+}
+
+func (a *App) connectToHost(hostKey string) (*mongo.Client, context.Context, func(), error) {
+	h := hosts[hostKey]
+	if len(h.URI) == 0 {
+		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+			Type:    runtime.InfoDialog,
+			Title:   "Invalid uri",
+			Message: "You haven't specified a valid uri for the selected host.",
+		})
+		return nil, nil, nil, errors.New("invalid uri")
+	}
+
+	client, err := mongo.NewClient(mongoOptions.Client().ApplyURI(h.URI))
+
+	if err != nil {
+		fmt.Println(err.Error())
+		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+			Type:    runtime.ErrorDialog,
+			Title:   "Could not connect",
+			Message: "Failed to establish a connection with " + h.Name,
+		})
+		return nil, nil, nil, errors.New("could not establish a connection with " + h.Name)
+	}
+
+	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+	client.Connect(ctx)
+	return client, ctx, func() {
+		client.Disconnect(ctx)
+		cancel()
+	}, nil
+}
diff --git a/main.go b/main.go
index 8c75935..d46fb5a 100644
--- a/main.go
+++ b/main.go
@@ -1,337 +1,19 @@
 package main
 
 import (
-	"context"
 	"embed"
-	"encoding/json"
-	"errors"
-	"fmt"
-	"strings"
-	"time"
 
+	"github.com/garraflavatra/mongodup/internal/app"
 	"github.com/wailsapp/wails/v2"
 	"github.com/wailsapp/wails/v2/pkg/options"
 	"github.com/wailsapp/wails/v2/pkg/options/assetserver"
-	"github.com/wailsapp/wails/v2/pkg/runtime"
-
-	"go.mongodb.org/mongo-driver/bson"
-	"go.mongodb.org/mongo-driver/mongo"
-	mongoOptions "go.mongodb.org/mongo-driver/mongo/options"
 )
 
 //go:embed all:frontend/dist
 var assets embed.FS
 
-type Host struct {
-	Name string `json:"name"`
-	URI  string `json:"uri"`
-}
-
-var hosts = map[string]Host{
-	"localhost": {Name: "Localhost", URI: "mongodb://localhost:27017"},
-	"tig":       {Name: "cmdb.myinfra.nl"},
-	"vbt":       {Name: "vbtverhuurmakelaars.nl"},
-}
-
-type App struct {
-	ctx context.Context
-}
-
-func NewApp() *App {
-	return &App{}
-}
-
-func (a *App) Startup(ctx context.Context) {
-	a.ctx = ctx
-}
-
-func (a *App) Hosts() map[string]Host {
-	return hosts
-}
-
-func (a *App) connectToHost(hostKey string) (*mongo.Client, context.Context, func(), error) {
-	h := hosts[hostKey]
-	if len(h.URI) == 0 {
-		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
-			Type:    runtime.InfoDialog,
-			Title:   "Invalid uri",
-			Message: "You haven't specified a valid uri for the selected host.",
-		})
-		return nil, nil, nil, errors.New("invalid uri")
-	}
-
-	client, err := mongo.NewClient(mongoOptions.Client().ApplyURI(h.URI))
-
-	if err != nil {
-		fmt.Println(err.Error())
-		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
-			Type:    runtime.ErrorDialog,
-			Title:   "Could not connect",
-			Message: "Failed to establish a connection with " + h.Name,
-		})
-		return nil, nil, nil, errors.New("could not establish a connection with " + h.Name)
-	}
-
-	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
-	client.Connect(ctx)
-	return client, ctx, func() {
-		client.Disconnect(ctx)
-		cancel()
-	}, nil
-}
-
-func (a *App) OpenConnection(hostKey string) (databases []string) {
-	client, ctx, close, err := a.connectToHost(hostKey)
-	if err != nil {
-		fmt.Println(err.Error())
-		return nil
-	}
-	databases, err = client.ListDatabaseNames(ctx, bson.M{})
-	if err != nil {
-		fmt.Println(err.Error())
-		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
-			Type:    runtime.ErrorDialog,
-			Title:   "Could not retrieve database list",
-			Message: err.Error(),
-		})
-		return nil
-	}
-	defer close()
-	return databases
-}
-
-func (a *App) OpenDatabase(hostKey, dbKey string) (collections []string) {
-	client, ctx, close, err := a.connectToHost(hostKey)
-	if err != nil {
-		fmt.Println(err.Error())
-		return nil
-	}
-	collections, err = client.Database(dbKey).ListCollectionNames(ctx, bson.D{})
-	if err != nil {
-		fmt.Println(err.Error())
-		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
-			Type:    runtime.ErrorDialog,
-			Title:   "Could not retrieve collection list for " + dbKey,
-			Message: err.Error(),
-		})
-		return nil
-	}
-	defer close()
-	return collections
-}
-
-func (a *App) DropDatabase(hostKey, dbKey string) bool {
-	sure, _ := runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
-		Title:         "Confirm",
-		Message:       "Are you sure you want to drop " + dbKey + "?",
-		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).Drop(ctx)
-	if err != nil {
-		fmt.Println(err.Error())
-		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
-			Type:    runtime.ErrorDialog,
-			Title:   "Could not drop " + dbKey,
-			Message: err.Error(),
-		})
-		return false
-	}
-	defer close()
-	return true
-}
-
-func (a *App) OpenCollection(hostKey, dbKey, collKey string) (result bson.M) {
-	client, ctx, close, err := a.connectToHost(hostKey)
-	if err != nil {
-		fmt.Println(err.Error())
-		return nil
-	}
-	command := bson.M{"collStats": collKey}
-	err = client.Database(dbKey).RunCommand(ctx, command).Decode(&result)
-	if err != nil {
-		fmt.Println(err.Error())
-		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
-			Type:    runtime.ErrorDialog,
-			Title:   "Could not retrieve collection list for " + dbKey,
-			Message: err.Error(),
-		})
-		return nil
-	}
-	defer close()
-	return result
-}
-
-type findResult struct {
-	Total   int64       `json:"total"`
-	Results interface{} `json:"results"`
-}
-
-func (a *App) PerformFind(hostKey, dbKey, collKey string, formJson string) findResult {
-	var out findResult
-	var form struct {
-		Fields string `json:"fields"`
-		Limit  int64  `json:"limit"`
-		Query  string `json:"query"`
-		Skip   int64  `json:"skip"`
-		Sort   string `json:"sort"`
-	}
-
-	err := json.Unmarshal([]byte(formJson), &form)
-	if err != nil {
-		fmt.Println(err.Error())
-		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
-			Type:    runtime.ErrorDialog,
-			Title:   "Couldn't parse form",
-			Message: err.Error(),
-		})
-		return out
-	}
-
-	client, ctx, close, err := a.connectToHost(hostKey)
-	if err != nil {
-		fmt.Println(err.Error())
-		return out
-	}
-
-	defer close()
-	var query bson.M
-	var projection bson.M
-	var sort bson.M
-
-	err = json.Unmarshal([]byte(form.Query), &query)
-	if err != nil {
-		fmt.Println(err.Error())
-		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
-			Type:    runtime.ErrorDialog,
-			Title:   "Invalid query",
-			Message: err.Error(),
-		})
-		return out
-	}
-
-	err = json.Unmarshal([]byte(form.Fields), &projection)
-	if err != nil {
-		fmt.Println(err.Error())
-		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
-			Type:    runtime.ErrorDialog,
-			Title:   "Invalid projection",
-			Message: err.Error(),
-		})
-		return out
-	}
-
-	err = json.Unmarshal([]byte(form.Sort), &sort)
-	if err != nil {
-		fmt.Println(err.Error())
-		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
-			Type:    runtime.ErrorDialog,
-			Title:   "Invalid sort",
-			Message: err.Error(),
-		})
-		return out
-	}
-
-	opt := mongoOptions.FindOptions{
-		Limit:      &form.Limit,
-		Projection: projection,
-		Skip:       &form.Skip,
-		Sort:       sort,
-	}
-
-	total, err := client.Database(dbKey).Collection(collKey).CountDocuments(ctx, query, nil)
-	if err != nil {
-		fmt.Println(err.Error())
-		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
-			Type:    runtime.ErrorDialog,
-			Title:   "Encountered an error while counting documents",
-			Message: err.Error(),
-		})
-		return out
-	}
-
-	cur, err := client.Database(dbKey).Collection(collKey).Find(ctx, query, &opt)
-	if err != nil {
-		fmt.Println(err.Error())
-		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
-			Type:    runtime.ErrorDialog,
-			Title:   "Encountered an error while performing query",
-			Message: err.Error(),
-		})
-		return out
-	}
-
-	defer cur.Close(ctx)
-	var results []bson.M
-	err = cur.All(ctx, &results)
-
-	if err != nil {
-		fmt.Println(err.Error())
-		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
-			Type:    runtime.ErrorDialog,
-			Title:   "Encountered an error while performing query",
-			Message: err.Error(),
-		})
-		return out
-	}
-
-	out.Results = results
-	out.Total = total
-	return out
-}
-
-func (a *App) PerformInsert(hostKey, dbKey, collKey, jsonData string) interface{} {
-	var data []interface{}
-
-	jsonData = strings.TrimSpace(jsonData)
-	if strings.HasPrefix(jsonData, "{") {
-		jsonData = "[" + jsonData + "]"
-	}
-
-	err := json.Unmarshal([]byte(jsonData), &data)
-	if err != nil {
-		fmt.Println(err.Error())
-		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
-			Type:    runtime.ErrorDialog,
-			Title:   "Couldn't parse JSON",
-			Message: err.Error(),
-		})
-		return nil
-	}
-
-	client, ctx, close, err := a.connectToHost(hostKey)
-	if err != nil {
-		fmt.Println(err.Error())
-		return nil
-	}
-
-	defer close()
-	res, err := client.Database(dbKey).Collection(collKey).InsertMany(ctx, data)
-	if err != nil {
-		fmt.Println(err.Error())
-		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
-			Type:    runtime.ErrorDialog,
-			Title:   "Encountered an error while performing query",
-			Message: err.Error(),
-		})
-		return nil
-	}
-
-	return res.InsertedIDs
-}
-
 func main() {
-	app := NewApp()
+	app := app.NewApp()
 
 	err := wails.Run(&options.App{
 		Title:  "Mongodup",