type checking and error handling for builtins improved.
This commit is contained in:
@@ -5,7 +5,7 @@ steps:
|
|||||||
- name: build
|
- name: build
|
||||||
image: golang:latest
|
image: golang:latest
|
||||||
commands:
|
commands:
|
||||||
- make test
|
- go test -v -short -cover -coverprofile=coverage.txt ./...
|
||||||
|
|
||||||
- name: coverage
|
- name: coverage
|
||||||
image: plugins/codecov
|
image: plugins/codecov
|
||||||
|
|||||||
@@ -1,20 +1,24 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
// Abs ...
|
// Abs ...
|
||||||
func Abs(args ...object.Object) object.Object {
|
func Abs(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1",
|
"abs", args,
|
||||||
len(args))
|
typing.ExactArgs(1),
|
||||||
|
typing.WithTypes(object.INTEGER_OBJ),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
if i, ok := args[0].(*object.Integer); ok {
|
i := args[0].(*object.Integer)
|
||||||
value := i.Value
|
value := i.Value
|
||||||
if value < 0 {
|
if value < 0 {
|
||||||
value = value * -1
|
value = value * -1
|
||||||
}
|
}
|
||||||
return &object.Integer{Value: value}
|
return &object.Integer{Value: value}
|
||||||
}
|
}
|
||||||
return newError("argument to `abs` not supported, got %s", args[0].Type())
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -1,9 +1,19 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
// Args ...
|
// Args ...
|
||||||
func Args(args ...object.Object) object.Object {
|
func Args(args ...object.Object) object.Object {
|
||||||
|
if err := typing.Check(
|
||||||
|
"args", args,
|
||||||
|
typing.ExactArgs(0),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
elements := make([]object.Object, len(object.Arguments))
|
elements := make([]object.Object, len(object.Arguments))
|
||||||
for i, arg := range object.Arguments {
|
for i, arg := range object.Arguments {
|
||||||
elements[i] = &object.String{Value: arg}
|
elements[i] = &object.String{Value: arg}
|
||||||
|
|||||||
@@ -1,6 +1,9 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
@@ -9,17 +12,12 @@ import (
|
|||||||
|
|
||||||
// Assert ...
|
// Assert ...
|
||||||
func Assert(args ...object.Object) object.Object {
|
func Assert(args ...object.Object) object.Object {
|
||||||
if len(args) != 2 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=2",
|
"assert", args,
|
||||||
len(args))
|
typing.ExactArgs(2),
|
||||||
}
|
typing.WithTypes(object.BOOLEAN_OBJ, object.STRING_OBJ),
|
||||||
if args[0].Type() != object.BOOLEAN_OBJ {
|
); err != nil {
|
||||||
return newError("argument #1 to `assert` must be BOOLEAN, got %s",
|
return newError(err.Error())
|
||||||
args[0].Type())
|
|
||||||
}
|
|
||||||
if args[1].Type() != object.STRING_OBJ {
|
|
||||||
return newError("argument #2 to `assert` must be STRING, got %s",
|
|
||||||
args[0].Type())
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if !args[0].(*object.Boolean).Value {
|
if !args[0].(*object.Boolean).Value {
|
||||||
|
|||||||
@@ -3,18 +3,20 @@ package builtins
|
|||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"monkey/object"
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
"strconv"
|
"strconv"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Bin ...
|
// Bin ...
|
||||||
func Bin(args ...object.Object) object.Object {
|
func Bin(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1",
|
"bin", args,
|
||||||
len(args))
|
typing.ExactArgs(1),
|
||||||
|
typing.WithTypes(object.INTEGER_OBJ),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
if i, ok := args[0].(*object.Integer); ok {
|
i := args[0].(*object.Integer)
|
||||||
return &object.String{Value: fmt.Sprintf("0b%s", strconv.FormatInt(i.Value, 2))}
|
return &object.String{Value: fmt.Sprintf("0b%s", strconv.FormatInt(i.Value, 2))}
|
||||||
}
|
}
|
||||||
return newError("argument to `bin` not supported, got %s", args[0].Type())
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -1,40 +1,18 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
// Bool ...
|
// Bool ...
|
||||||
func Bool(args ...object.Object) object.Object {
|
func Bool(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1", len(args))
|
"bool", args,
|
||||||
|
typing.ExactArgs(1),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
switch arg := args[0].(type) {
|
return &object.Boolean{Value: args[0].Bool()}
|
||||||
case *object.Null:
|
|
||||||
return &object.Boolean{Value: false}
|
|
||||||
case *object.Boolean:
|
|
||||||
return arg
|
|
||||||
case *object.Integer:
|
|
||||||
if arg.Value == 0 {
|
|
||||||
return &object.Boolean{Value: false}
|
|
||||||
}
|
|
||||||
return &object.Boolean{Value: true}
|
|
||||||
case *object.String:
|
|
||||||
if len(arg.Value) > 0 {
|
|
||||||
return &object.Boolean{Value: true}
|
|
||||||
}
|
|
||||||
return &object.Boolean{Value: false}
|
|
||||||
case *object.Array:
|
|
||||||
if len(arg.Elements) > 0 {
|
|
||||||
return &object.Boolean{Value: true}
|
|
||||||
}
|
|
||||||
return &object.Boolean{Value: false}
|
|
||||||
case *object.Hash:
|
|
||||||
if len(arg.Pairs) > 0 {
|
|
||||||
return &object.Boolean{Value: true}
|
|
||||||
}
|
|
||||||
return &object.Boolean{Value: false}
|
|
||||||
|
|
||||||
default:
|
|
||||||
return newError("argument to `bool` not supported, got=%s", args[0].Type())
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,17 +3,19 @@ package builtins
|
|||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"monkey/object"
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Chr ...
|
// Chr ...
|
||||||
func Chr(args ...object.Object) object.Object {
|
func Chr(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1",
|
"chr", args,
|
||||||
len(args))
|
typing.ExactArgs(1),
|
||||||
|
typing.WithTypes(object.INTEGER_OBJ),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
if i, ok := args[0].(*object.Integer); ok {
|
i := args[0].(*object.Integer)
|
||||||
return &object.String{Value: fmt.Sprintf("%c", rune(i.Value))}
|
return &object.String{Value: fmt.Sprintf("%c", rune(i.Value))}
|
||||||
}
|
}
|
||||||
return newError("argument to `chr` not supported, got %s", args[0].Type())
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -1,24 +1,24 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
// Divmod ...
|
// Divmod ...
|
||||||
func Divmod(args ...object.Object) object.Object {
|
func Divmod(args ...object.Object) object.Object {
|
||||||
if len(args) != 2 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=2",
|
"divmod", args,
|
||||||
len(args))
|
typing.ExactArgs(2),
|
||||||
|
typing.WithTypes(object.INTEGER_OBJ, object.INTEGER_OBJ),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
if a, ok := args[0].(*object.Integer); ok {
|
a := args[0].(*object.Integer)
|
||||||
if b, ok := args[1].(*object.Integer); ok {
|
b := args[1].(*object.Integer)
|
||||||
elements := make([]object.Object, 2)
|
elements := make([]object.Object, 2)
|
||||||
elements[0] = &object.Integer{Value: a.Value / b.Value}
|
elements[0] = &object.Integer{Value: a.Value / b.Value}
|
||||||
elements[1] = &object.Integer{Value: a.Value % b.Value}
|
elements[1] = &object.Integer{Value: a.Value % b.Value}
|
||||||
return &object.Array{Elements: elements}
|
return &object.Array{Elements: elements}
|
||||||
} else {
|
|
||||||
return newError("expected argument #2 to `divmod` to be `int` got=%s", args[1].Type())
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
return newError("expected argument #1 to `divmod` to be `int` got=%s", args[0].Type())
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,15 +1,22 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
// Exit ...
|
// Exit ...
|
||||||
func Exit(args ...object.Object) object.Object {
|
func Exit(args ...object.Object) object.Object {
|
||||||
|
if err := typing.Check(
|
||||||
|
"exit", args,
|
||||||
|
typing.RangeOfArgs(0, 1),
|
||||||
|
typing.WithTypes(object.INTEGER_OBJ),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
var status int
|
var status int
|
||||||
if len(args) == 1 {
|
if len(args) == 1 {
|
||||||
if args[0].Type() != object.INTEGER_OBJ {
|
|
||||||
return newError("argument to `exit` must be INTEGER, got %s",
|
|
||||||
args[0].Type())
|
|
||||||
}
|
|
||||||
status = int(args[0].(*object.Integer).Value)
|
status = int(args[0].(*object.Integer).Value)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,9 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
@@ -9,22 +12,16 @@ import (
|
|||||||
|
|
||||||
// FFI ...
|
// FFI ...
|
||||||
func FFI(args ...object.Object) object.Object {
|
func FFI(args ...object.Object) object.Object {
|
||||||
if len(args) != 2 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=2",
|
"ffi", args,
|
||||||
len(args))
|
typing.ExactArgs(2),
|
||||||
|
typing.WithTypes(object.STRING_OBJ, object.STRING_OBJ),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
arg, ok := args[0].(*object.String)
|
name := args[0].(*object.String).Value
|
||||||
if !ok {
|
symbol := args[1].(*object.String).Value
|
||||||
return newError("argument #1 to `ffi` expected to be `str` got=%T", args[0].Type())
|
|
||||||
}
|
|
||||||
name := arg.Value
|
|
||||||
|
|
||||||
arg, ok = args[1].(*object.String)
|
|
||||||
if !ok {
|
|
||||||
return newError("argument #2 to `ffi` expected to be `str` got=%T", args[0].Type())
|
|
||||||
}
|
|
||||||
symbol := arg.Value
|
|
||||||
|
|
||||||
p, err := plugin.Open(fmt.Sprintf("%s.so", name))
|
p, err := plugin.Open(fmt.Sprintf("%s.so", name))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -36,5 +33,5 @@ func FFI(args ...object.Object) object.Object {
|
|||||||
return newError("error finding symbol: %s", err)
|
return newError("error finding symbol: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return &object.Builtin{Name: symbol, Fn: v.(func(...object.Object) object.Object)}
|
return &object.Builtin{Name: symbol, Fn: v.(object.BuiltinFunction)}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,6 +2,7 @@ package builtins
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"monkey/object"
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
"sort"
|
"sort"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -11,19 +12,29 @@ import (
|
|||||||
|
|
||||||
// Find ...
|
// Find ...
|
||||||
func Find(args ...object.Object) object.Object {
|
func Find(args ...object.Object) object.Object {
|
||||||
if len(args) != 2 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=2",
|
"find", args,
|
||||||
len(args))
|
typing.ExactArgs(2),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// find substring in string
|
||||||
if haystack, ok := args[0].(*object.String); ok {
|
if haystack, ok := args[0].(*object.String); ok {
|
||||||
if needle, ok := args[1].(*object.String); ok {
|
if err := typing.Check(
|
||||||
|
"find", args,
|
||||||
|
typing.WithTypes(object.STRING_OBJ, object.STRING_OBJ),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
needle := args[1].(*object.String)
|
||||||
index := strings.Index(haystack.Value, needle.Value)
|
index := strings.Index(haystack.Value, needle.Value)
|
||||||
return &object.Integer{Value: int64(index)}
|
return &object.Integer{Value: int64(index)}
|
||||||
} else {
|
|
||||||
return newError("expected arg #2 to be `str` got got=%T", args[1])
|
|
||||||
}
|
}
|
||||||
} else if haystack, ok := args[0].(*object.Array); ok {
|
|
||||||
|
// find in array
|
||||||
|
if haystack, ok := args[0].(*object.Array); ok {
|
||||||
needle := args[1].(object.Comparable)
|
needle := args[1].(object.Comparable)
|
||||||
i := sort.Search(len(haystack.Elements), func(i int) bool {
|
i := sort.Search(len(haystack.Elements), func(i int) bool {
|
||||||
return needle.Compare(haystack.Elements[i]) == 0
|
return needle.Compare(haystack.Elements[i]) == 0
|
||||||
@@ -32,7 +43,10 @@ func Find(args ...object.Object) object.Object {
|
|||||||
return &object.Integer{Value: int64(i)}
|
return &object.Integer{Value: int64(i)}
|
||||||
}
|
}
|
||||||
return &object.Integer{Value: -1}
|
return &object.Integer{Value: -1}
|
||||||
} else {
|
|
||||||
return newError("expected arg #1 to be `str` or `array` got got=%T", args[0])
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
return newError(
|
||||||
|
"TypeError: find() expected argument #1 to be `array` or `str` got `%s`",
|
||||||
|
args[0].Type(),
|
||||||
|
)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,16 +1,18 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
// First ...
|
// First ...
|
||||||
func First(args ...object.Object) object.Object {
|
func First(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1",
|
"first", args,
|
||||||
len(args))
|
typing.ExactArgs(1),
|
||||||
}
|
typing.WithTypes(object.ARRAY_OBJ),
|
||||||
if args[0].Type() != object.ARRAY_OBJ {
|
); err != nil {
|
||||||
return newError("argument to `first` must be array, got %s",
|
return newError(err.Error())
|
||||||
args[0].Type())
|
|
||||||
}
|
}
|
||||||
|
|
||||||
arr := args[0].(*object.Array)
|
arr := args[0].(*object.Array)
|
||||||
|
|||||||
@@ -1,16 +1,22 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
// HashOf ...
|
// HashOf ...
|
||||||
func HashOf(args ...object.Object) object.Object {
|
func HashOf(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1",
|
"hash", args,
|
||||||
len(args))
|
typing.ExactArgs(1),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
if hash, ok := args[0].(object.Hashable); ok {
|
if hash, ok := args[0].(object.Hashable); ok {
|
||||||
return &object.Integer{Value: int64(hash.HashKey().Value)}
|
return &object.Integer{Value: int64(hash.HashKey().Value)}
|
||||||
}
|
}
|
||||||
return newError("argument #1 to `hash()` is not hashable: %s", args[0].Inspect())
|
|
||||||
|
return newError("TypeError: hash() expected argument #1 to be hashable")
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,18 +3,20 @@ package builtins
|
|||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"monkey/object"
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
"strconv"
|
"strconv"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Hex ...
|
// Hex ...
|
||||||
func Hex(args ...object.Object) object.Object {
|
func Hex(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1",
|
"hex", args,
|
||||||
len(args))
|
typing.ExactArgs(1),
|
||||||
|
typing.WithTypes(object.INTEGER_OBJ),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
if i, ok := args[0].(*object.Integer); ok {
|
i := args[0].(*object.Integer)
|
||||||
return &object.String{Value: fmt.Sprintf("0x%s", strconv.FormatInt(i.Value, 16))}
|
return &object.String{Value: fmt.Sprintf("0x%s", strconv.FormatInt(i.Value, 16))}
|
||||||
}
|
}
|
||||||
return newError("argument to `hex` not supported, got %s", args[0].Type())
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -3,13 +3,16 @@ package builtins
|
|||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"monkey/object"
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
)
|
)
|
||||||
|
|
||||||
// IdOf ...
|
// IdOf ...
|
||||||
func IdOf(args ...object.Object) object.Object {
|
func IdOf(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1",
|
"id", args,
|
||||||
len(args))
|
typing.ExactArgs(1),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
arg := args[0]
|
arg := args[0]
|
||||||
@@ -32,7 +35,7 @@ func IdOf(args ...object.Object) object.Object {
|
|||||||
return &object.String{Value: fmt.Sprintf("%p", c)}
|
return &object.String{Value: fmt.Sprintf("%p", c)}
|
||||||
} else if b, ok := arg.(*object.Builtin); ok {
|
} else if b, ok := arg.(*object.Builtin); ok {
|
||||||
return &object.String{Value: fmt.Sprintf("%p", b)}
|
return &object.String{Value: fmt.Sprintf("%p", b)}
|
||||||
} else {
|
|
||||||
return newError("argument 31 to `id()` unsupported got=%T", arg.Type())
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,9 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bufio"
|
"bufio"
|
||||||
@@ -11,15 +14,17 @@ import (
|
|||||||
|
|
||||||
// Input ...
|
// Input ...
|
||||||
func Input(args ...object.Object) object.Object {
|
func Input(args ...object.Object) object.Object {
|
||||||
if len(args) > 0 {
|
if err := typing.Check(
|
||||||
obj, ok := args[0].(*object.String)
|
"input", args,
|
||||||
if !ok {
|
typing.RangeOfArgs(0, 1),
|
||||||
return newError(
|
typing.WithTypes(object.STRING_OBJ),
|
||||||
"argument to `input` not supported, got %s",
|
); err != nil {
|
||||||
args[0].Type(),
|
return newError(err.Error())
|
||||||
)
|
|
||||||
}
|
}
|
||||||
fmt.Fprintf(os.Stdout, obj.Value)
|
|
||||||
|
if len(args) == 1 {
|
||||||
|
prompt := args[0].(*object.String).Value
|
||||||
|
fmt.Fprintf(os.Stdout, prompt)
|
||||||
}
|
}
|
||||||
|
|
||||||
buffer := bufio.NewReader(os.Stdin)
|
buffer := bufio.NewReader(os.Stdin)
|
||||||
|
|||||||
@@ -1,13 +1,19 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
import "strconv"
|
import "strconv"
|
||||||
|
|
||||||
// Int ...
|
// Int ...
|
||||||
func Int(args ...object.Object) object.Object {
|
func Int(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1", len(args))
|
"int", args,
|
||||||
|
typing.ExactArgs(1),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
switch arg := args[0].(type) {
|
switch arg := args[0].(type) {
|
||||||
@@ -23,9 +29,8 @@ func Int(args ...object.Object) object.Object {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return newError("could not parse string to int: %s", err)
|
return newError("could not parse string to int: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return &object.Integer{Value: n}
|
return &object.Integer{Value: n}
|
||||||
default:
|
default:
|
||||||
return newError("argument to `int` not supported, got=%s", args[0].Type())
|
return &object.Integer{}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,9 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"strings"
|
"strings"
|
||||||
@@ -8,22 +11,19 @@ import (
|
|||||||
|
|
||||||
// Join ...
|
// Join ...
|
||||||
func Join(args ...object.Object) object.Object {
|
func Join(args ...object.Object) object.Object {
|
||||||
if len(args) != 2 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1",
|
"join", args,
|
||||||
len(args))
|
typing.ExactArgs(2),
|
||||||
|
typing.WithTypes(object.ARRAY_OBJ, object.STRING_OBJ),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
if arr, ok := args[0].(*object.Array); ok {
|
arr := args[0].(*object.Array)
|
||||||
if sep, ok := args[1].(*object.String); ok {
|
sep := args[1].(*object.String)
|
||||||
a := make([]string, len(arr.Elements))
|
a := make([]string, len(arr.Elements))
|
||||||
for i, el := range arr.Elements {
|
for i, el := range arr.Elements {
|
||||||
a[i] = el.String()
|
a[i] = el.String()
|
||||||
}
|
}
|
||||||
return &object.String{Value: strings.Join(a, sep.Value)}
|
return &object.String{Value: strings.Join(a, sep.Value)}
|
||||||
} else {
|
|
||||||
return newError("expected arg #2 to be `str` got got=%T", args[1])
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
return newError("expected arg #1 to be `array` got got=%T", args[0])
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,16 +1,18 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
// Last ...
|
// Last ...
|
||||||
func Last(args ...object.Object) object.Object {
|
func Last(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1",
|
"last", args,
|
||||||
len(args))
|
typing.ExactArgs(1),
|
||||||
}
|
typing.WithTypes(object.ARRAY_OBJ),
|
||||||
if args[0].Type() != object.ARRAY_OBJ {
|
); err != nil {
|
||||||
return newError("argument to `last` must be array, got %s",
|
return newError(err.Error())
|
||||||
args[0].Type())
|
|
||||||
}
|
}
|
||||||
|
|
||||||
arr := args[0].(*object.Array)
|
arr := args[0].(*object.Array)
|
||||||
|
|||||||
@@ -1,23 +1,21 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
import "unicode/utf8"
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
// Len ...
|
// Len ...
|
||||||
func Len(args ...object.Object) object.Object {
|
func Len(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1",
|
"len", args,
|
||||||
len(args))
|
typing.ExactArgs(1),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
switch arg := args[0].(type) {
|
if size, ok := args[0].(object.Sizeable); ok {
|
||||||
case *object.Array:
|
return &object.Integer{Value: int64(size.Len())}
|
||||||
return &object.Integer{Value: int64(len(arg.Elements))}
|
|
||||||
case *object.String:
|
|
||||||
return &object.Integer{Value: int64(utf8.RuneCountInString(arg.Value))}
|
|
||||||
default:
|
|
||||||
return newError("argument to `len` not supported, got %s",
|
|
||||||
args[0].Type())
|
|
||||||
}
|
}
|
||||||
|
return newError("TypeError: object of type '%s' has no len()", args[0].Type())
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,18 +1,22 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
import "strings"
|
import "strings"
|
||||||
|
|
||||||
// Lower ...
|
// Lower ...
|
||||||
func Lower(args ...object.Object) object.Object {
|
func Lower(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1",
|
"lower", args,
|
||||||
len(args))
|
typing.ExactArgs(1),
|
||||||
|
typing.WithTypes(object.STRING_OBJ),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
if str, ok := args[0].(*object.String); ok {
|
str := args[0].(*object.String)
|
||||||
return &object.String{Value: strings.ToLower(str.Value)}
|
return &object.String{Value: strings.ToLower(str.Value)}
|
||||||
}
|
}
|
||||||
return newError("expected `str` argument to `lower` got=%T", args[0])
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -2,17 +2,21 @@ package builtins
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"monkey/object"
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
"sort"
|
"sort"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Max ...
|
// Max ...
|
||||||
func Max(args ...object.Object) object.Object {
|
func Max(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1",
|
"max", args,
|
||||||
len(args))
|
typing.ExactArgs(1),
|
||||||
|
typing.WithTypes(object.ARRAY_OBJ),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
if a, ok := args[0].(*object.Array); ok {
|
a := args[0].(*object.Array)
|
||||||
// TODO: Make this more generic
|
// TODO: Make this more generic
|
||||||
xs := make([]int, len(a.Elements))
|
xs := make([]int, len(a.Elements))
|
||||||
for n, e := range a.Elements {
|
for n, e := range a.Elements {
|
||||||
@@ -25,5 +29,3 @@ func Max(args ...object.Object) object.Object {
|
|||||||
sort.Ints(xs)
|
sort.Ints(xs)
|
||||||
return &object.Integer{Value: int64(xs[len(xs)-1])}
|
return &object.Integer{Value: int64(xs[len(xs)-1])}
|
||||||
}
|
}
|
||||||
return newError("argument #1 to `max` expected to be `array` got=%T", args[0].Type())
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -2,17 +2,21 @@ package builtins
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"monkey/object"
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
"sort"
|
"sort"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Min ...
|
// Min ...
|
||||||
func Min(args ...object.Object) object.Object {
|
func Min(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1",
|
"min", args,
|
||||||
len(args))
|
typing.ExactArgs(1),
|
||||||
|
typing.WithTypes(object.ARRAY_OBJ),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
if a, ok := args[0].(*object.Array); ok {
|
a := args[0].(*object.Array)
|
||||||
// TODO: Make this more generic
|
// TODO: Make this more generic
|
||||||
xs := make([]int, len(a.Elements))
|
xs := make([]int, len(a.Elements))
|
||||||
for n, e := range a.Elements {
|
for n, e := range a.Elements {
|
||||||
@@ -25,5 +29,3 @@ func Min(args ...object.Object) object.Object {
|
|||||||
sort.Ints(xs)
|
sort.Ints(xs)
|
||||||
return &object.Integer{Value: int64(xs[0])}
|
return &object.Integer{Value: int64(xs[0])}
|
||||||
}
|
}
|
||||||
return newError("argument #1 to `min` expected to be `array` got=%T", args[0].Type())
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -3,18 +3,20 @@ package builtins
|
|||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"monkey/object"
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
"strconv"
|
"strconv"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Oct ...
|
// Oct ...
|
||||||
func Oct(args ...object.Object) object.Object {
|
func Oct(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1",
|
"oct", args,
|
||||||
len(args))
|
typing.ExactArgs(1),
|
||||||
|
typing.WithTypes(object.INTEGER_OBJ),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
if i, ok := args[0].(*object.Integer); ok {
|
i := args[0].(*object.Integer)
|
||||||
return &object.String{Value: fmt.Sprintf("0%s", strconv.FormatInt(i.Value, 8))}
|
return &object.String{Value: fmt.Sprintf("0%s", strconv.FormatInt(i.Value, 8))}
|
||||||
}
|
}
|
||||||
return newError("argument to `oct` not supported, got %s", args[0].Type())
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -1,19 +1,26 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
// Ord ...
|
// Ord ...
|
||||||
func Ord(args ...object.Object) object.Object {
|
func Ord(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1",
|
"ord", args,
|
||||||
len(args))
|
typing.ExactArgs(1),
|
||||||
|
typing.WithTypes(object.STRING_OBJ),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
if s, ok := args[0].(*object.String); ok {
|
s := args[0].(*object.String)
|
||||||
if len(s.Value) == 1 {
|
if len(s.Value) == 1 {
|
||||||
return &object.Integer{Value: int64(s.Value[0])}
|
return &object.Integer{Value: int64(s.Value[0])}
|
||||||
}
|
}
|
||||||
return newError("`ord()` expected a character but got string of length %d", len(s.Value))
|
return newError(
|
||||||
}
|
"TypeError: ord() expected a single character `str` got=%s",
|
||||||
return newError("argument to `ord` not supported, got %s", args[0].Type())
|
s.Inspect(),
|
||||||
|
)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,23 +1,25 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
// Pop ...
|
// Pop ...
|
||||||
func Pop(args ...object.Object) object.Object {
|
func Pop(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1",
|
"pop", args,
|
||||||
len(args))
|
typing.ExactArgs(1),
|
||||||
}
|
typing.WithTypes(object.ARRAY_OBJ),
|
||||||
if args[0].Type() != object.ARRAY_OBJ {
|
); err != nil {
|
||||||
return newError("argument to `pop` must be array, got %s",
|
return newError(err.Error())
|
||||||
args[0].Type())
|
|
||||||
}
|
}
|
||||||
|
|
||||||
arr := args[0].(*object.Array)
|
arr := args[0].(*object.Array)
|
||||||
length := len(arr.Elements)
|
length := len(arr.Elements)
|
||||||
|
|
||||||
if length == 0 {
|
if length == 0 {
|
||||||
return newError("cannot pop from an empty array")
|
return newError("IndexError: pop from an empty array")
|
||||||
}
|
}
|
||||||
|
|
||||||
element := arr.Elements[length-1]
|
element := arr.Elements[length-1]
|
||||||
|
|||||||
@@ -1,6 +1,9 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
func pow(x, y int64) int64 {
|
func pow(x, y int64) int64 {
|
||||||
p := int64(1)
|
p := int64(1)
|
||||||
@@ -16,19 +19,16 @@ func pow(x, y int64) int64 {
|
|||||||
|
|
||||||
// Pow ...
|
// Pow ...
|
||||||
func Pow(args ...object.Object) object.Object {
|
func Pow(args ...object.Object) object.Object {
|
||||||
if len(args) != 2 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=2",
|
"pow", args,
|
||||||
len(args))
|
typing.ExactArgs(2),
|
||||||
|
typing.WithTypes(object.INTEGER_OBJ, object.INTEGER_OBJ),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
if x, ok := args[0].(*object.Integer); ok {
|
x := args[0].(*object.Integer)
|
||||||
if y, ok := args[1].(*object.Integer); ok {
|
y := args[1].(*object.Integer)
|
||||||
value := pow(x.Value, y.Value)
|
value := pow(x.Value, y.Value)
|
||||||
return &object.Integer{Value: value}
|
return &object.Integer{Value: value}
|
||||||
} else {
|
|
||||||
return newError("expected argument #2 to `pow` to be `int` got=%s", args[1].Type())
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
return newError("expected argument #1 to `pow` to be `int` got=%s", args[0].Type())
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,14 +1,23 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
import "fmt"
|
import "fmt"
|
||||||
|
|
||||||
// Print ...
|
// Print ...
|
||||||
func Print(args ...object.Object) object.Object {
|
func Print(args ...object.Object) object.Object {
|
||||||
for _, arg := range args {
|
if err := typing.Check(
|
||||||
fmt.Println(arg.String())
|
"print", args,
|
||||||
|
typing.MinimumArgs(1),
|
||||||
|
typing.WithTypes(object.STRING_OBJ),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fmt.Println(args[0].String())
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,28 +1,22 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
// Push ...
|
// Push ...
|
||||||
func Push(args ...object.Object) object.Object {
|
func Push(args ...object.Object) object.Object {
|
||||||
if len(args) != 2 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=2",
|
"push", args,
|
||||||
len(args))
|
typing.ExactArgs(2),
|
||||||
}
|
typing.WithTypes(object.ARRAY_OBJ),
|
||||||
if args[0].Type() != object.ARRAY_OBJ {
|
); err != nil {
|
||||||
return newError("argument to `push` must be array, got %s",
|
return newError(err.Error())
|
||||||
args[0].Type())
|
|
||||||
}
|
}
|
||||||
|
|
||||||
arr := args[0].(*object.Array)
|
arr := args[0].(*object.Array)
|
||||||
length := len(arr.Elements)
|
newArray := arr.Copy()
|
||||||
|
newArray.Append(args[1])
|
||||||
newElements := make([]object.Object, length+1)
|
return newArray
|
||||||
copy(newElements, arr.Elements)
|
|
||||||
if immutable, ok := args[1].(object.Immutable); ok {
|
|
||||||
newElements[length] = immutable.Clone()
|
|
||||||
} else {
|
|
||||||
newElements[length] = args[1]
|
|
||||||
}
|
|
||||||
|
|
||||||
return &object.Array{Elements: newElements}
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,27 +1,25 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"os"
|
"io/ioutil"
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Read ...
|
// Read ...
|
||||||
func Read(args ...object.Object) object.Object {
|
func Read(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1",
|
"read", args,
|
||||||
len(args))
|
typing.ExactArgs(1),
|
||||||
|
typing.WithTypes(object.STRING_OBJ),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
arg, ok := args[0].(*object.String)
|
filename := args[0].(*object.String).Value
|
||||||
if !ok {
|
data, err := ioutil.ReadFile(filename)
|
||||||
return newError("argument to `read` expected to be `str` got=%T", args[0].Type())
|
|
||||||
}
|
|
||||||
|
|
||||||
filename := arg.Value
|
|
||||||
data, err := os.ReadFile(filename)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return newError("error reading file: %s", err)
|
return newError("IOError: error reading from file %s: %s", filename, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return &object.String{Value: string(data)}
|
return &object.String{Value: string(data)}
|
||||||
|
|||||||
@@ -1,25 +1,22 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
// Rest ...
|
// Rest ...
|
||||||
func Rest(args ...object.Object) object.Object {
|
func Rest(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1",
|
"rest", args,
|
||||||
len(args))
|
typing.ExactArgs(1),
|
||||||
}
|
typing.WithTypes(object.ARRAY_OBJ),
|
||||||
if args[0].Type() != object.ARRAY_OBJ {
|
); err != nil {
|
||||||
return newError("argument to `rest` must be array, got %s",
|
return newError(err.Error())
|
||||||
args[0].Type())
|
|
||||||
}
|
}
|
||||||
|
|
||||||
arr := args[0].(*object.Array)
|
arr := args[0].(*object.Array)
|
||||||
length := len(arr.Elements)
|
newArray := arr.Copy()
|
||||||
if length > 0 {
|
newArray.PopLeft()
|
||||||
newElements := make([]object.Object, length-1, length-1)
|
return newArray
|
||||||
copy(newElements, arr.Elements[1:length])
|
|
||||||
return &object.Array{Elements: newElements}
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,19 +2,21 @@ package builtins
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"monkey/object"
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Reversed ...
|
// Reversed ...
|
||||||
func Reversed(args ...object.Object) object.Object {
|
func Reversed(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1",
|
"reversed", args,
|
||||||
len(args))
|
typing.ExactArgs(1),
|
||||||
|
typing.WithTypes(object.ARRAY_OBJ),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
if a, ok := args[0].(*object.Array); ok {
|
arr := args[0].(*object.Array)
|
||||||
newArray := a.Copy()
|
newArray := arr.Copy()
|
||||||
newArray.Reverse()
|
newArray.Reverse()
|
||||||
return newArray
|
return newArray
|
||||||
}
|
}
|
||||||
return newError("argument #1 to `reversed` expected to be `array` got=%T", args[0].Type())
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -2,20 +2,22 @@ package builtins
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"monkey/object"
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
"sort"
|
"sort"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Sorted ...
|
// Sorted ...
|
||||||
func Sorted(args ...object.Object) object.Object {
|
func Sorted(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1",
|
"sort", args,
|
||||||
len(args))
|
typing.ExactArgs(1),
|
||||||
|
typing.WithTypes(object.ARRAY_OBJ),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
if a, ok := args[0].(*object.Array); ok {
|
arr := args[0].(*object.Array)
|
||||||
newArray := a.Copy()
|
newArray := arr.Copy()
|
||||||
sort.Sort(newArray)
|
sort.Sort(newArray)
|
||||||
return newArray
|
return newArray
|
||||||
}
|
}
|
||||||
return newError("argument #1 to `sorted` expected to be `array` got=%T", args[0].Type())
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -1,6 +1,9 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"strings"
|
"strings"
|
||||||
@@ -8,22 +11,19 @@ import (
|
|||||||
|
|
||||||
// Split ...
|
// Split ...
|
||||||
func Split(args ...object.Object) object.Object {
|
func Split(args ...object.Object) object.Object {
|
||||||
if len(args) < 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1",
|
"split", args,
|
||||||
len(args))
|
typing.RangeOfArgs(1, 2),
|
||||||
|
typing.WithTypes(object.STRING_OBJ, object.STRING_OBJ),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
if obj, ok := args[0].(*object.String); ok {
|
|
||||||
var sep string
|
var sep string
|
||||||
|
s := args[0].(*object.String).Value
|
||||||
s := obj.Value
|
|
||||||
|
|
||||||
if len(args) == 2 {
|
if len(args) == 2 {
|
||||||
if obj, ok := args[1].(*object.String); ok {
|
sep = args[1].(*object.String).Value
|
||||||
sep = obj.Value
|
|
||||||
} else {
|
|
||||||
return newError("expected arg #2 to be `str` got=%T", args[1])
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
tokens := strings.Split(s, sep)
|
tokens := strings.Split(s, sep)
|
||||||
@@ -32,7 +32,4 @@ func Split(args ...object.Object) object.Object {
|
|||||||
elements[i] = &object.String{Value: token}
|
elements[i] = &object.String{Value: token}
|
||||||
}
|
}
|
||||||
return &object.Array{Elements: elements}
|
return &object.Array{Elements: elements}
|
||||||
} else {
|
|
||||||
return newError("expected arg #1 to be `str` got got=%T", args[0])
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,21 +1,18 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Str ...
|
// Str ...
|
||||||
func Str(args ...object.Object) object.Object {
|
func Str(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1", len(args))
|
"str", args,
|
||||||
|
typing.ExactArgs(1),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
arg, ok := args[0].(fmt.Stringer)
|
return &object.String{Value: args[0].String()}
|
||||||
if !ok {
|
|
||||||
return newError("argument to `str` not supported, got %s", args[0].Type())
|
|
||||||
}
|
|
||||||
|
|
||||||
return &object.String{Value: arg.String()}
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,11 +1,17 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
|
)
|
||||||
|
|
||||||
// TypeOf ...
|
// TypeOf ...
|
||||||
func TypeOf(args ...object.Object) object.Object {
|
func TypeOf(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1", len(args))
|
"type", args,
|
||||||
|
typing.ExactArgs(1),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
return &object.String{Value: string(args[0].Type())}
|
return &object.String{Value: string(args[0].Type())}
|
||||||
|
|||||||
@@ -1,18 +1,20 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
import (
|
||||||
|
"monkey/object"
|
||||||
import "strings"
|
"monkey/typing"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
// Upper ...
|
// Upper ...
|
||||||
func Upper(args ...object.Object) object.Object {
|
func Upper(args ...object.Object) object.Object {
|
||||||
if len(args) != 1 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=1",
|
"upper", args,
|
||||||
len(args))
|
typing.ExactArgs(1),
|
||||||
|
typing.WithTypes(object.STRING_OBJ),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
if str, ok := args[0].(*object.String); ok {
|
return &object.String{Value: strings.ToUpper(args[0].(*object.String).Value)}
|
||||||
return &object.String{Value: strings.ToUpper(str.Value)}
|
|
||||||
}
|
|
||||||
return newError("expected `str` argument to `upper` got=%T", args[0])
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,33 +1,27 @@
|
|||||||
package builtins
|
package builtins
|
||||||
|
|
||||||
import "monkey/object"
|
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"os"
|
"io/ioutil"
|
||||||
|
"monkey/object"
|
||||||
|
"monkey/typing"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Write ...
|
// Write ...
|
||||||
func Write(args ...object.Object) object.Object {
|
func Write(args ...object.Object) object.Object {
|
||||||
if len(args) != 2 {
|
if err := typing.Check(
|
||||||
return newError("wrong number of arguments. got=%d, want=2",
|
"write", args,
|
||||||
len(args))
|
typing.ExactArgs(2),
|
||||||
|
typing.WithTypes(object.STRING_OBJ, object.STRING_OBJ),
|
||||||
|
); err != nil {
|
||||||
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
arg, ok := args[0].(*object.String)
|
filename := args[0].(*object.String).Value
|
||||||
if !ok {
|
data := []byte(args[1].(*object.String).Value)
|
||||||
return newError("argument #1 to `write` expected to be `str` got=%T", args[0].Type())
|
|
||||||
}
|
|
||||||
filename := arg.Value
|
|
||||||
|
|
||||||
arg, ok = args[1].(*object.String)
|
err := ioutil.WriteFile(filename, data, 0755)
|
||||||
if !ok {
|
|
||||||
return newError("argument #2 to `write` expected to be `str` got=%T", args[1].Type())
|
|
||||||
}
|
|
||||||
data := []byte(arg.Value)
|
|
||||||
|
|
||||||
err := os.WriteFile(filename, data, 0755)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return newError("error writing file: %s", err)
|
return newError("IOError: error writing file %s: %s", filename, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return &object.Null{}
|
return &object.Null{}
|
||||||
|
|||||||
@@ -398,24 +398,24 @@ func TestBuiltinFunctions(t *testing.T) {
|
|||||||
{`len("")`, 0},
|
{`len("")`, 0},
|
||||||
{`len("four")`, 4},
|
{`len("four")`, 4},
|
||||||
{`len("hello world")`, 11},
|
{`len("hello world")`, 11},
|
||||||
{`len(1)`, errors.New("argument to `len` not supported, got int")},
|
{`len(1)`, errors.New("TypeError: object of type 'int' has no len()")},
|
||||||
{`len("one", "two")`, errors.New("wrong number of arguments. got=2, want=1")},
|
{`len("one", "two")`, errors.New("TypeError: len() takes exactly 1 argument (2 given)")},
|
||||||
{`len("∑")`, 1},
|
{`len("∑")`, 1},
|
||||||
{`len([1, 2, 3])`, 3},
|
{`len([1, 2, 3])`, 3},
|
||||||
{`len([])`, 0},
|
{`len([])`, 0},
|
||||||
{`first([1, 2, 3])`, 1},
|
{`first([1, 2, 3])`, 1},
|
||||||
{`first([])`, nil},
|
{`first([])`, nil},
|
||||||
{`first(1)`, errors.New("argument to `first` must be array, got int")},
|
{`first(1)`, errors.New("TypeError: first() expected argument #1 to be `array` got `int`")},
|
||||||
{`last([1, 2, 3])`, 3},
|
{`last([1, 2, 3])`, 3},
|
||||||
{`last([])`, nil},
|
{`last([])`, nil},
|
||||||
{`last(1)`, errors.New("argument to `last` must be array, got int")},
|
{`last(1)`, errors.New("TypeError: last() expected argument #1 to be `array` got `int`")},
|
||||||
{`rest([1, 2, 3])`, []int{2, 3}},
|
{`rest([1, 2, 3])`, []int{2, 3}},
|
||||||
{`rest([])`, nil},
|
{`rest([])`, nil},
|
||||||
{`push([], 1)`, []int{1}},
|
{`push([], 1)`, []int{1}},
|
||||||
{`push(1, 1)`, errors.New("argument to `push` must be array, got int")},
|
{`push(1, 1)`, errors.New("TypeError: push() expected argument #1 to be `array` got `int`")},
|
||||||
{`print("Hello World")`, nil},
|
{`print("Hello World")`, nil},
|
||||||
{`input()`, ""},
|
{`input()`, ""},
|
||||||
{`pop([])`, errors.New("cannot pop from an empty array")},
|
{`pop([])`, errors.New("IndexError: pop from an empty array")},
|
||||||
{`pop([1])`, 1},
|
{`pop([1])`, 1},
|
||||||
{`bool(1)`, true},
|
{`bool(1)`, true},
|
||||||
{`bool(0)`, false},
|
{`bool(0)`, false},
|
||||||
|
|||||||
@@ -14,6 +14,36 @@ func (ao *Array) Type() ObjectType {
|
|||||||
return ARRAY_OBJ
|
return ARRAY_OBJ
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (ao *Array) Bool() bool {
|
||||||
|
return len(ao.Elements) > 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *Array) PopLeft() Object {
|
||||||
|
if len(a.Elements) > 0 {
|
||||||
|
e := a.Elements[0]
|
||||||
|
a.Elements = a.Elements[1:]
|
||||||
|
return e
|
||||||
|
}
|
||||||
|
return &Null{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *Array) PopRight() Object {
|
||||||
|
if len(a.Elements) > 0 {
|
||||||
|
e := a.Elements[(len(a.Elements) - 1)]
|
||||||
|
a.Elements = a.Elements[:(len(a.Elements) - 1)]
|
||||||
|
return e
|
||||||
|
}
|
||||||
|
return &Null{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *Array) Prepend(obj Object) {
|
||||||
|
a.Elements = append([]Object{obj}, a.Elements...)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *Array) Append(obj Object) {
|
||||||
|
a.Elements = append(a.Elements, obj)
|
||||||
|
}
|
||||||
|
|
||||||
func (ao *Array) Inspect() string {
|
func (ao *Array) Inspect() string {
|
||||||
var out bytes.Buffer
|
var out bytes.Buffer
|
||||||
|
|
||||||
@@ -32,9 +62,9 @@ func (ao *Array) String() string {
|
|||||||
return ao.Inspect()
|
return ao.Inspect()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *Array) Less(i, j int) bool {
|
func (ao *Array) Less(i, j int) bool {
|
||||||
if cmp, ok := a.Elements[i].(Comparable); ok {
|
if cmp, ok := ao.Elements[i].(Comparable); ok {
|
||||||
return cmp.Compare(a.Elements[j]) == -1
|
return cmp.Compare(ao.Elements[j]) == -1
|
||||||
}
|
}
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -8,6 +8,10 @@ type Boolean struct {
|
|||||||
Value bool
|
Value bool
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (b *Boolean) Bool() bool {
|
||||||
|
return b.Value
|
||||||
|
}
|
||||||
|
|
||||||
func (b *Boolean) Type() ObjectType {
|
func (b *Boolean) Type() ObjectType {
|
||||||
return BOOLEAN_OBJ
|
return BOOLEAN_OBJ
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -7,12 +7,18 @@ type Builtin struct {
|
|||||||
Fn BuiltinFunction
|
Fn BuiltinFunction
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (b *Builtin) Bool() bool {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
func (b *Builtin) Type() ObjectType {
|
func (b *Builtin) Type() ObjectType {
|
||||||
return BUILTIN_OBJ
|
return BUILTIN_OBJ
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Builtin) Inspect() string {
|
func (b *Builtin) Inspect() string {
|
||||||
return fmt.Sprintf("<built-in function %s>", b.Name)
|
return fmt.Sprintf("<built-in function %s>", b.Name)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Builtin) String() string {
|
func (b *Builtin) String() string {
|
||||||
return b.Inspect()
|
return b.Inspect()
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -11,12 +11,18 @@ type CompiledFunction struct {
|
|||||||
NumParameters int
|
NumParameters int
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (cf *CompiledFunction) Bool() bool {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
func (cf *CompiledFunction) Type() ObjectType {
|
func (cf *CompiledFunction) Type() ObjectType {
|
||||||
return COMPILED_FUNCTION_OBJ
|
return COMPILED_FUNCTION_OBJ
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cf *CompiledFunction) Inspect() string {
|
func (cf *CompiledFunction) Inspect() string {
|
||||||
return fmt.Sprintf("CompiledFunction[%p]", cf)
|
return fmt.Sprintf("CompiledFunction[%p]", cf)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cf *CompiledFunction) String() string {
|
func (cf *CompiledFunction) String() string {
|
||||||
return cf.Inspect()
|
return cf.Inspect()
|
||||||
}
|
}
|
||||||
@@ -26,12 +32,18 @@ type Closure struct {
|
|||||||
Free []Object
|
Free []Object
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (c *Closure) Bool() bool {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
func (c *Closure) Type() ObjectType {
|
func (c *Closure) Type() ObjectType {
|
||||||
return CLOSURE_OBJ
|
return CLOSURE_OBJ
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Closure) Inspect() string {
|
func (c *Closure) Inspect() string {
|
||||||
return fmt.Sprintf("Closure[%p]", c)
|
return fmt.Sprintf("Closure[%p]", c)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Closure) String() string {
|
func (c *Closure) String() string {
|
||||||
return c.Inspect()
|
return c.Inspect()
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -4,15 +4,22 @@ type Error struct {
|
|||||||
Message string
|
Message string
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (e *Error) Bool() bool {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
func (e *Error) Type() ObjectType {
|
func (e *Error) Type() ObjectType {
|
||||||
return ERROR_OBJ
|
return ERROR_OBJ
|
||||||
}
|
}
|
||||||
|
|
||||||
func (e *Error) Inspect() string {
|
func (e *Error) Inspect() string {
|
||||||
return "Error: " + e.Message
|
return "Error: " + e.Message
|
||||||
}
|
}
|
||||||
|
|
||||||
func (e *Error) Clone() Object {
|
func (e *Error) Clone() Object {
|
||||||
return &Error{Message: e.Message}
|
return &Error{Message: e.Message}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (e *Error) String() string {
|
func (e *Error) String() string {
|
||||||
return e.Message
|
return e.Message
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -12,9 +12,14 @@ type Function struct {
|
|||||||
Env *Environment
|
Env *Environment
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (f *Function) Bool() bool {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
func (f *Function) Type() ObjectType {
|
func (f *Function) Type() ObjectType {
|
||||||
return FUNCTION_OBJ
|
return FUNCTION_OBJ
|
||||||
}
|
}
|
||||||
|
|
||||||
func (f *Function) Inspect() string {
|
func (f *Function) Inspect() string {
|
||||||
var out bytes.Buffer
|
var out bytes.Buffer
|
||||||
|
|
||||||
@@ -32,6 +37,7 @@ func (f *Function) Inspect() string {
|
|||||||
|
|
||||||
return out.String()
|
return out.String()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (f *Function) String() string {
|
func (f *Function) String() string {
|
||||||
return f.Inspect()
|
return f.Inspect()
|
||||||
}
|
}
|
||||||
@@ -40,12 +46,18 @@ type ReturnValue struct {
|
|||||||
Value Object
|
Value Object
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (rv *ReturnValue) Bool() bool {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
func (rv *ReturnValue) Type() ObjectType {
|
func (rv *ReturnValue) Type() ObjectType {
|
||||||
return RETURN_VALUE_OBJ
|
return RETURN_VALUE_OBJ
|
||||||
}
|
}
|
||||||
|
|
||||||
func (rv *ReturnValue) Inspect() string {
|
func (rv *ReturnValue) Inspect() string {
|
||||||
return rv.Value.Inspect()
|
return rv.Value.Inspect()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (rv *ReturnValue) String() string {
|
func (rv *ReturnValue) String() string {
|
||||||
return rv.Inspect()
|
return rv.Inspect()
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -44,6 +44,14 @@ type Hash struct {
|
|||||||
Pairs map[HashKey]HashPair
|
Pairs map[HashKey]HashPair
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (h *Hash) Len() int {
|
||||||
|
return len(h.Pairs)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Hash) Bool() bool {
|
||||||
|
return len(h.Pairs) > 0
|
||||||
|
}
|
||||||
|
|
||||||
func (h *Hash) Type() ObjectType {
|
func (h *Hash) Type() ObjectType {
|
||||||
return HASH_OBJ
|
return HASH_OBJ
|
||||||
}
|
}
|
||||||
@@ -62,6 +70,7 @@ func (h *Hash) Inspect() string {
|
|||||||
|
|
||||||
return out.String()
|
return out.String()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *Hash) String() string {
|
func (h *Hash) String() string {
|
||||||
return h.Inspect()
|
return h.Inspect()
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -6,6 +6,10 @@ type Integer struct {
|
|||||||
Value int64
|
Value int64
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (i *Integer) Bool() bool {
|
||||||
|
return i.Value != 0
|
||||||
|
}
|
||||||
|
|
||||||
func (i *Integer) Type() ObjectType {
|
func (i *Integer) Type() ObjectType {
|
||||||
return INTEGER_OBJ
|
return INTEGER_OBJ
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,6 +2,10 @@ package object
|
|||||||
|
|
||||||
type Null struct{}
|
type Null struct{}
|
||||||
|
|
||||||
|
func (n *Null) Bool() bool {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
func (n *Null) Type() ObjectType {
|
func (n *Null) Type() ObjectType {
|
||||||
return NULL_OBJ
|
return NULL_OBJ
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,5 +1,7 @@
|
|||||||
package object
|
package object
|
||||||
|
|
||||||
|
import "fmt"
|
||||||
|
|
||||||
// Type represents the type of an object
|
// Type represents the type of an object
|
||||||
type ObjectType string
|
type ObjectType string
|
||||||
|
|
||||||
@@ -25,6 +27,13 @@ type Comparable interface {
|
|||||||
Compare(other Object) int
|
Compare(other Object) int
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Sizeable is the interface for returning the size of an Object.
|
||||||
|
// Object(s) that have a valid size must implement this interface and the
|
||||||
|
// Len() method.
|
||||||
|
type Sizeable interface {
|
||||||
|
Len() int
|
||||||
|
}
|
||||||
|
|
||||||
// Immutable is the interface for all immutable objects which must implement
|
// Immutable is the interface for all immutable objects which must implement
|
||||||
// the Clone() method used by binding names to values.
|
// the Clone() method used by binding names to values.
|
||||||
type Immutable interface {
|
type Immutable interface {
|
||||||
@@ -34,8 +43,9 @@ type Immutable interface {
|
|||||||
// Object represents a value and implementations are expected to implement
|
// Object represents a value and implementations are expected to implement
|
||||||
// `Type()` and `Inspect()` functions
|
// `Type()` and `Inspect()` functions
|
||||||
type Object interface {
|
type Object interface {
|
||||||
|
fmt.Stringer
|
||||||
Type() ObjectType
|
Type() ObjectType
|
||||||
String() string
|
Bool() bool
|
||||||
Inspect() string
|
Inspect() string
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,11 +1,22 @@
|
|||||||
package object
|
package object
|
||||||
|
|
||||||
import "fmt"
|
import (
|
||||||
|
"fmt"
|
||||||
|
"unicode/utf8"
|
||||||
|
)
|
||||||
|
|
||||||
type String struct {
|
type String struct {
|
||||||
Value string
|
Value string
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (s *String) Len() int {
|
||||||
|
return utf8.RuneCountInString(s.Value)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *String) Bool() bool {
|
||||||
|
return s.Value != ""
|
||||||
|
}
|
||||||
|
|
||||||
func (s *String) Type() ObjectType {
|
func (s *String) Type() ObjectType {
|
||||||
return STRING_OBJ
|
return STRING_OBJ
|
||||||
}
|
}
|
||||||
|
|||||||
56
typing/typing.go
Normal file
56
typing/typing.go
Normal file
@@ -0,0 +1,56 @@
|
|||||||
|
package typing
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"monkey/object"
|
||||||
|
)
|
||||||
|
|
||||||
|
type CheckFunc func(name string, args []object.Object) error
|
||||||
|
|
||||||
|
func Check(name string, args []object.Object, checks ...CheckFunc) error {
|
||||||
|
for _, check := range checks {
|
||||||
|
if err := check(name, args); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func ExactArgs(n int) CheckFunc {
|
||||||
|
return func(name string, args []object.Object) error {
|
||||||
|
if len(args) != n {
|
||||||
|
return fmt.Errorf("TypeError: %s() takes exactly %d argument (%d given)", name, n, len(args))
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func MinimumArgs(n int) CheckFunc {
|
||||||
|
return func(name string, args []object.Object) error {
|
||||||
|
if len(args) < n {
|
||||||
|
return fmt.Errorf("TypeError: %s() takes a minimum %d arguments (%d given)", name, n, len(args))
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func RangeOfArgs(n, m int) CheckFunc {
|
||||||
|
return func(name string, args []object.Object) error {
|
||||||
|
if len(args) < n || len(args) > m {
|
||||||
|
return fmt.Errorf("TypeError: %s() takes at least %d arguments at most %d (%d given)", name, n, m, len(args))
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func WithTypes(types ...object.ObjectType) CheckFunc {
|
||||||
|
return func(name string, args []object.Object) error {
|
||||||
|
for i, t := range types {
|
||||||
|
if i < len(args) && args[i].Type() != t {
|
||||||
|
return fmt.Errorf("TypeError: %s() expected argument #%d to be `%s` got `%s`", name, i+1, t, args[i].Type())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -709,12 +709,12 @@ func TestBuiltinFunctions(t *testing.T) {
|
|||||||
{
|
{
|
||||||
`len(1)`,
|
`len(1)`,
|
||||||
&object.Error{
|
&object.Error{
|
||||||
Message: "argument to `len` not supported, got int",
|
Message: "TypeError: object of type 'int' has no len()",
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
{`len("one", "two")`,
|
{`len("one", "two")`,
|
||||||
&object.Error{
|
&object.Error{
|
||||||
Message: "wrong number of arguments. got=2, want=1",
|
Message: "TypeError: len() takes exactly 1 argument (2 given)",
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
{`len([1, 2, 3])`, 3},
|
{`len([1, 2, 3])`, 3},
|
||||||
@@ -724,27 +724,27 @@ func TestBuiltinFunctions(t *testing.T) {
|
|||||||
{`first([])`, Null},
|
{`first([])`, Null},
|
||||||
{`first(1)`,
|
{`first(1)`,
|
||||||
&object.Error{
|
&object.Error{
|
||||||
Message: "argument to `first` must be array, got int",
|
Message: "TypeError: first() expected argument #1 to be `array` got `int`",
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
{`last([1, 2, 3])`, 3},
|
{`last([1, 2, 3])`, 3},
|
||||||
{`last([])`, Null},
|
{`last([])`, Null},
|
||||||
{`last(1)`,
|
{`last(1)`,
|
||||||
&object.Error{
|
&object.Error{
|
||||||
Message: "argument to `last` must be array, got int",
|
Message: "TypeError: last() expected argument #1 to be `array` got `int`",
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
{`rest([1, 2, 3])`, []int{2, 3}},
|
{`rest([1, 2, 3])`, []int{2, 3}},
|
||||||
{`rest([])`, Null},
|
{`rest([])`, []int{}},
|
||||||
{`push([], 1)`, []int{1}},
|
{`push([], 1)`, []int{1}},
|
||||||
{`push(1, 1)`,
|
{`push(1, 1)`,
|
||||||
&object.Error{
|
&object.Error{
|
||||||
Message: "argument to `push` must be array, got int",
|
Message: "TypeError: push() expected argument #1 to be `array` got `int`",
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
{`input()`, ""},
|
{`input()`, ""},
|
||||||
{`pop([])`, &object.Error{
|
{`pop([])`, &object.Error{
|
||||||
Message: "cannot pop from an empty array",
|
Message: "IndexError: pop from an empty array",
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
{`pop([1])`, 1},
|
{`pop([1])`, 1},
|
||||||
|
|||||||
Reference in New Issue
Block a user