mirror of
https://github.com/Xevion/go-ha.git
synced 2025-12-05 23:15:07 -06:00
115 lines
3.0 KiB
Go
115 lines
3.0 KiB
Go
package gomeassistant
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/golang-module/carbon"
|
|
|
|
"saml.dev/gome-assistant/internal/http"
|
|
)
|
|
|
|
type State interface {
|
|
AfterSunrise(...DurationString) bool
|
|
BeforeSunrise(...DurationString) bool
|
|
AfterSunset(...DurationString) bool
|
|
BeforeSunset(...DurationString) bool
|
|
ListEntities() ([]EntityState, error)
|
|
Get(entityId string) (EntityState, error)
|
|
Equals(entityId, state string) (bool, error)
|
|
}
|
|
|
|
// State is used to retrieve state from Home Assistant.
|
|
type StateImpl struct {
|
|
httpClient *http.HttpClient
|
|
latitude float64
|
|
longitude float64
|
|
}
|
|
|
|
type EntityState struct {
|
|
EntityID string `json:"entity_id"`
|
|
State string `json:"state"`
|
|
Attributes map[string]any `json:"attributes"`
|
|
LastChanged time.Time `json:"last_changed"`
|
|
}
|
|
|
|
func newState(c *http.HttpClient, homeZoneEntityId string) (*StateImpl, error) {
|
|
state := &StateImpl{httpClient: c}
|
|
err := state.getLatLong(c, homeZoneEntityId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return state, nil
|
|
}
|
|
|
|
func (s *StateImpl) getLatLong(c *http.HttpClient, homeZoneEntityId string) error {
|
|
resp, err := s.Get(homeZoneEntityId)
|
|
if err != nil {
|
|
return fmt.Errorf("couldn't get latitude/longitude from home assistant entity '%s'. Did you type it correctly? It should be a zone like 'zone.home'", homeZoneEntityId)
|
|
}
|
|
|
|
if resp.Attributes["latitude"] != nil {
|
|
s.latitude = resp.Attributes["latitude"].(float64)
|
|
} else {
|
|
return errors.New("server returned nil latitude")
|
|
}
|
|
|
|
if resp.Attributes["longitude"] != nil {
|
|
s.longitude = resp.Attributes["longitude"].(float64)
|
|
} else {
|
|
return errors.New("server returned nil longitude")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *StateImpl) Get(entityId string) (EntityState, error) {
|
|
resp, err := s.httpClient.GetState(entityId)
|
|
if err != nil {
|
|
return EntityState{}, err
|
|
}
|
|
es := EntityState{}
|
|
err = json.Unmarshal(resp, &es)
|
|
return es, err
|
|
}
|
|
|
|
// ListEntities returns a list of all entities in Home Assistant.
|
|
// see rest documentation for more details: https://developers.home-assistant.io/docs/api/rest/#actions
|
|
func (s *StateImpl) ListEntities() ([]EntityState, error) {
|
|
resp, err := s.httpClient.States()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
es := []EntityState{}
|
|
err = json.Unmarshal(resp, &es)
|
|
return es, err
|
|
}
|
|
|
|
func (s *StateImpl) Equals(entityId string, expectedState string) (bool, error) {
|
|
currentState, err := s.Get(entityId)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
return currentState.State == expectedState, nil
|
|
}
|
|
|
|
func (s *StateImpl) BeforeSunrise(offset ...DurationString) bool {
|
|
sunrise := getSunriseSunset(s /* sunrise = */, true, carbon.Now(), offset...)
|
|
return carbon.Now().Lt(sunrise)
|
|
}
|
|
|
|
func (s *StateImpl) AfterSunrise(offset ...DurationString) bool {
|
|
return !s.BeforeSunrise(offset...)
|
|
}
|
|
|
|
func (s *StateImpl) BeforeSunset(offset ...DurationString) bool {
|
|
sunset := getSunriseSunset(s /* sunrise = */, false, carbon.Now(), offset...)
|
|
return carbon.Now().Lt(sunset)
|
|
}
|
|
|
|
func (s *StateImpl) AfterSunset(offset ...DurationString) bool {
|
|
return !s.BeforeSunset(offset...)
|
|
}
|