package configure

import (
	"gitee.com/kristas/booting-go/framework/announce"
	"gitee.com/kristas/booting-go/framework/common/util/converter"
	"gitee.com/kristas/booting-go/framework/common/util/reflectx"
	"gopkg.in/yaml.v3"
	"io/ioutil"
	"reflect"
	"strings"
)

var _config = make(map[string]interface{})

func GetConfig() map[string]interface{} {
	return _config
}

func LoadLocalFileConfig(environment string) {
	configPaths := configProducer(environment)
	for _, path := range configPaths {
		bytes, err := ioutil.ReadFile(path)
		if err != nil {
			continue
		}
		ReceiveConfig(bytes)
	}
}

func configProducer(environment string) []string {
	return []string{
		"config.yml",
		"config.yaml",
		"config-" + environment + ".yml",
		"config-" + environment + ".yaml",
	}
}

func ReceiveConfig(config []byte) {
	var tmp = make(map[string]interface{})
	err := yaml.Unmarshal(config, tmp)
	if err != nil {
		panic(err)
	}
	_config = converter.MergeMap(_config, tmp)
}

func ParserString(valuePath string) interface{} {
	result := walkConfig(valuePath, _config)
	return result
}

func Inject(valuePath string, o interface{}) error {
	result := ParserString(valuePath)
	bytes, err := yaml.Marshal(result)
	if err != nil {
		return err
	}
	err = yaml.Unmarshal(bytes, o)
	if err != nil {
		return err
	}
	return nil
}

func InjectConfiguration(o announce.Configuration) error {
	result := ParserString(o.Prefix())
	bytes, err := yaml.Marshal(result)
	if err != nil {
		return err
	}
	err = yaml.Unmarshal(bytes, o)
	if err != nil {
		return err
	}
	return nil
}

func InjectStructure(structure interface{}) {
	reflectx.WalkStructField(structure, func(field reflect.StructField, value reflect.Value) {
		if tag, ok := field.Tag.Lookup("value"); ok {
			if value.CanSet() {
				tempVal := reflect.New(field.Type).Interface()
				Inject(tag, tempVal)
				value.Set(reflect.ValueOf(tempVal).Elem())
			}
		}
	}, true)
}

func walkConfig(s string, m map[string]interface{}) interface{} {
	if !strings.Contains(s, ".") {
		return m[s]
	}
	arr := strings.SplitN(s, ".", 2)
	if m[arr[0]] == nil {
		panic("config parse error: " + s)
	}
	innerMap := m[arr[0]].(map[string]interface{})
	return walkConfig(arr[1], innerMap)
}