Method GetSessionWS

internal/api/handlers/management/sessions.go:130–175

GetSessionWS upgrades to WebSocket and streams command results for a session.

(c *gin.Context)

// GetSessionWS upgrades to WebSocket and streams command results for a session.
func (h *Handler) GetSessionWS(c *gin.Context) {
	id := c.Param("id")
	sess := sessions.Global().Get(id)
	if sess == nil {
		c.JSON(404, gin.H{"error": "session not found"})
		return
	}

	conn, err := wsUpgrader.Upgrade(c.Writer, c.Request, nil)
	if err != nil {
		return
	}
	defer conn.Close()

	subID := sessions.GenerateCommandID() // reuse as unique sub ID
	ch := sessions.Global().Subscribe(id, subID)
	if ch == nil {
		return
	}
	defer sessions.Global().Unsubscribe(id, subID)

	// Read goroutine to detect client disconnect.
	done := make(chan struct{})
	go func() {
		defer close(done)
		for {
			if _, _, err := conn.ReadMessage(); err != nil {
				return
			}
		}
	}()

	for {
		select {
		case result, ok := <-ch:
			if !ok {
				return
			}
			if err := conn.WriteJSON(result); err != nil {
				return
			}
		case <-done:
			return
		}
	}
}