go-ha/internal/websocket/reader.go

package websocket

import (
	"encoding/json"
	"log/slog"

	"github.com/gorilla/websocket"
)

type BaseMessage struct {
	Type    string `json:"type"`
	Id      int64  `json:"id"`
	Success bool   `json:"success"`
}

type ChanMsg struct {
	Id      int64
	Type    string
	Success bool
	Raw     []byte
}

func ListenWebsocket(conn *websocket.Conn, c chan ChanMsg) {
	for {
		bytes, err := ReadMessage(conn)
		if err != nil {
			slog.Error("Error reading from websocket", "err", err)
			close(c)
			break
		}

		base := BaseMessage{
			// default to true for messages that don't include "success" at all
			Success: true,
		}
		_ = json.Unmarshal(bytes, &base)
		if !base.Success {
			slog.Warn("Received unsuccessful response", "response", string(bytes))
		}
		chanMsg := ChanMsg{
			Type:    base.Type,
			Id:      base.Id,
			Success: base.Success,
			Raw:     bytes,
		}

		// Use non-blocking send to avoid hanging on closed channel
		select {
		case c <- chanMsg:
			// Message sent successfully
		default:
			// Channel is full or closed, break out of loop
			slog.Warn("Websocket message channel is full or closed, stopping listener")
			close(c)
			return
		}
	}
}