study_xxqg/conf/config.go

package conf

import (
	_ "embed"
	"os"

	log "github.com/sirupsen/logrus"
	"gopkg.in/yaml.v3"
)

// Config
//  @Description:
//
type Config struct {
	Model       int    `json:"model" yaml:"model"`
	LogLevel    string `json:"log_level" yaml:"log_level"`
	ShowBrowser bool   `json:"show_browser" yaml:"show_browser"`
	Scheme      string `json:"scheme" yaml:"scheme"`
	Push        struct {
		Ding struct {
			Enable      bool   `json:"enable" yaml:"enable"`
			AccessToken string `json:"access_token" yaml:"access_token"`
			Secret      string `json:"secret" yaml:"secret"`
		} `json:"ding" yaml:"ding"`
		PushPlus struct {
			Enable bool   `json:"enable" yaml:"enable"`
			Token  string `json:"token" yaml:"token"`
		} `json:"push_plus" yaml:"push_plus"`
	} `json:"push" yaml:"push"`
	TG struct {
		Enable bool   `json:"enable" yaml:"enable"`
		Token  string `json:"token" yaml:"token"`
		ChatID int64  `json:"chat_id" yaml:"chat_id"`
		Proxy  string `json:"proxy" yaml:"proxy"`
	} `json:"tg" yaml:"tg"`
	QQ struct {
	}
	Web struct {
		Enable   bool   `json:"enable" yaml:"enable"`
		Account  string `json:"account" yaml:"account"`
		Password string `json:"password" yaml:"password"`
		Host     string `json:"host" yaml:"host"`
		Port     int    `json:"port" yaml:"port"`
	} `json:"web"`
	Cron      string `json:"cron" yaml:"cron"`
	EdgePath  string `json:"edge_path" yaml:"edge_path"`
	QrCOde    bool   `json:"qr_code" yaml:"qr_code"`
	StartWait int    `json:"start_wait" yaml:"start_wait"`
	// cookie强制过期时间，单位为h
	ForceExpiration int `json:"force_expiration" yaml:"force_expiration"`
	Retry           struct {
		// 重试次数
		Times int `json:"times" yaml:"times"`
		// 重试时间
		Intervals int `json:"intervals" yaml:"intervals"`
	} `json:"retry" yaml:"retry"`

	Wechat struct {
		Enable        bool   `json:"enable" yaml:"enable"`
		Token         string `json:"token" yaml:"token"`
		Secret        string `json:"secret" yaml:"secret"`
		AppID         string `json:"app_id" yaml:"app_id"`
		LoginTempID   string `json:"login_temp_id" yaml:"login_temp_id"`
		NormalTempID  string `json:"normal_temp_id" yaml:"normal_temp_id"`
		PushLoginWarn bool   `json:"push_login_warn" yaml:"push_login_warn"`
	} `json:"wechat" yaml:"wechat"`
	// 专项答题可接受的最小值
	SpecialMinScore int `json:"special_min_score" yaml:"special_min_score"`

	JiGuangPush struct {
		Enable bool   `json:"enable" yaml:"enable"`
		Secret string `json:"secret" yaml:"secret"`
		AppKey string `json:"app_key" yaml:"app_key"`
	} `json:"ji_guang_push" yaml:"ji_guang_push"`
}

var (
	config = Config{
		Model: 1,
	}
)

//go:embed config_default.yml
var defaultConfig []byte

// GetConfig
/**
 * @Description:
 * @return Config
 */
func GetConfig() Config {
	file, err := os.ReadFile("./config/config.yml")
	if err != nil {
		log.Warningln("检测到配置文件可能不存在")
		err := os.WriteFile("./config/config.yml", defaultConfig, 0666)
		if err != nil {
			log.Errorln("写入到配置文件出现错误")
			log.Errorln(err.Error())
			return Config{}
		}
		log.Infoln("成功写入到配置文件,请重启应用")
		os.Exit(3)
	}
	err = yaml.Unmarshal(file, &config)
	if err != nil {
		log.Errorln(err.Error())
		return Config{}
	}
	if config.Scheme == "" {
		config.Scheme = "https://johlanse.github.io/study_xxqg/scheme.html?"
	}
	if config.SpecialMinScore == 0 {
		config.SpecialMinScore = 10
	}
	return config
}