添加代理路由和配置版本的控制器与服务，更新数据库模型，完善API路由

2026-10-05 23:26:38 +08:00 · 2026-03-09 22:53:51 +08:00
parent 34f317fe6f
commit 8f55d83b34
10 changed files with 721 additions and 1 deletions
@@ -0,0 +1,140 @@
+package service
+
+import (
+	"crypto/sha256"
+	"encoding/hex"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"gin-template/model"
+	"strings"
+	"time"
+
+	"gorm.io/gorm"
+)
+
+type ReleaseResult struct {
+	Version *model.ConfigVersion `json:"version"`
+	Routes  []*model.ProxyRoute  `json:"routes"`
+}
+
+func ListConfigVersions() ([]*model.ConfigVersion, error) {
+	return model.ListConfigVersions()
+}
+
+func GetActiveConfigVersion() (*model.ConfigVersion, error) {
+	return model.GetActiveConfigVersion()
+}
+
+func PublishConfigVersion(createdBy string) (*ReleaseResult, error) {
+	routes, err := model.GetEnabledProxyRoutes()
+	if err != nil {
+		return nil, err
+	}
+	if len(routes) == 0 {
+		return nil, errors.New("没有可发布的启用规则")
+	}
+	snapshotJSON, err := renderSnapshot(routes)
+	if err != nil {
+		return nil, err
+	}
+	renderedConfig := renderNginxConfig(routes)
+	version, err := nextVersionNumber(time.Now())
+	if err != nil {
+		return nil, err
+	}
+	record := &model.ConfigVersion{
+		Version:        version,
+		SnapshotJSON:   snapshotJSON,
+		RenderedConfig: renderedConfig,
+		Checksum:       checksum(renderedConfig),
+		IsActive:       true,
+		CreatedBy:      createdBy,
+	}
+	err = model.DB.Transaction(func(tx *gorm.DB) error {
+		if err := tx.Model(&model.ConfigVersion{}).Where("is_active = ?", true).Update("is_active", false).Error; err != nil {
+			return err
+		}
+		if err := tx.Create(record).Error; err != nil {
+			return err
+		}
+		return nil
+	})
+	if err != nil {
+		if isUniqueConstraintError(err) {
+			return nil, errors.New("版本号生成冲突，请重试")
+		}
+		return nil, err
+	}
+	return &ReleaseResult{
+		Version: record,
+		Routes:  routes,
+	}, nil
+}
+
+func ActivateConfigVersion(id uint) (*model.ConfigVersion, error) {
+	version, err := model.GetConfigVersionByID(id)
+	if err != nil {
+		return nil, err
+	}
+	err = model.DB.Transaction(func(tx *gorm.DB) error {
+		if err := tx.Model(&model.ConfigVersion{}).Where("is_active = ?", true).Update("is_active", false).Error; err != nil {
+			return err
+		}
+		if err := tx.Model(version).Update("is_active", true).Error; err != nil {
+			return err
+		}
+		return nil
+	})
+	if err != nil {
+		return nil, err
+	}
+	version.IsActive = true
+	return version, nil
+}
+
+func renderSnapshot(routes []*model.ProxyRoute) (string, error) {
+	type snapshotRoute struct {
+		Domain    string `json:"domain"`
+		OriginURL string `json:"origin_url"`
+		Enabled   bool   `json:"enabled"`
+		Remark    string `json:"remark,omitempty"`
+	}
+	items := make([]snapshotRoute, 0, len(routes))
+	for _, route := range routes {
+		items = append(items, snapshotRoute{
+			Domain:    route.Domain,
+			OriginURL: route.OriginURL,
+			Enabled:   route.Enabled,
+			Remark:    route.Remark,
+		})
+	}
+	data, err := json.Marshal(items)
+	if err != nil {
+		return "", err
+	}
+	return string(data), nil
+}
+
+func renderNginxConfig(routes []*model.ProxyRoute) string {
+	var builder strings.Builder
+	builder.WriteString("# This file is generated by ATSFlare. Do not edit manually.\n")
+	for _, route := range routes {
+		builder.WriteString(fmt.Sprintf("server {\n    listen 80;\n    server_name %s;\n\n    location / {\n        proxy_set_header Host $host;\n        proxy_set_header X-Real-IP $remote_addr;\n        proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;\n        proxy_set_header X-Forwarded-Proto $scheme;\n        proxy_pass %s;\n    }\n}\n\n", route.Domain, route.OriginURL))
+	}
+	return builder.String()
+}
+
+func checksum(content string) string {
+	sum := sha256.Sum256([]byte(content))
+	return hex.EncodeToString(sum[:])
+}
+
+func nextVersionNumber(now time.Time) (string, error) {
+	prefix := now.Format("20060102")
+	var count int64
+	if err := model.DB.Model(&model.ConfigVersion{}).Where("version LIKE ?", prefix+"-%").Count(&count).Error; err != nil {
+		return "", err
+	}
+	return fmt.Sprintf("%s-%03d", prefix, count+1), nil
+}
@@ -0,0 +1,103 @@
+package service
+
+import (
+	"errors"
+	"gin-template/model"
+	"net/url"
+	"strings"
+)
+
+type ProxyRouteInput struct {
+	Domain    string `json:"domain"`
+	OriginURL string `json:"origin_url"`
+	Enabled   bool   `json:"enabled"`
+	Remark    string `json:"remark"`
+}
+
+func ListProxyRoutes() ([]*model.ProxyRoute, error) {
+	return model.ListProxyRoutes()
+}
+
+func CreateProxyRoute(input ProxyRouteInput) (*model.ProxyRoute, error) {
+	route, err := buildProxyRoute(nil, input)
+	if err != nil {
+		return nil, err
+	}
+	if err = route.Insert(); err != nil {
+		if isUniqueConstraintError(err) {
+			return nil, errors.New("域名已存在")
+		}
+		return nil, err
+	}
+	return route, nil
+}
+
+func UpdateProxyRoute(id uint, input ProxyRouteInput) (*model.ProxyRoute, error) {
+	route, err := model.GetProxyRouteByID(id)
+	if err != nil {
+		return nil, err
+	}
+	route, err = buildProxyRoute(route, input)
+	if err != nil {
+		return nil, err
+	}
+	if err = route.Update(); err != nil {
+		if isUniqueConstraintError(err) {
+			return nil, errors.New("域名已存在")
+		}
+		return nil, err
+	}
+	return route, nil
+}
+
+func DeleteProxyRoute(id uint) error {
+	route, err := model.GetProxyRouteByID(id)
+	if err != nil {
+		return err
+	}
+	return route.Delete()
+}
+
+func buildProxyRoute(route *model.ProxyRoute, input ProxyRouteInput) (*model.ProxyRoute, error) {
+	domain := strings.ToLower(strings.TrimSpace(input.Domain))
+	originURL := strings.TrimSpace(input.OriginURL)
+	remark := strings.TrimSpace(input.Remark)
+	if domain == "" {
+		return nil, errors.New("域名不能为空")
+	}
+	if strings.Contains(domain, "://") || strings.Contains(domain, "/") {
+		return nil, errors.New("域名格式不合法")
+	}
+	if err := validateOriginURL(originURL); err != nil {
+		return nil, err
+	}
+	if route == nil {
+		route = &model.ProxyRoute{}
+	}
+	route.Domain = domain
+	route.OriginURL = originURL
+	route.Enabled = input.Enabled
+	route.Remark = remark
+	return route, nil
+}
+
+func validateOriginURL(raw string) error {
+	if raw == "" {
+		return errors.New("源站地址不能为空")
+	}
+	parsed, err := url.ParseRequestURI(raw)
+	if err != nil {
+		return errors.New("源站地址格式不合法")
+	}
+	if parsed.Scheme != "http" && parsed.Scheme != "https" {
+		return errors.New("源站地址必须以 http:// 或 https:// 开头")
+	}
+	if parsed.Host == "" {
+		return errors.New("源站地址格式不合法")
+	}
+	return nil
+}
+
+func isUniqueConstraintError(err error) bool {
+	return err != nil && strings.Contains(strings.ToLower(err.Error()), "unique")
+}