package config

import (
	"time"

	"github.com/spf13/viper"
)

// VictoriaMetricsConfig 核心配置：对接VM的基础信息
type VictoriaMetricsConfig struct {
	Addr        string        `mapstructure:"addr" validate:"required,url"` // VM服务地址，必填+URL格式校验
	Timeout     time.Duration `mapstructure:"timeout" default:"10s"`        // 请求超时，默认10秒
	AuthToken   string        `mapstructure:"auth_token"`                   // 认证Token
	MetricPath  string        `mapstructure:"metric_path" default:"/api/v1/write"` // 指标写入路径
}

// CollectorConfig 采集器配置：定义数据采集规则
type CollectorConfig struct {
	Interval  time.Duration `mapstructure:"interval" default:"15s"` // 采集间隔，默认15秒
	Metrics   []string      `mapstructure:"metrics"`                // 需采集的指标列表
	Labels    map[string]string `mapstructure:"labels"`            // 全局标签
	IgnoreErr bool          `mapstructure:"ignore_err" default:"false"` // 是否忽略采集错误
}

// AgentConfig 整合所有配置模块
type AgentConfig struct {
	VM        VictoriaMetricsConfig `mapstructure:"vm"`
	Collector CollectorConfig       `mapstructure:"collector"`
	Log       LogConfig             `mapstructure:"log"`
	Global    GlobalConfig          `mapstructure:"global"`
}

// LogConfig 日志配置
type LogConfig struct {
	Level string `mapstructure:"level" default:"info"`
	Path  string `mapstructure:"path" default:"./logs/agent.log"`
}

// GlobalConfig 全局配置
type GlobalConfig struct {
	MaxRetry int `mapstructure:"max_retry" default:"3"`
	Debug    bool `mapstructure:"debug" default:"false"`
}

package config

import (
	"flag"
	"fmt"
	"os"

	"github.com/go-playground/validator/v10"
	"github.com/spf13/viper"
)

var (
	cfgFile string
	validate *validator.Validate
)

// 初始化命令行参数
func init() {
	flag.StringVar(&cfgFile, "config", "./config/agent.yaml", "agent配置文件路径")
	validate = validator.New()
	// 注册自定义校验规则（如端口范围、VM地址合法性）
	_ = validate.RegisterValidation("vm_addr", validateVMAddr)
}

// LoadConfig 加载并解析配置
func LoadConfig() (*AgentConfig, error) {
	// 1. 解析命令行参数
	flag.Parse()

	// 2. 初始化viper，配置多源加载规则
	v := viper.New()
	// 设置配置文件格式
	v.SetConfigType("yaml")
	// 优先读取命令行指定的配置文件
	if cfgFile != "" {
		v.SetConfigFile(cfgFile)
	} else {
		// 默认配置路径
		v.AddConfigPath("./config")
		v.SetConfigName("agent")
	}

	// 3. 读取配置文件
	if err := v.ReadInConfig(); err != nil {
		return nil, fmt.Errorf("读取配置文件失败: %v", err)
	}

	// 4. 环境变量覆盖：支持通过环境变量修改配置（适配容器化部署）
	// 环境变量前缀为 MCP_VM_，例如 MCP_VM_VM_ADDR=192.168.1.100:8428
	v.AutomaticEnv()
	v.SetEnvPrefix("MCP_VM")
	v.SetEnvKeyReplacer(viper.NewDefaultEnvKeyReplacer())

	// 5. 映射配置到结构体
	var agentCfg AgentConfig
	if err := v.Unmarshal(&agentCfg); err != nil {
		return nil, fmt.Errorf("配置解析失败: %v", err)
	}

	// 6. 配置校验
	if err := validate.Struct(&agentCfg); err != nil {
		return nil, fmt.Errorf("配置校验失败: %v", err)
	}

	// 7. 调试模式下打印配置（便于运维排查）
	if agentCfg.Global.Debug {
		fmt.Printf("加载配置完成: %+v\n", agentCfg)
	}

	return &agentCfg, nil
}

// 自定义校验规则：校验VM地址合法性（示例）
func validateVMAddr(fl validator.FieldLevel) bool {
	addr := fl.Field().String()
	// 简单校验：包含IP/域名+端口，且端口为数字
	if addr == "" {
		return false
	}
	// 实际场景可补充更严格的校验逻辑
	return true
}

package config

import (
	"log"
	"time"

	"github.com/fsnotify/fsnotify"
	"github.com/spf13/viper"
)

// WatchConfig 监听配置文件变更，实现热更新
func (c *AgentConfig) WatchConfig(v *viper.Viper, onChange func(*AgentConfig)) {
	v.WatchConfig()
	v.OnConfigChange(func(e fsnotify.Event) {
		log.Printf("配置文件%s发生变更，开始热更新", e.Name)
		// 重新加载配置
		var newCfg AgentConfig
		if err := v.Unmarshal(&newCfg); err != nil {
			log.Printf("配置热更新失败: %v", err)
			return
		}
		// 校验新配置
		if err := validate.Struct(&newCfg); err != nil {
			log.Printf("新配置校验失败，放弃更新: %v", err)
			return
		}
		// 更新全局配置
		*c = newCfg
		// 执行配置变更后的回调（如重启采集器、更新VM连接）
		onChange(c)
		log.Println("配置热更新完成")
	})
	// 防止goroutine退出
	select {}
}

三、SRE 自定义 Agent 配置的核心技巧总结