updates

internal/settings/settings.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"log/slog"
 	"os"
+	"path/filepath"
 	"vctp/internal/utils"
 
 	"gopkg.in/yaml.v2"
@@ -19,10 +20,29 @@ type Settings struct {
 // SettingsYML struct holds various runtime data that is too cumbersome to specify via command line, eg replacement properties
 type SettingsYML struct {
 	Settings struct {
-		TenantsToFilter    []string `yaml:"tenants_to_filter"`
-		NodeChargeClusters []string `yaml:"node_charge_clusters"`
-		SrmActiveActiveVms []string `yaml:"srm_activeactive_vms"`
-		VcenterAddresses   []string `yaml:"vcenter_addresses"`
+		LogLevel                         string   `yaml:"log_level"`
+		LogOutput                        string   `yaml:"log_output"`
+		DatabaseDriver                   string   `yaml:"database_driver"`
+		DatabaseURL                      string   `yaml:"database_url"`
+		BindIP                           string   `yaml:"bind_ip"`
+		BindPort                         int      `yaml:"bind_port"`
+		BindDisableTLS                   bool     `yaml:"bind_disable_tls"`
+		TLSCertFilename                  string   `yaml:"tls_cert_filename"`
+		TLSKeyFilename                   string   `yaml:"tls_key_filename"`
+		VcenterUsername                  string   `yaml:"vcenter_username"`
+		VcenterPassword                  string   `yaml:"vcenter_password"`
+		VcenterInsecure                  bool     `yaml:"vcenter_insecure"`
+		VcenterEventPollingSeconds       int      `yaml:"vcenter_event_polling_seconds"`
+		VcenterInventoryPollingSeconds   int      `yaml:"vcenter_inventory_polling_seconds"`
+		VcenterInventorySnapshotSeconds  int      `yaml:"vcenter_inventory_snapshot_seconds"`
+		VcenterInventoryAggregateSeconds int      `yaml:"vcenter_inventory_aggregate_seconds"`
+		HourlySnapshotMaxAgeDays         int      `yaml:"hourly_snapshot_max_age_days"`
+		DailySnapshotMaxAgeMonths        int      `yaml:"daily_snapshot_max_age_months"`
+		SnapshotCleanupCron              string   `yaml:"snapshot_cleanup_cron"`
+		TenantsToFilter                  []string `yaml:"tenants_to_filter"`
+		NodeChargeClusters               []string `yaml:"node_charge_clusters"`
+		SrmActiveActiveVms               []string `yaml:"srm_activeactive_vms"`
+		VcenterAddresses                 []string `yaml:"vcenter_addresses"`
 	} `yaml:"settings"`
 }
 
@@ -65,3 +85,49 @@ func (s *Settings) ReadYMLSettings() error {
 
 	return nil
 }
+
+func (s *Settings) WriteYMLSettings() error {
+	if s.Values == nil {
+		return errors.New("settings are not loaded")
+	}
+	if len(s.SettingsPath) == 0 {
+		return errors.New("settings file path not specified")
+	}
+
+	data, err := yaml.Marshal(s.Values)
+	if err != nil {
+		return fmt.Errorf("unable to encode settings file: %w", err)
+	}
+
+	mode := os.FileMode(0o644)
+	if info, err := os.Stat(s.SettingsPath); err == nil {
+		mode = info.Mode().Perm()
+	}
+
+	dir := filepath.Dir(s.SettingsPath)
+	tmp, err := os.CreateTemp(dir, "vctp-settings-*.yml")
+	if err != nil {
+		return fmt.Errorf("unable to create temp settings file: %w", err)
+	}
+	tmpName := tmp.Name()
+	defer func() {
+		_ = os.Remove(tmpName)
+	}()
+
+	if _, err := tmp.Write(data); err != nil {
+		_ = tmp.Close()
+		return fmt.Errorf("unable to write temp settings file: %w", err)
+	}
+	if err := tmp.Chmod(mode); err != nil {
+		_ = tmp.Close()
+		return fmt.Errorf("unable to set temp settings permissions: %w", err)
+	}
+	if err := tmp.Close(); err != nil {
+		return fmt.Errorf("unable to close temp settings file: %w", err)
+	}
+	if err := os.Rename(tmpName, s.SettingsPath); err != nil {
+		return fmt.Errorf("unable to replace settings file: %w", err)
+	}
+
+	return nil
+}