mirror of
https://github.com/Xevion/go-ha.git
synced 2025-12-06 01:15:10 -06:00
02b6c413f1
Additionally, removed the context that gets passed into the Services but isn't used in one of them. The websockets APIs also don't have any use for context.
131 lines
3.0 KiB
Go
131 lines
3.0 KiB
Go
package services
|
|
|
|
import (
|
|
ws "saml.dev/gome-assistant/internal/websocket"
|
|
)
|
|
|
|
/* Structs */
|
|
|
|
type Vacuum struct {
|
|
conn *ws.WebsocketWriter
|
|
}
|
|
|
|
/* Public API */
|
|
|
|
// Tell the vacuum cleaner to do a spot clean-up.
|
|
// Takes an entityId.
|
|
func (v Vacuum) CleanSpot(entityId string) error {
|
|
req := NewBaseServiceRequest(entityId)
|
|
req.Domain = "vacuum"
|
|
req.Service = "clean_spot"
|
|
|
|
return v.conn.WriteMessage(req)
|
|
}
|
|
|
|
// Locate the vacuum cleaner robot.
|
|
// Takes an entityId.
|
|
func (v Vacuum) Locate(entityId string) error {
|
|
req := NewBaseServiceRequest(entityId)
|
|
req.Domain = "vacuum"
|
|
req.Service = "locate"
|
|
|
|
return v.conn.WriteMessage(req)
|
|
}
|
|
|
|
// Pause the cleaning task.
|
|
// Takes an entityId.
|
|
func (v Vacuum) Pause(entityId string) error {
|
|
req := NewBaseServiceRequest(entityId)
|
|
req.Domain = "vacuum"
|
|
req.Service = "pause"
|
|
|
|
return v.conn.WriteMessage(req)
|
|
}
|
|
|
|
// Tell the vacuum cleaner to return to its dock.
|
|
// Takes an entityId.
|
|
func (v Vacuum) ReturnToBase(entityId string) error {
|
|
req := NewBaseServiceRequest(entityId)
|
|
req.Domain = "vacuum"
|
|
req.Service = "return_to_base"
|
|
|
|
return v.conn.WriteMessage(req)
|
|
}
|
|
|
|
// Send a raw command to the vacuum cleaner. Takes an entityId and an optional
|
|
// map that is translated into service_data.
|
|
func (v Vacuum) SendCommand(entityId string, serviceData ...map[string]any) error {
|
|
req := NewBaseServiceRequest(entityId)
|
|
req.Domain = "vacuum"
|
|
req.Service = "send_command"
|
|
if len(serviceData) != 0 {
|
|
req.ServiceData = serviceData[0]
|
|
}
|
|
|
|
return v.conn.WriteMessage(req)
|
|
}
|
|
|
|
// Set the fan speed of the vacuum cleaner. Takes an entityId and an optional
|
|
// map that is translated into service_data.
|
|
func (v Vacuum) SetFanSpeed(entityId string, serviceData ...map[string]any) error {
|
|
req := NewBaseServiceRequest(entityId)
|
|
req.Domain = "vacuum"
|
|
req.Service = "set_fan_speed"
|
|
|
|
if len(serviceData) != 0 {
|
|
req.ServiceData = serviceData[0]
|
|
}
|
|
|
|
return v.conn.WriteMessage(req)
|
|
}
|
|
|
|
// Start or resume the cleaning task.
|
|
// Takes an entityId.
|
|
func (v Vacuum) Start(entityId string) error {
|
|
req := NewBaseServiceRequest(entityId)
|
|
req.Domain = "vacuum"
|
|
req.Service = "start"
|
|
|
|
return v.conn.WriteMessage(req)
|
|
}
|
|
|
|
// Start, pause, or resume the cleaning task.
|
|
// Takes an entityId.
|
|
func (v Vacuum) StartPause(entityId string) error {
|
|
req := NewBaseServiceRequest(entityId)
|
|
req.Domain = "vacuum"
|
|
req.Service = "start_pause"
|
|
|
|
return v.conn.WriteMessage(req)
|
|
}
|
|
|
|
// Stop the current cleaning task.
|
|
// Takes an entityId.
|
|
func (v Vacuum) Stop(entityId string) error {
|
|
req := NewBaseServiceRequest(entityId)
|
|
req.Domain = "vacuum"
|
|
req.Service = "stop"
|
|
|
|
return v.conn.WriteMessage(req)
|
|
}
|
|
|
|
// Stop the current cleaning task and return to home.
|
|
// Takes an entityId.
|
|
func (v Vacuum) TurnOff(entityId string) error {
|
|
req := NewBaseServiceRequest(entityId)
|
|
req.Domain = "vacuum"
|
|
req.Service = "turn_off"
|
|
|
|
return v.conn.WriteMessage(req)
|
|
}
|
|
|
|
// Start a new cleaning task.
|
|
// Takes an entityId.
|
|
func (v Vacuum) TurnOn(entityId string) error {
|
|
req := NewBaseServiceRequest(entityId)
|
|
req.Domain = "vacuum"
|
|
req.Service = "turn_on"
|
|
|
|
return v.conn.WriteMessage(req)
|
|
}
|