spray/internal/config.go

package internal

import (
	"fmt"
	"github.com/goccy/go-yaml"
	"github.com/gookit/config/v2"
	"reflect"
	"strconv"
)

//var (
//	defaultConfigPath = ".config/spray/"
//	defaultConfigFile = "config.yaml"
//)
//
//func LoadDefault(v interface{}) {
//	dir, err := os.UserHomeDir()
//	if err != nil {
//		logs.Log.Error(err.Error())
//		return
//	}
//	if !files.IsExist(filepath.Join(dir, defaultConfigPath, defaultConfigFile)) {
//		err := os.MkdirAll(filepath.Join(dir, defaultConfigPath), 0o700)
//		if err != nil {
//			logs.Log.Error(err.Error())
//			return
//		}
//		f, err := os.Create(filepath.Join(dir, defaultConfigPath, defaultConfigFile))
//		if err != nil {
//			logs.Log.Error(err.Error())
//			return
//		}
//		err = LoadConfig(filepath.Join(dir, defaultConfigPath, defaultConfigFile), v)
//		if err != nil {
//			logs.Log.Error(err.Error())
//			return
//		}
//		var buf bytes.Buffer
//		_, err = config.DumpTo(&buf, config.Yaml)
//		if err != nil {
//			logs.Log.Error(err.Error())
//			return
//		}
//		fmt.Println(buf.String())
//		f.Sync()
//	}
//}

func LoadConfig(filename string, v interface{}) error {
	err := config.LoadFiles(filename)
	if err != nil {
		return err
	}
	err = config.Decode(v)
	if err != nil {
		return err
	}
	return nil
}

func convertToFieldType(fieldType reflect.StructField, defaultVal string) interface{} {
	switch fieldType.Type.Kind() {
	case reflect.Bool:
		val, err := strconv.ParseBool(defaultVal)
		if err == nil {
			return val
		}
	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
		val, err := strconv.ParseInt(defaultVal, 10, 64)
		if err == nil {
			return val
		}
	case reflect.Float32, reflect.Float64:
		val, err := strconv.ParseFloat(defaultVal, 64)
		if err == nil {
			return val
		}
	case reflect.String:
		return defaultVal
		// 可以根据需要扩展其他类型
	}
	return nil // 如果转换失败或类型不受支持，返回nil
}

func setFieldValue(field reflect.Value) interface{} {
	switch field.Kind() {
	case reflect.Bool:
		return false
	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
		return 0
	case reflect.Float32, reflect.Float64:
		return 0.0
	case reflect.Slice, reflect.Array:
		return []interface{}{} // 返回一个空切片
	case reflect.String:
		return ""
	case reflect.Struct:
		return make(map[string]interface{})
	default:
		return nil
	}
}

// extractConfigAndDefaults 提取带有 `config` 和 `default` 标签的字段
func extractConfigAndDefaults(v reflect.Value, result map[string]interface{}) {
	t := v.Type()
	for i := 0; i < v.NumField(); i++ {
		field := v.Field(i)
		fieldType := t.Field(i)
		configTag := fieldType.Tag.Get("config")
		defaultTag := fieldType.Tag.Get("default")

		if configTag != "" {
			var value interface{}
			if defaultTag != "" {
				value = convertToFieldType(fieldType, defaultTag)
			} else {
				value = setFieldValue(field)
			}
			if field.Kind() == reflect.Struct {
				nestedResult := make(map[string]interface{})
				extractConfigAndDefaults(field, nestedResult)
				result[configTag] = nestedResult
			} else {
				result[configTag] = value
			}
		}
	}
}

func initDefaultConfig(cfg interface{}) (string, error) {
	v := reflect.ValueOf(cfg)
	if v.Kind() != reflect.Struct {
		return "", fmt.Errorf("expected a struct, got %s", v.Kind())
	}

	result := make(map[string]interface{})
	extractConfigAndDefaults(v, result)

	yamlData, err := yaml.Marshal(result)
	if err != nil {
		return "", err
	}

	return string(yamlData), nil
}