update
This commit is contained in:
@@ -1,8 +1,14 @@
|
|||||||
package core
|
package core
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"io"
|
||||||
|
"io/ioutil"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
|
|
||||||
|
"github.com/astaxie/beego/logs"
|
||||||
|
"github.com/beego/beego/v2/adapter/httplib"
|
||||||
|
"gopkg.in/yaml.v2"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Yaml struct {
|
type Yaml struct {
|
||||||
@@ -12,3 +18,33 @@ type Yaml struct {
|
|||||||
var ExecPath, _ = filepath.Abs(filepath.Dir(os.Args[0]))
|
var ExecPath, _ = filepath.Abs(filepath.Dir(os.Args[0]))
|
||||||
|
|
||||||
var Config Yaml
|
var Config Yaml
|
||||||
|
|
||||||
|
func ReadYaml(confDir string, conf interface{}, url string) {
|
||||||
|
path := confDir + "config.yaml"
|
||||||
|
if _, err := os.Stat(confDir); err != nil {
|
||||||
|
os.MkdirAll(confDir, os.ModePerm)
|
||||||
|
}
|
||||||
|
f, err := os.OpenFile(path, os.O_RDWR|os.O_CREATE, 0777)
|
||||||
|
if err != nil {
|
||||||
|
logs.Warn(err)
|
||||||
|
}
|
||||||
|
s, _ := ioutil.ReadAll(f)
|
||||||
|
if len(s) == 0 {
|
||||||
|
logs.Info("下载配置%s", url)
|
||||||
|
r, err := httplib.Get("https://ghproxy.com/" + url).Response()
|
||||||
|
if err == nil {
|
||||||
|
io.Copy(f, r.Body)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
f.Close()
|
||||||
|
content, err := ioutil.ReadFile(path)
|
||||||
|
if err != nil {
|
||||||
|
logs.Warn("解析配置文件%s读取错误: %v", path, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if yaml.Unmarshal(content, conf) != nil {
|
||||||
|
logs.Warn("解析配置文件%s出错: %v", path, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
logs.Info("解析配置文件%s", path)
|
||||||
|
}
|
||||||
|
|||||||
+1
-1
@@ -19,6 +19,7 @@ func init() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
initStore()
|
initStore()
|
||||||
|
ReadYaml(ExecPath+"/conf/", &Config, "https://raw.githubusercontent.com/cdle/sillyGirl/main/conf/demo_config.yaml")
|
||||||
InitReplies()
|
InitReplies()
|
||||||
initToHandleMessage()
|
initToHandleMessage()
|
||||||
file, err := os.Open("/etc/sillyGirl/sets.conf")
|
file, err := os.Open("/etc/sillyGirl/sets.conf")
|
||||||
@@ -36,5 +37,4 @@ func init() {
|
|||||||
}
|
}
|
||||||
initSys()
|
initSys()
|
||||||
Duration = time.Duration(sillyGirl.GetInt("duration", 5)) * time.Second
|
Duration = time.Duration(sillyGirl.GetInt("duration", 5)) * time.Second
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user