evcc-io/provider/const.go
2024-10-12 11:28:24 +02:00

173 lines
3.3 KiB
Go

package provider
import (
"context"
"encoding/hex"
"strconv"
"strings"
"github.com/evcc-io/evcc/provider/pipeline"
"github.com/evcc-io/evcc/util"
)
type constProvider struct {
ctx context.Context
str string
set Config
}
func init() {
registry.AddCtx("const", NewConstFromConfig)
}
// NewConstFromConfig creates const provider
func NewConstFromConfig(ctx context.Context, other map[string]interface{}) (Provider, error) {
var cc struct {
Value string
pipeline.Settings `mapstructure:",squash"`
Set Config
}
if err := util.DecodeOther(other, &cc); err != nil {
return nil, err
}
pipe, err := pipeline.New(nil, cc.Settings)
if err != nil {
return nil, err
}
b, err := pipe.Process([]byte(cc.Value))
if err != nil {
return nil, err
}
o := &constProvider{
ctx: ctx,
str: string(b),
set: cc.Set,
}
return o, nil
}
var _ StringProvider = (*constProvider)(nil)
func (o *constProvider) StringGetter() (func() (string, error), error) {
return func() (string, error) {
return o.str, nil
}, nil
}
var _ IntProvider = (*constProvider)(nil)
func (o *constProvider) IntGetter() (func() (int64, error), error) {
val, err := strconv.ParseInt(o.str, 10, 64)
if err != nil && o.str == "" {
err = nil
}
return func() (int64, error) {
return val, err
}, err
}
var _ FloatProvider = (*constProvider)(nil)
func (o *constProvider) FloatGetter() (func() (float64, error), error) {
val, err := strconv.ParseFloat(o.str, 64)
if err != nil && o.str == "" {
err = nil
}
return func() (float64, error) {
return val, err
}, err
}
var _ BoolProvider = (*constProvider)(nil)
func (o *constProvider) BoolGetter() (func() (bool, error), error) {
val, err := strconv.ParseBool(o.str)
if err != nil && o.str == "" {
err = nil
}
return func() (bool, error) {
return val, err
}, err
}
var _ SetIntProvider = (*constProvider)(nil)
func (o *constProvider) IntSetter(param string) (func(int64) error, error) {
set, err := NewIntSetterFromConfig(o.ctx, param, o.set)
if err != nil {
return nil, err
}
val, err := strconv.ParseInt(o.str, 10, 64)
if err != nil && o.str == "" {
err = nil
}
return func(_ int64) error {
return set(val)
}, err
}
var _ SetFloatProvider = (*constProvider)(nil)
func (o *constProvider) FloatSetter(param string) (func(float64) error, error) {
set, err := NewFloatSetterFromConfig(o.ctx, param, o.set)
if err != nil {
return nil, err
}
val, err := strconv.ParseFloat(o.str, 64)
if err != nil && o.str == "" {
err = nil
}
return func(_ float64) error {
return set(val)
}, err
}
var _ SetBoolProvider = (*constProvider)(nil)
func (o *constProvider) BoolSetter(param string) (func(bool) error, error) {
set, err := NewBoolSetterFromConfig(o.ctx, param, o.set)
if err != nil {
return nil, err
}
val, err := strconv.ParseBool(o.str)
if err != nil && o.str == "" {
err = nil
}
return func(_ bool) error {
return set(val)
}, err
}
var _ SetBytesProvider = (*constProvider)(nil)
func (o *constProvider) BytesSetter(param string) (func([]byte) error, error) {
set, err := NewBytesSetterFromConfig(o.ctx, param, o.set)
if err != nil {
return nil, err
}
str := strings.ReplaceAll(strings.TrimPrefix(o.str, "0x"), "_", "")
val, err := hex.DecodeString(str)
if err != nil {
err = nil
}
return func(_ []byte) error {
return set(val)
}, err
}