启动的时候尝试从备份文件中恢复配置

internal/configs/api_config.go

@@ -1,9 +1,12 @@
 package configs
 
 import (
+	teaconst "github.com/TeaOSLab/EdgeAdmin/internal/const"
 	"github.com/go-yaml/yaml"
 	"github.com/iwind/TeaGo/Tea"
 	"io/ioutil"
+	"os"
+	"path/filepath"
 )
 
 // API配置
@@ -17,7 +20,26 @@ type APIConfig struct {
 
 // 加载API配置
 func LoadAPIConfig() (*APIConfig, error) {
-	data, err := ioutil.ReadFile(Tea.ConfigFile("api.yaml"))
+	// 候选文件
+	localFile := Tea.ConfigFile("api.yaml")
+	isFromLocal := false
+	paths := []string{localFile}
+	homeDir, err := os.UserHomeDir()
+	if err == nil {
+		paths = append(paths, homeDir+"/."+teaconst.ProcessName+"/api.yaml")
+	}
+	paths = append(paths, "/etc/"+teaconst.ProcessName+"/api.yaml")
+
+	var data []byte
+	for _, path := range paths {
+		data, err = ioutil.ReadFile(path)
+		if err == nil {
+			if path == localFile {
+				isFromLocal = true
+			}
+			break
+		}
+	}
 	if err != nil {
 		return nil, err
 	}
@@ -28,6 +50,11 @@ func LoadAPIConfig() (*APIConfig, error) {
 		return nil, err
 	}
 
+	if !isFromLocal {
+		// 恢复文件
+		_ = ioutil.WriteFile(localFile, data, 0666)
+	}
+
 	return config, nil
 }
 
@@ -37,5 +64,38 @@ func (this *APIConfig) WriteFile(path string) error {
 	if err != nil {
 		return err
 	}
-	return ioutil.WriteFile(path, data, 0666)
+	err = ioutil.WriteFile(path, data, 0666)
+
+	// 写入 ~/ 和 /etc/ 目录，因为是备份需要，所以不需要提示错误信息
+	// 写入 ~/.edge-admin/
+	filename := filepath.Base(path)
+	homeDir, err := os.UserHomeDir()
+	if err == nil {
+		dir := homeDir + "/." + teaconst.ProcessName
+		stat, err := os.Stat(dir)
+		if err == nil && stat.IsDir() {
+			_ = ioutil.WriteFile(dir+"/"+filename, data, 0666)
+		} else if err != nil && os.IsNotExist(err) {
+			err = os.Mkdir(dir, 0777)
+			if err == nil {
+				_ = ioutil.WriteFile(dir+"/"+filename, data, 0666)
+			}
+		}
+	}
+
+	// 写入 /etc/.edge-admin
+	{
+		dir := "/etc/" + teaconst.ProcessName
+		stat, err := os.Stat(dir)
+		if err == nil && stat.IsDir() {
+			_ = ioutil.WriteFile(dir+"/"+filename, data, 0666)
+		} else if err != nil && os.IsNotExist(err) {
+			err = os.Mkdir(dir, 0777)
+			if err == nil {
+				_ = ioutil.WriteFile(dir+"/"+filename, data, 0666)
+			}
+		}
+	}
+
+	return err
 }