[新增] 添加 API 反向代理功能支持

2026-10-03 15:06:36 +08:00 · 2026-06-03 22:48:04 +08:00
parent 68d70bbbe8
commit 3eecd31868
11 changed files with 358 additions and 2 deletions
@@ -104,6 +104,10 @@ type snapshotPagesDeployment struct {
 	EntryFile          string `json:"entry_file"`
 	SPAFallbackEnabled bool   `json:"spa_fallback_enabled"`
 	SPAFallbackPath    string `json:"spa_fallback_path"`
+	APIProxyEnabled    bool   `json:"api_proxy_enabled"`
+	APIProxyPath       string `json:"api_proxy_path"`
+	APIProxyPass       string `json:"api_proxy_pass"`
+	APIProxyRewrite    string `json:"api_proxy_rewrite"`
 	LocalRoot          string `json:"local_root"`
 }

@@ -615,6 +619,10 @@ func buildSnapshotPagesDeployment(projectID *uint) (*snapshotPagesDeployment, er
 		EntryFile:          deployment.EntryFile,
 		SPAFallbackEnabled: project.SPAFallbackEnabled,
 		SPAFallbackPath:    normalizeStoredPagesFallbackPath(project.SPAFallbackPath),
+		APIProxyEnabled:    project.APIProxyEnabled,
+		APIProxyPath:       project.APIProxyPath,
+		APIProxyPass:       project.APIProxyPass,
+		APIProxyRewrite:    project.APIProxyRewrite,
 		LocalRoot:          fmt.Sprintf("%s/deployments/%d/current", openrestyrender.PagesDirPlaceholder, deployment.ID),
 	}, nil
 }
@@ -8,6 +8,7 @@ import (
 	"fmt"
 	"io"
 	"mime/multipart"
+	"net/url"
 	"openflare/common"
 	"openflare/model"
 	"os"
@@ -36,6 +37,10 @@ type PagesProjectInput struct {
 	Enabled            bool   `json:"enabled"`
 	SPAFallbackEnabled bool   `json:"spa_fallback_enabled"`
 	SPAFallbackPath    string `json:"spa_fallback_path"`
+	APIProxyEnabled    bool   `json:"api_proxy_enabled"`
+	APIProxyPath       string `json:"api_proxy_path"`
+	APIProxyPass       string `json:"api_proxy_pass"`
+	APIProxyRewrite    string `json:"api_proxy_rewrite"`
 }

 type PagesProjectView struct {
@@ -46,6 +51,10 @@ type PagesProjectView struct {
 	Enabled            bool                 `json:"enabled"`
 	SPAFallbackEnabled bool                 `json:"spa_fallback_enabled"`
 	SPAFallbackPath    string               `json:"spa_fallback_path"`
+	APIProxyEnabled    bool                 `json:"api_proxy_enabled"`
+	APIProxyPath       string               `json:"api_proxy_path"`
+	APIProxyPass       string               `json:"api_proxy_pass"`
+	APIProxyRewrite    string               `json:"api_proxy_rewrite"`
 	ActiveDeploymentID *uint                `json:"active_deployment_id"`
 	ActiveDeployment   *PagesDeploymentView `json:"active_deployment,omitempty"`
 	DeploymentCount    int64                `json:"deployment_count"`
@@ -137,6 +146,10 @@ func UpdatePagesProject(id uint, input PagesProjectInput) (*PagesProjectView, er
 		"enabled":              project.Enabled,
 		"spa_fallback_enabled": project.SPAFallbackEnabled,
 		"spa_fallback_path":    project.SPAFallbackPath,
+		"api_proxy_enabled":    project.APIProxyEnabled,
+		"api_proxy_path":       project.APIProxyPath,
+		"api_proxy_pass":       project.APIProxyPass,
+		"api_proxy_rewrite":    project.APIProxyRewrite,
 	}).Error; err != nil {
 		if model.IsUniqueConstraintError(err) {
 			return nil, errors.New("Pages 项目标识已存在")
@@ -413,6 +426,31 @@ func buildPagesProject(project *model.PagesProject, input PagesProjectInput) (*m
 		return nil, err
 	}
 	project.SPAFallbackPath = fallbackPath
+
+	project.APIProxyEnabled = input.APIProxyEnabled
+	apiProxyPath := strings.TrimSpace(input.APIProxyPath)
+	apiProxyPass := strings.TrimSpace(input.APIProxyPass)
+	apiProxyRewrite := strings.TrimSpace(input.APIProxyRewrite)
+
+	if project.APIProxyEnabled {
+		if apiProxyPath == "" {
+			return nil, errors.New("启用 API 反代时，匹配路径不能为空")
+		}
+		if !strings.HasPrefix(apiProxyPath, "/") {
+			return nil, errors.New("API 反代匹配路径必须以 '/' 开头")
+		}
+		if apiProxyPass == "" {
+			return nil, errors.New("启用 API 反代时，后端服务地址不能为空")
+		}
+		parsedURL, err := url.Parse(apiProxyPass)
+		if err != nil || (parsedURL.Scheme != "http" && parsedURL.Scheme != "https") || parsedURL.Host == "" {
+			return nil, errors.New("API 反代后端服务地址必须是有效的 HTTP/HTTPS URL")
+		}
+	}
+	project.APIProxyPath = apiProxyPath
+	project.APIProxyPass = apiProxyPass
+	project.APIProxyRewrite = apiProxyRewrite
+
 	return project, nil
 }

@@ -428,6 +466,10 @@ func buildPagesProjectView(project *model.PagesProject) (*PagesProjectView, erro
 		Enabled:            project.Enabled,
 		SPAFallbackEnabled: project.SPAFallbackEnabled,
 		SPAFallbackPath:    normalizeStoredPagesFallbackPath(project.SPAFallbackPath),
+		APIProxyEnabled:    project.APIProxyEnabled,
+		APIProxyPath:       project.APIProxyPath,
+		APIProxyPass:       project.APIProxyPass,
+		APIProxyRewrite:    project.APIProxyRewrite,
 		ActiveDeploymentID: project.ActiveDeploymentID,
 		CreatedAt:          project.CreatedAt,
 		UpdatedAt:          project.UpdatedAt,
@@ -271,3 +271,57 @@ func TestUploadPagesDeploymentWithTopLevelFolder(t *testing.T) {
 		t.Fatalf("expected EntryFile to be index.html, got %q", deployment.EntryFile)
 	}
 }
+
+func TestPagesProjectAPIProxyValidation(t *testing.T) {
+	setupServiceTestDB(t)
+
+	// 1. Invalid configuration: enabled but empty fields
+	_, err := CreatePagesProject(PagesProjectInput{
+		Name:            "API Proxy 1",
+		Enabled:         true,
+		APIProxyEnabled: true,
+	})
+	if err == nil || !strings.Contains(err.Error(), "匹配路径不能为空") {
+		t.Fatalf("expected error for empty match path, got: %v", err)
+	}
+
+	// 2. Invalid path: must start with '/'
+	_, err = CreatePagesProject(PagesProjectInput{
+		Name:            "API Proxy 2",
+		Enabled:         true,
+		APIProxyEnabled: true,
+		APIProxyPath:    "api",
+		APIProxyPass:    "http://127.0.0.1:8080",
+	})
+	if err == nil || !strings.Contains(err.Error(), "必须以 '/' 开头") {
+		t.Fatalf("expected error for path not starting with /, got: %v", err)
+	}
+
+	// 3. Invalid target URL
+	_, err = CreatePagesProject(PagesProjectInput{
+		Name:            "API Proxy 3",
+		Enabled:         true,
+		APIProxyEnabled: true,
+		APIProxyPath:    "/api",
+		APIProxyPass:    "127.0.0.1:8080",
+	})
+	if err == nil || !strings.Contains(err.Error(), "有效的 HTTP/HTTPS URL") {
+		t.Fatalf("expected error for invalid pass URL, got: %v", err)
+	}
+
+	// 4. Valid configuration
+	project, err := CreatePagesProject(PagesProjectInput{
+		Name:            "API Proxy Valid",
+		Enabled:         true,
+		APIProxyEnabled: true,
+		APIProxyPath:    "/api",
+		APIProxyPass:    "http://127.0.0.1:8080",
+		APIProxyRewrite: "/",
+	})
+	if err != nil {
+		t.Fatalf("unexpected error creating valid project: %v", err)
+	}
+	if !project.APIProxyEnabled || project.APIProxyPath != "/api" || project.APIProxyPass != "http://127.0.0.1:8080" || project.APIProxyRewrite != "/" {
+		t.Fatalf("unexpected project state: %+v", project)
+	}
+}