mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2026-02-12 10:10:25 +00:00
🎨 improve code
This commit is contained in:
@@ -1,47 +0,0 @@
|
||||
package utils
|
||||
|
||||
import (
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
func PathExecute() string {
|
||||
dir, _ := os.Getwd()
|
||||
return dir + "\\"
|
||||
}
|
||||
|
||||
func PathExists(path string) bool {
|
||||
_, err := os.Stat(path)
|
||||
return err == nil || os.IsExist(err)
|
||||
}
|
||||
|
||||
func ReadAllText(path string) string {
|
||||
b, err := ioutil.ReadFile(path)
|
||||
if err != nil {
|
||||
return ""
|
||||
}
|
||||
return string(b)
|
||||
}
|
||||
|
||||
func WriteAllText(path, text string) {
|
||||
_ = ioutil.WriteFile(path, []byte(text), 0644)
|
||||
}
|
||||
|
||||
func CreatePath(path string) error {
|
||||
err := os.MkdirAll(path, 0644)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func Int2Str(val int64) string {
|
||||
str := strconv.FormatInt(val, 10)
|
||||
return str
|
||||
}
|
||||
|
||||
func Str2Int(str string) int64 {
|
||||
val, _ := strconv.ParseInt(str, 10, 64)
|
||||
return val
|
||||
}
|
||||
@@ -1,58 +0,0 @@
|
||||
package utils
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"gopkg.in/yaml.v3"
|
||||
)
|
||||
|
||||
type YamlConfig struct {
|
||||
Version string `yaml:"插件版本"`
|
||||
Host string `yaml:"监听地址"`
|
||||
Port string `yaml:"监听端口"`
|
||||
AccessToken string `yaml:"Token"`
|
||||
Master []string `yaml:"主人QQ"`
|
||||
}
|
||||
|
||||
func DefaultConfig() *YamlConfig {
|
||||
return &YamlConfig{
|
||||
Version: "1",
|
||||
Host: "127.0.0.1",
|
||||
Port: "8080",
|
||||
AccessToken: "",
|
||||
Master: []string{
|
||||
"66666666",
|
||||
"88888888",
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func Load(p string) *YamlConfig {
|
||||
if !PathExists(p) {
|
||||
DefaultConfig().Save(p)
|
||||
}
|
||||
c := YamlConfig{}
|
||||
err := yaml.Unmarshal([]byte(ReadAllText(p)), &c)
|
||||
if err != nil {
|
||||
fmt.Println("[GroupManager] 尝试加载配置文件失败: 读取文件失败")
|
||||
fmt.Println("[GroupManager] 原配置文件已备份")
|
||||
os.Rename(p, p+".backup"+strconv.FormatInt(time.Now().Unix(), 10))
|
||||
DefaultConfig().Save(p)
|
||||
}
|
||||
c = YamlConfig{}
|
||||
yaml.Unmarshal([]byte(ReadAllText(p)), &c)
|
||||
return &c
|
||||
}
|
||||
|
||||
func (c *YamlConfig) Save(p string) error {
|
||||
data, err := yaml.Marshal(c)
|
||||
if err != nil {
|
||||
fmt.Println("[GroupManager] 写入新的配置文件失败")
|
||||
return err
|
||||
}
|
||||
WriteAllText(p, string(data))
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user