package router_ws

import (
	"encoding/json"
	"fmt"
	"gitee.com/os-lee/easy-paas/common"
	"gitee.com/os-lee/easy-paas/common/logger"
	"gitee.com/os-lee/easy-paas/gateway/config"
	"gitee.com/os-lee/easy-paas/gateway/internal/dao"
	"gitee.com/os-lee/easy-paas/gateway/internal/load_balancer"
	"gitee.com/os-lee/easy-paas/gateway/internal/service_manager"
	ws_middleware "gitee.com/os-lee/easy-paas/gateway/middleware_ws"
	"github.com/gin-gonic/gin"
)

func NewRouter(middlewares ...gin.HandlerFunc) *gin.Engine {
	router := gin.Default()
	router.Use(middlewares...)
	router.GET("/ping", func(c *gin.Context) {
		logger.Log.Println("print watcher")
		logger.Log.Println("1.ServiceManagerHandler")
		service_manager.Handler.ServiceMap.Range(func(key, value any) bool {
			sd := value.(*dao.ServiceDetail)
			marshal, _ := json.Marshal(sd)
			logger.Log.Printf("key: %s value: %v", key, string(marshal))
			return true
		})
		logger.Log.Println("2.LoadBalancerHandler")
		load_balancer.Handler.LoadBalancerMap.Range(func(key, value any) bool {
			sd := value.(load_balancer.Balancer)
			marshal, _ := json.Marshal(sd)
			logger.Log.Printf("key: %s value: %v \n", key, string(marshal))
			return true
		})
		pong := fmt.Sprintf("name:%s, protocol:%s, version:%s, message:%s",
			config.Conf.Server.Name, common.ProtocolTypeMap[common.ProtocolTypeWebSocket], config.Conf.Server.Version, "pong")
		common.ResponseSuccess(c, pong)
	})

	// Http 路由
	router.Use(
		ws_middleware.AccessMode(),
		ws_middleware.StripUri(),
		ws_middleware.UrlRewrite(),
		ws_middleware.Proxy(),
	)
	return router
}