Started working on shell feature (WIP)

internal/app/app.go

@@ -26,6 +26,7 @@ type EnvironmentInfo struct {
 
 	HasMongoExport bool `json:"hasMongoExport"`
 	HasMongoDump   bool `json:"hasMongoDump"`
+	HasMongoShell  bool `json:"hasMongoShell"`
 
 	HomeDirectory     string `json:"homeDirectory"`
 	DataDirectory     string `json:"dataDirectory"`
@@ -50,6 +51,9 @@ func NewApp(version string) *App {
 	_, err = exec.LookPath("mongoexport")
 	a.Env.HasMongoExport = err == nil
 
+	_, err = exec.LookPath("mongosh")
+	a.Env.HasMongoShell = err == nil
+
 	a.Env.HomeDirectory, err = os.UserHomeDir()
 	if err != nil {
 		panic(errors.New("encountered an error while getting home directory"))

internal/app/collection_shell.go

@@ -0,0 +1,83 @@
+package app
+
+import (
+	"fmt"
+	"os"
+	"os/exec"
+	"path"
+
+	"github.com/google/uuid"
+	"github.com/wailsapp/wails/v2/pkg/runtime"
+)
+
+type ExecuteShellScriptResult struct {
+	Output           string `json:"output"`
+	Status           int    `json:"status"`
+	ErrorTitle       string `json:"errorTitle"`
+	ErrorDescription string `json:"errorDescription"`
+}
+
+func (a *App) ExecuteShellScript(hostKey, dbKey, collKey, script string) (result ExecuteShellScriptResult) {
+	if !a.Env.HasMongoShell {
+		result.ErrorTitle = "mongosh not found"
+		result.ErrorDescription = "The mongosh executable is required to run a shell script. Please see https://www.mongodb.com/docs/mongodb-shell/install/"
+		return
+	}
+
+	hosts, err := a.Hosts()
+	if err != nil {
+		result.ErrorTitle = "Could not get hosts"
+		result.ErrorDescription = err.Error()
+		return
+	}
+
+	host, hostFound := hosts[hostKey]
+	if !hostFound {
+		result.ErrorTitle = "The specified host does not seem to exist"
+		return
+	}
+
+	id, err := uuid.NewRandom()
+	if err != nil {
+		runtime.LogErrorf(a.ctx, "Shell: failed to generate a UUID: %s", err.Error())
+		result.ErrorTitle = "Could not generate UUID"
+		result.ErrorDescription = err.Error()
+		return
+	}
+
+	dirname := path.Join(a.Env.DataDirectory, "Shell Scripts")
+	fname := path.Join(dirname, fmt.Sprintf("%s.js", id.String()))
+
+	if err := os.MkdirAll(dirname, os.ModePerm); err != nil {
+		runtime.LogWarningf(a.ctx, "Shell: failed to mkdir %s", err.Error())
+		result.ErrorTitle = "Could not create temporary directory"
+		result.ErrorDescription = err.Error()
+		return
+	}
+
+	script = fmt.Sprintf("db = connect('%s');\n\n%s", host.URI, script)
+
+	if err := os.WriteFile(fname, []byte(script), os.ModePerm); err != nil {
+		runtime.LogWarningf(a.ctx, "Shell: failed to write script to %s", err.Error())
+		result.ErrorTitle = "Could not create temporary script file"
+		result.ErrorDescription = err.Error()
+		return
+	}
+
+	cmd := exec.Command("mongosh", "--file", fname)
+	stdout, err := cmd.Output()
+
+	if exiterr, ok := err.(*exec.ExitError); ok {
+		result.Status = exiterr.ExitCode()
+	} else if err != nil {
+		runtime.LogWarningf(a.ctx, "Shell: failed to execute: mongosh --file %s: %s", fname, err.Error())
+		result.ErrorTitle = "Could not execute script"
+		result.ErrorDescription = err.Error()
+		return
+	} else {
+		result.Status = 0
+	}
+
+	result.Output = string(stdout)
+	return
+}