first commit: one-api base code + SAAS plan document

2025-12-29 22:52:27 +08:00
commit cb7c48bfa7
564 changed files with 61468 additions and 0 deletions
--- a/router/main.go
+++ b/router/main.go
@@ -0,0 +1,31 @@
+package router
+
+import (
+	"embed"
+	"fmt"
+	"github.com/gin-gonic/gin"
+	"github.com/songquanpeng/one-api/common/config"
+	"github.com/songquanpeng/one-api/common/logger"
+	"net/http"
+	"os"
+	"strings"
+)
+
+func SetRouter(router *gin.Engine, buildFS embed.FS) {
+	SetApiRouter(router)
+	SetDashboardRouter(router)
+	SetRelayRouter(router)
+	frontendBaseUrl := os.Getenv("FRONTEND_BASE_URL")
+	if config.IsMasterNode && frontendBaseUrl != "" {
+		frontendBaseUrl = ""
+		logger.SysLog("FRONTEND_BASE_URL is ignored on master node")
+	}
+	if frontendBaseUrl == "" {
+		SetWebRouter(router, buildFS)
+	} else {
+		frontendBaseUrl = strings.TrimSuffix(frontendBaseUrl, "/")
+		router.NoRoute(func(c *gin.Context) {
+			c.Redirect(http.StatusMovedPermanently, fmt.Sprintf("%s%s", frontendBaseUrl, c.Request.RequestURI))
+		})
+	}
+}