64 lines
1.3 KiB
Go
64 lines
1.3 KiB
Go
package config
|
||
|
||
import (
|
||
"github.com/fsnotify/fsnotify"
|
||
"github.com/spf13/viper"
|
||
"strconv"
|
||
"strings"
|
||
)
|
||
|
||
const (
|
||
LogBasePath string = "./log/"
|
||
WelcomeContent string = "欢迎使用E5SubBot!"
|
||
HelpContent string = `
|
||
命令：
|
||
/my 查看已绑定账户信息
|
||
/bind 绑定新账户
|
||
/unbind 解绑账户
|
||
/export 导出账户信息(JSON)
|
||
/help 帮助
|
||
源码及使用方法：https://github.com/iyear/E5SubBot
|
||
`
|
||
)
|
||
|
||
var (
|
||
ErrorTimes map[string]int //错误次数
|
||
BindMaxNum int
|
||
ErrMaxTimes int
|
||
Notice string
|
||
Admins []int64
|
||
)
|
||
|
||
func InitConfig() {
|
||
viper.SetConfigName("config")
|
||
viper.AddConfigPath(".")
|
||
err := viper.ReadInConfig()
|
||
if err != nil {
|
||
|
||
}
|
||
viper.SetDefault("errlimit", 5)
|
||
viper.SetDefault("bindmax", 5)
|
||
|
||
BindMaxNum = viper.GetInt("bindmax")
|
||
ErrMaxTimes = viper.GetInt("errlimit")
|
||
Notice = viper.GetString("notice")
|
||
Admins = getAdmins()
|
||
|
||
viper.WatchConfig()
|
||
viper.OnConfigChange(func(e fsnotify.Event) {
|
||
BindMaxNum = viper.GetInt("bindmax")
|
||
ErrMaxTimes = viper.GetInt("errlimit")
|
||
Notice = viper.GetString("notice")
|
||
Admins = getAdmins()
|
||
})
|
||
}
|
||
func getAdmins() []int64 {
|
||
var result []int64
|
||
admins := strings.Split(viper.GetString("admin"), ",")
|
||
for _, v := range admins {
|
||
id, _ := strconv.ParseInt(v, 10, 64)
|
||
result = append(result, id)
|
||
}
|
||
return result
|
||
}
|