100 lines
1.9 KiB
Go
100 lines
1.9 KiB
Go
package objects
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/d5/tengo/compiler/token"
|
|
)
|
|
|
|
// Array represents an array of objects.
|
|
type Array struct {
|
|
Value []Object
|
|
}
|
|
|
|
// TypeName returns the name of the type.
|
|
func (o *Array) TypeName() string {
|
|
return "array"
|
|
}
|
|
|
|
func (o *Array) String() string {
|
|
var elements []string
|
|
for _, e := range o.Value {
|
|
elements = append(elements, e.String())
|
|
}
|
|
|
|
return fmt.Sprintf("[%s]", strings.Join(elements, ", "))
|
|
}
|
|
|
|
// BinaryOp returns another object that is the result of
|
|
// a given binary operator and a right-hand side object.
|
|
func (o *Array) BinaryOp(op token.Token, rhs Object) (Object, error) {
|
|
if rhs, ok := rhs.(*Array); ok {
|
|
switch op {
|
|
case token.Add:
|
|
if len(rhs.Value) == 0 {
|
|
return o, nil
|
|
}
|
|
return &Array{Value: append(o.Value, rhs.Value...)}, nil
|
|
}
|
|
}
|
|
|
|
return nil, ErrInvalidOperator
|
|
}
|
|
|
|
// Copy returns a copy of the type.
|
|
func (o *Array) Copy() Object {
|
|
var c []Object
|
|
for _, elem := range o.Value {
|
|
c = append(c, elem.Copy())
|
|
}
|
|
|
|
return &Array{Value: c}
|
|
}
|
|
|
|
// IsFalsy returns true if the value of the type is falsy.
|
|
func (o *Array) 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 *Array) Equals(x Object) bool {
|
|
t, ok := x.(*Array)
|
|
if !ok {
|
|
return false
|
|
}
|
|
|
|
if len(o.Value) != len(t.Value) {
|
|
return false
|
|
}
|
|
|
|
for i, e := range o.Value {
|
|
if !e.Equals(t.Value[i]) {
|
|
return false
|
|
}
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
// Get returns an element at a given index.
|
|
func (o *Array) Get(index int) (Object, error) {
|
|
if index < 0 || index >= len(o.Value) {
|
|
return nil, errors.New("array index out of bounds")
|
|
}
|
|
|
|
return o.Value[index], nil
|
|
}
|
|
|
|
// Set sets an element at a given index.
|
|
func (o *Array) Set(index int, value Object) error {
|
|
if index < 0 || index >= len(o.Value) {
|
|
return errors.New("array index out of bounds")
|
|
}
|
|
|
|
o.Value[index] = value
|
|
|
|
return nil
|
|
}
|