getter.go

package provider

import (
	"strconv"

	"github.com/spf13/cast"
)

type getter struct {
	sp    StringProvider
	scale float64
}

func defaultGetters(sp StringProvider, scale float64) *getter {
	return &getter{
		sp:    sp,
		scale: scale,
	}
}

var _ FloatProvider = (*getter)(nil)

// FloatGetter parses float from exec result
func (p *getter) FloatGetter() (func() (float64, error), error) {
	g, err := p.sp.StringGetter()

	return func() (float64, error) {
		s, err := g()
		if err != nil {
			return 0, err
		}

		f, err := strconv.ParseFloat(s, 64)
		if err != nil {
			return 0, err
		}

		return f * p.scale, nil
	}, err
}

var _ IntProvider = (*getter)(nil)

// IntGetter parses int64 from exec result
func (p *getter) IntGetter() (func() (int64, error), error) {
	g, err := p.FloatGetter()

	return func() (int64, error) {
		f, err := g()
		if err != nil {
			return 0, err
		}

		return int64(f), nil
	}, err
}

var _ BoolProvider = (*getter)(nil)

// BoolGetter parses bool from exec result. "on", "true" and 1 are considered truish.
func (p *getter) BoolGetter() (func() (bool, error), error) {
	g, err := p.sp.StringGetter()

	return func() (bool, error) {
		s, err := g()
		if err != nil {
			return false, err
		}

		return cast.ToBoolE(s)
	}, err
}