package routers

import (
	"runtime"
	"server/config"
	"server/modules/customer/middleware"

	"github.com/gin-gonic/gin"
	"github.com/liangdas/mqant/log"
)

func SetUpRouter() *gin.Engine {
	release := config.GetConfig().Release
	if release {
		gin.SetMode(gin.ReleaseMode)
		// 禁用控制台颜色
		gin.DisableConsoleColor()
	} else {
		gin.SetMode(gin.DebugMode)
	}
	r := gin.New()
	// 跨域处理
	r.Use(middleware.CrosHandler())
	r.Use(middleware.TokenMiddleWare())
	r.Use(middleware.PowerMiddleWare())
	r.Use(gin.RecoveryWithWriter(log.LogBeego(), func(c *gin.Context, err interface{}) {
		buf := make([]byte, 1024)
		runtime.Stack(buf, false)
		log.Error("panic(%+v), stack:\n%s", err, string(buf))
	}))

	gmHandle(r)
	account(r)
	power(r)
	common(r)
	chat(r)
	mail(r)
	guser(r)

	return r
}