router/socket_router.go

package router

import (
	"whatsapp-backend/provider"

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

func SocketRouter(router *gin.Engine, controller provider.ControllerProvider, middleware provider.MiddlewareProvider) {
	socketController := controller.ProvideSocketController()
	authMiddleware := middleware.ProvideAuthMiddleware()

	// WebSocket endpoint needs to be authenticated
	routerGroup := router.Group("/api/whatsapp", authMiddleware.RequireAuth())
	// routerGroup.Use(gzip.Gzip(gzip.DefaultCompression)) // gzip might interfere with websocket upgrade
	{
		routerGroup.GET("/socket", socketController.HandleWebSocket)
	}
}