refactor(ws): ws logging and channel manager added no auth currently

This commit is contained in:
2025-07-29 11:29:59 -05:00
parent 6a631be909
commit a1a30cffd1
12 changed files with 693 additions and 319 deletions

View File

@@ -1,25 +1,55 @@
package socketio
package websocket
import (
"github.com/gin-gonic/gin"
"net/http"
channelmgt "lst.net/cmd/services/websocket/channelMGT"
"lst.net/utils/db"
"github.com/gin-gonic/gin"
logging "lst.net/utils/logger"
)
var broadcaster = make(chan logging.Message) // define broadcaster here so its accessible
var (
broadcaster = make(chan logging.Message)
)
func RegisterSocketRoutes(r *gin.Engine) {
sqlDB, err := db.DB.DB()
if err != nil {
panic(err)
// Initialize all channels
InitializeChannels()
// Start channel processors
StartAllChannels()
// Start background services
go LogServices(broadcaster)
go StartBroadcasting(broadcaster, channels)
// WebSocket route
r.GET("/ws", func(c *gin.Context) {
SocketHandler(c, channels)
})
r.GET("/ws/clients", AdminAuthMiddleware(), handleGetClients)
}
func handleGetClients(c *gin.Context) {
channel := c.Query("channel")
var clientList []*Client
if channel != "" {
clientList = GetClientsByChannel(channel)
} else {
clientList = GetAllClients()
}
// channels
go channelmgt.AllLogs(sqlDB, broadcaster)
go StartBroadcasting(broadcaster)
wsGroup := r.Group("/ws")
wsGroup.GET("/connect", SocketHandler)
c.JSON(http.StatusOK, gin.H{
"count": len(clientList),
"clients": clientList,
})
}
func AdminAuthMiddleware() gin.HandlerFunc {
return func(c *gin.Context) {
// Implement your admin authentication logic
// Example: Check API key or JWT token
c.Next()
}
}