package option

import (
	"gopkg.in/yaml.v2"
	"io"
	"log"
	"os"
)

//自定义选项

type Option struct {
	UDesk struct {
		Filters []struct {
			Filter7D  string `yaml:"filter7d,omitempty"`
			Filter14D string `yaml:"filter14d,omitempty"`
		} `yaml:"filters"`
		Macros []string `yaml:"macros"`
	} `yaml:"uDesk"`
	VolumePath string `yaml:"volumePath"`
	FileServer struct {
		Address      string `yaml:"address"`
		Preservation int    `yaml:"preservation"`
	} `yaml:"fileServer"`
	ModelConf struct {
		API   string `yaml:"api"`
		Token string `yaml:"token"`
	} `yaml:"modelConf"`
	Wechat struct {
		Webhook string `yaml:"webhook"`
	} `yaml:"wechat"`
}

// 新选项对象
func newOptionObj() *Option {
	return &Option{}
}

func (o *Option) readYaml(yamlFile string) *Option {
	file, err := os.Open(yamlFile)
	if err != nil {
		log.Printf("打开yaml文件错误，错误是：%v", err)
		return nil
	}
	defer file.Close()

	data, err := io.ReadAll(file)
	if err != nil {
		log.Printf("读取yaml文件错误，错误是：%v", err)
		return nil
	}
	var option Option

	err = yaml.Unmarshal(data, &option)

	if err != nil {
		log.Printf("yaml文件内容反序列化失败，错误是：%v", err)
		return nil
	}
	return &option
}

func loadALLOption() *Option {
	option := newOptionObj()
	option = option.readYaml("/conf/option.yaml")
	return option
}

var (
	UDESK_FILTER7D     = loadALLOption().UDesk.Filters[0].Filter7D
	UDSK_FILTER14D     = loadALLOption().UDesk.Filters[1].Filter14D
	UDESK_MACROS       = loadALLOption().UDesk.Macros
	FILESERVER_ADDRESS = loadALLOption().FileServer.Address
	PRESERVATION       = loadALLOption().FileServer.Preservation
	VOLUMEPATH         = loadALLOption().VolumePath
	MODEL_API          = loadALLOption().ModelConf.API
	MODEL_TOKEN        = loadALLOption().ModelConf.Token
	WECHAT_WEBHOOK     = loadALLOption().Wechat.Webhook
)