mirror of
https://github.com/Xevion/go-ha.git
synced 2025-12-06 01:15:10 -06:00
120 lines
3.3 KiB
Go
120 lines
3.3 KiB
Go
package gomeassistant
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/dromara/carbon/v2"
|
|
|
|
"github.com/Xevion/go-ha/internal"
|
|
"github.com/Xevion/go-ha/types"
|
|
)
|
|
|
|
type State interface {
|
|
AfterSunrise(...types.DurationString) bool
|
|
BeforeSunrise(...types.DurationString) bool
|
|
AfterSunset(...types.DurationString) bool
|
|
BeforeSunset(...types.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 *internal.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 *internal.HttpClient, homeZoneEntityId string) (*StateImpl, error) {
|
|
state := &StateImpl{httpClient: c}
|
|
|
|
// Ensure the zone exists and has required attributes
|
|
entity, err := state.Get(homeZoneEntityId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("home zone entity '%s' not found: %w", homeZoneEntityId, err)
|
|
}
|
|
|
|
// Ensure it's a zone entity
|
|
if !strings.HasPrefix(homeZoneEntityId, "zone.") {
|
|
return nil, fmt.Errorf("entity '%s' is not a zone entity (must start with zone.)", homeZoneEntityId)
|
|
}
|
|
|
|
// Verify and extract latitude and longitude
|
|
if entity.Attributes == nil {
|
|
return nil, fmt.Errorf("home zone entity '%s' has no attributes", homeZoneEntityId)
|
|
}
|
|
|
|
if lat, ok := entity.Attributes["latitude"].(float64); ok {
|
|
state.latitude = lat
|
|
} else {
|
|
return nil, fmt.Errorf("home zone entity '%s' missing valid latitude attribute", homeZoneEntityId)
|
|
}
|
|
|
|
if long, ok := entity.Attributes["longitude"].(float64); ok {
|
|
state.longitude = long
|
|
} else {
|
|
return nil, fmt.Errorf("home zone entity '%s' missing valid longitude attribute", homeZoneEntityId)
|
|
}
|
|
|
|
return state, 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.GetStates()
|
|
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 ...types.DurationString) bool {
|
|
sunrise := getSunriseSunset(s, true, carbon.Now(), offset...)
|
|
return carbon.Now().Lt(sunrise)
|
|
}
|
|
|
|
func (s *StateImpl) AfterSunrise(offset ...types.DurationString) bool {
|
|
return !s.BeforeSunrise(offset...)
|
|
}
|
|
|
|
func (s *StateImpl) BeforeSunset(offset ...types.DurationString) bool {
|
|
sunset := getSunriseSunset(s, false, carbon.Now(), offset...)
|
|
return carbon.Now().Lt(sunset)
|
|
}
|
|
|
|
func (s *StateImpl) AfterSunset(offset ...types.DurationString) bool {
|
|
return !s.BeforeSunset(offset...)
|
|
}
|