synctv/internal/settings/floate64.go

package settings

import (
	"fmt"
	"math"
	"strconv"
	"sync/atomic"

	"github.com/synctv-org/synctv/internal/db"
	"github.com/synctv-org/synctv/internal/model"
)

type Float64Setting interface {
	Setting
	Set(float64) error
	Get() float64
	Default() float64
	Parse(string) (float64, error)
	Stringify(float64) string
	SetBeforeInit(func(Float64Setting, float64) (float64, error))
	SetBeforeSet(func(Float64Setting, float64) (float64, error))
	SetAfterGet(func(Float64Setting, float64) float64)
}

var _ Float64Setting = (*Float64)(nil)

type Float64 struct {
	validator  func(float64) error
	beforeInit func(Float64Setting, float64) (float64, error)
	beforeSet  func(Float64Setting, float64) (float64, error)
	afterInit  func(Float64Setting, float64)
	afterSet   func(Float64Setting, float64)
	afterGet   func(Float64Setting, float64) float64
	setting
	value        uint64
	defaultValue float64
}

type Float64SettingOption func(*Float64)

func WithInitPriorityFloat64(priority int) Float64SettingOption {
	return func(s *Float64) {
		s.SetInitPriority(priority)
	}
}

func WithValidatorFloat64(validator func(float64) error) Float64SettingOption {
	return func(s *Float64) {
		s.validator = validator
	}
}

func WithBeforeInitFloat64(beforeInit func(Float64Setting, float64) (float64, error)) Float64SettingOption {
	return func(s *Float64) {
		s.SetBeforeInit(beforeInit)
	}
}

func WithBeforeSetFloat64(beforeSet func(Float64Setting, float64) (float64, error)) Float64SettingOption {
	return func(s *Float64) {
		s.SetBeforeSet(beforeSet)
	}
}

func WithAfterInitFloat64(afterInit func(Float64Setting, float64)) Float64SettingOption {
	return func(s *Float64) {
		s.SetAfterInit(afterInit)
	}
}

func WithAfterSetFloat64(afterSet func(Float64Setting, float64)) Float64SettingOption {
	return func(s *Float64) {
		s.SetAfterSet(afterSet)
	}
}

func WithAfterGetFloat64(afterGet func(Float64Setting, float64) float64) Float64SettingOption {
	return func(s *Float64) {
		s.SetAfterGet(afterGet)
	}
}

func newFloat64(name string, value float64, group model.SettingGroup, options ...Float64SettingOption) *Float64 {
	f := &Float64{
		setting: setting{
			name:        name,
			group:       group,
			settingType: model.SettingTypeFloat64,
		},
		defaultValue: value,
	}
	for _, option := range options {
		option(f)
	}
	f.set(value)
	return f
}

func (f *Float64) SetBeforeInit(beforeInit func(Float64Setting, float64) (float64, error)) {
	f.beforeInit = beforeInit
}

func (f *Float64) SetBeforeSet(beforeSet func(Float64Setting, float64) (float64, error)) {
	f.beforeSet = beforeSet
}

func (f *Float64) SetAfterInit(afterInit func(Float64Setting, float64)) {
	f.afterInit = afterInit
}

func (f *Float64) SetAfterSet(afterSet func(Float64Setting, float64)) {
	f.afterSet = afterSet
}

func (f *Float64) SetAfterGet(afterGet func(Float64Setting, float64) float64) {
	f.afterGet = afterGet
}

func (f *Float64) Parse(value string) (float64, error) {
	v, err := strconv.ParseFloat(value, 64)
	if err != nil {
		return 0, err
	}
	if f.validator != nil {
		return v, f.validator(v)
	}
	return v, nil
}

func (f *Float64) Stringify(value float64) string {
	return strconv.FormatFloat(value, 'f', -1, 64)
}

func (f *Float64) Init(value string) error {
	if f.Inited() {
		return ErrSettingAlreadyInited
	}

	v, err := f.Parse(value)
	if err != nil {
		return err
	}

	if f.beforeInit != nil {
		v, err = f.beforeInit(f, v)
		if err != nil {
			return err
		}
	}

	f.set(v)

	if f.afterInit != nil {
		f.afterInit(f, v)
	}

	f.inited = true

	return nil
}

func (f *Float64) String() string {
	return f.Stringify(f.Get())
}

func (f *Float64) Default() float64 {
	return f.defaultValue
}

func (f *Float64) DefaultString() string {
	return f.Stringify(f.defaultValue)
}

func (f *Float64) DefaultInterface() any {
	return f.Default()
}

func (f *Float64) SetString(value string) error {
	v, err := f.Parse(value)
	if err != nil {
		return err
	}

	if f.beforeSet != nil {
		v, err = f.beforeSet(f, v)
		if err != nil {
			return err
		}
	}

	err = db.UpdateSettingItemValue(f.name, f.Stringify(v))
	if err != nil {
		return err
	}

	f.set(v)

	if f.afterSet != nil {
		f.afterSet(f, v)
	}

	return nil
}

func (f *Float64) set(value float64) {
	atomic.StoreUint64(&f.value, math.Float64bits(value))
}

func (f *Float64) Set(v float64) (err error) {
	if f.validator != nil {
		err = f.validator(v)
		if err != nil {
			return err
		}
	}

	if f.beforeSet != nil {
		v, err = f.beforeSet(f, v)
		if err != nil {
			return err
		}
	}

	err = db.UpdateSettingItemValue(f.name, f.Stringify(v))
	if err != nil {
		return err
	}

	f.set(v)

	if f.afterSet != nil {
		f.afterSet(f, v)
	}

	return
}

func (f *Float64) Get() float64 {
	v := math.Float64frombits(atomic.LoadUint64(&f.value))
	if f.afterGet != nil {
		v = f.afterGet(f, v)
	}
	return v
}

func (f *Float64) Interface() any {
	return f.Get()
}

func NewFloat64Setting(k string, v float64, g model.SettingGroup, options ...Float64SettingOption) Float64Setting {
	_, loaded := Settings[k]
	if loaded {
		panic(fmt.Sprintf("setting %s already exists", k))
	}
	return CoverFloat64Setting(k, v, g, options...)
}

func CoverFloat64Setting(k string, v float64, g model.SettingGroup, options ...Float64SettingOption) Float64Setting {
	f := newFloat64(k, v, g, options...)
	Settings[k] = f
	if GroupSettings[g] == nil {
		GroupSettings[g] = make(map[model.SettingGroup]Setting)
	}
	GroupSettings[g][k] = f
	pushNeedInit(f)
	return f
}

func LoadFloat64Setting(k string) (Float64Setting, bool) {
	s, ok := Settings[k]
	if !ok {
		return nil, false
	}
	f, ok := s.(Float64Setting)
	return f, ok
}

func LoadOrNewFloat64Setting(k string, v float64, g model.SettingGroup, options ...Float64SettingOption) Float64Setting {
	s, ok := LoadFloat64Setting(k)
	if ok {
		return s
	}
	return CoverFloat64Setting(k, v, g, options...)
}