封装路由

internal/router/router.go

@@ -0,0 +1,38 @@
+package router
+
+import (
+	"net/http"
+	"strings"
+
+	"lmvpn/internal/handler"
+	"lmvpn/internal/middleware"
+	"lmvpn/internal/vpn"
+
+	"github.com/gin-gonic/gin"
+)
+
+func Setup(r *gin.Engine) {
+	r.GET("/ws", vpn.HandleWS)
+
+	r.POST("/api/login", handler.Login)
+
+	auth := r.Group("/api")
+	auth.Use(middleware.AuthMiddleware())
+	{
+		auth.GET("/me", handler.Me)
+	}
+
+	fs := http.FileServer(http.Dir("./dist"))
+	r.Use(func(c *gin.Context) {
+		if strings.HasPrefix(c.Request.URL.Path, "/ws") {
+			return
+		}
+		if strings.HasPrefix(c.Request.URL.Path, "/api") {
+			c.Next()
+			return
+		}
+
+		fs.ServeHTTP(c.Writer, c.Request)
+		c.Abort()
+	})
+}