Initial commit (code only without large binaries)

EdgeAPI/internal/configs/api_config.go

@@ -0,0 +1,198 @@
+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
+
+// ClickHouseConfig 仅用于访问日志列表只读查询（logs_ingest）
+type ClickHouseConfig struct {
+	Host          string `yaml:"host" json:"host"`
+	Port          int    `yaml:"port" json:"port"`
+	User          string `yaml:"user" json:"user"`
+	Password      string `yaml:"password" json:"password"`
+	Database      string `yaml:"database" json:"database"`
+	Scheme        string `yaml:"scheme" json:"scheme"`
+	TLSSkipVerify bool   `yaml:"tlsSkipVerify" json:"tlsSkipVerify"`
+	TLSServerName string `yaml:"tlsServerName" json:"tlsServerName"`
+}
+
+// APIConfig API节点配置
+type APIConfig struct {
+	NodeId     string            `yaml:"nodeId" json:"nodeId"`
+	Secret     string            `yaml:"secret" json:"secret"`
+	ClickHouse *ClickHouseConfig `yaml:"clickhouse,omitempty" json:"clickhouse,omitempty"`
+
+	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
+}

EdgeAPI/internal/configs/api_config_test.go

@@ -0,0 +1,14 @@
+package configs
+
+import (
+	_ "github.com/iwind/TeaGo/bootstrap"
+	"testing"
+)
+
+func TestSharedAPIConfig(t *testing.T) {
+	config, err := SharedAPIConfig()
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log(config)
+}

EdgeAPI/internal/configs/db_config.go

@@ -0,0 +1,26 @@
+// Copyright 2024 GoEdge CDN goedge.cdn@gmail.com. All rights reserved. Official site: https://goedge.cn .
+
+package configs
+
+import (
+	"errors"
+	"os"
+
+	"github.com/iwind/TeaGo/Tea"
+	"github.com/iwind/TeaGo/dbs"
+	"gopkg.in/yaml.v3"
+)
+
+func LoadDBConfig() (*dbs.Config, error) {
+	var config = &dbs.Config{}
+	for _, filename := range []string{".db.yaml", "db.yaml"} {
+		configData, err := os.ReadFile(Tea.ConfigFile(filename))
+		if err != nil {
+			continue
+		}
+		err = yaml.Unmarshal(configData, config)
+		return config, err
+	}
+
+	return nil, errors.New("could not find database config file '.db.yaml' or 'db.yaml'")
+}

EdgeAPI/internal/configs/locker.go

@@ -0,0 +1,5 @@
+package configs
+
+import "sync"
+
+var sharedLocker = &sync.RWMutex{}

EdgeAPI/internal/configs/simple_db_config.go

@@ -0,0 +1,60 @@
+// Copyright 2024 GoEdge CDN goedge.cdn@gmail.com. All rights reserved. Official site: https://goedge.cn .
+
+package configs
+
+import (
+	"fmt"
+	"net/url"
+	"os"
+
+	"github.com/iwind/TeaGo/Tea"
+	"github.com/iwind/TeaGo/dbs"
+	"gopkg.in/yaml.v3"
+)
+
+type SimpleDBConfig struct {
+	User       string   `yaml:"user"`
+	Password   string   `yaml:"password"`
+	Database   string   `yaml:"database"`
+	Host       string   `yaml:"host"`
+	BoolFields []string `yaml:"boolFields,omitempty"`
+}
+
+func ParseSimpleDBConfig(data []byte) (*SimpleDBConfig, error) {
+	var config = &SimpleDBConfig{}
+	err := yaml.Unmarshal(data, config)
+	return config, err
+}
+
+func (this *SimpleDBConfig) GenerateOldConfig() error {
+	var dbConfig = &dbs.DBConfig{
+		Driver: "mysql",
+		Dsn:    url.QueryEscape(this.User) + ":" + url.QueryEscape(this.Password) + "@tcp(" + this.Host + ")/" + url.PathEscape(this.Database) + "?charset=utf8mb4&timeout=30s&multiStatements=true",
+		Prefix: "edge",
+	}
+	dbConfig.Models.Package = "internal/db/models"
+
+	var config = &dbs.Config{
+		DBs: map[string]*dbs.DBConfig{
+			"dev":  dbConfig,
+			"prod": dbConfig,
+		},
+	}
+	config.Default.DB = "prod"
+	config.Fields = map[string][]string{
+		"bool": this.BoolFields,
+	}
+
+	oldConfigYAML, encodeErr := yaml.Marshal(config)
+	if encodeErr != nil {
+		return encodeErr
+	}
+
+	var targetFile = Tea.ConfigFile(".db.yaml")
+	err := os.WriteFile(targetFile, oldConfigYAML, 0666)
+	if err != nil {
+		return fmt.Errorf("create database config file failed: %w", err)
+	}
+
+	return nil
+}