first commit

agent/internal/config/config.go

@@ -0,0 +1,45 @@
+package config
+
+import (
+	"os"
+	"time"
+
+	"gopkg.in/yaml.v3"
+)
+
+const ConfigPath = "/etc/keymanager/config.yaml"
+
+type Config struct {
+	ServerURL    string        `yaml:"server_url"`
+	ServerID     string        `yaml:"server_id"`
+	PreRegToken  string        `yaml:"pre_reg_token"`
+	AgentToken   string        `yaml:"agent_token"`
+	PollInterval time.Duration `yaml:"poll_interval"`
+	TLS          bool          `yaml:"tls"`
+}
+
+func Load() (*Config, error) {
+	data, err := os.ReadFile(ConfigPath)
+	if err != nil {
+		return nil, err
+	}
+	var cfg Config
+	if err := yaml.Unmarshal(data, &cfg); err != nil {
+		return nil, err
+	}
+	if cfg.PollInterval == 0 {
+		cfg.PollInterval = 30 * time.Second
+	}
+	return &cfg, nil
+}
+
+func Save(cfg *Config) error {
+	data, err := yaml.Marshal(cfg)
+	if err != nil {
+		return err
+	}
+	if err := os.MkdirAll("/etc/keymanager", 0700); err != nil {
+		return err
+	}
+	return os.WriteFile(ConfigPath, data, 0600)
+}