mirror of
https://github.com/Mrs4s/go-cqhttp.git
synced 2025-05-05 19:43:49 +08:00
3ea9c2547c
return on filter file does not exist refactor: event filter 将event filter独立出来 refact: rate limit 将限速改成连接独立控制 switch to yaml config
130 lines
3.1 KiB
Go
130 lines
3.1 KiB
Go
package config
|
||
|
||
import (
|
||
_ "embed" // embed the default config file
|
||
"os"
|
||
"path"
|
||
|
||
log "github.com/sirupsen/logrus"
|
||
"gopkg.in/yaml.v3"
|
||
)
|
||
|
||
// DefaultConfig 默认配置文件
|
||
//go:embed default_config.yml
|
||
var DefaultConfig string
|
||
|
||
var currentPath = getCurrentPath()
|
||
|
||
// DefaultConfigFile 默认配置文件路径
|
||
var DefaultConfigFile = path.Join(currentPath, "config.yml")
|
||
|
||
// Config 总配置文件
|
||
type Config struct {
|
||
Account struct {
|
||
Uin int64 `yaml:"uin"`
|
||
Password string `yaml:"password"`
|
||
Encrypt bool `yaml:"encrypt"`
|
||
ReLogin struct {
|
||
Disabled bool `yaml:"disabled"`
|
||
Delay int `yaml:"delay"`
|
||
MaxTimes uint `yaml:"max-times"`
|
||
Interval int `yaml:"interval"`
|
||
}
|
||
UseSSOAddress bool `yaml:"use-sso-address"`
|
||
} `yaml:"account"`
|
||
|
||
Heartbeat struct {
|
||
Disabled bool `yaml:"disabled"`
|
||
Interval int `yaml:"interval"`
|
||
} `yaml:"heartbeat"`
|
||
|
||
Message struct {
|
||
PostFormat string `yaml:"post-format"`
|
||
IgnoreInvalidCQCode bool `yaml:"ignore-invalid-cqcode"`
|
||
ForceFragment bool `yaml:"force-fragment"`
|
||
FixURL bool `yaml:"fix-url"`
|
||
ProxyRewrite string `yaml:"proxy-rewrite"`
|
||
ReportSelfMessage bool `yaml:"report-self-message"`
|
||
} `yaml:"message"`
|
||
|
||
Output struct {
|
||
LogLevel string `yaml:"log-level"`
|
||
Debug bool `yaml:"debug"`
|
||
} `yaml:"output"`
|
||
|
||
Servers []map[string]yaml.Node `yaml:"servers"`
|
||
Database map[string]yaml.Node `yaml:"database"`
|
||
}
|
||
|
||
type MiddleWares struct {
|
||
AccessToken string `yaml:"access-token"`
|
||
Filter string `yaml:"filter"`
|
||
RateLimit struct {
|
||
Enabled bool `yaml:"enabled"`
|
||
Frequency float64 `yaml:"frequency"`
|
||
Bucket int `yaml:"bucket"`
|
||
} `yaml:"rate-limit"`
|
||
}
|
||
|
||
// HTTPServer HTTP通信相关配置
|
||
type HTTPServer struct {
|
||
Disabled bool `yaml:"disabled"`
|
||
Host string `yaml:"host"`
|
||
Port int `yaml:"port"`
|
||
Timeout int32 `yaml:"timeout"`
|
||
Post []struct {
|
||
URL string `yaml:"url"`
|
||
Secret string `yaml:"secret"`
|
||
}
|
||
|
||
MiddleWares `yaml:"middlewares"`
|
||
}
|
||
|
||
// WebsocketServer 正向WS相关配置
|
||
type WebsocketServer struct {
|
||
Disabled bool `yaml:"disabled"`
|
||
Host string `yaml:"host"`
|
||
Port int `yaml:"port"`
|
||
|
||
MiddleWares `yaml:"middlewares"`
|
||
}
|
||
|
||
// WebsocketReverse 反向WS相关配置
|
||
type WebsocketReverse struct {
|
||
Disabled bool `yaml:"disabled"`
|
||
Universal string `yaml:"universal"`
|
||
API string `yaml:"api"`
|
||
Event string `yaml:"event"`
|
||
ReconnectInterval int `yaml:"reconnect-interval"`
|
||
|
||
MiddleWares `yaml:"middlewares"`
|
||
}
|
||
|
||
// LevelDBConfig leveldb 相关配置
|
||
type LevelDBConfig struct {
|
||
Enable bool `yaml:"enable"`
|
||
}
|
||
|
||
// Get 从默认配置文件路径中获取
|
||
func Get() *Config {
|
||
file, err := os.Open(DefaultConfigFile)
|
||
if err != nil {
|
||
log.Error("获取配置文件失败: ", err)
|
||
return nil
|
||
}
|
||
config := &Config{}
|
||
if yaml.NewDecoder(file).Decode(config) != nil {
|
||
log.Fatal("配置文件不合法!", err)
|
||
}
|
||
return config
|
||
}
|
||
|
||
// getCurrentPath 获取当前文件的路径，直接返回string
|
||
func getCurrentPath() string {
|
||
cwd, e := os.Getwd()
|
||
if e != nil {
|
||
panic(e)
|
||
}
|
||
return cwd
|
||
}
|