package config

import (
	"fmt"
	"os"
	"path/filepath"
	"strings"
	"sync"

	"gopkg.in/yaml.v3"
)

var (
	config *Config
	once   sync.Once
)

func Init(args []string) {
	_initConfig(args)
}

type Config struct {
	Server struct {
		Host string `yaml:"host"`
		Port int    `yaml:"port"`
		Env  string `yaml:"env"`
	} `yaml:"server"`
	// 内部加工配置
	LogFile string
}

func GetConfig() *Config {
	return _initConfig(nil)
}

func _initConfig(args []string) *Config {
	if config == nil {
		once.Do(func() {
			c, err := loadConfig(args)
			if err != nil {
				panic(err)
			}
			config = c
		})
	}
	return config
}

// 获取配置文件的绝对路径
func getConfigFile(confFile string) string {
	if confFile != "" {
		if _, err := os.Stat(confFile); err == nil {
			return confFile
		}
	}
	// 方案：优先从工作目录查找（适配 GoLand 直接运行的场景）
	workDir, err := os.Getwd()
	if err == nil {
		workDirConfig := filepath.Join(workDir, "output/conf/conf.yml")
		if _, err := os.Stat(workDirConfig); err == nil {
			return workDirConfig
		}
		workDirConfig = filepath.Join(workDir, "conf/conf.yml")
		if _, err := os.Stat(workDirConfig); err == nil {
			return workDirConfig
		}
		if strings.Contains(workDirConfig, "starter") {
			values := strings.Split(workDirConfig, "/starter")
			if len(values) > 1 {
				v := filepath.Join(values[0], "starter", "conf/conf.yml")
				if _, err := os.Stat(v); err == nil {
					return v
				}
			}
		}
	}

	// 方案：从程序执行路径推导（适配部署环境）
	exePath, err := os.Executable()
	if err != nil {
		panic(fmt.Sprintf("获取程序路径失败: %v", err))
	}
	exeDir := filepath.Dir(exePath)
	// 判断是否是部署环境（bin 目录下）
	if filepath.Base(exeDir) == "bin" {
		// 部署环境：output/bin/ -> output/conf/conf.yml
		return filepath.Join(exeDir, "../conf/conf.yml")
	}

	// 兜底：程序目录下的 conf/conf.yaml
	return filepath.Join(exeDir, "conf/conf.yml")
}

// 加载配置文件
func loadConfig(args []string) (*Config, error) {
	confFile := getValueFromArgs(args, "config")
	configFile := getConfigFile(confFile)
	fmt.Println("configFile:", configFile)
	// 读取配置文件内容
	data, err := os.ReadFile(configFile)
	if err != nil {
		return nil, fmt.Errorf("读取配置文件失败: %v", err)
	}

	// 解析 YAML 到结构体
	var c Config
	if err := yaml.Unmarshal(data, &c); err != nil {
		return nil, fmt.Errorf("解析 YAML 失败: %v", err)
	}
	prefix := ""
	if !strings.Contains(configFile, "/output/conf/") {
		prefix = "/output"
	}
	pathArr := strings.Split(configFile, "/conf/")
	if len(pathArr) == 0 {
		panic("配置文件需要在`conf`目录下")
	}
	logDir := pathArr[0] + prefix + "/logs"
	fmt.Println("logDir:", logDir)
	// 自动创建日志目录（关键：避免目录不存在导致写入失败）
	if err := os.MkdirAll(logDir, 0755); err != nil {
		panic(fmt.Sprintf("创建日志目录失败: %v", err))
	}
	c.LogFile = filepath.Join(logDir, "starter.log")
	return &c, nil
}

func getValueFromArgs(args []string, key string) string {
	for _, arg := range args {
		if strings.HasPrefix(arg, fmt.Sprintf("%s=", key)) {
			value := strings.ReplaceAll(arg, fmt.Sprintf("%s=", key), "")
			return value
		}
	}
	return ""
}