52 lines
1.1 KiB
Go
52 lines
1.1 KiB
Go
package api
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/gorilla/websocket"
|
|
|
|
"hightube/internal/chat"
|
|
)
|
|
|
|
var upgrader = websocket.Upgrader{
|
|
CheckOrigin: func(r *http.Request) bool {
|
|
return true // Allow all connections
|
|
},
|
|
}
|
|
|
|
// WSHandler handles websocket requests from clients
|
|
func WSHandler(c *gin.Context) {
|
|
roomID := c.Param("room_id")
|
|
username := c.DefaultQuery("username", "Anonymous")
|
|
|
|
conn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
|
|
if err != nil {
|
|
fmt.Printf("[WS ERROR] Failed to upgrade: %v\n", err)
|
|
return
|
|
}
|
|
|
|
client := &chat.Client{
|
|
Hub: chat.MainHub,
|
|
Conn: conn,
|
|
Send: make(chan []byte, 256),
|
|
RoomID: roomID,
|
|
Username: username,
|
|
}
|
|
|
|
client.Hub.RegisterClient(client)
|
|
|
|
// Start reading and writing loops in goroutines
|
|
go client.WritePump()
|
|
go client.ReadPump()
|
|
|
|
// Optionally broadcast a system message: User Joined
|
|
chat.MainHub.BroadcastToRoom(chat.Message{
|
|
Type: "system",
|
|
Username: "System",
|
|
Content: fmt.Sprintf("%s joined the room", username),
|
|
RoomID: roomID,
|
|
})
|
|
}
|