/
conf.go
75 lines (66 loc) · 1.53 KB
/
conf.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
package conf
import (
"io/ioutil"
"os"
"sync"
"time"
log "github.com/sirupsen/logrus"
"gopkg.in/yaml.v2"
)
type UserData struct {
UserName string `yaml:"User_Name"`
NowKey string `yaml:"Now_Key"`
AwsSecret map[string]*AwsSecret `yaml:"Aws_Secret"`
}
type AwsSecret struct {
Id string `yaml:"id"`
Secret string `yaml:"secret"`
Proxy string `yaml:"Proxy"`
}
type Conf struct {
LogLevel string `yaml:"Log_Level"`
BotToken string `yaml:"Bot_Token"`
UserInfo map[int]*UserData `yaml:"User_Info"`
}
func New() *Conf {
return &Conf{}
}
var Lock = sync.RWMutex{}
func (c *Conf) LoadConfig() error {
r, readErr := ioutil.ReadFile("./config.yml")
if readErr != nil {
if os.IsNotExist(readErr) {
log.Error("Config file not found")
log.Error("Write default config file")
c.LogLevel = "error"
c.UserInfo = map[int]*UserData{0: {}}
c.BotToken = "Tg Bot Token"
writeErr := c.SaveConfig()
if writeErr != nil {
log.Error("Write file error: ", "writeErr")
os.Exit(1)
}
log.Error("已将默认配置文件写出,请填写bot token后重新启动")
log.Error("将于10秒后退出")
time.Sleep(time.Second * 10)
os.Exit(1)
}
return readErr
}
unmErr := yaml.Unmarshal(r, c)
if unmErr != nil {
return unmErr
}
return nil
}
func (c *Conf) SaveConfig() error {
rt, marErr := yaml.Marshal(c)
if marErr != nil {
return marErr
}
writeErr := ioutil.WriteFile("./config.yml", rt, 0644)
if writeErr != nil {
return writeErr
}
return nil
}