package handler

import (
	"net/http"

	"github.com/gin-gonic/gin"
	"kefu-sys/server/internal/middleware"
	"kefu-sys/server/internal/ws"
)

type WsHandler struct{}

func NewWsHandler() *WsHandler { return &WsHandler{} }

func (h *WsHandler) Connect(c *gin.Context) {
	userID := middleware.GetUserID(c)
	tenantID := middleware.GetTenantID(c)

	client, err := ws.UpgradeAgent(c.Writer, c.Request, userID, tenantID, middleware.GetRole(c))
	if err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{"code": 500, "message": "升级连接失败"})
		return
	}

	ws.HandleWebSocket(client)
}