Custom views!

internal/app/collection_find_views.go

@@ -0,0 +1,182 @@
+package app
+
+import (
+	"encoding/json"
+	"errors"
+	"fmt"
+	"io/ioutil"
+	"os"
+
+	"github.com/wailsapp/wails/v2/pkg/runtime"
+)
+
+type ViewType string
+
+const (
+	TableView ViewType = "table"
+	ListView  ViewType = "list"
+)
+
+type ViewColumn struct {
+	Key   string `json:"key"`
+	Width int64  `json:"width"`
+}
+
+type View struct {
+	Name                 string       `json:"name"`
+	Host                 string       `json:"host"`
+	Database             string       `json:"database"`
+	Collection           string       `json:"collection"`
+	Type                 ViewType     `json:"type"`
+	HideObjectIndicators bool         `json:"hideObjectIndicators"`
+	Columns              []ViewColumn `json:"columns"`
+}
+
+var BuiltInListView = View{
+	Name: "List",
+	Type: ListView,
+}
+
+type ViewStore map[string]View
+
+func updateViewStore(newData ViewStore) error {
+	filePath, err := appDataFilePath("views.json")
+	if err != nil {
+		return err
+	}
+
+	jsonData, err := json.MarshalIndent(newData, "", "\t")
+	if err != nil {
+		return err
+	}
+
+	err = ioutil.WriteFile(filePath, jsonData, os.ModePerm)
+	return err
+}
+
+func (a *App) Views() (ViewStore, error) {
+	views := make(ViewStore, 0)
+	filePath, err := appDataFilePath("views.json")
+	if err != nil {
+		fmt.Println(err.Error())
+		return views, err
+	}
+
+	jsonData, err := ioutil.ReadFile(filePath)
+	if err != nil {
+		// It's ok if the file cannot be opened, for example if it is not accessible.
+		// Therefore no error is returned.
+		fmt.Println(err.Error())
+		return views, nil
+	}
+
+	if len(jsonData) > 0 {
+		err = json.Unmarshal(jsonData, &views)
+		if err != nil {
+			fmt.Println(err.Error())
+			return nil, errors.New("views.json file contains malformatted JSON data")
+		}
+	}
+
+	views["list"] = BuiltInListView
+	return views, nil
+}
+
+// func (a *App) AddView(jsonData string) error {
+// 	views, err := a.Views()
+// 	if err != nil {
+// 		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+// 			Type:  runtime.InfoDialog,
+// 			Title: "Could not retrieve views",
+// 		})
+// 		return errors.New("could not retrieve existing view store")
+// 	}
+
+// 	var newView View
+// 	err = json.Unmarshal([]byte(jsonData), &newView)
+// 	if err != nil {
+// 		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+// 			Type:  runtime.InfoDialog,
+// 			Title: "Malformed JSON",
+// 		})
+// 		return errors.New("invalid JSON")
+// 	}
+
+// 	id, err := uuid.NewRandom()
+// 	if err != nil {
+// 		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+// 			Type:  runtime.InfoDialog,
+// 			Title: "Failed to generate a UUID",
+// 		})
+// 		return errors.New("could not generate a UUID")
+// 	}
+
+// 	views[id.String()] = newView
+// 	err = updateViewStore(views)
+// 	if err != nil {
+// 		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+// 			Type:  runtime.InfoDialog,
+// 			Title: "Could not update view store",
+// 		})
+// 		return errors.New("could not update view store")
+// 	}
+
+// 	return nil
+// }
+
+func (a *App) UpdateViewStore(jsonData string) error {
+	var viewStore ViewStore
+	err := json.Unmarshal([]byte(jsonData), &viewStore)
+	if err != nil {
+		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+			Type:  runtime.InfoDialog,
+			Title: "Malformed JSON",
+		})
+		return errors.New("invalid JSON")
+	}
+
+	err = updateViewStore(viewStore)
+	if err != nil {
+		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+			Type:  runtime.InfoDialog,
+			Title: "Could not update view store",
+		})
+		return errors.New("could not update view store")
+	}
+
+	return nil
+}
+
+func (a *App) RemoveView(viewKey string) error {
+	views, err := a.Views()
+	if err != nil {
+		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+			Type:  runtime.InfoDialog,
+			Title: "Could not retrieve views",
+		})
+		return errors.New("could not retrieve existing view store")
+	}
+
+	sure, _ := runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+		Title:         "Confirm",
+		Message:       "Are you sure you want to remove " + views[viewKey].Name + "?",
+		Buttons:       []string{"Yes", "No"},
+		DefaultButton: "Yes",
+		CancelButton:  "No",
+	})
+	if sure != "Yes" {
+		return errors.New("operation aborted")
+	}
+
+	delete(views, viewKey)
+	err = updateViewStore(views)
+
+	if err != nil {
+		runtime.MessageDialog(a.ctx, runtime.MessageDialogOptions{
+			Type:  runtime.InfoDialog,
+			Title: "Could not update view store",
+		})
+		return errors.New("could not update view store")
+	}
+	return nil
+}

internal/app/hosts.go

@@ -12,19 +12,8 @@ import (
 )
 
 type Host struct {
-	Name      string `json:"name"`
-	URI       string `json:"uri"`
-	Databases map[string]struct {
-		Collections map[string]struct {
-			ViewConfig struct {
-				HideObjectIndicators bool `json:"hideObjectIndicators"`
-				Columns              []struct {
-					Key   string `json:"key"`
-					Width int64  `json:"width"`
-				} `json:"columns"`
-			} `json:"viewConfig"`
-		} `json:"collections"`
-	} `json:"databases"`
+	Name string `json:"name"`
+	URI  string `json:"uri"`
 }
 
 func updateHostsFile(newData map[string]Host) error {