wavelet init

Wavelet/internal/router/root/custom.go

@@ -0,0 +1,14 @@
+// Copyright 2026 Arctel.net
+// SPDX-License-Identifier: Apache-2.0
+
+// Package root registers custom business routes and frontend serving.
+package root
+
+import (
+	"github.com/gin-gonic/gin"
+)
+
+// RegisterCustomRootRoutes registers custom business routes that belong to the root path.
+func RegisterCustomRootRoutes(_ *gin.Engine) {
+	// Add custom root routes here
+}

Wavelet/internal/router/root/default.go

@@ -0,0 +1,32 @@
+// Copyright 2026 Arctel.net
+// SPDX-License-Identifier: Apache-2.0
+
+package root
+
+import (
+	_ "github.com/Rain-kl/Wavelet/docs" // Swagger documentation generation setup
+	publicconfig "github.com/Rain-kl/Wavelet/internal/apps/config"
+	"github.com/Rain-kl/Wavelet/internal/apps/health"
+	"github.com/Rain-kl/Wavelet/internal/apps/upload"
+	"github.com/Rain-kl/Wavelet/internal/config"
+	"github.com/gin-gonic/gin"
+	swaggerFiles "github.com/swaggo/files"
+	ginSwagger "github.com/swaggo/gin-swagger"
+)
+
+// RegisterDefaultRootRoutes registers default routes that belong to the root path.
+func RegisterDefaultRootRoutes(r *gin.Engine) {
+	// 1. Serve files by ID
+	r.GET("/f/:id", upload.ServeFileByID)
+
+	// 2. Dynamic robots.txt serving
+	r.GET("/robots.txt", publicconfig.GetRobotsTXT)
+
+	// 3. Swagger routes (Non-production only)
+	if !config.Config.App.IsProduction() {
+		r.GET(config.Config.App.APIPrefix+"/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler))
+	}
+
+	// 4. Health check
+	r.GET(config.Config.App.APIPrefix+"/health", health.Health)
+}

Wavelet/internal/router/root/frontend.go

@@ -0,0 +1,110 @@
+//go:build embed_frontend
+
+// Copyright 2026 Arctel.net
+// SPDX-License-Identifier: Apache-2.0
+
+package root
+
+import (
+	"embed"
+	"io"
+	"io/fs"
+	"net/http"
+	"strings"
+
+	"github.com/gin-gonic/gin"
+)
+
+//go:embed all:dist
+var frontendFS embed.FS
+
+func serveFileDirect(c *gin.Context, subFS fs.FS, filePath string) bool {
+	file, err := subFS.Open(filePath)
+	if err != nil {
+		return false
+	}
+	defer file.Close()
+
+	stat, err := file.Stat()
+	if err != nil {
+		return false
+	}
+
+	if stat.IsDir() {
+		return false
+	}
+
+	seeker, ok := file.(io.ReadSeeker)
+	if !ok {
+		return false
+	}
+
+	// 使用 http.ServeContent 直接输出文件内容，不进行路径规范化重定向
+	http.ServeContent(c.Writer, c.Request, filePath, stat.ModTime(), seeker)
+	return true
+}
+
+func init() {
+	RegisterFrontend = func(r *gin.Engine) {
+		subFS, err := fs.Sub(frontendFS, "dist")
+		if err != nil {
+			panic(err)
+		}
+
+		r.NoRoute(func(c *gin.Context) {
+			path := c.Request.URL.Path
+
+			// API 接口路由或文件服务路由 -> 直接返回，由 Gin 处理标准 404
+			if strings.HasPrefix(path, "/api/") || strings.HasPrefix(path, "/f/") {
+				return
+			}
+
+			// 只处理 GET 和 HEAD 请求
+			if c.Request.Method != http.MethodGet && c.Request.Method != http.MethodHead {
+				c.JSON(http.StatusMethodNotAllowed, gin.H{"error_msg": "Method not allowed"})
+				return
+			}
+
+			// 移除开头的斜杠以在嵌入文件系统中查找
+			cleanPath := strings.TrimPrefix(path, "/")
+
+			// 1. 根路径 -> 直接输出 index.html
+			if cleanPath == "" {
+				if serveFileDirect(c, subFS, "index.html") {
+					return
+				}
+			}
+
+			// 2. 精确匹配（如果对应的文件存在，直接输出）
+			if serveFileDirect(c, subFS, cleanPath) {
+				return
+			}
+
+			// 如果是个目录（例如请求了 "/login"，同时 dist 目录下存在一个叫 "login" 的文件夹目录），
+			// 则查找是否有对应的 ".html" 文件（例如 "login.html"）并进行输出。
+			if cleanPath != "" {
+				htmlPath := cleanPath + ".html"
+				if serveFileDirect(c, subFS, htmlPath) {
+					return
+				}
+			}
+
+			// 3. Next.js Clean URLs 兜底逻辑（例如访问 /settings/security -> 实际映射输出 settings/security.html）
+			if !strings.Contains(cleanPath, ".") {
+				htmlPath := cleanPath + ".html"
+				if serveFileDirect(c, subFS, htmlPath) {
+					return
+				}
+				indexPath := cleanPath + "/index.html"
+				if serveFileDirect(c, subFS, indexPath) {
+					return
+				}
+			}
+
+			// 4. 单页应用（SPA）前端路由兜底：返回 index.html
+			if serveFileDirect(c, subFS, "index.html") {
+				return
+			}
+		})
+	}
+}

Wavelet/internal/router/root/root.go

@@ -0,0 +1,26 @@
+// Copyright 2026 Arctel.net
+// SPDX-License-Identifier: Apache-2.0
+
+// Package root registers custom business routes and frontend serving.
+package root
+
+import (
+	"github.com/gin-gonic/gin"
+)
+
+// RegisterFrontend is a package-level variable overridden by frontend.go when built with embed_frontend.
+var RegisterFrontend = func(_ *gin.Engine) {
+	// No-op by default
+}
+
+// RegisterRootRoutes registers custom business routes that belong to the root path.
+func RegisterRootRoutes(r *gin.Engine) {
+	// 1. Default root routes (/f/:id, /robots.txt, and /swagger/*any)
+	RegisterDefaultRootRoutes(r)
+
+	// 2. Register custom serving
+	RegisterCustomRootRoutes(r)
+
+	// 3. Register frontend serving
+	RegisterFrontend(r)
+}