130 lines
3.2 KiB
Go
130 lines
3.2 KiB
Go
// Copyright (c) 2023 Joshua Rich <joshua.rich@gmail.com>
|
|
//
|
|
// This software is released under the MIT License.
|
|
// https://opensource.org/licenses/MIT
|
|
|
|
package hass
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/cenkalti/backoff/v4"
|
|
"github.com/joshuar/go-hass-agent/internal/config"
|
|
"github.com/rs/zerolog/log"
|
|
"nhooyr.io/websocket"
|
|
"nhooyr.io/websocket/wsjson"
|
|
)
|
|
|
|
type HassWebsocket struct {
|
|
conn *websocket.Conn
|
|
ReadCh chan *WebsocketResponse
|
|
WriteCh chan interface{}
|
|
}
|
|
|
|
type WebsocketResponse struct {
|
|
Type string `json:"type"`
|
|
Success bool `json:"success,omitempty"`
|
|
Error struct {
|
|
Code string `json:"code"`
|
|
Message string `json:"message"`
|
|
} `json:"error,omitempty"`
|
|
ID int `json:"id,omitempty"`
|
|
Result interface{} `json:"result,omitempty"`
|
|
HAVersion string `json:"ha_version,omitempty"`
|
|
Notification struct {
|
|
Message string `json:"message"`
|
|
Title string `json:"title,omitempty"`
|
|
Target []string `json:"target,omitempty"`
|
|
Data interface{} `json:"data,omitempty"`
|
|
ConfirmID string `json:"confirm_id,omitempty"`
|
|
} `json:"event,omitempty"`
|
|
}
|
|
|
|
func (ws *HassWebsocket) Read(ctx context.Context) {
|
|
for {
|
|
response := &WebsocketResponse{
|
|
Success: true,
|
|
}
|
|
err := wsjson.Read(ctx, ws.conn, &response)
|
|
if err != nil {
|
|
log.Debug().Err(err).Caller().
|
|
Msg("Unable to read from websocket.")
|
|
close(ws.ReadCh)
|
|
return
|
|
}
|
|
select {
|
|
case <-ctx.Done():
|
|
log.Debug().Caller().
|
|
Msg("Stopping reading from websocket.")
|
|
close(ws.ReadCh)
|
|
return
|
|
case ws.ReadCh <- response:
|
|
}
|
|
}
|
|
}
|
|
|
|
func (ws *HassWebsocket) Write(ctx context.Context) {
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
log.Debug().Caller().
|
|
Msg("Stopping writing to websocket.")
|
|
ws.conn.Close(websocket.StatusNormalClosure, "")
|
|
close(ws.WriteCh)
|
|
return
|
|
case data := <-ws.WriteCh:
|
|
writeCtx, writeCancel := context.WithTimeout(ctx, time.Minute)
|
|
defer writeCancel()
|
|
err := wsjson.Write(writeCtx, ws.conn, data)
|
|
if err != nil {
|
|
log.Debug().Err(err).Caller().
|
|
Msg("Unable to write to websocket.")
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (ws *HassWebsocket) Close() {
|
|
ws.conn.Close(websocket.StatusNormalClosure, "requested websocket close")
|
|
}
|
|
|
|
func NewWebsocket(ctx context.Context) *HassWebsocket {
|
|
|
|
agentConfig, validConfig := config.FromContext(ctx)
|
|
if !validConfig {
|
|
log.Debug().Caller().Msg("Could not retrieve valid config from context.")
|
|
return nil
|
|
}
|
|
|
|
log.Debug().Caller().Msgf("Using %s for websocket connection.", agentConfig.WebSocketURL)
|
|
ctxConnect, cancelConnect := context.WithTimeout(ctx, time.Minute)
|
|
defer cancelConnect()
|
|
var conn *websocket.Conn
|
|
var err error
|
|
retryFunc := func() error {
|
|
conn, _, err = websocket.Dial(ctxConnect, agentConfig.WebSocketURL, nil)
|
|
if err != nil {
|
|
log.Debug().Err(err).Caller().
|
|
Msg("Could not connect to websocket.")
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
err = backoff.Retry(retryFunc, backoff.WithContext(backoff.NewExponentialBackOff(), ctxConnect))
|
|
if err != nil {
|
|
log.Debug().Err(err).Caller().
|
|
Msg("Could not connect to websocket.")
|
|
cancelConnect()
|
|
return nil
|
|
}
|
|
ws := &HassWebsocket{
|
|
conn: conn,
|
|
ReadCh: make(chan *WebsocketResponse),
|
|
WriteCh: make(chan interface{}),
|
|
}
|
|
go ws.Read(ctx)
|
|
go ws.Write(ctx)
|
|
return ws
|
|
}
|