-
Notifications
You must be signed in to change notification settings - Fork 19
/
number.go
123 lines (95 loc) · 3.5 KB
/
number.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
package types
import (
"math"
"github.com/nyaruka/goflow/utils"
"github.com/shopspring/decimal"
)
func init() {
decimal.MarshalJSONWithoutQuotes = true
}
// XNumber is any whole or fractional number
type XNumber struct {
native decimal.Decimal
}
// NewXNumber creates a new XNumber
func NewXNumber(value decimal.Decimal) XNumber {
return XNumber{native: value}
}
// NewXNumberFromInt creates a new XNumber from the given int
func NewXNumberFromInt(value int) XNumber {
return NewXNumber(decimal.New(int64(value), 0))
}
// NewXNumberFromInt64 creates a new XNumber from the given int
func NewXNumberFromInt64(value int64) XNumber {
return NewXNumber(decimal.New(value, 0))
}
// RequireXNumberFromString creates a new XNumber from the given string
func RequireXNumberFromString(value string) XNumber {
return NewXNumber(decimal.RequireFromString(value))
}
// Describe returns a representation of this type for error messages
func (x XNumber) Describe() string { return x.ToXText(nil).Native() }
// Reduce returns the primitive version of this type (i.e. itself)
func (x XNumber) Reduce(env utils.Environment) XPrimitive { return x }
// ToXText converts this type to text
func (x XNumber) ToXText(env utils.Environment) XText { return NewXText(x.Native().String()) }
// ToXBoolean converts this type to a bool
func (x XNumber) ToXBoolean(env utils.Environment) XBoolean {
return NewXBoolean(!x.Equals(XNumberZero))
}
// ToXJSON is called when this type is passed to @(json(...))
func (x XNumber) ToXJSON(env utils.Environment) XText { return MustMarshalToXText(x.Native()) }
// Native returns the native value of this type
func (x XNumber) Native() decimal.Decimal { return x.native }
// String returns the native string representation of this type
func (x XNumber) String() string { return x.ToXText(nil).Native() }
// Equals determines equality for this type
func (x XNumber) Equals(other XNumber) bool {
return x.Native().Equals(other.Native())
}
// Compare compares this number to another
func (x XNumber) Compare(other XNumber) int {
return x.Native().Cmp(other.Native())
}
// MarshalJSON is called when a struct containing this type is marshaled
func (x XNumber) MarshalJSON() ([]byte, error) {
return x.Native().MarshalJSON()
}
// UnmarshalJSON is called when a struct containing this type is unmarshaled
func (x *XNumber) UnmarshalJSON(data []byte) error {
nativePtr := &x.native
return nativePtr.UnmarshalJSON(data)
}
// XNumberZero is the zero number value
var XNumberZero = NewXNumber(decimal.Zero)
var _ XPrimitive = XNumberZero
// ToXNumber converts the given value to a number or returns an error if that isn't possible
func ToXNumber(env utils.Environment, x XValue) (XNumber, XError) {
if !utils.IsNil(x) {
x = x.Reduce(env)
switch typed := x.(type) {
case XError:
return XNumberZero, typed
case XNumber:
return typed, nil
case XText:
parsed, err := decimal.NewFromString(typed.Native())
if err == nil {
return NewXNumber(parsed), nil
}
}
}
return XNumberZero, NewXErrorf("unable to convert %s to a number", Describe(x))
}
// ToInteger tries to convert the passed in value to an integer or returns an error if that isn't possible
func ToInteger(env utils.Environment, x XValue) (int, XError) {
number, err := ToXNumber(env, x)
if err != nil {
return 0, err
}
intPart := number.Native().IntPart()
if intPart < math.MinInt32 || intPart > math.MaxInt32 {
return 0, NewXErrorf("number value %s is out of range for an integer", number.ToXText(env).Native())
}
return int(intPart), nil
}