visionA/local-agent/server/internal/api/ws/device_events_ws.go

package ws

import (
	"visiona-agent/server/internal/device"

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

// upgrader 是所有 WS handler 共用的 gorilla upgrader。
// M8-8：CheckOrigin 改為 ws.CheckOrigin（loopback 白名單），
// 不再接受任意 Origin（v1 模式是 wails:// 才放行所有 origin）。
var upgrader = websocket.Upgrader{
	CheckOrigin: CheckOrigin,
}

func DeviceEventsHandler(hub *Hub, deviceMgr *device.Manager) gin.HandlerFunc {
	return func(c *gin.Context) {
		conn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
		if err != nil {
			return
		}
		defer conn.Close()

		client := &Client{Conn: conn, Send: make(chan []byte, 10)}
		room := "device-events"
		sub := &Subscription{Client: client, Room: room}
		hub.Register(sub)
		defer hub.Unregister(sub)

		// Forward device events to this WebSocket room
		go func() {
			for event := range deviceMgr.Events() {
				hub.BroadcastToRoom(room, event)
			}
		}()

		// Write pump
		for msg := range client.Send {
			if err := conn.WriteMessage(websocket.TextMessage, msg); err != nil {
				return
			}
		}
	}
}