diff options
Diffstat (limited to 'vendor/github.com/d5/tengo/script/variable.go')
-rw-r--r-- | vendor/github.com/d5/tengo/script/variable.go | 149 |
1 files changed, 149 insertions, 0 deletions
diff --git a/vendor/github.com/d5/tengo/script/variable.go b/vendor/github.com/d5/tengo/script/variable.go new file mode 100644 index 00000000..c5e01bd9 --- /dev/null +++ b/vendor/github.com/d5/tengo/script/variable.go @@ -0,0 +1,149 @@ +package script + +import ( + "errors" + + "github.com/d5/tengo/objects" +) + +// Variable is a user-defined variable for the script. +type Variable struct { + name string + value *objects.Object +} + +// NewVariable creates a Variable. +func NewVariable(name string, value interface{}) (*Variable, error) { + obj, err := objects.FromInterface(value) + if err != nil { + return nil, err + } + + return &Variable{ + name: name, + value: &obj, + }, nil +} + +// Name returns the name of the variable. +func (v *Variable) Name() string { + return v.name +} + +// Value returns an empty interface of the variable value. +func (v *Variable) Value() interface{} { + return objectToInterface(*v.value) +} + +// ValueType returns the name of the value type. +func (v *Variable) ValueType() string { + return (*v.value).TypeName() +} + +// Int returns int value of the variable value. +// It returns 0 if the value is not convertible to int. +func (v *Variable) Int() int { + c, _ := objects.ToInt(*v.value) + + return c +} + +// Int64 returns int64 value of the variable value. +// It returns 0 if the value is not convertible to int64. +func (v *Variable) Int64() int64 { + c, _ := objects.ToInt64(*v.value) + + return c +} + +// Float returns float64 value of the variable value. +// It returns 0.0 if the value is not convertible to float64. +func (v *Variable) Float() float64 { + c, _ := objects.ToFloat64(*v.value) + + return c +} + +// Char returns rune value of the variable value. +// It returns 0 if the value is not convertible to rune. +func (v *Variable) Char() rune { + c, _ := objects.ToRune(*v.value) + + return c +} + +// Bool returns bool value of the variable value. +// It returns 0 if the value is not convertible to bool. +func (v *Variable) Bool() bool { + c, _ := objects.ToBool(*v.value) + + return c +} + +// Array returns []interface value of the variable value. +// It returns 0 if the value is not convertible to []interface. +func (v *Variable) Array() []interface{} { + switch val := (*v.value).(type) { + case *objects.Array: + var arr []interface{} + for _, e := range val.Value { + arr = append(arr, objectToInterface(e)) + } + return arr + } + + return nil +} + +// Map returns map[string]interface{} value of the variable value. +// It returns 0 if the value is not convertible to map[string]interface{}. +func (v *Variable) Map() map[string]interface{} { + switch val := (*v.value).(type) { + case *objects.Map: + kv := make(map[string]interface{}) + for mk, mv := range val.Value { + kv[mk] = objectToInterface(mv) + } + return kv + } + + return nil +} + +// String returns string value of the variable value. +// It returns 0 if the value is not convertible to string. +func (v *Variable) String() string { + c, _ := objects.ToString(*v.value) + + return c +} + +// Bytes returns a byte slice of the variable value. +// It returns nil if the value is not convertible to byte slice. +func (v *Variable) Bytes() []byte { + c, _ := objects.ToByteSlice(*v.value) + + return c +} + +// Error returns an error if the underlying value is error object. +// If not, this returns nil. +func (v *Variable) Error() error { + err, ok := (*v.value).(*objects.Error) + if ok { + return errors.New(err.String()) + } + + return nil +} + +// Object returns an underlying Object of the variable value. +// Note that returned Object is a copy of an actual Object used in the script. +func (v *Variable) Object() objects.Object { + return *v.value +} + +// IsUndefined returns true if the underlying value is undefined. +func (v *Variable) IsUndefined() bool { + return *v.value == objects.UndefinedValue +} |