evcc-io/server/mqtt_setter.go
2025-12-31 13:25:58 +01:00

92 lines
2 KiB
Go

package server
import (
"encoding/json"
"slices"
"strconv"
"time"
"github.com/evcc-io/evcc/api"
"github.com/evcc-io/evcc/util"
"github.com/spf13/cast"
)
type setter struct {
topic string
fun func(string) error
}
func isEmpty(payload string) bool {
return slices.Contains([]string{"-", "nil", "null", "none"}, payload)
}
func setterFunc[T any](conv func(string) (T, error), set func(T) error) func(string) error {
return func(payload string) error {
val, err := conv(payload)
if err == nil {
err = set(val)
}
return err
}
}
// ptrSetter updates pointer api
func ptrSetter[T any](conv func(string) (T, error), set func(*T) error) func(string) error {
return func(payload string) error {
var val *T
v, err := conv(payload)
if err == nil {
val = &v
} else if !isEmpty(payload) {
return err
}
return set(val)
}
}
func floatSetter(set func(float64) error) func(string) error {
return setterFunc(parseFloat, set)
}
func floatPtrSetter(set func(*float64) error) func(string) error {
return ptrSetter(parseFloat, set)
}
func intSetter(set func(int) error) func(string) error {
return setterFunc(strconv.Atoi, set)
}
func boolSetter(set func(bool) error) func(string) error {
return setterFunc(func(v string) (bool, error) {
return cast.ToBoolE(v)
}, set)
}
func durationSetter(set func(time.Duration) error) func(string) error {
return setterFunc(util.ParseDuration, set)
}
func planStrategySetter(set func(api.PlanStrategy) error) func(string) error {
return func(payload string) error {
var planStrategy planStrategyPayload
if err := json.Unmarshal([]byte(payload), &planStrategy); err != nil {
return err
}
return set(api.PlanStrategy{
Continuous: planStrategy.Continuous,
Precondition: time.Duration(planStrategy.Precondition) * time.Second,
})
}
}
func planGoalSetter[T any](set func(time.Time, T) error) func(string) error {
return func(payload string) error {
var plan planGoal[T]
if err := json.Unmarshal([]byte(payload), &plan); err != nil {
return err
}
return set(plan.Time, plan.Value)
}
}