1
0
Fork 0
hats/pkg/homeassistant/thermostat.go

111 lines
3.0 KiB
Go
Raw Permalink Normal View History

2023-12-13 17:31:33 +00:00
package homeassistant
import (
"fmt"
)
type ThermostatMode string
const ThermostatModeOff ThermostatMode = "off"
const ThermostatModeHeat ThermostatMode = "heat"
const ThermostatModeCool ThermostatMode = "cool"
const ThermostatModeHeatCool ThermostatMode = "heat_cool"
type ThermostatFanMode string
const ThermostatFanModeAuto ThermostatFanMode = "Auto low"
const ThermostatFanModeLow ThermostatFanMode = "Low"
type ThermostatState struct {
Mode ThermostatMode
FanMode ThermostatFanMode
TargetTempHigh float64
TargetTempLow float64
}
func (c *RestClient) GetThermostatState(entityId string) (output ThermostatState, err error) {
output = ThermostatState{
Mode: ThermostatModeOff,
}
state, err := c.GetState(entityId)
if err != nil {
return output, err
}
defer func() {
if panicErr := recover(); panicErr != nil {
err = fmt.Errorf("error parsing attribute value: %v", panicErr)
}
}()
switch state.State {
case string(ThermostatModeHeat):
output.Mode = ThermostatModeHeat
output.TargetTempLow = state.Attributes["temperature"].(float64)
case string(ThermostatModeCool):
output.Mode = ThermostatModeCool
output.TargetTempHigh = state.Attributes["temperature"].(float64)
case string(ThermostatModeHeatCool):
output.Mode = ThermostatModeHeatCool
output.TargetTempHigh = state.Attributes["target_temp_high"].(float64)
output.TargetTempLow = state.Attributes["target_temp_low"].(float64)
default:
output.Mode = ThermostatModeOff
}
switch state.Attributes["fan_mode"].(string) {
case string(ThermostatFanModeLow):
output.FanMode = ThermostatFanModeLow
default:
output.FanMode = ThermostatFanModeAuto
}
return output, err
}
func (c *RestClient) SetThermostatState(entityId string, desiredState ThermostatState) error {
if desiredState.Mode == ThermostatModeOff {
return c.CallService(entityId, Services.TurnOff)
}
err := c.CallService(entityId, Services.TurnOn)
if err != nil {
return fmt.Errorf("error turning on thermostat: %w", err)
}
extras := map[string]any{
"hvac_mode": string(desiredState.Mode),
}
switch desiredState.Mode {
case ThermostatModeCool:
extras["temperature"] = desiredState.TargetTempHigh
case ThermostatModeHeat:
extras["temperature"] = desiredState.TargetTempLow
case ThermostatModeHeatCool:
extras["target_temp_high"] = desiredState.TargetTempHigh
extras["target_temp_low"] = desiredState.TargetTempLow
}
serviceCalls := []*CallServiceInput{{
EntityID: entityId, Service: Services.SetTemperature, Extras: extras,
}}
if desiredState.FanMode != "" {
serviceCalls = append(serviceCalls, &CallServiceInput{
EntityID: entityId, Service: Services.SetFanMode, Extras: map[string]any{
ExtraProps.FanMode: string(desiredState.FanMode),
},
})
}
return c.CallServices(serviceCalls...)
}
func (c *RestClient) SetThermostatFanMode(entityId string, mode ThermostatFanMode) error {
return c.CallService(entityId, Services.SetFanMode, map[string]any{
ExtraProps.FanMode: string(mode),
})
}