66 lines
1.7 KiB
Go
66 lines
1.7 KiB
Go
package handlers
|
|
|
|
import (
|
|
"github.com/coder/websocket"
|
|
"github.com/gin-gonic/gin"
|
|
"go.uber.org/zap"
|
|
|
|
apperrors "veza-backend-api/internal/errors"
|
|
"veza-backend-api/internal/services"
|
|
chatws "veza-backend-api/internal/websocket/chat"
|
|
)
|
|
|
|
type ChatWebSocketHandler struct {
|
|
chatService *services.ChatService
|
|
hub *chatws.Hub
|
|
handler *chatws.MessageHandler
|
|
logger *zap.Logger
|
|
}
|
|
|
|
func NewChatWebSocketHandler(chatService *services.ChatService, hub *chatws.Hub, handler *chatws.MessageHandler, logger *zap.Logger) *ChatWebSocketHandler {
|
|
if logger == nil {
|
|
logger = zap.NewNop()
|
|
}
|
|
return &ChatWebSocketHandler{
|
|
chatService: chatService,
|
|
hub: hub,
|
|
handler: handler,
|
|
logger: logger,
|
|
}
|
|
}
|
|
|
|
func (h *ChatWebSocketHandler) HandleWebSocket(c *gin.Context) {
|
|
tokenString := c.Query("token")
|
|
if tokenString == "" {
|
|
RespondWithAppError(c, apperrors.NewUnauthorizedError("missing token"))
|
|
return
|
|
}
|
|
|
|
claims, err := h.chatService.ValidateChatToken(tokenString)
|
|
if err != nil {
|
|
h.logger.Warn("Invalid chat token",
|
|
zap.Error(err))
|
|
RespondWithAppError(c, apperrors.NewUnauthorizedError("invalid or expired token"))
|
|
return
|
|
}
|
|
|
|
conn, err := websocket.Accept(c.Writer, c.Request, &websocket.AcceptOptions{
|
|
InsecureSkipVerify: true,
|
|
})
|
|
if err != nil {
|
|
h.logger.Error("Failed to accept WebSocket",
|
|
zap.Error(err),
|
|
zap.String("user_id", claims.UserID.String()))
|
|
return
|
|
}
|
|
|
|
client := chatws.NewClient(h.hub, conn, claims.UserID, claims.Username, h.handler, h.logger)
|
|
|
|
h.hub.Register(client)
|
|
|
|
client.SendJSON(chatws.NewActionConfirmedResponse("connected", true))
|
|
|
|
ctx := c.Request.Context()
|
|
go client.WritePump(ctx)
|
|
go client.ReadPump(ctx)
|
|
}
|