110 lines
2.5 KiB
Go
110 lines
2.5 KiB
Go
package models
|
|
|
|
import (
|
|
"bufio"
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
|
|
"github.com/beego/beego/v2/client/httplib"
|
|
"github.com/beego/beego/v2/core/logs"
|
|
"gopkg.in/yaml.v2"
|
|
)
|
|
|
|
type Container struct {
|
|
Type string
|
|
Name string
|
|
Default bool
|
|
Address string
|
|
Username string
|
|
Password string
|
|
Path string
|
|
Version string
|
|
Token string
|
|
Available bool
|
|
Delete []string
|
|
Weigth int
|
|
Mode string
|
|
Reader *bufio.Reader
|
|
Config string
|
|
Limit int
|
|
}
|
|
type Yaml struct {
|
|
Containers []Container
|
|
Qrcode string
|
|
Master string
|
|
Mode string
|
|
Static string
|
|
Database string
|
|
QywxKey string `yaml:"qywx_key"`
|
|
Resident string
|
|
UserAgent string `yaml:"user_agent"`
|
|
Theme string
|
|
TelegramBotToken string `yaml:"telegram_bot_token"`
|
|
TelegramUserID int `yaml:"telegram_user_id"`
|
|
QQID int64 `yaml:"qquid"`
|
|
QQGroupID int64 `yaml:"qqgid"`
|
|
DefaultPriority int `yaml:"default_priority"`
|
|
NoGhproxy bool `yaml:"no_ghproxy"`
|
|
QbotPublicMode bool `yaml:"qbot_public_mode"`
|
|
DailyAssetPushCron string `yaml:"daily_asset_push_cron"`
|
|
Version string `yaml:"version"`
|
|
}
|
|
|
|
var Balance = "balance"
|
|
var Parallel = "parallel"
|
|
var GhProxy = "https://ghproxy.com/"
|
|
var Cdle = false
|
|
|
|
var Config Yaml
|
|
|
|
func initConfig() {
|
|
confDir := ExecPath + "/conf"
|
|
if _, err := os.Stat(confDir); err != nil {
|
|
os.MkdirAll(confDir, os.ModePerm)
|
|
}
|
|
for _, name := range []string{"app.conf", "config.yaml"} {
|
|
f, err := os.OpenFile(ExecPath+"/conf/"+name, os.O_RDWR|os.O_CREATE, 0777)
|
|
if err != nil {
|
|
logs.Warn(err)
|
|
}
|
|
s, _ := ioutil.ReadAll(f)
|
|
if len(s) == 0 {
|
|
logs.Info("下载配置%s", name)
|
|
r, err := httplib.Get(GhProxy + "https://raw.githubusercontent.com/cdle/xdd/main/conf/" + name).Response()
|
|
if err == nil {
|
|
io.Copy(f, r.Body)
|
|
}
|
|
}
|
|
f.Close()
|
|
}
|
|
content, err := ioutil.ReadFile(ExecPath + "/conf/config.yaml")
|
|
if err != nil {
|
|
logs.Warn("解析config.yaml读取错误: %v", err)
|
|
}
|
|
if yaml.Unmarshal(content, &Config) != nil {
|
|
logs.Warn("解析config.yaml出错: %v", err)
|
|
}
|
|
if Config.Master == "" {
|
|
Config.Master = "xxxx"
|
|
}
|
|
if Config.Mode != Parallel {
|
|
Config.Mode = Balance
|
|
}
|
|
if Config.Qrcode != "" {
|
|
Config.Theme = Config.Qrcode
|
|
}
|
|
//测试
|
|
if ExecPath == "/Users/cdle/Desktop/xdd" {
|
|
Cdle = true
|
|
Config.QQID = 17745270
|
|
Config.QQGroupID = 610790654
|
|
}
|
|
if Config.NoGhproxy {
|
|
GhProxy = ""
|
|
}
|
|
if Config.Database == "" {
|
|
Config.Database = ExecPath + "/.xdd.db"
|
|
}
|
|
}
|