119 lines
2.1 KiB
Go
119 lines
2.1 KiB
Go
package objects
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/d5/tengo/compiler/token"
|
|
)
|
|
|
|
// Map represents a map of objects.
|
|
type Map struct {
|
|
Value map[string]Object
|
|
}
|
|
|
|
// TypeName returns the name of the type.
|
|
func (o *Map) TypeName() string {
|
|
return "map"
|
|
}
|
|
|
|
func (o *Map) String() string {
|
|
var pairs []string
|
|
for k, v := range o.Value {
|
|
pairs = append(pairs, fmt.Sprintf("%s: %s", k, v.String()))
|
|
}
|
|
|
|
return fmt.Sprintf("{%s}", strings.Join(pairs, ", "))
|
|
}
|
|
|
|
// BinaryOp returns another object that is the result of
|
|
// a given binary operator and a right-hand side object.
|
|
func (o *Map) BinaryOp(op token.Token, rhs Object) (Object, error) {
|
|
return nil, ErrInvalidOperator
|
|
}
|
|
|
|
// Copy returns a copy of the type.
|
|
func (o *Map) Copy() Object {
|
|
c := make(map[string]Object)
|
|
for k, v := range o.Value {
|
|
c[k] = v.Copy()
|
|
}
|
|
|
|
return &Map{Value: c}
|
|
}
|
|
|
|
// IsFalsy returns true if the value of the type is falsy.
|
|
func (o *Map) IsFalsy() bool {
|
|
return len(o.Value) == 0
|
|
}
|
|
|
|
// Equals returns true if the value of the type
|
|
// is equal to the value of another object.
|
|
func (o *Map) Equals(x Object) bool {
|
|
var xVal map[string]Object
|
|
switch x := x.(type) {
|
|
case *Map:
|
|
xVal = x.Value
|
|
case *ImmutableMap:
|
|
xVal = x.Value
|
|
default:
|
|
return false
|
|
}
|
|
|
|
if len(o.Value) != len(xVal) {
|
|
return false
|
|
}
|
|
|
|
for k, v := range o.Value {
|
|
tv := xVal[k]
|
|
if !v.Equals(tv) {
|
|
return false
|
|
}
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
// IndexGet returns the value for the given key.
|
|
func (o *Map) IndexGet(index Object) (res Object, err error) {
|
|
strIdx, ok := index.(*String)
|
|
if !ok {
|
|
err = ErrInvalidIndexType
|
|
return
|
|
}
|
|
|
|
val, ok := o.Value[strIdx.Value]
|
|
if !ok {
|
|
val = UndefinedValue
|
|
}
|
|
|
|
return val, nil
|
|
}
|
|
|
|
// IndexSet sets the value for the given key.
|
|
func (o *Map) IndexSet(index, value Object) (err error) {
|
|
strIdx, ok := ToString(index)
|
|
if !ok {
|
|
err = ErrInvalidIndexType
|
|
return
|
|
}
|
|
|
|
o.Value[strIdx] = value
|
|
|
|
return nil
|
|
}
|
|
|
|
// Iterate creates a map iterator.
|
|
func (o *Map) Iterate() Iterator {
|
|
var keys []string
|
|
for k := range o.Value {
|
|
keys = append(keys, k)
|
|
}
|
|
|
|
return &MapIterator{
|
|
v: o.Value,
|
|
k: keys,
|
|
l: len(keys),
|
|
}
|
|
}
|