🎨 improve code

This commit is contained in:
Yiwen-Chan
2021-02-13 20:42:53 +08:00
parent dbf8aa27f2
commit 2ad5102297
16 changed files with 340 additions and 586 deletions

View File

@@ -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
}

View File

@@ -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
}