Renamed dump.go to database_dump.go

internal/app/database_dump.go

@@ -0,0 +1,71 @@
+package app
+
+import (
+	"encoding/json"
+	"fmt"
+	"os/exec"
+	"path"
+	"strings"
+
+	"github.com/ncruces/zenity"
+	"github.com/wailsapp/wails/v2/pkg/runtime"
+)
+
+type DumpInfo struct {
+	OutDir   string   `json:"outdir"`
+	Filename string   `json:"filename"`
+	HostKey  string   `json:"hostKey"`
+	DbKey    string   `json:"dbKey"`
+	CollKeys []string `json:"collKeys"`
+}
+
+func (a *App) PerformDump(jsonData string) bool {
+	var info DumpInfo
+	err := json.Unmarshal([]byte(jsonData), &info)
+	if err != nil {
+		runtime.LogError(a.ctx, "Could not unmarshal dump form")
+		runtime.LogError(a.ctx, err.Error())
+		zenity.Error(err.Error(), zenity.Title("Could not parse JSON"), zenity.ErrorIcon)
+		return false
+	}
+
+	hosts, err := a.Hosts()
+	if err != nil {
+		zenity.Error(err.Error(), zenity.Title("Error while getting hosts"), zenity.ErrorIcon)
+		return false
+	}
+	host := hosts[info.HostKey]
+
+	if !a.Env.HasMongoDump {
+		zenity.Error("You need to install mongodump to perform a dump.", zenity.Title("Additional tooling required"), zenity.ErrorIcon)
+		return false
+	}
+
+	args := make([]string, 0)
+	args = append(args, fmt.Sprintf(`--uri="%v"`, host.URI))
+
+	if info.DbKey != "" {
+		args = append(args, fmt.Sprintf(`--db="%v"`, info.DbKey))
+
+		if info.CollKeys != nil {
+			args = append(args, fmt.Sprintf(`--collection="%v"`, info.CollKeys[0]))
+		}
+	}
+
+	args = append(args, fmt.Sprintf(`--out="%v"`, path.Join(info.OutDir, info.Filename)))
+	cmd := exec.Command("mongodump", args...)
+	var stdout strings.Builder
+	var stderr strings.Builder
+	cmd.Stdout = &stdout
+	cmd.Stderr = &stderr
+	err = cmd.Run()
+
+	runtime.LogInfo(a.ctx, "Performing dump, executing command: mongodump "+strings.Join(args, " "))
+	runtime.LogInfo(a.ctx, "mongodump stdout: "+stdout.String())
+	runtime.LogInfo(a.ctx, "mongodump sterr: "+stderr.String())
+	if err != nil {
+		runtime.LogWarning(a.ctx, "Error while executing mongodump: "+err.Error())
+	}
+
+	return err == nil
+}