feat: publish v1.0.0-rc.36-custom.1

router/main.go

@@ -0,0 +1,41 @@
+package router
+
+import (
+	"fmt"
+	"net/http"
+	"os"
+	"strings"
+
+	"github.com/QuantumNous/new-api/common"
+	"github.com/QuantumNous/new-api/middleware"
+
+	"github.com/gin-gonic/gin"
+)
+
+func SetRouter(router *gin.Engine, assets WebAssets) {
+	SetApiRouter(router)
+	SetDashboardRouter(router)
+	SetRelayRouter(router)
+	SetTaskPluginProtocolRouter(router)
+	SetVideoRouter(router)
+	SetTaskRouter(router)
+	pluginDispatcher := SetPluginRouter(router)
+	frontendBaseUrl := os.Getenv("FRONTEND_BASE_URL")
+	if common.IsMasterNode && frontendBaseUrl != "" {
+		frontendBaseUrl = ""
+		common.SysLog("FRONTEND_BASE_URL is ignored on master node")
+	}
+	if frontendBaseUrl == "" {
+		SetWebRouter(router, assets, pluginDispatcher)
+	} else {
+		frontendBaseUrl = strings.TrimSuffix(frontendBaseUrl, "/")
+		router.NoRoute(
+			pluginDispatcher,
+			middleware.RouteTag("web"),
+			middleware.AccessTokenAudit(),
+			func(c *gin.Context) {
+				c.Redirect(http.StatusMovedPermanently, fmt.Sprintf("%s%s", frontendBaseUrl, c.Request.RequestURI))
+			},
+		)
+	}
+}