package router

import (
	"toutoukan/controllers/search"
	"toutoukan/controllers/system"
	"toutoukan/controllers/test"
	"toutoukan/controllers/user"
	"toutoukan/init/ratelimit"
	"toutoukan/kills"
	"toutoukan/socket"
	"toutoukan/utill"

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

func SetupRouter() *gin.Engine {
	r := gin.Default()
	apiGroup := r.Group("/user")
	//apiGroup.Use(hystrix.CircuitBreakerMiddleware("user_api"))
	{
		apiGroup.POST("/login", user.UserLogin)
		apiGroup.POST("/test", utill.JWTAuthMiddleware(), test.Testjwt)
		apiGroup.POST("/kill", ratelimit.RateLimitMiddleware(), kills.Userkill)

	}
	r.GET("/socket", utill.JWTAuthMiddleware(), func(c *gin.Context) {
		socket.WebsocketHandler(c)
	})
	systemGroup := r.Group("/system")
	{
		systemGroup.POST("/sendMsg", system.SendMsg)
	}
	searchGroup := r.Group("/search")
	{
		searchGroup.POST("/usersearch", search.DataSearch)
		searchGroup.POST("/insert", search.InsertDocument)
	}

	return r
}