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)

	bodyLimit := middleware.BodyLimit(1 << 20) // 1 MiB

	r.POST("/api/login", bodyLimit, middleware.LoginRateLimit(), handler.Login)
	r.GET("/api/version", handler.GetVersion)

	auth := r.Group("/api")
	auth.Use(middleware.AuthMiddleware(), bodyLimit)
	{
		auth.GET("/me", handler.Me)
		auth.PUT("/me/password", handler.ChangePassword)
		auth.GET("/me/sessions", handler.ListMySessions)
		auth.DELETE("/me/sessions/:sessionId", handler.RevokeMySession)
		auth.GET("/me/vpn/connections", handler.GetMyVpnConnections)
		auth.GET("/me/traffic/today", handler.GetMyTrafficToday)
		auth.GET("/me/traffic", handler.GetMyTrafficHistory)
	}

	admin := r.Group("/api/admin")
	admin.Use(middleware.AuthMiddleware(), middleware.AdminMiddleware(), bodyLimit)
	{
		admin.GET("/stats", handler.GetAdminStats)
		admin.GET("/users/count", handler.GetUserCount)
		admin.GET("/users", handler.ListUsers)
		admin.POST("/users", handler.CreateUser)
		admin.PUT("/users/:id", handler.UpdateUser)
		admin.DELETE("/users/:id", handler.DeleteUser)
		admin.DELETE("/users/:id/sessions", handler.AdminRevokeUserSessions)

		admin.GET("/vpn/settings", handler.GetVpnSettings)
		admin.PUT("/vpn/settings", handler.UpdateVpnSettings)
		admin.GET("/vpn/status", handler.GetVpnStatus)
		admin.GET("/vpn/diag", handler.GetVpnDiag)
		admin.GET("/vpn/reservations", handler.ListVpnReservations)
		admin.POST("/vpn/reservations", handler.CreateVpnReservation)
		admin.DELETE("/vpn/reservations/:id", handler.DeleteVpnReservation)
		admin.DELETE("/vpn/clients/:id", handler.KickUserClient)

		admin.GET("/traffic/today", handler.GetAdminTrafficToday)
		admin.GET("/traffic/history", handler.GetAdminTrafficHistory)
		admin.GET("/traffic/users/:id", handler.GetAdminUserTraffic)
	}

	distDir := http.Dir("./dist")
	fs := http.FileServer(distDir)
	r.NoRoute(func(c *gin.Context) {
		path := c.Request.URL.Path
		if strings.HasPrefix(path, "/api") || strings.HasPrefix(path, "/ws") {
			c.JSON(http.StatusNotFound, gin.H{"error": "not found"})
			return
		}
		f, err := distDir.Open(path)
		if err != nil {
			c.Header("Content-Type", "text/html")
			c.File("./dist/index.html")
			return
		}
		f.Close()
		fs.ServeHTTP(c.Writer, c.Request)
	})
}