rearrange builtins
This commit is contained in:
5
.gitignore
vendored
5
.gitignore
vendored
@@ -78,8 +78,13 @@ fabric.properties
|
||||
.idea/caches/build_file_checksums.ser
|
||||
|
||||
*~
|
||||
*-e
|
||||
*.so
|
||||
*.bak
|
||||
*.txt
|
||||
*.swo
|
||||
*.swn
|
||||
.DS_Store
|
||||
|
||||
/dist
|
||||
/monkey-lang
|
||||
12
builtins/args.go
Normal file
12
builtins/args.go
Normal file
@@ -0,0 +1,12 @@
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
// Args ...
|
||||
func Args(args ...object.Object) object.Object {
|
||||
elements := make([]object.Object, len(object.Arguments))
|
||||
for i, arg := range object.Arguments {
|
||||
elements[i] = &object.String{Value: arg}
|
||||
}
|
||||
return &object.Array{Elements: elements}
|
||||
}
|
||||
@@ -1,4 +1,6 @@
|
||||
package object
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
@@ -6,22 +8,22 @@ import (
|
||||
)
|
||||
|
||||
// Assert ...
|
||||
func Assert(args ...Object) Object {
|
||||
func Assert(args ...object.Object) object.Object {
|
||||
if len(args) != 2 {
|
||||
return newError("wrong number of arguments. got=%d, want=2",
|
||||
len(args))
|
||||
}
|
||||
if args[0].Type() != BOOLEAN_OBJ {
|
||||
if args[0].Type() != object.BOOLEAN_OBJ {
|
||||
return newError("argument #1 to `assert` must be BOOLEAN, got %s",
|
||||
args[0].Type())
|
||||
}
|
||||
if args[1].Type() != STRING_OBJ {
|
||||
if args[1].Type() != object.STRING_OBJ {
|
||||
return newError("argument #2 to `assert` must be STRING, got %s",
|
||||
args[0].Type())
|
||||
}
|
||||
|
||||
if !args[0].(*Boolean).Value {
|
||||
fmt.Printf("Assertion Error: %s", args[1].(*String).Value)
|
||||
if !args[0].(*object.Boolean).Value {
|
||||
fmt.Printf("Assertion Error: %s", args[1].(*object.String).Value)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
40
builtins/bool.go
Normal file
40
builtins/bool.go
Normal file
@@ -0,0 +1,40 @@
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
// Bool ...
|
||||
func Bool(args ...object.Object) object.Object {
|
||||
if len(args) != 1 {
|
||||
return newError("wrong number of arguments. got=%d, want=1", len(args))
|
||||
}
|
||||
|
||||
switch arg := args[0].(type) {
|
||||
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())
|
||||
}
|
||||
}
|
||||
@@ -1,12 +1,13 @@
|
||||
package object
|
||||
package builtins
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"monkey/object"
|
||||
"sort"
|
||||
)
|
||||
|
||||
// Builtins ...
|
||||
var Builtins = map[string]*Builtin{
|
||||
var Builtins = map[string]*object.Builtin{
|
||||
"len": {Name: "len", Fn: Len},
|
||||
"input": {Name: "input", Fn: Input},
|
||||
"print": {Name: "print", Fn: Print},
|
||||
@@ -20,7 +21,7 @@ var Builtins = map[string]*Builtin{
|
||||
"bool": {Name: "bool", Fn: Bool},
|
||||
"int": {Name: "int", Fn: Int},
|
||||
"str": {Name: "str", Fn: Str},
|
||||
"typeof": {Name: "typeof", Fn: TypeOf},
|
||||
"type": {Name: "type", Fn: TypeOf},
|
||||
"args": {Name: "args", Fn: Args},
|
||||
"lower": {Name: "lower", Fn: Lower},
|
||||
"upper": {Name: "upper", Fn: Upper},
|
||||
@@ -29,10 +30,11 @@ var Builtins = map[string]*Builtin{
|
||||
"find": {Name: "find", Fn: Find},
|
||||
"read": {Name: "read", Fn: Read},
|
||||
"write": {Name: "write", Fn: Write},
|
||||
"ffi": {Name: "ffi", Fn: FFI},
|
||||
}
|
||||
|
||||
// BuiltinsIndex ...
|
||||
var BuiltinsIndex []*Builtin
|
||||
var BuiltinsIndex []*object.Builtin
|
||||
|
||||
func init() {
|
||||
var keys []string
|
||||
@@ -46,6 +48,6 @@ func init() {
|
||||
}
|
||||
}
|
||||
|
||||
func newError(format string, a ...interface{}) *Error {
|
||||
return &Error{Message: fmt.Sprintf(format, a...)}
|
||||
func newError(format string, a ...interface{}) *object.Error {
|
||||
return &object.Error{Message: fmt.Sprintf(format, a...)}
|
||||
}
|
||||
19
builtins/exit.go
Normal file
19
builtins/exit.go
Normal file
@@ -0,0 +1,19 @@
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
// Exit ...
|
||||
func Exit(args ...object.Object) object.Object {
|
||||
var status int
|
||||
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)
|
||||
}
|
||||
|
||||
object.ExitFunction(status)
|
||||
|
||||
return nil
|
||||
}
|
||||
40
builtins/ffi.go
Normal file
40
builtins/ffi.go
Normal file
@@ -0,0 +1,40 @@
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"plugin"
|
||||
)
|
||||
|
||||
// FFI ...
|
||||
func FFI(args ...object.Object) object.Object {
|
||||
if len(args) != 2 {
|
||||
return newError("wrong number of arguments. got=%d, want=2",
|
||||
len(args))
|
||||
}
|
||||
|
||||
arg, ok := args[0].(*object.String)
|
||||
if !ok {
|
||||
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))
|
||||
if err != nil {
|
||||
return newError("error loading plugin: %s", err)
|
||||
}
|
||||
|
||||
v, err := p.Lookup(symbol)
|
||||
if err != nil {
|
||||
return newError("error finding symbol: %s", err)
|
||||
}
|
||||
|
||||
return &object.Builtin{Name: symbol, Fn: v.(func(...object.Object) object.Object)}
|
||||
}
|
||||
@@ -1,33 +1,35 @@
|
||||
package object
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
import (
|
||||
"strings"
|
||||
)
|
||||
|
||||
// Find ...
|
||||
func Find(args ...Object) Object {
|
||||
func Find(args ...object.Object) object.Object {
|
||||
if len(args) != 2 {
|
||||
return newError("wrong number of arguments. got=%d, want=2",
|
||||
len(args))
|
||||
}
|
||||
|
||||
if haystack, ok := args[0].(*String); ok {
|
||||
if needle, ok := args[1].(*String); ok {
|
||||
if haystack, ok := args[0].(*object.String); ok {
|
||||
if needle, ok := args[1].(*object.String); ok {
|
||||
index := strings.Index(haystack.Value, needle.Value)
|
||||
return &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].(*Array); ok {
|
||||
} else if haystack, ok := args[0].(*object.Array); ok {
|
||||
needle := args[1]
|
||||
index := -1
|
||||
for i, el := range haystack.Elements {
|
||||
if cmp, ok := el.(Comparable); ok && cmp.Equal(needle) {
|
||||
if cmp, ok := el.(object.Comparable); ok && cmp.Equal(needle) {
|
||||
index = i
|
||||
break
|
||||
}
|
||||
}
|
||||
return &Integer{Value: int64(index)}
|
||||
return &object.Integer{Value: int64(index)}
|
||||
} else {
|
||||
return newError("expected arg #1 to be `str` or `array` got got=%T", args[0])
|
||||
}
|
||||
@@ -1,17 +1,19 @@
|
||||
package object
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
// First ...
|
||||
func First(args ...Object) Object {
|
||||
func First(args ...object.Object) object.Object {
|
||||
if len(args) != 1 {
|
||||
return newError("wrong number of arguments. got=%d, want=1",
|
||||
len(args))
|
||||
}
|
||||
if args[0].Type() != ARRAY_OBJ {
|
||||
if args[0].Type() != object.ARRAY_OBJ {
|
||||
return newError("argument to `first` must be array, got %s",
|
||||
args[0].Type())
|
||||
}
|
||||
|
||||
arr := args[0].(*Array)
|
||||
arr := args[0].(*object.Array)
|
||||
if len(arr.Elements) > 0 {
|
||||
return arr.Elements[0]
|
||||
}
|
||||
@@ -1,4 +1,6 @@
|
||||
package object
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
@@ -8,9 +10,9 @@ import (
|
||||
)
|
||||
|
||||
// Input ...
|
||||
func Input(args ...Object) Object {
|
||||
func Input(args ...object.Object) object.Object {
|
||||
if len(args) > 0 {
|
||||
obj, ok := args[0].(*String)
|
||||
obj, ok := args[0].(*object.String)
|
||||
if !ok {
|
||||
return newError(
|
||||
"argument to `input` not supported, got %s",
|
||||
@@ -26,5 +28,5 @@ func Input(args ...Object) Object {
|
||||
if err != nil && err != io.EOF {
|
||||
return newError(fmt.Sprintf("error reading input from stdin: %s", err))
|
||||
}
|
||||
return &String{Value: string(line)}
|
||||
return &object.String{Value: string(line)}
|
||||
}
|
||||
@@ -1,28 +1,30 @@
|
||||
package object
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
import "strconv"
|
||||
|
||||
// Int ...
|
||||
func Int(args ...Object) Object {
|
||||
func Int(args ...object.Object) object.Object {
|
||||
if len(args) != 1 {
|
||||
return newError("wrong number of arguments. got=%d, want=1", len(args))
|
||||
}
|
||||
|
||||
switch arg := args[0].(type) {
|
||||
case *Boolean:
|
||||
case *object.Boolean:
|
||||
if arg.Value {
|
||||
return &Integer{Value: 1}
|
||||
return &object.Integer{Value: 1}
|
||||
}
|
||||
return &Integer{Value: 0}
|
||||
case *Integer:
|
||||
return &object.Integer{Value: 0}
|
||||
case *object.Integer:
|
||||
return arg
|
||||
case *String:
|
||||
case *object.String:
|
||||
n, err := strconv.ParseInt(arg.Value, 10, 64)
|
||||
if err != nil {
|
||||
return newError("could not parse string to int: %s", err)
|
||||
}
|
||||
|
||||
return &Integer{Value: n}
|
||||
return &object.Integer{Value: n}
|
||||
default:
|
||||
return newError("argument to `int` not supported, got=%s", args[0].Type())
|
||||
}
|
||||
@@ -1,23 +1,25 @@
|
||||
package object
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
import (
|
||||
"strings"
|
||||
)
|
||||
|
||||
// Join ...
|
||||
func Join(args ...Object) Object {
|
||||
func Join(args ...object.Object) object.Object {
|
||||
if len(args) != 2 {
|
||||
return newError("wrong number of arguments. got=%d, want=1",
|
||||
len(args))
|
||||
}
|
||||
|
||||
if arr, ok := args[0].(*Array); ok {
|
||||
if sep, ok := args[1].(*String); ok {
|
||||
if arr, ok := args[0].(*object.Array); ok {
|
||||
if sep, ok := args[1].(*object.String); ok {
|
||||
a := make([]string, len(arr.Elements))
|
||||
for i, el := range arr.Elements {
|
||||
a[i] = el.String()
|
||||
}
|
||||
return &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])
|
||||
}
|
||||
@@ -1,17 +1,19 @@
|
||||
package object
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
// Last ...
|
||||
func Last(args ...Object) Object {
|
||||
func Last(args ...object.Object) object.Object {
|
||||
if len(args) != 1 {
|
||||
return newError("wrong number of arguments. got=%d, want=1",
|
||||
len(args))
|
||||
}
|
||||
if args[0].Type() != ARRAY_OBJ {
|
||||
if args[0].Type() != object.ARRAY_OBJ {
|
||||
return newError("argument to `last` must be array, got %s",
|
||||
args[0].Type())
|
||||
}
|
||||
|
||||
arr := args[0].(*Array)
|
||||
arr := args[0].(*object.Array)
|
||||
length := len(arr.Elements)
|
||||
if length > 0 {
|
||||
return arr.Elements[length-1]
|
||||
@@ -1,19 +1,21 @@
|
||||
package object
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
import "unicode/utf8"
|
||||
|
||||
// Len ...
|
||||
func Len(args ...Object) Object {
|
||||
func Len(args ...object.Object) object.Object {
|
||||
if len(args) != 1 {
|
||||
return newError("wrong number of arguments. got=%d, want=1",
|
||||
len(args))
|
||||
}
|
||||
|
||||
switch arg := args[0].(type) {
|
||||
case *Array:
|
||||
return &Integer{Value: int64(len(arg.Elements))}
|
||||
case *String:
|
||||
return &Integer{Value: int64(utf8.RuneCountInString(arg.Value))}
|
||||
case *object.Array:
|
||||
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())
|
||||
@@ -1,16 +1,18 @@
|
||||
package object
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
import "strings"
|
||||
|
||||
// Lower ...
|
||||
func Lower(args ...Object) Object {
|
||||
func Lower(args ...object.Object) object.Object {
|
||||
if len(args) != 1 {
|
||||
return newError("wrong number of arguments. got=%d, want=1",
|
||||
len(args))
|
||||
}
|
||||
|
||||
if str, ok := args[0].(*String); ok {
|
||||
return &String{Value: strings.ToLower(str.Value)}
|
||||
if str, ok := args[0].(*object.String); ok {
|
||||
return &object.String{Value: strings.ToLower(str.Value)}
|
||||
}
|
||||
return newError("expected `str` argument to `lower` got=%T", args[0])
|
||||
}
|
||||
@@ -1,17 +1,19 @@
|
||||
package object
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
// Pop ...
|
||||
func Pop(args ...Object) Object {
|
||||
func Pop(args ...object.Object) object.Object {
|
||||
if len(args) != 1 {
|
||||
return newError("wrong number of arguments. got=%d, want=1",
|
||||
len(args))
|
||||
}
|
||||
if args[0].Type() != ARRAY_OBJ {
|
||||
if args[0].Type() != object.ARRAY_OBJ {
|
||||
return newError("argument to `pop` must be array, got %s",
|
||||
args[0].Type())
|
||||
}
|
||||
|
||||
arr := args[0].(*Array)
|
||||
arr := args[0].(*object.Array)
|
||||
length := len(arr.Elements)
|
||||
|
||||
if length == 0 {
|
||||
@@ -1,9 +1,11 @@
|
||||
package object
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
import "fmt"
|
||||
|
||||
// Print ...
|
||||
func Print(args ...Object) Object {
|
||||
func Print(args ...object.Object) object.Object {
|
||||
for _, arg := range args {
|
||||
fmt.Println(arg.String())
|
||||
}
|
||||
@@ -1,26 +1,28 @@
|
||||
package object
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
// Push ...
|
||||
func Push(args ...Object) Object {
|
||||
func Push(args ...object.Object) object.Object {
|
||||
if len(args) != 2 {
|
||||
return newError("wrong number of arguments. got=%d, want=2",
|
||||
len(args))
|
||||
}
|
||||
if args[0].Type() != ARRAY_OBJ {
|
||||
if args[0].Type() != object.ARRAY_OBJ {
|
||||
return newError("argument to `push` must be array, got %s",
|
||||
args[0].Type())
|
||||
}
|
||||
|
||||
arr := args[0].(*Array)
|
||||
arr := args[0].(*object.Array)
|
||||
length := len(arr.Elements)
|
||||
|
||||
newElements := make([]Object, length+1)
|
||||
newElements := make([]object.Object, length+1)
|
||||
copy(newElements, arr.Elements)
|
||||
if immutable, ok := args[1].(Immutable); ok {
|
||||
if immutable, ok := args[1].(object.Immutable); ok {
|
||||
newElements[length] = immutable.Clone()
|
||||
} else {
|
||||
newElements[length] = args[1]
|
||||
}
|
||||
|
||||
return &Array{Elements: newElements}
|
||||
return &object.Array{Elements: newElements}
|
||||
}
|
||||
@@ -1,17 +1,19 @@
|
||||
package object
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
import (
|
||||
"os"
|
||||
)
|
||||
|
||||
// Read ...
|
||||
func Read(args ...Object) Object {
|
||||
func Read(args ...object.Object) object.Object {
|
||||
if len(args) != 1 {
|
||||
return newError("wrong number of arguments. got=%d, want=1",
|
||||
len(args))
|
||||
}
|
||||
|
||||
arg, ok := args[0].(*String)
|
||||
arg, ok := args[0].(*object.String)
|
||||
if !ok {
|
||||
return newError("argument to `read` expected to be `str` got=%T", args[0].Type())
|
||||
}
|
||||
@@ -22,5 +24,5 @@ func Read(args ...Object) Object {
|
||||
return newError("error reading file: %s", err)
|
||||
}
|
||||
|
||||
return &String{Value: string(data)}
|
||||
return &object.String{Value: string(data)}
|
||||
}
|
||||
@@ -1,22 +1,24 @@
|
||||
package object
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
// Rest ...
|
||||
func Rest(args ...Object) Object {
|
||||
func Rest(args ...object.Object) object.Object {
|
||||
if len(args) != 1 {
|
||||
return newError("wrong number of arguments. got=%d, want=1",
|
||||
len(args))
|
||||
}
|
||||
if args[0].Type() != ARRAY_OBJ {
|
||||
if args[0].Type() != object.ARRAY_OBJ {
|
||||
return newError("argument to `rest` must be array, got %s",
|
||||
args[0].Type())
|
||||
}
|
||||
|
||||
arr := args[0].(*Array)
|
||||
arr := args[0].(*object.Array)
|
||||
length := len(arr.Elements)
|
||||
if length > 0 {
|
||||
newElements := make([]Object, length-1, length-1)
|
||||
newElements := make([]object.Object, length-1, length-1)
|
||||
copy(newElements, arr.Elements[1:length])
|
||||
return &Array{Elements: newElements}
|
||||
return &object.Array{Elements: newElements}
|
||||
}
|
||||
|
||||
return nil
|
||||
@@ -1,23 +1,25 @@
|
||||
package object
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
import (
|
||||
"strings"
|
||||
)
|
||||
|
||||
// Split ...
|
||||
func Split(args ...Object) Object {
|
||||
func Split(args ...object.Object) object.Object {
|
||||
if len(args) < 1 {
|
||||
return newError("wrong number of arguments. got=%d, want=1",
|
||||
len(args))
|
||||
}
|
||||
|
||||
if obj, ok := args[0].(*String); ok {
|
||||
if obj, ok := args[0].(*object.String); ok {
|
||||
var sep string
|
||||
|
||||
s := obj.Value
|
||||
|
||||
if len(args) == 2 {
|
||||
if obj, ok := args[1].(*String); ok {
|
||||
if obj, ok := args[1].(*object.String); ok {
|
||||
sep = obj.Value
|
||||
} else {
|
||||
return newError("expected arg #2 to be `str` got=%T", args[1])
|
||||
@@ -25,11 +27,11 @@ func Split(args ...Object) Object {
|
||||
}
|
||||
|
||||
tokens := strings.Split(s, sep)
|
||||
elements := make([]Object, len(tokens))
|
||||
elements := make([]object.Object, len(tokens))
|
||||
for i, token := range tokens {
|
||||
elements[i] = &String{Value: token}
|
||||
elements[i] = &object.String{Value: token}
|
||||
}
|
||||
return &Array{Elements: elements}
|
||||
return &object.Array{Elements: elements}
|
||||
} else {
|
||||
return newError("expected arg #1 to be `str` got got=%T", args[0])
|
||||
}
|
||||
@@ -1,11 +1,13 @@
|
||||
package object
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
)
|
||||
|
||||
// Str ...
|
||||
func Str(args ...Object) Object {
|
||||
func Str(args ...object.Object) object.Object {
|
||||
if len(args) != 1 {
|
||||
return newError("wrong number of arguments. got=%d, want=1", len(args))
|
||||
}
|
||||
@@ -15,5 +17,5 @@ func Str(args ...Object) Object {
|
||||
return newError("argument to `str` not supported, got %s", args[0].Type())
|
||||
}
|
||||
|
||||
return &String{Value: arg.String()}
|
||||
return &object.String{Value: arg.String()}
|
||||
}
|
||||
12
builtins/typeof.go
Normal file
12
builtins/typeof.go
Normal file
@@ -0,0 +1,12 @@
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
// TypeOf ...
|
||||
func TypeOf(args ...object.Object) object.Object {
|
||||
if len(args) != 1 {
|
||||
return newError("wrong number of arguments. got=%d, want=1", len(args))
|
||||
}
|
||||
|
||||
return &object.String{Value: string(args[0].Type())}
|
||||
}
|
||||
@@ -1,16 +1,18 @@
|
||||
package object
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
import "strings"
|
||||
|
||||
// Upper ...
|
||||
func Upper(args ...Object) Object {
|
||||
func Upper(args ...object.Object) object.Object {
|
||||
if len(args) != 1 {
|
||||
return newError("wrong number of arguments. got=%d, want=1",
|
||||
len(args))
|
||||
}
|
||||
|
||||
if str, ok := args[0].(*String); ok {
|
||||
return &String{Value: strings.ToUpper(str.Value)}
|
||||
if str, ok := args[0].(*object.String); ok {
|
||||
return &object.String{Value: strings.ToUpper(str.Value)}
|
||||
}
|
||||
return newError("expected `str` argument to `upper` got=%T", args[0])
|
||||
}
|
||||
@@ -1,23 +1,25 @@
|
||||
package object
|
||||
package builtins
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
import (
|
||||
"os"
|
||||
)
|
||||
|
||||
// Write ...
|
||||
func Write(args ...Object) Object {
|
||||
func Write(args ...object.Object) object.Object {
|
||||
if len(args) != 2 {
|
||||
return newError("wrong number of arguments. got=%d, want=2",
|
||||
len(args))
|
||||
}
|
||||
|
||||
arg, ok := args[0].(*String)
|
||||
arg, ok := args[0].(*object.String)
|
||||
if !ok {
|
||||
return newError("argument #1 to `write` expected to be `str` got=%T", args[0].Type())
|
||||
}
|
||||
filename := arg.Value
|
||||
|
||||
arg, ok = args[1].(*String)
|
||||
arg, ok = args[1].(*object.String)
|
||||
if !ok {
|
||||
return newError("argument #2 to `write` expected to be `str` got=%T", args[1].Type())
|
||||
}
|
||||
@@ -28,5 +30,5 @@ func Write(args ...Object) Object {
|
||||
return newError("error writing file: %s", err)
|
||||
}
|
||||
|
||||
return &Null{}
|
||||
return &object.Null{}
|
||||
}
|
||||
@@ -4,6 +4,7 @@ import (
|
||||
"fmt"
|
||||
"log"
|
||||
"monkey/ast"
|
||||
"monkey/builtins"
|
||||
"monkey/code"
|
||||
"monkey/object"
|
||||
"sort"
|
||||
@@ -42,7 +43,7 @@ func New() *Compiler {
|
||||
|
||||
symbolTable := NewSymbolTable()
|
||||
|
||||
for i, builtin := range object.BuiltinsIndex {
|
||||
for i, builtin := range builtins.BuiltinsIndex {
|
||||
symbolTable.DefineBuiltin(i, builtin.Name)
|
||||
}
|
||||
|
||||
|
||||
@@ -879,11 +879,11 @@ func TestBuiltins(t *testing.T) {
|
||||
`,
|
||||
expectedConstants: []interface{}{1},
|
||||
expectedInstructions: []code.Instructions{
|
||||
code.Make(code.OpGetBuiltin, 10),
|
||||
code.Make(code.OpGetBuiltin, 11),
|
||||
code.Make(code.OpArray, 0),
|
||||
code.Make(code.OpCall, 1),
|
||||
code.Make(code.OpPop),
|
||||
code.Make(code.OpGetBuiltin, 14),
|
||||
code.Make(code.OpGetBuiltin, 15),
|
||||
code.Make(code.OpArray, 0),
|
||||
code.Make(code.OpConstant, 0),
|
||||
code.Make(code.OpCall, 2),
|
||||
@@ -894,7 +894,7 @@ func TestBuiltins(t *testing.T) {
|
||||
input: `fn() { return len([]) }`,
|
||||
expectedConstants: []interface{}{
|
||||
[]code.Instructions{
|
||||
code.Make(code.OpGetBuiltin, 10),
|
||||
code.Make(code.OpGetBuiltin, 11),
|
||||
code.Make(code.OpArray, 0),
|
||||
code.Make(code.OpCall, 1),
|
||||
code.Make(code.OpReturn),
|
||||
|
||||
@@ -1,7 +0,0 @@
|
||||
package evaluator
|
||||
|
||||
import (
|
||||
"monkey/object"
|
||||
)
|
||||
|
||||
var builtins = object.Builtins
|
||||
@@ -3,6 +3,7 @@ package evaluator
|
||||
import (
|
||||
"fmt"
|
||||
"monkey/ast"
|
||||
"monkey/builtins"
|
||||
"monkey/object"
|
||||
"strings"
|
||||
)
|
||||
@@ -487,7 +488,7 @@ func evalIdentifier(node *ast.Identifier, env *object.Environment) object.Object
|
||||
return val
|
||||
}
|
||||
|
||||
if builtin, ok := builtins[node.Value]; ok {
|
||||
if builtin, ok := builtins.Builtins[node.Value]; ok {
|
||||
return builtin
|
||||
}
|
||||
|
||||
|
||||
51
object/array.go
Normal file
51
object/array.go
Normal file
@@ -0,0 +1,51 @@
|
||||
package object
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// Array is the array literal type that holds a slice of Object(s)
|
||||
type Array struct {
|
||||
Elements []Object
|
||||
}
|
||||
|
||||
func (ao *Array) Type() ObjectType {
|
||||
return ARRAY_OBJ
|
||||
}
|
||||
func (ao *Array) Inspect() string {
|
||||
var out bytes.Buffer
|
||||
|
||||
var elements []string
|
||||
for _, e := range ao.Elements {
|
||||
elements = append(elements, e.Inspect())
|
||||
}
|
||||
|
||||
out.WriteString("[")
|
||||
out.WriteString(strings.Join(elements, ", "))
|
||||
out.WriteString("]")
|
||||
|
||||
return out.String()
|
||||
}
|
||||
func (ao *Array) String() string {
|
||||
return ao.Inspect()
|
||||
}
|
||||
func (ao *Array) Equal(other Object) bool {
|
||||
if obj, ok := other.(*Array); ok {
|
||||
if len(ao.Elements) != len(obj.Elements) {
|
||||
return false
|
||||
}
|
||||
for i, el := range ao.Elements {
|
||||
cmp, ok := el.(Comparable)
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
if !cmp.Equal(obj.Elements[i]) {
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
28
object/bool.go
Normal file
28
object/bool.go
Normal file
@@ -0,0 +1,28 @@
|
||||
package object
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
)
|
||||
|
||||
type Boolean struct {
|
||||
Value bool
|
||||
}
|
||||
|
||||
func (b *Boolean) Type() ObjectType {
|
||||
return BOOLEAN_OBJ
|
||||
}
|
||||
func (b *Boolean) Inspect() string {
|
||||
return fmt.Sprintf("%t", b.Value)
|
||||
}
|
||||
func (b *Boolean) Clone() Object {
|
||||
return &Boolean{Value: b.Value}
|
||||
}
|
||||
func (b *Boolean) String() string {
|
||||
return b.Inspect()
|
||||
}
|
||||
func (b *Boolean) Equal(other Object) bool {
|
||||
if obj, ok := other.(*Boolean); ok {
|
||||
return b.Value == obj.Value
|
||||
}
|
||||
return false
|
||||
}
|
||||
18
object/builtin.go
Normal file
18
object/builtin.go
Normal file
@@ -0,0 +1,18 @@
|
||||
package object
|
||||
|
||||
import "fmt"
|
||||
|
||||
type Builtin struct {
|
||||
Name string
|
||||
Fn BuiltinFunction
|
||||
}
|
||||
|
||||
func (b *Builtin) Type() ObjectType {
|
||||
return BUILTIN_OBJ
|
||||
}
|
||||
func (b *Builtin) Inspect() string {
|
||||
return fmt.Sprintf("<built-in function %s>", b.Name)
|
||||
}
|
||||
func (b *Builtin) String() string {
|
||||
return b.Inspect()
|
||||
}
|
||||
@@ -1,10 +0,0 @@
|
||||
package object
|
||||
|
||||
// Args ...
|
||||
func Args(args ...Object) Object {
|
||||
elements := make([]Object, len(Arguments))
|
||||
for i, arg := range Arguments {
|
||||
elements[i] = &String{Value: arg}
|
||||
}
|
||||
return &Array{Elements: elements}
|
||||
}
|
||||
@@ -1,38 +0,0 @@
|
||||
package object
|
||||
|
||||
// Bool ...
|
||||
func Bool(args ...Object) Object {
|
||||
if len(args) != 1 {
|
||||
return newError("wrong number of arguments. got=%d, want=1", len(args))
|
||||
}
|
||||
|
||||
switch arg := args[0].(type) {
|
||||
case *Null:
|
||||
return &Boolean{Value: false}
|
||||
case *Boolean:
|
||||
return arg
|
||||
case *Integer:
|
||||
if arg.Value == 0 {
|
||||
return &Boolean{Value: false}
|
||||
}
|
||||
return &Boolean{Value: true}
|
||||
case *String:
|
||||
if len(arg.Value) > 0 {
|
||||
return &Boolean{Value: true}
|
||||
}
|
||||
return &Boolean{Value: false}
|
||||
case *Array:
|
||||
if len(arg.Elements) > 0 {
|
||||
return &Boolean{Value: true}
|
||||
}
|
||||
return &Boolean{Value: false}
|
||||
case *Hash:
|
||||
if len(arg.Pairs) > 0 {
|
||||
return &Boolean{Value: true}
|
||||
}
|
||||
return &Boolean{Value: false}
|
||||
|
||||
default:
|
||||
return newError("argument to `bool` not supported, got=%s", args[0].Type())
|
||||
}
|
||||
}
|
||||
@@ -1,17 +0,0 @@
|
||||
package object
|
||||
|
||||
// Exit ...
|
||||
func Exit(args ...Object) Object {
|
||||
var status int
|
||||
if len(args) == 1 {
|
||||
if args[0].Type() != INTEGER_OBJ {
|
||||
return newError("argument to `exit` must be INTEGER, got %s",
|
||||
args[0].Type())
|
||||
}
|
||||
status = int(args[0].(*Integer).Value)
|
||||
}
|
||||
|
||||
ExitFunction(status)
|
||||
|
||||
return nil
|
||||
}
|
||||
@@ -1,10 +0,0 @@
|
||||
package object
|
||||
|
||||
// TypeOf ...
|
||||
func TypeOf(args ...Object) Object {
|
||||
if len(args) != 1 {
|
||||
return newError("wrong number of arguments. got=%d, want=1", len(args))
|
||||
}
|
||||
|
||||
return &String{Value: string(args[0].Type())}
|
||||
}
|
||||
37
object/closure.go
Normal file
37
object/closure.go
Normal file
@@ -0,0 +1,37 @@
|
||||
package object
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"monkey/code"
|
||||
)
|
||||
|
||||
type CompiledFunction struct {
|
||||
Instructions code.Instructions
|
||||
NumLocals int
|
||||
NumParameters int
|
||||
}
|
||||
|
||||
func (cf *CompiledFunction) Type() ObjectType {
|
||||
return COMPILED_FUNCTION_OBJ
|
||||
}
|
||||
func (cf *CompiledFunction) Inspect() string {
|
||||
return fmt.Sprintf("CompiledFunction[%p]", cf)
|
||||
}
|
||||
func (cf *CompiledFunction) String() string {
|
||||
return cf.Inspect()
|
||||
}
|
||||
|
||||
type Closure struct {
|
||||
Fn *CompiledFunction
|
||||
Free []Object
|
||||
}
|
||||
|
||||
func (c *Closure) Type() ObjectType {
|
||||
return CLOSURE_OBJ
|
||||
}
|
||||
func (c *Closure) Inspect() string {
|
||||
return fmt.Sprintf("Closure[%p]", c)
|
||||
}
|
||||
func (c *Closure) String() string {
|
||||
return c.Inspect()
|
||||
}
|
||||
18
object/error.go
Normal file
18
object/error.go
Normal file
@@ -0,0 +1,18 @@
|
||||
package object
|
||||
|
||||
type Error struct {
|
||||
Message string
|
||||
}
|
||||
|
||||
func (e *Error) Type() ObjectType {
|
||||
return ERROR_OBJ
|
||||
}
|
||||
func (e *Error) Inspect() string {
|
||||
return "Error: " + e.Message
|
||||
}
|
||||
func (e *Error) Clone() Object {
|
||||
return &Error{Message: e.Message}
|
||||
}
|
||||
func (e *Error) String() string {
|
||||
return e.Message
|
||||
}
|
||||
51
object/function.go
Normal file
51
object/function.go
Normal file
@@ -0,0 +1,51 @@
|
||||
package object
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"monkey/ast"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type Function struct {
|
||||
Parameters []*ast.Identifier
|
||||
Body *ast.BlockStatement
|
||||
Env *Environment
|
||||
}
|
||||
|
||||
func (f *Function) Type() ObjectType {
|
||||
return FUNCTION_OBJ
|
||||
}
|
||||
func (f *Function) Inspect() string {
|
||||
var out bytes.Buffer
|
||||
|
||||
params := []string{}
|
||||
for _, p := range f.Parameters {
|
||||
params = append(params, p.String())
|
||||
}
|
||||
|
||||
out.WriteString("fn")
|
||||
out.WriteString("(")
|
||||
out.WriteString(strings.Join(params, ", "))
|
||||
out.WriteString(") {\n")
|
||||
out.WriteString(f.Body.String())
|
||||
out.WriteString("\n}")
|
||||
|
||||
return out.String()
|
||||
}
|
||||
func (f *Function) String() string {
|
||||
return f.Inspect()
|
||||
}
|
||||
|
||||
type ReturnValue struct {
|
||||
Value Object
|
||||
}
|
||||
|
||||
func (rv *ReturnValue) Type() ObjectType {
|
||||
return RETURN_VALUE_OBJ
|
||||
}
|
||||
func (rv *ReturnValue) Inspect() string {
|
||||
return rv.Value.Inspect()
|
||||
}
|
||||
func (rv *ReturnValue) String() string {
|
||||
return rv.Inspect()
|
||||
}
|
||||
91
object/hash.go
Normal file
91
object/hash.go
Normal file
@@ -0,0 +1,91 @@
|
||||
package object
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"hash/fnv"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type HashKey struct {
|
||||
Type ObjectType
|
||||
Value uint64
|
||||
}
|
||||
|
||||
func (b *Boolean) HashKey() HashKey {
|
||||
var value uint64
|
||||
|
||||
if b.Value {
|
||||
value = 1
|
||||
} else {
|
||||
value = 0
|
||||
}
|
||||
|
||||
return HashKey{Type: b.Type(), Value: value}
|
||||
}
|
||||
|
||||
func (i *Integer) HashKey() HashKey {
|
||||
return HashKey{Type: i.Type(), Value: uint64(i.Value)}
|
||||
}
|
||||
|
||||
func (s *String) HashKey() HashKey {
|
||||
h := fnv.New64a()
|
||||
h.Write([]byte(s.Value))
|
||||
|
||||
return HashKey{Type: s.Type(), Value: h.Sum64()}
|
||||
}
|
||||
|
||||
type HashPair struct {
|
||||
Key Object
|
||||
Value Object
|
||||
}
|
||||
|
||||
type Hash struct {
|
||||
Pairs map[HashKey]HashPair
|
||||
}
|
||||
|
||||
func (h *Hash) Type() ObjectType {
|
||||
return HASH_OBJ
|
||||
}
|
||||
func (h *Hash) Inspect() string {
|
||||
var out bytes.Buffer
|
||||
|
||||
pairs := []string{}
|
||||
for _, pair := range h.Pairs {
|
||||
pairs = append(pairs, fmt.Sprintf("%s: %s", pair.Key.Inspect(), pair.Value.Inspect()))
|
||||
}
|
||||
|
||||
out.WriteString("{")
|
||||
out.WriteString(strings.Join(pairs, ", "))
|
||||
out.WriteString("}")
|
||||
|
||||
return out.String()
|
||||
}
|
||||
func (h *Hash) String() string {
|
||||
return h.Inspect()
|
||||
}
|
||||
func (h *Hash) Equal(other Object) bool {
|
||||
if obj, ok := other.(*Hash); ok {
|
||||
if len(h.Pairs) != len(obj.Pairs) {
|
||||
return false
|
||||
}
|
||||
for _, pair := range h.Pairs {
|
||||
left := pair.Value
|
||||
hashed := left.(Hashable)
|
||||
right, ok := obj.Pairs[hashed.HashKey()]
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
cmp, ok := left.(Comparable)
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
if !cmp.Equal(right.Value) {
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
26
object/int.go
Normal file
26
object/int.go
Normal file
@@ -0,0 +1,26 @@
|
||||
package object
|
||||
|
||||
import "fmt"
|
||||
|
||||
type Integer struct {
|
||||
Value int64
|
||||
}
|
||||
|
||||
func (i *Integer) Type() ObjectType {
|
||||
return INTEGER_OBJ
|
||||
}
|
||||
func (i *Integer) Inspect() string {
|
||||
return fmt.Sprintf("%d", i.Value)
|
||||
}
|
||||
func (i *Integer) Clone() Object {
|
||||
return &Integer{Value: i.Value}
|
||||
}
|
||||
func (i *Integer) String() string {
|
||||
return i.Inspect()
|
||||
}
|
||||
func (i *Integer) Equal(other Object) bool {
|
||||
if obj, ok := other.(*Integer); ok {
|
||||
return i.Value == obj.Value
|
||||
}
|
||||
return false
|
||||
}
|
||||
17
object/null.go
Normal file
17
object/null.go
Normal file
@@ -0,0 +1,17 @@
|
||||
package object
|
||||
|
||||
type Null struct{}
|
||||
|
||||
func (n *Null) Type() ObjectType {
|
||||
return NULL_OBJ
|
||||
}
|
||||
func (n *Null) Inspect() string {
|
||||
return "null"
|
||||
}
|
||||
func (n *Null) String() string {
|
||||
return n.Inspect()
|
||||
}
|
||||
func (n *Null) Equal(other Object) bool {
|
||||
_, ok := other.(*Null)
|
||||
return ok
|
||||
}
|
||||
336
object/object.go
336
object/object.go
@@ -1,14 +1,6 @@
|
||||
package object
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"hash/fnv"
|
||||
"monkey/ast"
|
||||
"monkey/code"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// Type represents the type of an object
|
||||
type ObjectType string
|
||||
|
||||
const (
|
||||
@@ -39,333 +31,19 @@ type Immutable interface {
|
||||
Clone() Object
|
||||
}
|
||||
|
||||
// Object represents a value and implementations are expected to implement
|
||||
// `Type()` and `Inspect()` functions
|
||||
type Object interface {
|
||||
Type() ObjectType
|
||||
String() string
|
||||
Inspect() string
|
||||
}
|
||||
|
||||
type Integer struct {
|
||||
Value int64
|
||||
}
|
||||
|
||||
type CompiledFunction struct {
|
||||
Instructions code.Instructions
|
||||
NumLocals int
|
||||
NumParameters int
|
||||
}
|
||||
|
||||
func (i *Integer) Type() ObjectType {
|
||||
return INTEGER_OBJ
|
||||
}
|
||||
func (i *Integer) Inspect() string {
|
||||
return fmt.Sprintf("%d", i.Value)
|
||||
}
|
||||
func (i *Integer) Clone() Object {
|
||||
return &Integer{Value: i.Value}
|
||||
}
|
||||
func (i *Integer) String() string {
|
||||
return i.Inspect()
|
||||
}
|
||||
func (i *Integer) Equal(other Object) bool {
|
||||
if obj, ok := other.(*Integer); ok {
|
||||
return i.Value == obj.Value
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
type Boolean struct {
|
||||
Value bool
|
||||
}
|
||||
|
||||
func (b *Boolean) Type() ObjectType {
|
||||
return BOOLEAN_OBJ
|
||||
}
|
||||
func (b *Boolean) Inspect() string {
|
||||
return fmt.Sprintf("%t", b.Value)
|
||||
}
|
||||
func (b *Boolean) Clone() Object {
|
||||
return &Boolean{Value: b.Value}
|
||||
}
|
||||
func (b *Boolean) String() string {
|
||||
return b.Inspect()
|
||||
}
|
||||
func (b *Boolean) Equal(other Object) bool {
|
||||
if obj, ok := other.(*Boolean); ok {
|
||||
return b.Value == obj.Value
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
type Null struct{}
|
||||
|
||||
func (n *Null) Type() ObjectType {
|
||||
return NULL_OBJ
|
||||
}
|
||||
func (n *Null) Inspect() string {
|
||||
return "null"
|
||||
}
|
||||
func (n *Null) String() string {
|
||||
return n.Inspect()
|
||||
}
|
||||
func (n *Null) Equal(other Object) bool {
|
||||
_, ok := other.(*Null)
|
||||
return ok
|
||||
}
|
||||
|
||||
type ReturnValue struct {
|
||||
Value Object
|
||||
}
|
||||
|
||||
func (rv *ReturnValue) Type() ObjectType {
|
||||
return RETURN_VALUE_OBJ
|
||||
}
|
||||
func (rv *ReturnValue) Inspect() string {
|
||||
return rv.Value.Inspect()
|
||||
}
|
||||
func (rv *ReturnValue) String() string {
|
||||
return rv.Inspect()
|
||||
}
|
||||
|
||||
type Error struct {
|
||||
Message string
|
||||
}
|
||||
|
||||
func (e *Error) Type() ObjectType {
|
||||
return ERROR_OBJ
|
||||
}
|
||||
func (e *Error) Inspect() string {
|
||||
return "Error: " + e.Message
|
||||
}
|
||||
func (e *Error) Clone() Object {
|
||||
return &Error{Message: e.Message}
|
||||
}
|
||||
func (e *Error) String() string {
|
||||
return e.Message
|
||||
}
|
||||
|
||||
type Function struct {
|
||||
Parameters []*ast.Identifier
|
||||
Body *ast.BlockStatement
|
||||
Env *Environment
|
||||
}
|
||||
|
||||
func (f *Function) Type() ObjectType {
|
||||
return FUNCTION_OBJ
|
||||
}
|
||||
func (f *Function) Inspect() string {
|
||||
var out bytes.Buffer
|
||||
|
||||
params := []string{}
|
||||
for _, p := range f.Parameters {
|
||||
params = append(params, p.String())
|
||||
}
|
||||
|
||||
out.WriteString("fn")
|
||||
out.WriteString("(")
|
||||
out.WriteString(strings.Join(params, ", "))
|
||||
out.WriteString(") {\n")
|
||||
out.WriteString(f.Body.String())
|
||||
out.WriteString("\n}")
|
||||
|
||||
return out.String()
|
||||
}
|
||||
func (f *Function) String() string {
|
||||
return f.Inspect()
|
||||
}
|
||||
|
||||
type String struct {
|
||||
Value string
|
||||
}
|
||||
|
||||
func (s *String) Type() ObjectType {
|
||||
return STRING_OBJ
|
||||
}
|
||||
func (s *String) Inspect() string {
|
||||
return fmt.Sprintf("%#v", s.Value)
|
||||
}
|
||||
func (s *String) Clone() Object {
|
||||
return &String{Value: s.Value}
|
||||
}
|
||||
func (s *String) String() string {
|
||||
return s.Value
|
||||
}
|
||||
func (s *String) Equal(other Object) bool {
|
||||
if obj, ok := other.(*String); ok {
|
||||
return s.Value == obj.Value
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
type BuiltinFunction func(args ...Object) Object
|
||||
|
||||
type Builtin struct {
|
||||
Name string
|
||||
Fn BuiltinFunction
|
||||
}
|
||||
|
||||
func (b *Builtin) Type() ObjectType {
|
||||
return BUILTIN_OBJ
|
||||
}
|
||||
func (b *Builtin) Inspect() string {
|
||||
return fmt.Sprintf("<built-in function %s>", b.Name)
|
||||
}
|
||||
func (b *Builtin) String() string {
|
||||
return b.Inspect()
|
||||
}
|
||||
|
||||
type Array struct {
|
||||
Elements []Object
|
||||
}
|
||||
|
||||
func (ao *Array) Type() ObjectType {
|
||||
return ARRAY_OBJ
|
||||
}
|
||||
func (ao *Array) Inspect() string {
|
||||
var out bytes.Buffer
|
||||
|
||||
elements := []string{}
|
||||
for _, e := range ao.Elements {
|
||||
elements = append(elements, e.Inspect())
|
||||
}
|
||||
|
||||
out.WriteString("[")
|
||||
out.WriteString(strings.Join(elements, ", "))
|
||||
out.WriteString("]")
|
||||
|
||||
return out.String()
|
||||
}
|
||||
func (ao *Array) String() string {
|
||||
return ao.Inspect()
|
||||
}
|
||||
func (ao *Array) Equal(other Object) bool {
|
||||
if obj, ok := other.(*Array); ok {
|
||||
if len(ao.Elements) != len(obj.Elements) {
|
||||
return false
|
||||
}
|
||||
for i, el := range ao.Elements {
|
||||
cmp, ok := el.(Comparable)
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
if !cmp.Equal(obj.Elements[i]) {
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
type HashKey struct {
|
||||
Type ObjectType
|
||||
Value uint64
|
||||
}
|
||||
|
||||
func (b *Boolean) HashKey() HashKey {
|
||||
var value uint64
|
||||
|
||||
if b.Value {
|
||||
value = 1
|
||||
} else {
|
||||
value = 0
|
||||
}
|
||||
|
||||
return HashKey{Type: b.Type(), Value: value}
|
||||
}
|
||||
|
||||
// Hashable is the interface for all hashable objects which must implement
|
||||
// the HashKey() method which returns a HashKey result.
|
||||
type Hashable interface {
|
||||
HashKey() HashKey
|
||||
}
|
||||
|
||||
func (i *Integer) HashKey() HashKey {
|
||||
return HashKey{Type: i.Type(), Value: uint64(i.Value)}
|
||||
}
|
||||
|
||||
func (s *String) HashKey() HashKey {
|
||||
h := fnv.New64a()
|
||||
h.Write([]byte(s.Value))
|
||||
|
||||
return HashKey{Type: s.Type(), Value: h.Sum64()}
|
||||
}
|
||||
|
||||
type HashPair struct {
|
||||
Key Object
|
||||
Value Object
|
||||
}
|
||||
|
||||
type Hash struct {
|
||||
Pairs map[HashKey]HashPair
|
||||
}
|
||||
|
||||
func (h *Hash) Type() ObjectType {
|
||||
return HASH_OBJ
|
||||
}
|
||||
func (h *Hash) Inspect() string {
|
||||
var out bytes.Buffer
|
||||
|
||||
pairs := []string{}
|
||||
for _, pair := range h.Pairs {
|
||||
pairs = append(pairs, fmt.Sprintf("%s: %s", pair.Key.Inspect(), pair.Value.Inspect()))
|
||||
}
|
||||
|
||||
out.WriteString("{")
|
||||
out.WriteString(strings.Join(pairs, ", "))
|
||||
out.WriteString("}")
|
||||
|
||||
return out.String()
|
||||
}
|
||||
func (h *Hash) String() string {
|
||||
return h.Inspect()
|
||||
}
|
||||
func (h *Hash) Equal(other Object) bool {
|
||||
if obj, ok := other.(*Hash); ok {
|
||||
if len(h.Pairs) != len(obj.Pairs) {
|
||||
return false
|
||||
}
|
||||
for _, pair := range h.Pairs {
|
||||
left := pair.Value
|
||||
hashed := left.(Hashable)
|
||||
right, ok := obj.Pairs[hashed.HashKey()]
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
cmp, ok := left.(Comparable)
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
if !cmp.Equal(right.Value) {
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func (cf *CompiledFunction) Type() ObjectType {
|
||||
return COMPILED_FUNCTION_OBJ
|
||||
}
|
||||
func (cf *CompiledFunction) Inspect() string {
|
||||
return fmt.Sprintf("CompiledFunction[%p]", cf)
|
||||
}
|
||||
func (cf *CompiledFunction) String() string {
|
||||
return cf.Inspect()
|
||||
}
|
||||
|
||||
type Closure struct {
|
||||
Fn *CompiledFunction
|
||||
Free []Object
|
||||
}
|
||||
|
||||
func (c *Closure) Type() ObjectType {
|
||||
return CLOSURE_OBJ
|
||||
}
|
||||
func (c *Closure) Inspect() string {
|
||||
return fmt.Sprintf("Closure[%p]", c)
|
||||
}
|
||||
func (c *Closure) String() string {
|
||||
return c.Inspect()
|
||||
}
|
||||
// BuiltinFunction represents the builtin function type
|
||||
type BuiltinFunction func(args ...Object) Object
|
||||
|
||||
26
object/str.go
Normal file
26
object/str.go
Normal file
@@ -0,0 +1,26 @@
|
||||
package object
|
||||
|
||||
import "fmt"
|
||||
|
||||
type String struct {
|
||||
Value string
|
||||
}
|
||||
|
||||
func (s *String) Type() ObjectType {
|
||||
return STRING_OBJ
|
||||
}
|
||||
func (s *String) Inspect() string {
|
||||
return fmt.Sprintf("%#v", s.Value)
|
||||
}
|
||||
func (s *String) Clone() Object {
|
||||
return &String{Value: s.Value}
|
||||
}
|
||||
func (s *String) String() string {
|
||||
return s.Value
|
||||
}
|
||||
func (s *String) Equal(other Object) bool {
|
||||
if obj, ok := other.(*String); ok {
|
||||
return s.Value == obj.Value
|
||||
}
|
||||
return false
|
||||
}
|
||||
8
plugins/hello.go
Normal file
8
plugins/hello.go
Normal file
@@ -0,0 +1,8 @@
|
||||
package main
|
||||
|
||||
import "monkey/object"
|
||||
|
||||
// Hello ...
|
||||
func Hello(args ...object.Object) object.Object {
|
||||
return &object.String{Value: "Hello World!"}
|
||||
}
|
||||
@@ -8,6 +8,7 @@ import (
|
||||
"fmt"
|
||||
"io"
|
||||
"log"
|
||||
"monkey/builtins"
|
||||
"monkey/compiler"
|
||||
"monkey/evaluator"
|
||||
"monkey/lexer"
|
||||
@@ -49,7 +50,7 @@ type VMState struct {
|
||||
|
||||
func NewVMState() *VMState {
|
||||
symbolTable := compiler.NewSymbolTable()
|
||||
for i, builtin := range object.BuiltinsIndex {
|
||||
for i, builtin := range builtins.BuiltinsIndex {
|
||||
symbolTable.DefineBuiltin(i, builtin.Name)
|
||||
}
|
||||
|
||||
|
||||
3
vm/vm.go
3
vm/vm.go
@@ -3,6 +3,7 @@ package vm
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"monkey/builtins"
|
||||
"monkey/code"
|
||||
"monkey/compiler"
|
||||
"monkey/object"
|
||||
@@ -323,7 +324,7 @@ func (vm *VM) Run() error {
|
||||
builtinIndex := code.ReadUint8(ins[ip+1:])
|
||||
vm.currentFrame().ip += 1
|
||||
|
||||
builtin := object.BuiltinsIndex[builtinIndex]
|
||||
builtin := builtins.BuiltinsIndex[builtinIndex]
|
||||
|
||||
err := vm.push(builtin)
|
||||
if err != nil {
|
||||
|
||||
Reference in New Issue
Block a user