package router

import (
	"ckwk/internal/handler"
	"ckwk/internal/middleware"

	"github.com/gin-gonic/gin"
)

func SetupRouter() *gin.Engine {
	r := gin.Default()
	wkHandler := handler.NewWKHandler()
	sessionMiddleware := middleware.SessionMiddleware(wkHandler.Session)

	api := r.Group("/api")
	{
		api.POST("/login", wkHandler.Login)
		v1 := api.Group("/v1", sessionMiddleware)
		{
			v1.POST("/online", wkHandler.Online)
			v1.POST("/logout", wkHandler.Logout)
		}

		v2 := api.Group("/v2", sessionMiddleware)
		{
			v2.POST("/logout", wkHandler.Logout)
			v2.POST("/study", wkHandler.Study)
			v2.POST("/record", wkHandler.AllRecord)
		}
	}

	return r
}