1.4.5.2

EdgeAPI/internal/configs/api_config.go

@@ -0,0 +1,185 @@
+package configs
+
+import (
+	teaconst "github.com/TeaOSLab/EdgeAPI/internal/const"
+	"github.com/iwind/TeaGo/Tea"
+	"gopkg.in/yaml.v3"
+	"os"
+	"path/filepath"
+)
+
+var sharedAPIConfig *APIConfig = nil
+
+// APIConfig API节点配置
+type APIConfig struct {
+	NodeId string `yaml:"nodeId" json:"nodeId"`
+	Secret string `yaml:"secret" json:"secret"`
+
+	numberId int64 // 数字ID
+}
+
+// SharedAPIConfig 获取共享配置
+func SharedAPIConfig() (*APIConfig, error) {
+	sharedLocker.Lock()
+	defer sharedLocker.Unlock()
+
+	if sharedAPIConfig != nil {
+		return sharedAPIConfig, nil
+	}
+
+	// 候选文件
+	var config = &APIConfig{}
+	{
+		var localFile = Tea.ConfigFile("api.yaml")
+		var isFromLocal = false
+		var paths = []string{localFile}
+		homeDir, homeErr := os.UserHomeDir()
+		if homeErr == nil {
+			paths = append(paths, homeDir+"/."+teaconst.ProcessName+"/api.yaml")
+		}
+		paths = append(paths, "/etc/"+teaconst.ProcessName+"/api.yaml")
+
+		// 依次检查文件
+		var data []byte
+		var err error
+		var firstErr error
+		for _, path := range paths {
+			data, err = os.ReadFile(path)
+			if err != nil {
+				if firstErr == nil {
+					firstErr = err
+				}
+			} else {
+				if path == localFile {
+					isFromLocal = true
+				}
+				break
+			}
+		}
+		if firstErr != nil {
+			return nil, firstErr
+		}
+
+		// 解析内容
+		err = yaml.Unmarshal(data, config)
+		if err != nil {
+			return nil, err
+		}
+
+		if !isFromLocal {
+			// 恢复文件
+			_ = os.WriteFile(localFile, data, 0666)
+		}
+	}
+
+	// 恢复数据库文件
+	{
+		var dbConfigFile = Tea.ConfigFile("db.yaml")
+		_, err := os.Stat(dbConfigFile)
+		if err != nil {
+			var paths = []string{}
+			homeDir, homeErr := os.UserHomeDir()
+			if homeErr == nil {
+				paths = append(paths, homeDir+"/."+teaconst.ProcessName+"/db.yaml")
+			}
+			paths = append(paths, "/etc/"+teaconst.ProcessName+"/db.yaml")
+			for _, path := range paths {
+				_, err = os.Stat(path)
+				if err == nil {
+					data, err := os.ReadFile(path)
+					if err == nil {
+						_ = os.WriteFile(dbConfigFile, data, 0666)
+						break
+					}
+				}
+			}
+		}
+	}
+
+	sharedAPIConfig = config
+	return config, nil
+}
+
+// SetNumberId 设置数字ID
+func (this *APIConfig) SetNumberId(numberId int64) {
+	this.numberId = numberId
+	teaconst.NodeId = numberId
+}
+
+// NumberId 获取数字ID
+func (this *APIConfig) NumberId() int64 {
+	return this.numberId
+}
+
+// WriteFile 保存到文件
+func (this *APIConfig) WriteFile(path string) error {
+	data, err := yaml.Marshal(this)
+	if err != nil {
+		return err
+	}
+
+	// 生成备份文件
+	var filename = filepath.Base(path)
+	homeDir, _ := os.UserHomeDir()
+	var backupDirs = []string{"/etc/edge-api"}
+	if len(homeDir) > 0 {
+		backupDirs = append(backupDirs, homeDir+"/.edge-api")
+	}
+	for _, backupDir := range backupDirs {
+		stat, err := os.Stat(backupDir)
+		if err == nil && stat.IsDir() {
+			_ = os.WriteFile(backupDir+"/"+filename, data, 0666)
+		} else if err != nil && os.IsNotExist(err) {
+			err = os.Mkdir(backupDir, 0777)
+			if err == nil {
+				_ = os.WriteFile(backupDir+"/"+filename, data, 0666)
+			}
+		}
+	}
+
+	return os.WriteFile(path, data, 0666)
+}
+
+// ResetAPIConfig 重置配置
+func ResetAPIConfig() error {
+	for _, filename := range []string{"api.yaml", "db.yaml", ".db.yaml"} {
+		// 重置 configs/api.yaml
+		{
+			var configFile = Tea.ConfigFile(filename)
+			stat, err := os.Stat(configFile)
+			if err == nil && !stat.IsDir() {
+				err = os.Remove(configFile)
+				if err != nil {
+					return err
+				}
+			}
+		}
+
+		// 重置 ~/.edge-api/api.yaml
+		homeDir, homeErr := os.UserHomeDir()
+		if homeErr == nil {
+			var configFile = homeDir + "/." + teaconst.ProcessName + "/" + filename
+			stat, err := os.Stat(configFile)
+			if err == nil && !stat.IsDir() {
+				err = os.Remove(configFile)
+				if err != nil {
+					return err
+				}
+			}
+		}
+
+		// 重置 /etc/edge-api/api.yaml
+		{
+			var configFile = "/etc/" + teaconst.ProcessName + "/" + filename
+			stat, err := os.Stat(configFile)
+			if err == nil && !stat.IsDir() {
+				err = os.Remove(configFile)
+				if err != nil {
+					return err
+				}
+			}
+		}
+	}
+
+	return nil
+}