package rpcx

import (
	"fmt"
	"os"
	"regexp"

	"yunyan/lego/core"
	"yunyan/lego/utils/container/id"
	"yunyan/lego/utils/mapstructure"

	"gopkg.in/yaml.v2"
)

// 匹配 ${VAR} 与 ${VAR:-default} 两种占位符
var envVarRe = regexp.MustCompile(`\$\{([A-Za-z_][A-Za-z0-9_]*)(?::-([^}]*))?\}`)

// expandEnv 展开配置内容里的环境变量占位符，便于多服务共用同一份环境文件（docker compose env_file）：
//   - ${VAR}            取环境变量 VAR，未设置时替换为空串
//   - ${VAR:-default}   取环境变量 VAR，未设置或为空时使用 default
func expandEnv(content string) string {
	return envVarRe.ReplaceAllStringFunc(content, func(m string) string {
		sub := envVarRe.FindStringSubmatch(m)
		name, def := sub[1], sub[2]
		if v, ok := os.LookupEnv(name); ok && v != "" {
			return v
		}
		return def
	})
}

type RPCXConfig struct {
	Port           int32    //监听短裤
	Addr           string   //对外地址
	ETCDServers    []string //etcd 地址
	UpdateInterval int32    //保活间隔
}

type Option func(*Options)

type Options struct {
	ConfPath   string
	Version    string //服务版本
	Setting    core.ServiceSttings
	RPCXConfig RPCXConfig
}

func SetConfPath(v string) Option {
	return func(o *Options) {
		o.ConfPath = v
	}
}

func SetVersion(v string) Option {
	return func(o *Options) {
		o.Version = v
	}
}

func newOptions(option ...Option) *Options {
	options := &Options{
		ConfPath: "conf/cluster.yaml",
	}
	for _, o := range option {
		o(options)
	}
	yamlFile, err := os.ReadFile(options.ConfPath)
	if err != nil {
		panic(fmt.Sprintf("读取服务配置【%s】文件失败err:%v:", options.ConfPath, err))
	}
	// 先做环境变量替换，再解析。这样公共配置（如 Redis 地址、key 前缀）只需在环境文件改一处。
	err = yaml.Unmarshal([]byte(expandEnv(string(yamlFile))), &options.Setting)
	if err != nil {
		panic(fmt.Sprintf("读取服务配置【%s】文件失败err:%v:", options.ConfPath, err))
	}
	if len(options.Setting.Type) == 0 || len(options.Setting.Tag) == 0 {
		panic(fmt.Sprintf("[%s] 配置缺少必要配置: %+v", options.ConfPath, options))
	}
	if len(options.Setting.Id) == 0 {
		options.Setting.Id = fmt.Sprintf("%s-%s", options.Setting.Type, id.NewXId())
	}
	if err = mapstructure.Decode(options.Setting.Exte, &options.RPCXConfig); err != nil {
		panic(fmt.Sprintf("[%s] 配置缺少必要配置 Exte: %+v", options.ConfPath, options))
	}
	return options
}