39 lines
627 B
Go
39 lines
627 B
Go
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()
|
|
})
|
|
}
|