mirror of
https://github.com/garraflavatra/rolens.git
synced 2024-11-28 10:22:24 +01:00
125 lines
3.5 KiB
Go
125 lines
3.5 KiB
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"path"
|
|
"path/filepath"
|
|
"runtime"
|
|
|
|
"github.com/garraflavatra/rolens/internal/ui"
|
|
"github.com/ncruces/zenity"
|
|
wailsRuntime "github.com/wailsapp/wails/v2/pkg/runtime"
|
|
)
|
|
|
|
type EnvironmentInfo struct {
|
|
Arch string `json:"arch"`
|
|
BuildType string `json:"buildType"`
|
|
Platform string `json:"platform"`
|
|
|
|
HasMongoExport bool `json:"hasMongoExport"`
|
|
HasMongoDump bool `json:"hasMongoDump"`
|
|
|
|
HomeDirectory string `json:"homeDirectory"`
|
|
DataDirectory string `json:"dataDirectory"`
|
|
LogDirectory string `json:"logDirectory"`
|
|
DownloadDirectory string `json:"downloadDirectory"`
|
|
}
|
|
|
|
type App struct {
|
|
Env EnvironmentInfo
|
|
State map[string]string
|
|
ctx context.Context
|
|
ui *ui.UI
|
|
}
|
|
|
|
func NewApp() *App {
|
|
a := &App{
|
|
State: make(map[string]string),
|
|
}
|
|
|
|
_, err := exec.LookPath("mongodump")
|
|
a.Env.HasMongoDump = err == nil
|
|
|
|
_, err = exec.LookPath("mongoexport")
|
|
a.Env.HasMongoExport = err == nil
|
|
|
|
a.Env.HomeDirectory, err = os.UserHomeDir()
|
|
if err != nil {
|
|
panic(errors.New("encountered an error while getting home directory"))
|
|
}
|
|
|
|
switch runtime.GOOS {
|
|
case "windows":
|
|
a.Env.DataDirectory = filepath.Join(a.Env.HomeDirectory, "/AppData/Local/Rolens")
|
|
a.Env.LogDirectory = filepath.Join(a.Env.HomeDirectory, "/AppData/Local/Rolens/Logs")
|
|
a.Env.DownloadDirectory = filepath.Join(a.Env.HomeDirectory, "/Downloads")
|
|
case "darwin":
|
|
a.Env.DataDirectory = filepath.Join(a.Env.HomeDirectory, "/Library/Application Support/Rolens")
|
|
a.Env.LogDirectory = filepath.Join(a.Env.HomeDirectory, "/Library/Logs/Rolens")
|
|
a.Env.DownloadDirectory = filepath.Join(a.Env.HomeDirectory, "/Downloads")
|
|
case "linux":
|
|
a.Env.DataDirectory = filepath.Join(a.Env.HomeDirectory, "/.config/rolens")
|
|
a.Env.LogDirectory = filepath.Join(a.Env.HomeDirectory, "/.config/rolens/logs")
|
|
a.Env.DownloadDirectory = filepath.Join(a.Env.HomeDirectory, "/Downloads")
|
|
default:
|
|
panic(errors.New("unsupported platform"))
|
|
}
|
|
|
|
os.MkdirAll(a.Env.DataDirectory, os.ModePerm)
|
|
os.MkdirAll(a.Env.LogDirectory, os.ModePerm)
|
|
|
|
return a
|
|
}
|
|
|
|
func (a *App) Startup(ctx context.Context, ui *ui.UI) {
|
|
a.ctx = ctx
|
|
a.ui = ui
|
|
wailsRuntime.LogInfo(a.ctx, "Runcycle: Startup")
|
|
|
|
wailsEnv := wailsRuntime.Environment(a.ctx)
|
|
a.Env.Arch = wailsEnv.Arch
|
|
a.Env.BuildType = wailsEnv.BuildType
|
|
a.Env.Platform = wailsEnv.Platform
|
|
|
|
jsonEnv, err := json.MarshalIndent(a.Env, "", " ")
|
|
if err != nil {
|
|
wailsRuntime.LogWarningf(a.ctx, "Could not marshal environment info: %s", err.Error())
|
|
}
|
|
err = os.WriteFile(path.Join(a.Env.LogDirectory, "environment.json"), jsonEnv, 0644)
|
|
if err != nil {
|
|
wailsRuntime.LogWarningf(a.ctx, "Could not save environment.json: %s", err.Error())
|
|
}
|
|
}
|
|
|
|
func (a *App) Shutdown(ctx context.Context) {
|
|
wailsRuntime.LogInfo(a.ctx, "Runcycle: Shutdown")
|
|
}
|
|
|
|
func (a *App) Environment() EnvironmentInfo {
|
|
return a.Env
|
|
}
|
|
|
|
func (a *App) PurgeLogDirectory() {
|
|
err := zenity.Question("Are you sure you want to remove all logfiles?", zenity.Title("Confirm"), zenity.WarningIcon)
|
|
if err == zenity.ErrCanceled {
|
|
return
|
|
}
|
|
|
|
err = os.RemoveAll(a.Env.LogDirectory)
|
|
if err == nil {
|
|
zenity.Info("Successfully purged log directory.", zenity.InfoIcon)
|
|
} else {
|
|
zenity.Error(err.Error(), zenity.Title("Encountered an error while purging log directory."), zenity.WarningIcon)
|
|
}
|
|
}
|
|
|
|
func (a *App) ReportSharedStateVariable(key, value string) {
|
|
a.State[key] = value
|
|
wailsRuntime.LogDebug(a.ctx, fmt.Sprintf("State: %s=\"%s\"", key, value))
|
|
}
|