111 lines
2.0 KiB
Go
111 lines
2.0 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"pulsegate-gui/internal/models"
|
|
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
func DefaultPath() string {
|
|
configDir, err := os.UserConfigDir()
|
|
if err != nil {
|
|
home, _ := os.UserHomeDir()
|
|
return filepath.Join(home, ".config", "pulsegate", "config.yaml")
|
|
}
|
|
|
|
return filepath.Join(configDir, "pulsegate", "config.yaml")
|
|
}
|
|
|
|
func EnsureExists(path string) error {
|
|
if _, err := os.Stat(path); err == nil {
|
|
return nil
|
|
}
|
|
|
|
if err := os.MkdirAll(filepath.Dir(path), 0700); err != nil {
|
|
return err
|
|
}
|
|
|
|
cfg := models.AppConfig{
|
|
Settings: models.Settings{
|
|
Theme: "neon-green",
|
|
Terminal: models.TerminalSettings{
|
|
Term: "xterm-256color",
|
|
EnableKittyFix: true,
|
|
},
|
|
},
|
|
Servers: []models.Server{
|
|
{
|
|
Name: "Example Server",
|
|
Host: "10.0.0.10",
|
|
User: "root",
|
|
Port: 22,
|
|
Group: "Homelab",
|
|
Auth: "key",
|
|
Key: "~/.ssh/id_ed25519",
|
|
KittyFix: true,
|
|
},
|
|
},
|
|
QuickCommands: []models.QuickCommand{
|
|
{Name: "Disk Usage", Command: "df -h"},
|
|
{Name: "RAM Usage", Command: "free -h"},
|
|
{Name: "Uptime", Command: "uptime"},
|
|
},
|
|
}
|
|
|
|
return Save(path, cfg)
|
|
}
|
|
|
|
func Load(path string) (models.AppConfig, error) {
|
|
var cfg models.AppConfig
|
|
|
|
data, err := os.ReadFile(path)
|
|
if err != nil {
|
|
return cfg, err
|
|
}
|
|
|
|
if err := yaml.Unmarshal(data, &cfg); err != nil {
|
|
return cfg, err
|
|
}
|
|
|
|
normalize(&cfg)
|
|
return cfg, nil
|
|
}
|
|
|
|
func Save(path string, cfg models.AppConfig) error {
|
|
normalize(&cfg)
|
|
|
|
data, err := yaml.Marshal(&cfg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := os.MkdirAll(filepath.Dir(path), 0700); err != nil {
|
|
return err
|
|
}
|
|
|
|
return os.WriteFile(path, data, 0600)
|
|
}
|
|
|
|
func normalize(cfg *models.AppConfig) {
|
|
if cfg.Settings.Theme == "" {
|
|
cfg.Settings.Theme = "neon-green"
|
|
}
|
|
|
|
if cfg.Settings.Terminal.Term == "" {
|
|
cfg.Settings.Terminal.Term = "xterm-256color"
|
|
}
|
|
|
|
for i := range cfg.Servers {
|
|
if cfg.Servers[i].Port == 0 {
|
|
cfg.Servers[i].Port = 22
|
|
}
|
|
|
|
if cfg.Servers[i].Auth == "" {
|
|
cfg.Servers[i].Auth = "key"
|
|
}
|
|
}
|
|
}
|