Added lots of services and fixed bug with ID for websocket calls

This commit is contained in:
Sam Lewis
2022-10-13 02:17:36 -04:00
parent 7f9e346d34
commit b99c1e5925
13 changed files with 379 additions and 59 deletions

View File

@@ -3,21 +3,37 @@ package services
import (
"context"
"github.com/saml-dev/gome-assistant/internal/http"
ws "github.com/saml-dev/gome-assistant/internal/websocket"
"nhooyr.io/websocket"
)
type HomeAssistant struct {
conn *websocket.Conn
ctx context.Context
httpClient *http.HttpClient
conn *websocket.Conn
ctx context.Context
}
func (ha *HomeAssistant) TurnOn(entityId string) {
// TurnOn a Home Assistant entity. Takes an entityId and an optional
// map that is translated into service_data.
func (ha *HomeAssistant) TurnOn(entityId string, serviceData ...map[string]any) {
req := NewBaseServiceRequest(entityId)
req.Domain = "homeassistant"
req.Service = "turn_on"
if len(serviceData) != 0 {
req.ServiceData = serviceData[0]
}
ws.WriteMessage(req, ha.conn, ha.ctx)
}
// Toggle a Home Assistant entity. Takes an entityId and an optional
// map that is translated into service_data.
func (ha *HomeAssistant) Toggle(entityId string, serviceData ...map[string]any) {
req := NewBaseServiceRequest(entityId)
req.Domain = "homeassistant"
req.Service = "toggle"
if len(serviceData) != 0 {
req.ServiceData = serviceData[0]
}
ws.WriteMessage(req, ha.conn, ha.ctx)
}
@@ -29,11 +45,3 @@ func (ha *HomeAssistant) TurnOff(entityId string) {
ws.WriteMessage(req, ha.conn, ha.ctx)
}
func (ha *HomeAssistant) Toggle(entityId string) {
req := NewBaseServiceRequest(entityId)
req.Domain = "homeassistant"
req.Service = "toggle"
ws.WriteMessage(req, ha.conn, ha.ctx)
}