mirror of
https://github.com/fosrl/newt.git
synced 2026-02-08 05:56:40 +00:00
121 lines
3.2 KiB
Go
121 lines
3.2 KiB
Go
package websocket
|
|
|
|
import (
|
|
"encoding/json"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
|
|
"github.com/fosrl/newt/logger"
|
|
)
|
|
|
|
func getConfigPath(clientType string) string {
|
|
configFile := os.Getenv("CONFIG_FILE")
|
|
if configFile == "" {
|
|
var configDir string
|
|
switch runtime.GOOS {
|
|
case "darwin":
|
|
configDir = filepath.Join(os.Getenv("HOME"), "Library", "Application Support", clientType+"-client")
|
|
case "windows":
|
|
logDir := filepath.Join(os.Getenv("PROGRAMDATA"), "olm")
|
|
configDir = filepath.Join(logDir, clientType+"-client")
|
|
default: // linux and others
|
|
configDir = filepath.Join(os.Getenv("HOME"), ".config", clientType+"-client")
|
|
}
|
|
|
|
if err := os.MkdirAll(configDir, 0755); err != nil {
|
|
log.Printf("Failed to create config directory: %v", err)
|
|
}
|
|
|
|
return filepath.Join(configDir, "config.json")
|
|
}
|
|
|
|
return configFile
|
|
}
|
|
|
|
func (c *Client) loadConfig() error {
|
|
originalConfig := *c.config // Store original config to detect changes
|
|
configPath := getConfigPath(c.clientType)
|
|
|
|
if c.config.ID != "" && c.config.Secret != "" && c.config.Endpoint != "" {
|
|
logger.Debug("Config already provided, skipping loading from file")
|
|
// Check if config file exists, if not, we should save it
|
|
if _, err := os.Stat(configPath); os.IsNotExist(err) {
|
|
logger.Info("Config file does not exist at %s, will create it", configPath)
|
|
c.configNeedsSave = true
|
|
}
|
|
return nil
|
|
}
|
|
|
|
logger.Info("Loading config from: %s", configPath)
|
|
data, err := os.ReadFile(configPath)
|
|
if err != nil {
|
|
if os.IsNotExist(err) {
|
|
logger.Info("Config file does not exist at %s, will create it with provided values", configPath)
|
|
c.configNeedsSave = true
|
|
return nil
|
|
}
|
|
return err
|
|
}
|
|
|
|
var config Config
|
|
if err := json.Unmarshal(data, &config); err != nil {
|
|
return err
|
|
}
|
|
|
|
// Track what was loaded from file vs provided by CLI
|
|
fileHadID := c.config.ID == ""
|
|
fileHadSecret := c.config.Secret == ""
|
|
fileHadCert := c.config.TlsClientCert == ""
|
|
fileHadEndpoint := c.config.Endpoint == ""
|
|
|
|
if c.config.ID == "" {
|
|
c.config.ID = config.ID
|
|
}
|
|
if c.config.Secret == "" {
|
|
c.config.Secret = config.Secret
|
|
}
|
|
if c.config.TlsClientCert == "" {
|
|
c.config.TlsClientCert = config.TlsClientCert
|
|
}
|
|
if c.config.Endpoint == "" {
|
|
c.config.Endpoint = config.Endpoint
|
|
c.baseURL = config.Endpoint
|
|
}
|
|
|
|
// Check if CLI args provided values that override file values
|
|
if (!fileHadID && originalConfig.ID != "") ||
|
|
(!fileHadSecret && originalConfig.Secret != "") ||
|
|
(!fileHadCert && originalConfig.TlsClientCert != "") ||
|
|
(!fileHadEndpoint && originalConfig.Endpoint != "") {
|
|
logger.Info("CLI arguments provided, config will be updated")
|
|
c.configNeedsSave = true
|
|
}
|
|
|
|
logger.Debug("Loaded config from %s", configPath)
|
|
logger.Debug("Config: %+v", c.config)
|
|
|
|
return nil
|
|
}
|
|
|
|
func (c *Client) saveConfig() error {
|
|
if !c.configNeedsSave {
|
|
logger.Debug("Config has not changed, skipping save")
|
|
return nil
|
|
}
|
|
|
|
configPath := getConfigPath(c.clientType)
|
|
data, err := json.MarshalIndent(c.config, "", " ")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
logger.Info("Saving config to: %s", configPath)
|
|
err = os.WriteFile(configPath, data, 0644)
|
|
if err == nil {
|
|
c.configNeedsSave = false // Reset flag after successful save
|
|
}
|
|
return err
|
|
}
|