type checking and error handling for builtins improved.
Some checks failed
Build / build (push) Successful in 11m16s
Test / build (push) Failing after 17m0s

This commit is contained in:
Chuck Smith
2024-03-25 16:18:08 -04:00
parent 1c99d2198b
commit 6d234099d1
52 changed files with 650 additions and 433 deletions

View File

@@ -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

View File

@@ -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())
} }

View File

@@ -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}

View File

@@ -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 {

View File

@@ -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())
} }

View File

@@ -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())
}
} }

View File

@@ -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())
} }

View File

@@ -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())
}
} }

View File

@@ -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)
} }

View File

@@ -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)}
} }

View File

@@ -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(),
)
} }

View File

@@ -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)

View File

@@ -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")
} }

View File

@@ -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())
} }

View File

@@ -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
} }

View File

@@ -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)

View File

@@ -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{}
} }
} }

View File

@@ -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])
}
} }

View File

@@ -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)

View File

@@ -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())
} }

View File

@@ -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])
} }

View File

@@ -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 {
@@ -24,6 +28,4 @@ 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())
} }

View File

@@ -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 {
@@ -24,6 +28,4 @@ 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())
} }

View File

@@ -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())
} }

View File

@@ -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(),
)
} }

View File

@@ -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]

View File

@@ -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())
}
} }

View File

@@ -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
} }

View File

@@ -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}
} }

View File

@@ -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)}

View File

@@ -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
} }

View File

@@ -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())
} }

View File

@@ -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())
} }

View File

@@ -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])
}
} }

View File

@@ -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()}
} }

View File

@@ -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())}

View File

@@ -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])
} }

View File

@@ -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{}

View File

@@ -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},

View File

@@ -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
} }

View File

@@ -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
} }

View File

@@ -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()
} }

View File

@@ -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()
} }

View File

@@ -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
} }

View File

@@ -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()
} }

View File

@@ -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()
} }

View File

@@ -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
} }

View File

@@ -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
} }

View File

@@ -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
} }

View File

@@ -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
View 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
}
}

View File

@@ -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},