package core

import (
	"encoding/json"
	"flag"
	"fmt"
	"os"
	"runtime"

	"gitee.com/rachel_os/fastsearch/global"
	"gitee.com/rachel_os/fastsearch/utils"
	"github.com/fatih/structs"
	"gopkg.in/yaml.v2"
)

// Parser 解析器
func Parser() *global.Config {

	var addr = flag.String("addr", "0.0.0.0:5679", "设置监听地址和端口")
	//兼容windows
	dir := fmt.Sprintf(".%sdata/db", string(os.PathSeparator))
	negative_dir := fmt.Sprintf(".%sdata/negative_data", string(os.PathSeparator))

	var dataDir = flag.String("data", dir, "设置数据存储目录")
	var negative_data_dir = flag.String("negative_data", negative_dir, "设置负面词库存储目录")

	var debug = flag.Bool("debug", false, "设置是否开启调试模式")

	var dictionaryPath = flag.String("dictionary", "./data/dictionary.txt", "设置词典路径")

	var enableAdmin = flag.Bool("enableAdmin", true, "设置是否开启后台管理")

	var gomaxprocs = flag.Int("gomaxprocs", runtime.NumCPU()*2, "设置GOMAXPROCS")

	var auth = flag.String("auth", "", "开启认证，例如: admin:123456")

	var enableGzip = flag.Bool("enableGzip", true, "是否开启gzip压缩")
	var timeout = flag.Int64("timeout", 10*60, "数据库超时关闭时间(秒)")
	var bufferNum = flag.Int("bufferNum", 1000, "分片缓冲数量")
	var allowDrop = flag.Bool("allow_drop", false, "允许删除数据库")
	var info = flag.Bool("info", false, "输出信息")

	var configPath = flag.String("config", "", "配置文件路径，配置此项其他参数忽略")
	flag.Parse()

	_config := &global.Config{}
	config := &global.Config{
		Addr:          *addr,
		Data:          *dataDir,
		INFO:          *info,
		Negative_data: *negative_data_dir,
		Debug:         *debug,
		AllowDrop:     *allowDrop,
		Dictionary:    *dictionaryPath,
		EnableAdmin:   *enableAdmin,
		Gomaxprocs:    *gomaxprocs,
		Auth:          *auth,
		EnableGzip:    *enableGzip,
		Timeout:       *timeout,
		BufferNum:     *bufferNum,
	}

	if *configPath != "" {
		//解析配置文件
		//file, err := ioutil.ReadFile(*configPath)
		file, err := os.ReadFile(*configPath) //详情：https://github.com/golang/go/issues/42026
		if err != nil {
			panic(err)
		}
		err = yaml.Unmarshal(file, _config)
		if err != nil {
			panic(err)
		}

		return _config
	}
	return config
}
func MergeConfig(cfg1, cfg2 *global.Config) {
	m1 := structs.Map(cfg1)
	m2 := structs.Map(cfg2)
	m3 := utils.MergeMap(m1, m2)
	c, _ := json.Marshal(m3)
	json.Unmarshal(c, cfg1)
}