113 lines
3.8 KiB
Go
113 lines
3.8 KiB
Go
// 本工具仅在停服后调整指定配置目录的空白排版，并同步原有文件摘要。
|
||
package main
|
||
|
||
import (
|
||
"bytes"
|
||
"crypto/sha256"
|
||
"encoding/hex"
|
||
"encoding/json"
|
||
"errors"
|
||
"flag"
|
||
"fmt"
|
||
"os"
|
||
"path/filepath"
|
||
|
||
"git.pcux.cn/xuans/Smail/utils/atomicfile"
|
||
)
|
||
|
||
func main() {
|
||
if err := run(); err != nil {
|
||
fmt.Fprintln(os.Stderr, err)
|
||
os.Exit(1)
|
||
}
|
||
}
|
||
|
||
func run() error {
|
||
directory := flag.String("config", "", "已停止后端的配置目录")
|
||
backup := flag.String("backup", "", "保存原字节的独立备份目录")
|
||
apply := flag.Bool("apply", false, "确认停服，执行排版；默认只检查")
|
||
flag.Parse()
|
||
if !filepath.IsAbs(*directory) || !filepath.IsAbs(*backup) {
|
||
return errors.New("配置及备份目录必须使用明确的绝对路径")
|
||
}
|
||
info, err := os.Lstat(*directory)
|
||
if err != nil || !info.IsDir() || info.Mode()&os.ModeSymlink != 0 {
|
||
return errors.New("配置目录不可用或是链接")
|
||
}
|
||
names := []string{"smtp.json", "email.json", "ssl.json", "plugins.json", "config.json"}
|
||
original, formatted := map[string][]byte{}, map[string][]byte{}
|
||
for _, name := range names {
|
||
path := filepath.Join(*directory, name)
|
||
info, err := os.Lstat(path)
|
||
if err != nil || !info.Mode().IsRegular() || info.Size() > 4<<20 {
|
||
return errors.New("配置文件不可用、过大或不是普通文件")
|
||
}
|
||
data, err := os.ReadFile(path)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
var result bytes.Buffer
|
||
// 不解析再编码业务值，保留中文、密文、数字及转义的原始字节。
|
||
if err := json.Indent(&result, data, "", " "); err != nil {
|
||
return fmt.Errorf("%s 不是有效 JSON，未写入", name)
|
||
}
|
||
original[name], formatted[name] = data, result.Bytes()
|
||
}
|
||
var root struct {
|
||
Sections map[string]struct {
|
||
File string `json:"file"`
|
||
SHA256 string `json:"sha256"`
|
||
} `json:"sections"`
|
||
}
|
||
if err := json.Unmarshal(original["config.json"], &root); err != nil || len(root.Sections) != 4 {
|
||
return errors.New("仅支持已完成分区的五文件配置")
|
||
}
|
||
for _, name := range names[:4] {
|
||
reference := root.Sections[name[:len(name)-5]]
|
||
if reference.File != name || reference.SHA256 != digest(original[name]) {
|
||
return fmt.Errorf("%s 原有摘要不匹配，未写入", name)
|
||
}
|
||
old := []byte(`"sha256": "` + reference.SHA256 + `"`)
|
||
next := []byte(`"sha256": "` + digest(formatted[name]) + `"`)
|
||
if bytes.Count(formatted["config.json"], old) != 1 {
|
||
return errors.New("根配置摘要位置不唯一，未写入")
|
||
}
|
||
formatted["config.json"] = bytes.Replace(formatted["config.json"], old, next, 1)
|
||
}
|
||
changed := false
|
||
for _, name := range names {
|
||
changed = changed || !bytes.Equal(original[name], formatted[name])
|
||
}
|
||
if !*apply || !changed {
|
||
fmt.Printf("检查通过：五个 JSON 可读取，是否需要排版：%t。\n", changed)
|
||
return nil
|
||
}
|
||
// 备份目录必须不存在，拒绝覆盖之前的备份。
|
||
if err := os.Mkdir(*backup, 0700); err != nil {
|
||
return errors.New("无法创建独立备份目录，未改写配置")
|
||
}
|
||
for _, name := range names {
|
||
if err := atomicfile.Write(filepath.Join(*backup, name), original[name], 0600); err != nil {
|
||
return errors.New("备份未完成，未改写配置")
|
||
}
|
||
}
|
||
for _, name := range names {
|
||
data, err := os.ReadFile(filepath.Join(*directory, name))
|
||
if err != nil || !bytes.Equal(data, original[name]) {
|
||
return errors.New("备份期间配置发生变化，未改写配置")
|
||
}
|
||
}
|
||
for _, name := range names {
|
||
if err := atomicfile.Write(filepath.Join(*directory, name), formatted[name], 0600); err != nil {
|
||
return errors.New("写入未完成，请保持停服并使用本次备份恢复五个文件")
|
||
}
|
||
}
|
||
fmt.Println("五个 JSON 已分行缩进，摘要已同步；配置值、代次及密文未改变。")
|
||
return nil
|
||
}
|
||
|
||
func digest(data []byte) string {
|
||
value := sha256.Sum256(data)
|
||
return hex.EncodeToString(value[:])
|
||
}
|