Fix VM memory allocation optimizations by reducing what we allocate on the heap
This commit is contained in:
@@ -15,10 +15,10 @@ func Abs(args ...object.Object) object.Object {
|
|||||||
return newError(err.Error())
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
i := args[0].(*object.Integer)
|
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}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -21,12 +21,12 @@ func Accept(args ...object.Object) object.Object {
|
|||||||
err error
|
err error
|
||||||
)
|
)
|
||||||
|
|
||||||
fd := int(args[0].(*object.Integer).Value)
|
fd := int(args[0].(object.Integer).Value)
|
||||||
|
|
||||||
nfd, _, err = syscall.Accept(fd)
|
nfd, _, err = syscall.Accept(fd)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return newError("SocketError: %s", err)
|
return newError("SocketError: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return &object.Integer{Value: int64(nfd)}
|
return object.Integer{Value: int64(nfd)}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -16,7 +16,7 @@ func args(args ...object.Object) object.Object {
|
|||||||
|
|
||||||
elements := make([]object.Object, len(object.Args))
|
elements := make([]object.Object, len(object.Args))
|
||||||
for i, arg := range object.Args {
|
for i, arg := range object.Args {
|
||||||
elements[i] = &object.String{Value: arg}
|
elements[i] = object.String{Value: arg}
|
||||||
}
|
}
|
||||||
return &object.Array{Elements: elements}
|
return &object.Array{Elements: elements}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -17,8 +17,8 @@ func Assert(args ...object.Object) object.Object {
|
|||||||
return newError(err.Error())
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
if !args[0].(*object.Boolean).Value {
|
if !args[0].(object.Boolean).Value {
|
||||||
fmt.Printf("Assertion Error: %s", args[1].(*object.String).Value)
|
fmt.Printf("Assertion Error: %s", args[1].(object.String).Value)
|
||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -17,6 +17,6 @@ func Bin(args ...object.Object) object.Object {
|
|||||||
return newError(err.Error())
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
i := args[0].(*object.Integer)
|
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))}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -21,8 +21,8 @@ func Bind(args ...object.Object) object.Object {
|
|||||||
sockaddr syscall.Sockaddr
|
sockaddr syscall.Sockaddr
|
||||||
)
|
)
|
||||||
|
|
||||||
fd := int(args[0].(*object.Integer).Value)
|
fd := int(args[0].(object.Integer).Value)
|
||||||
address := args[1].(*object.String).Value
|
address := args[1].(object.String).Value
|
||||||
|
|
||||||
sockaddr, err = syscall.Getsockname(fd)
|
sockaddr, err = syscall.Getsockname(fd)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -50,5 +50,5 @@ func Bind(args ...object.Object) object.Object {
|
|||||||
return newError("SocketError: %s", err)
|
return newError("SocketError: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return &object.Null{}
|
return object.Null{}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -14,5 +14,5 @@ func Bool(args ...object.Object) object.Object {
|
|||||||
return newError(err.Error())
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
return &object.Boolean{Value: args[0].Bool()}
|
return object.Boolean{Value: args[0].Bool()}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -72,6 +72,6 @@ func init() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func newError(format string, a ...interface{}) *object.Error {
|
func newError(format string, a ...interface{}) object.Error {
|
||||||
return &object.Error{Message: fmt.Sprintf(format, a...)}
|
return object.Error{Message: fmt.Sprintf(format, a...)}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -16,6 +16,6 @@ func Chr(args ...object.Object) object.Object {
|
|||||||
return newError(err.Error())
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
i := args[0].(*object.Integer)
|
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))}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -16,12 +16,12 @@ func Close(args ...object.Object) object.Object {
|
|||||||
return newError(err.Error())
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
fd := int(args[0].(*object.Integer).Value)
|
fd := int(args[0].(object.Integer).Value)
|
||||||
|
|
||||||
err := syscall.Close(fd)
|
err := syscall.Close(fd)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return newError("IOError: %s", err)
|
return newError("IOError: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return &object.Null{}
|
return object.Null{}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -18,8 +18,8 @@ func Connect(args ...object.Object) object.Object {
|
|||||||
|
|
||||||
var sa syscall.Sockaddr
|
var sa syscall.Sockaddr
|
||||||
|
|
||||||
fd := int(args[0].(*object.Integer).Value)
|
fd := int(args[0].(object.Integer).Value)
|
||||||
address := args[1].(*object.String).Value
|
address := args[1].(object.String).Value
|
||||||
|
|
||||||
sockaddr, err := syscall.Getsockname(fd)
|
sockaddr, err := syscall.Getsockname(fd)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -46,5 +46,5 @@ func Connect(args ...object.Object) object.Object {
|
|||||||
return newError("SocketError: %s", err)
|
return newError("SocketError: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return &object.Null{}
|
return object.Null{}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -15,10 +15,10 @@ func Divmod(args ...object.Object) object.Object {
|
|||||||
return newError(err.Error())
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
a := args[0].(*object.Integer)
|
a := args[0].(object.Integer)
|
||||||
b := args[1].(*object.Integer)
|
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}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -17,7 +17,7 @@ func Exit(args ...object.Object) object.Object {
|
|||||||
|
|
||||||
var status int
|
var status int
|
||||||
if len(args) == 1 {
|
if len(args) == 1 {
|
||||||
status = int(args[0].(*object.Integer).Value)
|
status = int(args[0].(object.Integer).Value)
|
||||||
}
|
}
|
||||||
|
|
||||||
object.ExitFunction(status)
|
object.ExitFunction(status)
|
||||||
|
|||||||
@@ -20,8 +20,8 @@ func FFI(args ...object.Object) object.Object {
|
|||||||
return newError(err.Error())
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
name := args[0].(*object.String).Value
|
name := args[0].(object.String).Value
|
||||||
symbol := args[1].(*object.String).Value
|
symbol := args[1].(object.String).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 {
|
||||||
|
|||||||
@@ -20,7 +20,7 @@ func Find(args ...object.Object) object.Object {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// find substring in string
|
// find substring in string
|
||||||
if haystack, ok := args[0].(*object.String); ok {
|
if haystack, ok := args[0].(object.String); ok {
|
||||||
if err := typing.Check(
|
if err := typing.Check(
|
||||||
"find", args,
|
"find", args,
|
||||||
typing.WithTypes(object.StringType, object.StringType),
|
typing.WithTypes(object.StringType, object.StringType),
|
||||||
@@ -28,9 +28,9 @@ func Find(args ...object.Object) object.Object {
|
|||||||
return newError(err.Error())
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
needle := args[1].(*object.String)
|
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)}
|
||||||
}
|
}
|
||||||
|
|
||||||
// find in array
|
// find in array
|
||||||
@@ -40,9 +40,9 @@ func Find(args ...object.Object) object.Object {
|
|||||||
return needle.Compare(haystack.Elements[i]) == 0
|
return needle.Compare(haystack.Elements[i]) == 0
|
||||||
})
|
})
|
||||||
if i < len(haystack.Elements) && needle.Compare(haystack.Elements[i]) == 0 {
|
if i < len(haystack.Elements) && needle.Compare(haystack.Elements[i]) == 0 {
|
||||||
return &object.Integer{Value: int64(i)}
|
return object.Integer{Value: int64(i)}
|
||||||
}
|
}
|
||||||
return &object.Integer{Value: -1}
|
return object.Integer{Value: -1}
|
||||||
}
|
}
|
||||||
|
|
||||||
return newError(
|
return newError(
|
||||||
|
|||||||
@@ -15,7 +15,7 @@ func HashOf(args ...object.Object) object.Object {
|
|||||||
}
|
}
|
||||||
|
|
||||||
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("TypeError: hash() expected argument #1 to be hashable")
|
return newError("TypeError: hash() expected argument #1 to be hashable")
|
||||||
|
|||||||
@@ -17,6 +17,6 @@ func Hex(args ...object.Object) object.Object {
|
|||||||
return newError(err.Error())
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
i := args[0].(*object.Integer)
|
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))}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -17,25 +17,24 @@ func IdOf(args ...object.Object) object.Object {
|
|||||||
|
|
||||||
arg := args[0]
|
arg := args[0]
|
||||||
|
|
||||||
if n, ok := arg.(*object.Null); ok {
|
if n, ok := arg.(object.Null); ok {
|
||||||
return &object.String{Value: fmt.Sprintf("%p", n)}
|
return object.String{Value: fmt.Sprintf("%p", &n)}
|
||||||
} else if b, ok := arg.(*object.Boolean); ok {
|
} else if b, ok := arg.(object.Boolean); ok {
|
||||||
return &object.String{Value: fmt.Sprintf("%p", b)}
|
return object.String{Value: fmt.Sprintf("%p", &b)}
|
||||||
} else if i, ok := arg.(*object.Integer); ok {
|
} else if i, ok := arg.(object.Integer); ok {
|
||||||
return &object.String{Value: fmt.Sprintf("%p", i)}
|
return object.String{Value: fmt.Sprintf("%p", &i)}
|
||||||
} else if s, ok := arg.(*object.String); ok {
|
} else if s, ok := arg.(object.String); ok {
|
||||||
return &object.String{Value: fmt.Sprintf("%p", s)}
|
return object.String{Value: fmt.Sprintf("%p", &s)}
|
||||||
} else if a, ok := arg.(*object.Array); ok {
|
} else if a, ok := arg.(*object.Array); ok {
|
||||||
return &object.String{Value: fmt.Sprintf("%p", a)}
|
return object.String{Value: fmt.Sprintf("%p", a)}
|
||||||
} else if h, ok := arg.(*object.Hash); ok {
|
} else if h, ok := arg.(*object.Hash); ok {
|
||||||
return &object.String{Value: fmt.Sprintf("%p", h)}
|
return object.String{Value: fmt.Sprintf("%p", h)}
|
||||||
} else if f, ok := arg.(*object.Function); ok {
|
} else if f, ok := arg.(*object.Function); ok {
|
||||||
return &object.String{Value: fmt.Sprintf("%p", f)}
|
return object.String{Value: fmt.Sprintf("%p", f)}
|
||||||
} else if c, ok := arg.(*object.Closure); ok {
|
} else if c, ok := arg.(*object.Closure); ok {
|
||||||
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)}
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -20,7 +20,7 @@ func Input(args ...object.Object) object.Object {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if len(args) == 1 {
|
if len(args) == 1 {
|
||||||
prompt := args[0].(*object.String).Value
|
prompt := args[0].(object.String).Value
|
||||||
fmt.Fprintf(os.Stdout, prompt)
|
fmt.Fprintf(os.Stdout, prompt)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -30,5 +30,5 @@ func Input(args ...object.Object) object.Object {
|
|||||||
if err != nil && err != io.EOF {
|
if err != nil && err != io.EOF {
|
||||||
return newError(fmt.Sprintf("error reading input from stdin: %s", err))
|
return newError(fmt.Sprintf("error reading input from stdin: %s", err))
|
||||||
}
|
}
|
||||||
return &object.String{Value: string(line)}
|
return object.String{Value: string(line)}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -16,20 +16,20 @@ func Int(args ...object.Object) object.Object {
|
|||||||
}
|
}
|
||||||
|
|
||||||
switch arg := args[0].(type) {
|
switch arg := args[0].(type) {
|
||||||
case *object.Boolean:
|
case object.Boolean:
|
||||||
if arg.Value {
|
if arg.Value {
|
||||||
return &object.Integer{Value: 1}
|
return object.Integer{Value: 1}
|
||||||
}
|
}
|
||||||
return &object.Integer{Value: 0}
|
return object.Integer{Value: 0}
|
||||||
case *object.Integer:
|
case object.Integer:
|
||||||
return arg
|
return arg
|
||||||
case *object.String:
|
case object.String:
|
||||||
n, err := strconv.ParseInt(arg.Value, 10, 64)
|
n, err := strconv.ParseInt(arg.Value, 10, 64)
|
||||||
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 &object.Integer{}
|
return object.Integer{}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -17,10 +17,10 @@ func Join(args ...object.Object) object.Object {
|
|||||||
}
|
}
|
||||||
|
|
||||||
arr := args[0].(*object.Array)
|
arr := args[0].(*object.Array)
|
||||||
sep := args[1].(*object.String)
|
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)}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -15,7 +15,7 @@ func Len(args ...object.Object) object.Object {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if size, ok := args[0].(object.Sizeable); ok {
|
if size, ok := args[0].(object.Sizeable); ok {
|
||||||
return &object.Integer{Value: int64(size.Len())}
|
return object.Integer{Value: int64(size.Len())}
|
||||||
}
|
}
|
||||||
return newError("TypeError: object of type '%s' has no len()", args[0].Type())
|
return newError("TypeError: object of type '%s' has no len()", args[0].Type())
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -16,12 +16,12 @@ func Listen(args ...object.Object) object.Object {
|
|||||||
return newError(err.Error())
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
fd := int(args[0].(*object.Integer).Value)
|
fd := int(args[0].(object.Integer).Value)
|
||||||
backlog := int(args[1].(*object.Integer).Value)
|
backlog := int(args[1].(object.Integer).Value)
|
||||||
|
|
||||||
if err := syscall.Listen(fd, backlog); err != nil {
|
if err := syscall.Listen(fd, backlog); err != nil {
|
||||||
return newError("SocketError: %s", err)
|
return newError("SocketError: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return &object.Null{}
|
return object.Null{}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -16,6 +16,6 @@ func Lower(args ...object.Object) object.Object {
|
|||||||
return newError(err.Error())
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
str := args[0].(*object.String)
|
str := args[0].(object.String)
|
||||||
return &object.String{Value: strings.ToLower(str.Value)}
|
return object.String{Value: strings.ToLower(str.Value)}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -20,12 +20,12 @@ func Max(args ...object.Object) object.Object {
|
|||||||
// TODO: Make this more generic
|
// TODO: Make this more generic
|
||||||
xs := []int{} //make([]int, len(a.Elements))
|
xs := []int{} //make([]int, len(a.Elements))
|
||||||
for n, e := range a.Elements {
|
for n, e := range a.Elements {
|
||||||
if i, ok := e.(*object.Integer); ok {
|
if i, ok := e.(object.Integer); ok {
|
||||||
xs = append(xs, int(i.Value))
|
xs = append(xs, int(i.Value))
|
||||||
} else {
|
} else {
|
||||||
return newError("item #%d not an `int` got=%d", n, e.Type())
|
return newError("item #%d not an `int` got=%d", n, e.Type())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
sort.Ints(xs)
|
sort.Ints(xs)
|
||||||
return &object.Integer{Value: int64(xs[len(xs)-1])}
|
return object.Integer{Value: int64(xs[len(xs)-1])}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -20,12 +20,12 @@ func Min(args ...object.Object) object.Object {
|
|||||||
// TODO: Make this more generic
|
// TODO: Make this more generic
|
||||||
xs := []int{} //make([]int, len(a.Elements))
|
xs := []int{} //make([]int, len(a.Elements))
|
||||||
for n, e := range a.Elements {
|
for n, e := range a.Elements {
|
||||||
if i, ok := e.(*object.Integer); ok {
|
if i, ok := e.(object.Integer); ok {
|
||||||
xs = append(xs, int(i.Value))
|
xs = append(xs, int(i.Value))
|
||||||
} else {
|
} else {
|
||||||
return newError("item #%d not an `int` got=%d", n, e.Type())
|
return newError("item #%d not an `int` got=%d", n, e.Type())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
sort.Ints(xs)
|
sort.Ints(xs)
|
||||||
return &object.Integer{Value: int64(xs[0])}
|
return object.Integer{Value: int64(xs[0])}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -17,6 +17,6 @@ func Oct(args ...object.Object) object.Object {
|
|||||||
return newError(err.Error())
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
i := args[0].(*object.Integer)
|
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))}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -66,10 +66,10 @@ func Open(args ...object.Object) object.Object {
|
|||||||
perm uint32 = 0640
|
perm uint32 = 0640
|
||||||
)
|
)
|
||||||
|
|
||||||
filename = args[0].(*object.String).Value
|
filename = args[0].(object.String).Value
|
||||||
|
|
||||||
if len(args) == 2 {
|
if len(args) == 2 {
|
||||||
mode = args[1].(*object.String).Value
|
mode = args[1].(object.String).Value
|
||||||
}
|
}
|
||||||
|
|
||||||
flag, err := parseMode(mode)
|
flag, err := parseMode(mode)
|
||||||
@@ -82,5 +82,5 @@ func Open(args ...object.Object) object.Object {
|
|||||||
return newError("IOError: %s", err)
|
return newError("IOError: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return &object.Integer{Value: int64(fd)}
|
return object.Integer{Value: int64(fd)}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -15,9 +15,9 @@ func Ord(args ...object.Object) object.Object {
|
|||||||
return newError(err.Error())
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
s := args[0].(*object.String)
|
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(
|
return newError(
|
||||||
"TypeError: ord() expected a single character `str` got=%s",
|
"TypeError: ord() expected a single character `str` got=%s",
|
||||||
|
|||||||
@@ -27,8 +27,8 @@ func Pow(args ...object.Object) object.Object {
|
|||||||
return newError(err.Error())
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
x := args[0].(*object.Integer)
|
x := args[0].(object.Integer)
|
||||||
y := args[1].(*object.Integer)
|
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}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -24,10 +24,10 @@ func Read(args ...object.Object) object.Object {
|
|||||||
n = DefaultBufferSize
|
n = DefaultBufferSize
|
||||||
)
|
)
|
||||||
|
|
||||||
fd = int(args[0].(*object.Integer).Value)
|
fd = int(args[0].(object.Integer).Value)
|
||||||
|
|
||||||
if len(args) == 2 {
|
if len(args) == 2 {
|
||||||
n = int(args[1].(*object.Integer).Value)
|
n = int(args[1].(object.Integer).Value)
|
||||||
}
|
}
|
||||||
|
|
||||||
buf := make([]byte, n)
|
buf := make([]byte, n)
|
||||||
@@ -36,5 +36,5 @@ func Read(args ...object.Object) object.Object {
|
|||||||
return newError("IOError: %s", err)
|
return newError("IOError: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return &object.String{Value: string(buf[:n])}
|
return object.String{Value: string(buf[:n])}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -16,11 +16,11 @@ func ReadFile(args ...object.Object) object.Object {
|
|||||||
return newError(err.Error())
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
filename := args[0].(*object.String).Value
|
filename := args[0].(object.String).Value
|
||||||
data, err := os.ReadFile(filename)
|
data, err := os.ReadFile(filename)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return newError("IOError: error reading from file %s: %s", filename, err)
|
return newError("IOError: error reading from file %s: %s", filename, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return &object.String{Value: string(data)}
|
return object.String{Value: string(data)}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -21,11 +21,11 @@ func Seek(args ...object.Object) object.Object {
|
|||||||
whence = 0
|
whence = 0
|
||||||
)
|
)
|
||||||
|
|
||||||
fd = int(args[0].(*object.Integer).Value)
|
fd = int(args[0].(object.Integer).Value)
|
||||||
offset := args[1].(*object.Integer).Value
|
offset := args[1].(object.Integer).Value
|
||||||
|
|
||||||
if len(args) == 3 {
|
if len(args) == 3 {
|
||||||
whence = int(args[2].(*object.Integer).Value)
|
whence = int(args[2].(object.Integer).Value)
|
||||||
}
|
}
|
||||||
|
|
||||||
offset, err := syscall.Seek(fd, offset, whence)
|
offset, err := syscall.Seek(fd, offset, whence)
|
||||||
@@ -33,5 +33,5 @@ func Seek(args ...object.Object) object.Object {
|
|||||||
return newError("IOError: %s", err)
|
return newError("IOError: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return &object.Integer{Value: offset}
|
return object.Integer{Value: offset}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -23,7 +23,7 @@ func Socket(args ...object.Object) object.Object {
|
|||||||
proto int
|
proto int
|
||||||
)
|
)
|
||||||
|
|
||||||
arg := args[0].(*object.String).Value
|
arg := args[0].(object.String).Value
|
||||||
|
|
||||||
switch strings.ToLower(arg) {
|
switch strings.ToLower(arg) {
|
||||||
case "unix":
|
case "unix":
|
||||||
@@ -61,5 +61,5 @@ func Socket(args ...object.Object) object.Object {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return &object.Integer{Value: int64(fd)}
|
return object.Integer{Value: int64(fd)}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -17,16 +17,16 @@ func Split(args ...object.Object) object.Object {
|
|||||||
}
|
}
|
||||||
|
|
||||||
var sep string
|
var sep string
|
||||||
s := args[0].(*object.String).Value
|
s := args[0].(object.String).Value
|
||||||
|
|
||||||
if len(args) == 2 {
|
if len(args) == 2 {
|
||||||
sep = args[1].(*object.String).Value
|
sep = args[1].(object.String).Value
|
||||||
}
|
}
|
||||||
|
|
||||||
tokens := strings.Split(s, sep)
|
tokens := strings.Split(s, sep)
|
||||||
elements := make([]object.Object, len(tokens))
|
elements := make([]object.Object, len(tokens))
|
||||||
for i, token := range tokens {
|
for i, token := range tokens {
|
||||||
elements[i] = &object.String{Value: token}
|
elements[i] = object.String{Value: token}
|
||||||
}
|
}
|
||||||
return &object.Array{Elements: elements}
|
return &object.Array{Elements: elements}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -14,5 +14,5 @@ func Str(args ...object.Object) object.Object {
|
|||||||
return newError(err.Error())
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
return &object.String{Value: args[0].String()}
|
return object.String{Value: args[0].String()}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -14,5 +14,5 @@ func TypeOf(args ...object.Object) object.Object {
|
|||||||
return newError(err.Error())
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
return &object.String{Value: args[0].Type().String()}
|
return object.String{Value: args[0].Type().String()}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -16,5 +16,5 @@ func Upper(args ...object.Object) object.Object {
|
|||||||
return newError(err.Error())
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
return &object.String{Value: strings.ToUpper(args[0].(*object.String).Value)}
|
return object.String{Value: strings.ToUpper(args[0].(object.String).Value)}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -16,13 +16,13 @@ func Write(args ...object.Object) object.Object {
|
|||||||
return newError(err.Error())
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
fd := int(args[0].(*object.Integer).Value)
|
fd := int(args[0].(object.Integer).Value)
|
||||||
data := []byte(args[1].(*object.String).Value)
|
data := []byte(args[1].(object.String).Value)
|
||||||
|
|
||||||
n, err := syscall.Write(fd, data)
|
n, err := syscall.Write(fd, data)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return newError("IOError: %s", err)
|
return newError("IOError: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return &object.Integer{Value: int64(n)}
|
return object.Integer{Value: int64(n)}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -16,13 +16,13 @@ func WriteFile(args ...object.Object) object.Object {
|
|||||||
return newError(err.Error())
|
return newError(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
filename := args[0].(*object.String).Value
|
filename := args[0].(object.String).Value
|
||||||
data := []byte(args[1].(*object.String).Value)
|
data := []byte(args[1].(object.String).Value)
|
||||||
|
|
||||||
err := os.WriteFile(filename, data, 0755)
|
err := os.WriteFile(filename, data, 0755)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return newError("IOError: error writing file %s: %s", filename, err)
|
return newError("IOError: error writing file %s: %s", filename, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return &object.Null{}
|
return object.Null{}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -70,13 +70,13 @@ func (e *encoder) WriteObjects(objs ...object.Object) (err error) {
|
|||||||
err = errors.Join(err, e.WriteValue(obj.Type()))
|
err = errors.Join(err, e.WriteValue(obj.Type()))
|
||||||
|
|
||||||
switch o := obj.(type) {
|
switch o := obj.(type) {
|
||||||
case *object.Null:
|
case object.Null:
|
||||||
break
|
break
|
||||||
case *object.Boolean:
|
case object.Boolean:
|
||||||
err = errors.Join(err, e.WriteValue(o.Value))
|
err = errors.Join(err, e.WriteValue(o.Value))
|
||||||
case *object.Integer:
|
case object.Integer:
|
||||||
err = errors.Join(err, e.WriteValue(o.Value))
|
err = errors.Join(err, e.WriteValue(o.Value))
|
||||||
case *object.String:
|
case object.String:
|
||||||
err = errors.Join(err, e.WriteValue(len(o.Value)))
|
err = errors.Join(err, e.WriteValue(len(o.Value)))
|
||||||
err = errors.Join(err, e.WriteValue(o.Value))
|
err = errors.Join(err, e.WriteValue(o.Value))
|
||||||
case *object.CompiledFunction:
|
case *object.CompiledFunction:
|
||||||
@@ -146,13 +146,13 @@ func (d *decoder) Objects(len int) (o []object.Object) {
|
|||||||
for i := 0; i < len; i++ {
|
for i := 0; i < len; i++ {
|
||||||
switch t := object.Type(d.Int()); t {
|
switch t := object.Type(d.Int()); t {
|
||||||
case object.NullType:
|
case object.NullType:
|
||||||
o = append(o, &object.Null{})
|
o = append(o, object.Null{})
|
||||||
case object.BooleanType:
|
case object.BooleanType:
|
||||||
o = append(o, &object.Boolean{Value: d.Byte() == 1})
|
o = append(o, object.Boolean{Value: d.Byte() == 1})
|
||||||
case object.IntegerType:
|
case object.IntegerType:
|
||||||
o = append(o, &object.Integer{Value: d.Int64()})
|
o = append(o, object.Integer{Value: d.Int64()})
|
||||||
case object.StringType:
|
case object.StringType:
|
||||||
o = append(o, &object.String{Value: d.String(d.Int())})
|
o = append(o, object.String{Value: d.String(d.Int())})
|
||||||
case object.CFunctionType:
|
case object.CFunctionType:
|
||||||
// The order of the fields has to reflect the data layout in the encoded bytecode.
|
// The order of the fields has to reflect the data layout in the encoded bytecode.
|
||||||
o = append(o, &object.CompiledFunction{
|
o = append(o, &object.CompiledFunction{
|
||||||
|
|||||||
@@ -173,7 +173,7 @@ func (c *Compiler) Compile(node ast.Node) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
case *ast.IntegerLiteral:
|
case *ast.IntegerLiteral:
|
||||||
integer := &object.Integer{Value: node.Value}
|
integer := object.Integer{Value: node.Value}
|
||||||
c.emit(code.OpConstant, c.addConstant(integer))
|
c.emit(code.OpConstant, c.addConstant(integer))
|
||||||
|
|
||||||
case *ast.Null:
|
case *ast.Null:
|
||||||
@@ -356,7 +356,7 @@ func (c *Compiler) Compile(node ast.Node) error {
|
|||||||
c.loadSymbol(symbol)
|
c.loadSymbol(symbol)
|
||||||
|
|
||||||
case *ast.StringLiteral:
|
case *ast.StringLiteral:
|
||||||
str := &object.String{Value: node.Value}
|
str := object.String{Value: node.Value}
|
||||||
c.emit(code.OpConstant, c.addConstant(str))
|
c.emit(code.OpConstant, c.addConstant(str))
|
||||||
|
|
||||||
case *ast.ArrayLiteral:
|
case *ast.ArrayLiteral:
|
||||||
|
|||||||
@@ -1220,16 +1220,16 @@ func testConstants2(t *testing.T, expected []interface{}, actual []object.Object
|
|||||||
)
|
)
|
||||||
|
|
||||||
case string:
|
case string:
|
||||||
assert.Equal(constant, actual[i].(*object.String).Value)
|
assert.Equal(constant, actual[i].(object.String).Value)
|
||||||
|
|
||||||
case int:
|
case int:
|
||||||
assert.Equal(int64(constant), actual[i].(*object.Integer).Value)
|
assert.Equal(int64(constant), actual[i].(object.Integer).Value)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func testIntegerObject(expected int64, actual object.Object) interface{} {
|
func testIntegerObject(expected int64, actual object.Object) interface{} {
|
||||||
result, ok := actual.(*object.Integer)
|
result, ok := actual.(object.Integer)
|
||||||
if !ok {
|
if !ok {
|
||||||
return fmt.Errorf("object is not Integer. got=%T (%+v", actual, actual)
|
return fmt.Errorf("object is not Integer. got=%T (%+v", actual, actual)
|
||||||
}
|
}
|
||||||
@@ -1242,7 +1242,7 @@ func testIntegerObject(expected int64, actual object.Object) interface{} {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func testStringObject(expected string, actual object.Object) error {
|
func testStringObject(expected string, actual object.Object) error {
|
||||||
result, ok := actual.(*object.String)
|
result, ok := actual.(object.String)
|
||||||
if !ok {
|
if !ok {
|
||||||
return fmt.Errorf("object is not String. got %T (%+v", actual, actual)
|
return fmt.Errorf("object is not String. got %T (%+v", actual, actual)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -13,9 +13,9 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
NULL = &object.Null{}
|
NULL = object.Null{}
|
||||||
TRUE = &object.Boolean{Value: true}
|
TRUE = object.Boolean{Value: true}
|
||||||
FALSE = &object.Boolean{Value: false}
|
FALSE = object.Boolean{Value: false}
|
||||||
)
|
)
|
||||||
|
|
||||||
func isError(obj object.Object) bool {
|
func isError(obj object.Object) bool {
|
||||||
@@ -52,7 +52,7 @@ func Eval(node ast.Node, env *object.Environment) object.Object {
|
|||||||
if isError(val) {
|
if isError(val) {
|
||||||
return val
|
return val
|
||||||
}
|
}
|
||||||
return &object.ReturnValue{Value: val}
|
return object.ReturnValue{Value: val}
|
||||||
|
|
||||||
case *ast.BindExpression:
|
case *ast.BindExpression:
|
||||||
value := Eval(node.Value, env)
|
value := Eval(node.Value, env)
|
||||||
@@ -95,7 +95,7 @@ func Eval(node ast.Node, env *object.Environment) object.Object {
|
|||||||
if isError(index) {
|
if isError(index) {
|
||||||
return index
|
return index
|
||||||
}
|
}
|
||||||
if idx, ok := index.(*object.Integer); ok {
|
if idx, ok := index.(object.Integer); ok {
|
||||||
array.Elements[idx.Value] = value
|
array.Elements[idx.Value] = value
|
||||||
} else {
|
} else {
|
||||||
return newError("cannot index array with %#v", index)
|
return newError("cannot index array with %#v", index)
|
||||||
@@ -130,7 +130,7 @@ func Eval(node ast.Node, env *object.Environment) object.Object {
|
|||||||
|
|
||||||
// Expressions
|
// Expressions
|
||||||
case *ast.IntegerLiteral:
|
case *ast.IntegerLiteral:
|
||||||
return &object.Integer{Value: node.Value}
|
return object.Integer{Value: node.Value}
|
||||||
|
|
||||||
case *ast.Boolean:
|
case *ast.Boolean:
|
||||||
return nativeBoolToBooleanObject(node.Value)
|
return nativeBoolToBooleanObject(node.Value)
|
||||||
@@ -169,7 +169,7 @@ func Eval(node ast.Node, env *object.Environment) object.Object {
|
|||||||
return applyFunction(function, args)
|
return applyFunction(function, args)
|
||||||
|
|
||||||
case *ast.StringLiteral:
|
case *ast.StringLiteral:
|
||||||
return &object.String{Value: node.Value}
|
return object.String{Value: node.Value}
|
||||||
|
|
||||||
case *ast.ArrayLiteral:
|
case *ast.ArrayLiteral:
|
||||||
elements := evalExpressions(node.Elements, env)
|
elements := evalExpressions(node.Elements, env)
|
||||||
@@ -203,12 +203,12 @@ func evalImportExpression(ie *ast.ImportExpression, env *object.Environment) obj
|
|||||||
return name
|
return name
|
||||||
}
|
}
|
||||||
|
|
||||||
if s, ok := name.(*object.String); ok {
|
if s, ok := name.(object.String); ok {
|
||||||
attrs := EvalModule(s.Value)
|
attrs := EvalModule(s.Value)
|
||||||
if isError(attrs) {
|
if isError(attrs) {
|
||||||
return attrs
|
return attrs
|
||||||
}
|
}
|
||||||
return &object.Module{Name: s.Value, Attrs: attrs}
|
return object.Module{Name: s.Value, Attrs: attrs}
|
||||||
}
|
}
|
||||||
return newError("ImportError: invalid import path '%s'", name)
|
return newError("ImportError: invalid import path '%s'", name)
|
||||||
}
|
}
|
||||||
@@ -243,9 +243,9 @@ func evalProgram(program *ast.Program, env *object.Environment) object.Object {
|
|||||||
result = Eval(statement, env)
|
result = Eval(statement, env)
|
||||||
|
|
||||||
switch result := result.(type) {
|
switch result := result.(type) {
|
||||||
case *object.ReturnValue:
|
case object.ReturnValue:
|
||||||
return result.Value
|
return result.Value
|
||||||
case *object.Error:
|
case object.Error:
|
||||||
return result
|
return result
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -271,7 +271,7 @@ func evalBlockStatements(block *ast.BlockStatement, env *object.Environment) obj
|
|||||||
return result
|
return result
|
||||||
}
|
}
|
||||||
|
|
||||||
func nativeBoolToBooleanObject(input bool) object.Object {
|
func nativeBoolToBooleanObject(input bool) object.Boolean {
|
||||||
if input {
|
if input {
|
||||||
return TRUE
|
return TRUE
|
||||||
}
|
}
|
||||||
@@ -314,14 +314,14 @@ func evalIntegerPrefixOperatorExpression(operator string, right object.Object) o
|
|||||||
return newError("unknown operator: -%s", right.Type())
|
return newError("unknown operator: -%s", right.Type())
|
||||||
}
|
}
|
||||||
|
|
||||||
value := right.(*object.Integer).Value
|
value := right.(object.Integer).Value
|
||||||
switch operator {
|
switch operator {
|
||||||
case "!":
|
case "!":
|
||||||
return FALSE
|
return FALSE
|
||||||
case "~":
|
case "~":
|
||||||
return &object.Integer{Value: ^value}
|
return object.Integer{Value: ^value}
|
||||||
case "-":
|
case "-":
|
||||||
return &object.Integer{Value: -value}
|
return object.Integer{Value: -value}
|
||||||
default:
|
default:
|
||||||
return newError("unknown operator: %s", operator)
|
return newError("unknown operator: %s", operator)
|
||||||
}
|
}
|
||||||
@@ -354,7 +354,7 @@ func evalInfixExpression(operator string, left, right object.Object) object.Obje
|
|||||||
// [1] * 3
|
// [1] * 3
|
||||||
case operator == "*" && left.Type() == object.ArrayType && right.Type() == object.IntegerType:
|
case operator == "*" && left.Type() == object.ArrayType && right.Type() == object.IntegerType:
|
||||||
leftVal := left.(*object.Array).Elements
|
leftVal := left.(*object.Array).Elements
|
||||||
rightVal := int(right.(*object.Integer).Value)
|
rightVal := int(right.(object.Integer).Value)
|
||||||
elements := leftVal
|
elements := leftVal
|
||||||
for i := rightVal; i > 1; i-- {
|
for i := rightVal; i > 1; i-- {
|
||||||
elements = append(elements, leftVal...)
|
elements = append(elements, leftVal...)
|
||||||
@@ -363,7 +363,7 @@ func evalInfixExpression(operator string, left, right object.Object) object.Obje
|
|||||||
|
|
||||||
// 3 * [1]
|
// 3 * [1]
|
||||||
case operator == "*" && left.Type() == object.IntegerType && right.Type() == object.ArrayType:
|
case operator == "*" && left.Type() == object.IntegerType && right.Type() == object.ArrayType:
|
||||||
leftVal := int(left.(*object.Integer).Value)
|
leftVal := int(left.(object.Integer).Value)
|
||||||
rightVal := right.(*object.Array).Elements
|
rightVal := right.(*object.Array).Elements
|
||||||
elements := rightVal
|
elements := rightVal
|
||||||
for i := leftVal; i > 1; i-- {
|
for i := leftVal; i > 1; i-- {
|
||||||
@@ -373,15 +373,15 @@ func evalInfixExpression(operator string, left, right object.Object) object.Obje
|
|||||||
|
|
||||||
// " " * 4
|
// " " * 4
|
||||||
case operator == "*" && left.Type() == object.StringType && right.Type() == object.IntegerType:
|
case operator == "*" && left.Type() == object.StringType && right.Type() == object.IntegerType:
|
||||||
leftVal := left.(*object.String).Value
|
leftVal := left.(object.String).Value
|
||||||
rightVal := right.(*object.Integer).Value
|
rightVal := right.(object.Integer).Value
|
||||||
return &object.String{Value: strings.Repeat(leftVal, int(rightVal))}
|
return object.String{Value: strings.Repeat(leftVal, int(rightVal))}
|
||||||
|
|
||||||
// 4 * " "
|
// 4 * " "
|
||||||
case operator == "*" && left.Type() == object.IntegerType && right.Type() == object.StringType:
|
case operator == "*" && left.Type() == object.IntegerType && right.Type() == object.StringType:
|
||||||
leftVal := left.(*object.Integer).Value
|
leftVal := left.(object.Integer).Value
|
||||||
rightVal := right.(*object.String).Value
|
rightVal := right.(object.String).Value
|
||||||
return &object.String{Value: strings.Repeat(rightVal, int(leftVal))}
|
return object.String{Value: strings.Repeat(rightVal, int(leftVal))}
|
||||||
|
|
||||||
case operator == "==":
|
case operator == "==":
|
||||||
return nativeBoolToBooleanObject(left.(object.Comparable).Compare(right) == 0)
|
return nativeBoolToBooleanObject(left.(object.Comparable).Compare(right) == 0)
|
||||||
@@ -409,8 +409,8 @@ func evalInfixExpression(operator string, left, right object.Object) object.Obje
|
|||||||
}
|
}
|
||||||
|
|
||||||
func evalBooleanInfixExpression(operator string, left, right object.Object) object.Object {
|
func evalBooleanInfixExpression(operator string, left, right object.Object) object.Object {
|
||||||
leftVal := left.(*object.Boolean).Value
|
leftVal := left.(object.Boolean).Value
|
||||||
rightVal := right.(*object.Boolean).Value
|
rightVal := right.(object.Boolean).Value
|
||||||
|
|
||||||
switch operator {
|
switch operator {
|
||||||
case "&&":
|
case "&&":
|
||||||
@@ -423,42 +423,42 @@ func evalBooleanInfixExpression(operator string, left, right object.Object) obje
|
|||||||
}
|
}
|
||||||
|
|
||||||
func evalStringInfixExpression(operator string, left object.Object, right object.Object) object.Object {
|
func evalStringInfixExpression(operator string, left object.Object, right object.Object) object.Object {
|
||||||
leftVal := left.(*object.String).Value
|
leftVal := left.(object.String).Value
|
||||||
rightVal := right.(*object.String).Value
|
rightVal := right.(object.String).Value
|
||||||
|
|
||||||
switch operator {
|
switch operator {
|
||||||
case "+":
|
case "+":
|
||||||
return &object.String{Value: leftVal + rightVal}
|
return object.String{Value: leftVal + rightVal}
|
||||||
default:
|
default:
|
||||||
return newError("unknown operator: %s %s %s", left.Type(), operator, right.Type())
|
return newError("unknown operator: %s %s %s", left.Type(), operator, right.Type())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func evalIntegerInfixExpression(operator string, left, right object.Object) object.Object {
|
func evalIntegerInfixExpression(operator string, left, right object.Object) object.Object {
|
||||||
leftVal := left.(*object.Integer).Value
|
leftVal := left.(object.Integer).Value
|
||||||
rightVal := right.(*object.Integer).Value
|
rightVal := right.(object.Integer).Value
|
||||||
|
|
||||||
switch operator {
|
switch operator {
|
||||||
case "+":
|
case "+":
|
||||||
return &object.Integer{Value: leftVal + rightVal}
|
return object.Integer{Value: leftVal + rightVal}
|
||||||
case "-":
|
case "-":
|
||||||
return &object.Integer{Value: leftVal - rightVal}
|
return object.Integer{Value: leftVal - rightVal}
|
||||||
case "*":
|
case "*":
|
||||||
return &object.Integer{Value: leftVal * rightVal}
|
return object.Integer{Value: leftVal * rightVal}
|
||||||
case "/":
|
case "/":
|
||||||
return &object.Integer{Value: leftVal / rightVal}
|
return object.Integer{Value: leftVal / rightVal}
|
||||||
case "%":
|
case "%":
|
||||||
return &object.Integer{Value: leftVal % rightVal}
|
return object.Integer{Value: leftVal % rightVal}
|
||||||
case "|":
|
case "|":
|
||||||
return &object.Integer{Value: leftVal | rightVal}
|
return object.Integer{Value: leftVal | rightVal}
|
||||||
case "^":
|
case "^":
|
||||||
return &object.Integer{Value: leftVal ^ rightVal}
|
return object.Integer{Value: leftVal ^ rightVal}
|
||||||
case "&":
|
case "&":
|
||||||
return &object.Integer{Value: leftVal & rightVal}
|
return object.Integer{Value: leftVal & rightVal}
|
||||||
case "<<":
|
case "<<":
|
||||||
return &object.Integer{Value: leftVal << uint64(rightVal)}
|
return object.Integer{Value: leftVal << uint64(rightVal)}
|
||||||
case ">>":
|
case ">>":
|
||||||
return &object.Integer{Value: leftVal >> uint64(rightVal)}
|
return object.Integer{Value: leftVal >> uint64(rightVal)}
|
||||||
case "<":
|
case "<":
|
||||||
return nativeBoolToBooleanObject(leftVal < rightVal)
|
return nativeBoolToBooleanObject(leftVal < rightVal)
|
||||||
case "<=":
|
case "<=":
|
||||||
@@ -504,11 +504,11 @@ func isTruthy(obj object.Object) bool {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func newError(format string, a ...interface{}) *object.Error {
|
func newError(format string, a ...interface{}) object.Error {
|
||||||
return &object.Error{Message: fmt.Sprintf(format, a...)}
|
return object.Error{Message: fmt.Sprintf(format, a...)}
|
||||||
}
|
}
|
||||||
|
|
||||||
// EvalModule evaluates the named module and returns a *object.Module objec
|
// EvalModule evaluates the named module and returns a object.Module objec
|
||||||
func EvalModule(name string) object.Object {
|
func EvalModule(name string) object.Object {
|
||||||
filename := utils.FindModule(name)
|
filename := utils.FindModule(name)
|
||||||
|
|
||||||
@@ -588,7 +588,7 @@ func extendFunctionEnv(fn *object.Function, args []object.Object) *object.Enviro
|
|||||||
}
|
}
|
||||||
|
|
||||||
func unwrapReturnValue(obj object.Object) object.Object {
|
func unwrapReturnValue(obj object.Object) object.Object {
|
||||||
if returnValue, ok := obj.(*object.ReturnValue); ok {
|
if returnValue, ok := obj.(object.ReturnValue); ok {
|
||||||
return returnValue.Value
|
return returnValue.Value
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -611,20 +611,20 @@ func evalIndexExpression(left, index object.Object) object.Object {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func EvalModuleIndexExpression(module, index object.Object) object.Object {
|
func EvalModuleIndexExpression(module, index object.Object) object.Object {
|
||||||
moduleObject := module.(*object.Module)
|
moduleObject := module.(object.Module)
|
||||||
return evalHashIndexExpression(moduleObject.Attrs, index)
|
return evalHashIndexExpression(moduleObject.Attrs, index)
|
||||||
}
|
}
|
||||||
|
|
||||||
func evalStringIndexExpression(str, index object.Object) object.Object {
|
func evalStringIndexExpression(str, index object.Object) object.Object {
|
||||||
stringObject := str.(*object.String)
|
stringObject := str.(object.String)
|
||||||
idx := index.(*object.Integer).Value
|
idx := index.(object.Integer).Value
|
||||||
max := int64((len(stringObject.Value)) - 1)
|
max := int64((len(stringObject.Value)) - 1)
|
||||||
|
|
||||||
if idx < 0 || idx > max {
|
if idx < 0 || idx > max {
|
||||||
return &object.String{Value: ""}
|
return object.String{Value: ""}
|
||||||
}
|
}
|
||||||
|
|
||||||
return &object.String{Value: string(stringObject.Value[idx])}
|
return object.String{Value: string(stringObject.Value[idx])}
|
||||||
}
|
}
|
||||||
|
|
||||||
func evalHashIndexExpression(hash, index object.Object) object.Object {
|
func evalHashIndexExpression(hash, index object.Object) object.Object {
|
||||||
@@ -645,7 +645,7 @@ func evalHashIndexExpression(hash, index object.Object) object.Object {
|
|||||||
|
|
||||||
func evalArrayIndexExpression(array, index object.Object) object.Object {
|
func evalArrayIndexExpression(array, index object.Object) object.Object {
|
||||||
arrayObject := array.(*object.Array)
|
arrayObject := array.(*object.Array)
|
||||||
idx := index.(*object.Integer).Value
|
idx := index.(object.Integer).Value
|
||||||
maxInx := int64(len(arrayObject.Elements) - 1)
|
maxInx := int64(len(arrayObject.Elements) - 1)
|
||||||
|
|
||||||
if idx < 0 || idx > maxInx {
|
if idx < 0 || idx > maxInx {
|
||||||
|
|||||||
@@ -25,25 +25,25 @@ func assertEvaluated(t *testing.T, expected interface{}, actual object.Object) {
|
|||||||
|
|
||||||
switch expected.(type) {
|
switch expected.(type) {
|
||||||
case nil:
|
case nil:
|
||||||
if _, ok := actual.(*object.Null); ok {
|
if _, ok := actual.(object.Null); ok {
|
||||||
assert.True(ok)
|
assert.True(ok)
|
||||||
} else {
|
} else {
|
||||||
assert.Equal(expected, actual)
|
assert.Equal(expected, actual)
|
||||||
}
|
}
|
||||||
case int:
|
case int:
|
||||||
if i, ok := actual.(*object.Integer); ok {
|
if i, ok := actual.(object.Integer); ok {
|
||||||
assert.Equal(int64(expected.(int)), i.Value)
|
assert.Equal(int64(expected.(int)), i.Value)
|
||||||
} else {
|
} else {
|
||||||
assert.Equal(expected, actual)
|
assert.Equal(expected, actual)
|
||||||
}
|
}
|
||||||
case error:
|
case error:
|
||||||
if e, ok := actual.(*object.Integer); ok {
|
if e, ok := actual.(object.Integer); ok {
|
||||||
assert.Equal(expected.(error).Error(), e.Value)
|
assert.Equal(expected.(error).Error(), e.Value)
|
||||||
} else {
|
} else {
|
||||||
assert.Equal(expected, actual)
|
assert.Equal(expected, actual)
|
||||||
}
|
}
|
||||||
case string:
|
case string:
|
||||||
if s, ok := actual.(*object.String); ok {
|
if s, ok := actual.(object.String); ok {
|
||||||
assert.Equal(expected.(string), s.Value)
|
assert.Equal(expected.(string), s.Value)
|
||||||
} else {
|
} else {
|
||||||
assert.Equal(expected, actual)
|
assert.Equal(expected, actual)
|
||||||
@@ -161,7 +161,6 @@ func TestIfElseExpression(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
for _, tt := range tests {
|
for _, tt := range tests {
|
||||||
t.Log(tt.input)
|
|
||||||
evaluated := testEval(tt.input)
|
evaluated := testEval(tt.input)
|
||||||
integer, ok := tt.expected.(int)
|
integer, ok := tt.expected.(int)
|
||||||
if ok {
|
if ok {
|
||||||
@@ -279,7 +278,7 @@ func TestErrorHandling(t *testing.T) {
|
|||||||
for _, tt := range tests {
|
for _, tt := range tests {
|
||||||
evaluated := testEval(tt.input)
|
evaluated := testEval(tt.input)
|
||||||
|
|
||||||
errObj, ok := evaluated.(*object.Error)
|
errObj, ok := evaluated.(object.Error)
|
||||||
if !ok {
|
if !ok {
|
||||||
t.Errorf("no error object returned. got=%T(%+v)",
|
t.Errorf("no error object returned. got=%T(%+v)",
|
||||||
evaluated, evaluated)
|
evaluated, evaluated)
|
||||||
@@ -409,7 +408,7 @@ func TestStringLiteral(t *testing.T) {
|
|||||||
input := `"Hello World!"`
|
input := `"Hello World!"`
|
||||||
|
|
||||||
evaluated := testEval(input)
|
evaluated := testEval(input)
|
||||||
str, ok := evaluated.(*object.String)
|
str, ok := evaluated.(object.String)
|
||||||
if !ok {
|
if !ok {
|
||||||
t.Fatalf("object is not String. got=%T (+%v)", evaluated, evaluated)
|
t.Fatalf("object is not String. got=%T (+%v)", evaluated, evaluated)
|
||||||
}
|
}
|
||||||
@@ -423,7 +422,7 @@ func TestStringConcatenation(t *testing.T) {
|
|||||||
input := `"Hello" + " " + "World!"`
|
input := `"Hello" + " " + "World!"`
|
||||||
|
|
||||||
evaluated := testEval(input)
|
evaluated := testEval(input)
|
||||||
str, ok := evaluated.(*object.String)
|
str, ok := evaluated.(object.String)
|
||||||
if !ok {
|
if !ok {
|
||||||
t.Fatalf("object is not String. got=%T (+%v)", evaluated, evaluated)
|
t.Fatalf("object is not String. got=%T (+%v)", evaluated, evaluated)
|
||||||
}
|
}
|
||||||
@@ -495,7 +494,7 @@ func TestBuiltinFunctions(t *testing.T) {
|
|||||||
case string:
|
case string:
|
||||||
testStringObject(t, evaluated, expected)
|
testStringObject(t, evaluated, expected)
|
||||||
case error:
|
case error:
|
||||||
errObj, ok := evaluated.(*object.Error)
|
errObj, ok := evaluated.(object.Error)
|
||||||
if !ok {
|
if !ok {
|
||||||
t.Errorf("object is not Error. got=%T (%+v)",
|
t.Errorf("object is not Error. got=%T (%+v)",
|
||||||
evaluated, evaluated)
|
evaluated, evaluated)
|
||||||
@@ -641,12 +640,12 @@ func TestHashLiterals(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
expected := map[object.HashKey]int64{
|
expected := map[object.HashKey]int64{
|
||||||
(&object.String{Value: "one"}).HashKey(): 1,
|
(object.String{Value: "one"}).HashKey(): 1,
|
||||||
(&object.String{Value: "two"}).HashKey(): 2,
|
(object.String{Value: "two"}).HashKey(): 2,
|
||||||
(&object.String{Value: "three"}).HashKey(): 3,
|
(object.String{Value: "three"}).HashKey(): 3,
|
||||||
(&object.Integer{Value: 4}).HashKey(): 4,
|
(object.Integer{Value: 4}).HashKey(): 4,
|
||||||
TRUE.HashKey(): 5,
|
TRUE.HashKey(): 5,
|
||||||
FALSE.HashKey(): 6,
|
FALSE.HashKey(): 6,
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(result.Pairs) != len(expected) {
|
if len(result.Pairs) != len(expected) {
|
||||||
@@ -672,8 +671,8 @@ func TestHashMerging(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
expected := map[object.HashKey]int64{
|
expected := map[object.HashKey]int64{
|
||||||
(&object.String{Value: "a"}).HashKey(): 1,
|
(object.String{Value: "a"}).HashKey(): 1,
|
||||||
(&object.String{Value: "b"}).HashKey(): 2,
|
(object.String{Value: "b"}).HashKey(): 2,
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(result.Pairs) != len(expected) {
|
if len(result.Pairs) != len(expected) {
|
||||||
@@ -804,7 +803,7 @@ func testEval(input string) object.Object {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func testIntegerObject(t *testing.T, obj object.Object, expected int64) bool {
|
func testIntegerObject(t *testing.T, obj object.Object, expected int64) bool {
|
||||||
result, ok := obj.(*object.Integer)
|
result, ok := obj.(object.Integer)
|
||||||
if !ok {
|
if !ok {
|
||||||
t.Errorf("object is not Integer. got=%T (%+v)", obj, obj)
|
t.Errorf("object is not Integer. got=%T (%+v)", obj, obj)
|
||||||
return false
|
return false
|
||||||
@@ -823,7 +822,7 @@ func TestNullExpression(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func testBooleanObject(t *testing.T, obj object.Object, expected bool) bool {
|
func testBooleanObject(t *testing.T, obj object.Object, expected bool) bool {
|
||||||
result, ok := obj.(*object.Boolean)
|
result, ok := obj.(object.Boolean)
|
||||||
if !ok {
|
if !ok {
|
||||||
t.Errorf("object is not Boolean. got=%T (%+v)", obj, obj)
|
t.Errorf("object is not Boolean. got=%T (%+v)", obj, obj)
|
||||||
return false
|
return false
|
||||||
@@ -845,7 +844,7 @@ func testNullObject(t *testing.T, obj object.Object) bool {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func testStringObject(t *testing.T, obj object.Object, expected string) bool {
|
func testStringObject(t *testing.T, obj object.Object, expected string) bool {
|
||||||
result, ok := obj.(*object.String)
|
result, ok := obj.(object.String)
|
||||||
if !ok {
|
if !ok {
|
||||||
t.Errorf("object is not String. got=%T (%+v)", obj, obj)
|
t.Errorf("object is not String. got=%T (%+v)", obj, obj)
|
||||||
return false
|
return false
|
||||||
|
|||||||
@@ -232,7 +232,7 @@ func (l *Lexer) skipWhitespace() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (l *Lexer) readString() (string, error) {
|
func (l *Lexer) readString() (string, error) {
|
||||||
b := &strings.Builder{}
|
b := strings.Builder{}
|
||||||
for {
|
for {
|
||||||
l.readChar()
|
l.readChar()
|
||||||
if l.ch == '\\' {
|
if l.ch == '\\' {
|
||||||
|
|||||||
@@ -25,7 +25,7 @@ func (ao *Array) PopLeft() Object {
|
|||||||
ao.Elements = ao.Elements[1:]
|
ao.Elements = ao.Elements[1:]
|
||||||
return e
|
return e
|
||||||
}
|
}
|
||||||
return &Null{}
|
return Null{}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ao *Array) PopRight() Object {
|
func (ao *Array) PopRight() Object {
|
||||||
@@ -34,7 +34,7 @@ func (ao *Array) PopRight() Object {
|
|||||||
ao.Elements = ao.Elements[:(len(ao.Elements) - 1)]
|
ao.Elements = ao.Elements[:(len(ao.Elements) - 1)]
|
||||||
return e
|
return e
|
||||||
}
|
}
|
||||||
return &Null{}
|
return Null{}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ao *Array) Prepend(obj Object) {
|
func (ao *Array) Prepend(obj Object) {
|
||||||
@@ -89,7 +89,7 @@ func (ao *Array) Mul(other Object) (Object, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
var elements []Object
|
var elements []Object
|
||||||
N := int(other.(*Integer).Value)
|
N := int(other.(Integer).Value)
|
||||||
for i := 0; i < N; i++ {
|
for i := 0; i < N; i++ {
|
||||||
elements = append(elements, ao.Elements...)
|
elements = append(elements, ao.Elements...)
|
||||||
}
|
}
|
||||||
@@ -102,10 +102,10 @@ func (ao *Array) Get(index Object) (Object, error) {
|
|||||||
return nil, fmt.Errorf("invalid type for array index, expected Integer got %s", index.Type())
|
return nil, fmt.Errorf("invalid type for array index, expected Integer got %s", index.Type())
|
||||||
}
|
}
|
||||||
|
|
||||||
i := index.(*Integer).Value
|
i := index.(Integer).Value
|
||||||
N := int64(len(ao.Elements))
|
N := int64(len(ao.Elements))
|
||||||
if i < 0 || i >= N {
|
if i < 0 || i >= N {
|
||||||
return &Null{}, nil
|
return Null{}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
return ao.Elements[i], nil
|
return ao.Elements[i], nil
|
||||||
@@ -116,7 +116,7 @@ func (ao *Array) Set(index, other Object) error {
|
|||||||
return fmt.Errorf("invalid type for array index, expected Integer got %s", index.Type())
|
return fmt.Errorf("invalid type for array index, expected Integer got %s", index.Type())
|
||||||
}
|
}
|
||||||
|
|
||||||
i := index.(*Integer).Value
|
i := index.(Integer).Value
|
||||||
N := int64(len(ao.Elements))
|
N := int64(len(ao.Elements))
|
||||||
if i < 0 || i >= N {
|
if i < 0 || i >= N {
|
||||||
return fmt.Errorf("index out of bounds %d with array length %d", i, N)
|
return fmt.Errorf("index out of bounds %d with array length %d", i, N)
|
||||||
|
|||||||
@@ -8,53 +8,53 @@ type Boolean struct {
|
|||||||
Value bool
|
Value bool
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Boolean) Bool() bool {
|
func (b Boolean) Bool() bool {
|
||||||
return b.Value
|
return b.Value
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Boolean) Type() Type {
|
func (b Boolean) Type() Type {
|
||||||
return BooleanType
|
return BooleanType
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Boolean) Inspect() string {
|
func (b Boolean) Inspect() string {
|
||||||
return fmt.Sprintf("%t", b.Value)
|
return fmt.Sprintf("%t", b.Value)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Boolean) Clone() Object {
|
func (b Boolean) Clone() Object {
|
||||||
return &Boolean{Value: b.Value}
|
return Boolean{Value: b.Value}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Boolean) String() string {
|
func (b Boolean) String() string {
|
||||||
return b.Inspect()
|
return b.Inspect()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Boolean) Int() int {
|
func (b Boolean) Int() int {
|
||||||
if b.Value {
|
if b.Value {
|
||||||
return 1
|
return 1
|
||||||
}
|
}
|
||||||
return 0
|
return 0
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Boolean) LogicalAnd(other Object) (Object, error) {
|
func (b Boolean) LogicalAnd(other Object) (Object, error) {
|
||||||
if !AssertTypes(other, BooleanType, IntegerType) {
|
if !AssertTypes(other, BooleanType, IntegerType) {
|
||||||
return nil, NewBinaryOpError(b, other, "&&")
|
return nil, NewBinaryOpError(b, other, "&&")
|
||||||
}
|
}
|
||||||
return &Boolean{b.Value && other.Bool()}, nil
|
return Boolean{b.Value && other.Bool()}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Boolean) LogicalOr(other Object) (Object, error) {
|
func (b Boolean) LogicalOr(other Object) (Object, error) {
|
||||||
if !AssertTypes(other, BooleanType, IntegerType) {
|
if !AssertTypes(other, BooleanType, IntegerType) {
|
||||||
return nil, NewBinaryOpError(b, other, "||")
|
return nil, NewBinaryOpError(b, other, "||")
|
||||||
}
|
}
|
||||||
return &Boolean{b.Value || other.Bool()}, nil
|
return Boolean{b.Value || other.Bool()}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Boolean) LogicalNot() Object {
|
func (b Boolean) LogicalNot() Object {
|
||||||
return &Boolean{!b.Value}
|
return Boolean{!b.Value}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Boolean) Compare(other Object) int {
|
func (b Boolean) Compare(other Object) int {
|
||||||
if obj, ok := other.(*Boolean); ok {
|
if obj, ok := other.(Boolean); ok {
|
||||||
return b.Int() - obj.Int()
|
return b.Int() - obj.Int()
|
||||||
}
|
}
|
||||||
return 1
|
return 1
|
||||||
|
|||||||
@@ -39,7 +39,7 @@ func (e *Environment) ExportedHash() *Hash {
|
|||||||
pairs := make(map[HashKey]HashPair)
|
pairs := make(map[HashKey]HashPair)
|
||||||
for k, v := range e.store {
|
for k, v := range e.store {
|
||||||
if unicode.IsUpper(rune(k[0])) {
|
if unicode.IsUpper(rune(k[0])) {
|
||||||
s := &String{Value: k}
|
s := String{Value: k}
|
||||||
pairs[s.HashKey()] = HashPair{Key: s, Value: v}
|
pairs[s.HashKey()] = HashPair{Key: s, Value: v}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -6,22 +6,22 @@ type Error struct {
|
|||||||
Message string
|
Message string
|
||||||
}
|
}
|
||||||
|
|
||||||
func (e *Error) Bool() bool {
|
func (e Error) Bool() bool {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
func (e *Error) Type() Type {
|
func (e Error) Type() Type {
|
||||||
return ErrorType
|
return ErrorType
|
||||||
}
|
}
|
||||||
|
|
||||||
func (e *Error) Inspect() string {
|
func (e Error) Inspect() string {
|
||||||
return "Error: " + e.Message
|
return "Error: " + e.Message
|
||||||
}
|
}
|
||||||
|
|
||||||
func (e *Error) Clone() Object {
|
func (e Error) Clone() Object {
|
||||||
return &Error{Message: e.Message}
|
return Error{Message: e.Message}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (e *Error) String() string {
|
func (e Error) String() string {
|
||||||
return e.Message
|
return e.Message
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -12,15 +12,15 @@ type Function struct {
|
|||||||
Env *Environment
|
Env *Environment
|
||||||
}
|
}
|
||||||
|
|
||||||
func (f *Function) Bool() bool {
|
func (f Function) Bool() bool {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
func (f *Function) Type() Type {
|
func (f Function) Type() Type {
|
||||||
return FunctionType
|
return FunctionType
|
||||||
}
|
}
|
||||||
|
|
||||||
func (f *Function) Inspect() string {
|
func (f Function) Inspect() string {
|
||||||
var out bytes.Buffer
|
var out bytes.Buffer
|
||||||
|
|
||||||
params := []string{}
|
params := []string{}
|
||||||
@@ -38,7 +38,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()
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -46,18 +46,18 @@ type ReturnValue struct {
|
|||||||
Value Object
|
Value Object
|
||||||
}
|
}
|
||||||
|
|
||||||
func (rv *ReturnValue) Bool() bool {
|
func (rv ReturnValue) Bool() bool {
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
func (rv *ReturnValue) Type() Type {
|
func (rv ReturnValue) Type() Type {
|
||||||
return ReturnType
|
return ReturnType
|
||||||
}
|
}
|
||||||
|
|
||||||
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()
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -12,7 +12,7 @@ type HashKey struct {
|
|||||||
Value uint64
|
Value uint64
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Boolean) HashKey() HashKey {
|
func (b Boolean) HashKey() HashKey {
|
||||||
var value uint64
|
var value uint64
|
||||||
|
|
||||||
if b.Value {
|
if b.Value {
|
||||||
@@ -24,11 +24,11 @@ func (b *Boolean) HashKey() HashKey {
|
|||||||
return HashKey{Type: b.Type(), Value: value}
|
return HashKey{Type: b.Type(), Value: value}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (i *Integer) HashKey() HashKey {
|
func (i Integer) HashKey() HashKey {
|
||||||
return HashKey{Type: i.Type(), Value: uint64(i.Value)}
|
return HashKey{Type: i.Type(), Value: uint64(i.Value)}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *String) HashKey() HashKey {
|
func (s String) HashKey() HashKey {
|
||||||
h := fnv.New64a()
|
h := fnv.New64a()
|
||||||
h.Write([]byte(s.Value))
|
h.Write([]byte(s.Value))
|
||||||
|
|
||||||
@@ -98,7 +98,7 @@ func (h *Hash) Get(index Object) (Object, error) {
|
|||||||
|
|
||||||
pair, found := h.Pairs[key.HashKey()]
|
pair, found := h.Pairs[key.HashKey()]
|
||||||
if !found {
|
if !found {
|
||||||
return &Null{}, nil
|
return Null{}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
return pair.Value, nil
|
return pair.Value, nil
|
||||||
|
|||||||
@@ -6,44 +6,44 @@ type Integer struct {
|
|||||||
Value int64
|
Value int64
|
||||||
}
|
}
|
||||||
|
|
||||||
func (i *Integer) Bool() bool {
|
func (i Integer) Bool() bool {
|
||||||
return i.Value != 0
|
return i.Value != 0
|
||||||
}
|
}
|
||||||
|
|
||||||
func (i *Integer) Type() Type {
|
func (i Integer) Type() Type {
|
||||||
return IntegerType
|
return IntegerType
|
||||||
}
|
}
|
||||||
|
|
||||||
func (i *Integer) Inspect() string {
|
func (i Integer) Inspect() string {
|
||||||
return fmt.Sprintf("%d", i.Value)
|
return fmt.Sprintf("%d", i.Value)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (i *Integer) Clone() Object {
|
func (i Integer) Clone() Object {
|
||||||
return &Integer{Value: i.Value}
|
return Integer{Value: i.Value}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (i *Integer) String() string {
|
func (i Integer) String() string {
|
||||||
return i.Inspect()
|
return i.Inspect()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (i *Integer) Add(other Object) (Object, error) {
|
func (i Integer) Add(other Object) (Object, error) {
|
||||||
if !AssertTypes(other, IntegerType) {
|
if !AssertTypes(other, IntegerType) {
|
||||||
return nil, NewBinaryOpError(i, other, "+")
|
return nil, NewBinaryOpError(i, other, "+")
|
||||||
}
|
}
|
||||||
return &Integer{i.Value + other.(*Integer).Value}, nil
|
return Integer{i.Value + other.(Integer).Value}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (i *Integer) Sub(other Object) (Object, error) {
|
func (i Integer) Sub(other Object) (Object, error) {
|
||||||
if !AssertTypes(other, IntegerType) {
|
if !AssertTypes(other, IntegerType) {
|
||||||
return nil, NewBinaryOpError(i, other, "-")
|
return nil, NewBinaryOpError(i, other, "-")
|
||||||
}
|
}
|
||||||
return &Integer{i.Value - other.(*Integer).Value}, nil
|
return Integer{i.Value - other.(Integer).Value}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (i *Integer) Mul(other Object) (Object, error) {
|
func (i Integer) Mul(other Object) (Object, error) {
|
||||||
switch other.Type() {
|
switch other.Type() {
|
||||||
case IntegerType:
|
case IntegerType:
|
||||||
return &Integer{i.Value * other.(*Integer).Value}, nil
|
return Integer{i.Value * other.(Integer).Value}, nil
|
||||||
case StringType:
|
case StringType:
|
||||||
return other.(Mul).Mul(i)
|
return other.(Mul).Mul(i)
|
||||||
case ArrayType:
|
case ArrayType:
|
||||||
@@ -53,69 +53,69 @@ func (i *Integer) Mul(other Object) (Object, error) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (i *Integer) Div(other Object) (Object, error) {
|
func (i Integer) Div(other Object) (Object, error) {
|
||||||
if !AssertTypes(other, IntegerType) {
|
if !AssertTypes(other, IntegerType) {
|
||||||
return nil, NewBinaryOpError(i, other, "/")
|
return nil, NewBinaryOpError(i, other, "/")
|
||||||
}
|
}
|
||||||
return &Integer{i.Value / other.(*Integer).Value}, nil
|
return Integer{i.Value / other.(Integer).Value}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (i *Integer) Mod(other Object) (Object, error) {
|
func (i Integer) Mod(other Object) (Object, error) {
|
||||||
if !AssertTypes(other, IntegerType) {
|
if !AssertTypes(other, IntegerType) {
|
||||||
return nil, NewBinaryOpError(i, other, "%")
|
return nil, NewBinaryOpError(i, other, "%")
|
||||||
}
|
}
|
||||||
return &Integer{i.Value % other.(*Integer).Value}, nil
|
return Integer{i.Value % other.(Integer).Value}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (i *Integer) BitwiseOr(other Object) (Object, error) {
|
func (i Integer) BitwiseOr(other Object) (Object, error) {
|
||||||
if !AssertTypes(other, IntegerType) {
|
if !AssertTypes(other, IntegerType) {
|
||||||
return nil, NewBinaryOpError(i, other, "|")
|
return nil, NewBinaryOpError(i, other, "|")
|
||||||
}
|
}
|
||||||
return &Integer{i.Value | other.(*Integer).Value}, nil
|
return Integer{i.Value | other.(Integer).Value}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (i *Integer) BitwiseXor(other Object) (Object, error) {
|
func (i Integer) BitwiseXor(other Object) (Object, error) {
|
||||||
if !AssertTypes(other, IntegerType) {
|
if !AssertTypes(other, IntegerType) {
|
||||||
return nil, NewBinaryOpError(i, other, "^")
|
return nil, NewBinaryOpError(i, other, "^")
|
||||||
}
|
}
|
||||||
return &Integer{i.Value ^ other.(*Integer).Value}, nil
|
return Integer{i.Value ^ other.(Integer).Value}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (i *Integer) BitwiseAnd(other Object) (Object, error) {
|
func (i Integer) BitwiseAnd(other Object) (Object, error) {
|
||||||
if !AssertTypes(other, IntegerType) {
|
if !AssertTypes(other, IntegerType) {
|
||||||
return nil, NewBinaryOpError(i, other, "&")
|
return nil, NewBinaryOpError(i, other, "")
|
||||||
}
|
}
|
||||||
return &Integer{i.Value & other.(*Integer).Value}, nil
|
return Integer{i.Value & other.(Integer).Value}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (i *Integer) BitwiseNot() Object {
|
func (i Integer) BitwiseNot() Object {
|
||||||
return &Integer{^i.Value}
|
return Integer{^i.Value}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (i *Integer) LeftShift(other Object) (Object, error) {
|
func (i Integer) LeftShift(other Object) (Object, error) {
|
||||||
if !AssertTypes(other, IntegerType) {
|
if !AssertTypes(other, IntegerType) {
|
||||||
return nil, NewBinaryOpError(i, other, "<<")
|
return nil, NewBinaryOpError(i, other, "<<")
|
||||||
}
|
}
|
||||||
return &Integer{i.Value << other.(*Integer).Value}, nil
|
return Integer{i.Value << other.(Integer).Value}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (i *Integer) RightShift(other Object) (Object, error) {
|
func (i Integer) RightShift(other Object) (Object, error) {
|
||||||
if !AssertTypes(other, IntegerType) {
|
if !AssertTypes(other, IntegerType) {
|
||||||
return nil, NewBinaryOpError(i, other, ">>")
|
return nil, NewBinaryOpError(i, other, ">>")
|
||||||
}
|
}
|
||||||
return &Integer{i.Value >> other.(*Integer).Value}, nil
|
return Integer{i.Value >> other.(Integer).Value}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (i *Integer) LogicalNot() Object {
|
func (i Integer) LogicalNot() Object {
|
||||||
return &Boolean{!i.Bool()}
|
return Boolean{!i.Bool()}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (i *Integer) Negate() Object {
|
func (i Integer) Negate() Object {
|
||||||
return &Integer{-i.Value}
|
return Integer{-i.Value}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (i *Integer) Compare(other Object) int {
|
func (i Integer) Compare(other Object) int {
|
||||||
if obj, ok := other.(*Integer); ok {
|
if obj, ok := other.(Integer); ok {
|
||||||
switch {
|
switch {
|
||||||
case i.Value < obj.Value:
|
case i.Value < obj.Value:
|
||||||
return -1
|
return -1
|
||||||
|
|||||||
@@ -24,7 +24,7 @@ func (m Module) Inspect() string {
|
|||||||
return fmt.Sprintf("<module '%s'>", m.Name)
|
return fmt.Sprintf("<module '%s'>", m.Name)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *Module) Get(index Object) (Object, error) {
|
func (m Module) Get(index Object) (Object, error) {
|
||||||
key, ok := index.(Hashable)
|
key, ok := index.(Hashable)
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, fmt.Errorf("invalid module attribute %s", index.Type())
|
return nil, fmt.Errorf("invalid module attribute %s", index.Type())
|
||||||
@@ -32,7 +32,7 @@ func (m *Module) Get(index Object) (Object, error) {
|
|||||||
|
|
||||||
attr, found := m.Attrs.(*Hash).Pairs[key.HashKey()]
|
attr, found := m.Attrs.(*Hash).Pairs[key.HashKey()]
|
||||||
if !found {
|
if !found {
|
||||||
return &Null{}, nil
|
return Null{}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
return attr.Value, nil
|
return attr.Value, nil
|
||||||
|
|||||||
@@ -2,28 +2,28 @@ package object
|
|||||||
|
|
||||||
type Null struct{}
|
type Null struct{}
|
||||||
|
|
||||||
func (n *Null) Bool() bool {
|
func (n Null) Bool() bool {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
func (n *Null) Type() Type {
|
func (n Null) Type() Type {
|
||||||
return NullType
|
return NullType
|
||||||
}
|
}
|
||||||
|
|
||||||
func (n *Null) Inspect() string {
|
func (n Null) Inspect() string {
|
||||||
return "null"
|
return "null"
|
||||||
}
|
}
|
||||||
|
|
||||||
func (n *Null) String() string {
|
func (n Null) String() string {
|
||||||
return n.Inspect()
|
return n.Inspect()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (n *Null) LogicalNot() Object {
|
func (n Null) LogicalNot() Object {
|
||||||
return &Boolean{!n.Bool()}
|
return Boolean{!n.Bool()}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (n *Null) Compare(other Object) int {
|
func (n Null) Compare(other Object) int {
|
||||||
if _, ok := other.(*Null); ok {
|
if _, ok := other.(Null); ok {
|
||||||
return 0
|
return 0
|
||||||
}
|
}
|
||||||
return 1
|
return 1
|
||||||
|
|||||||
@@ -3,10 +3,10 @@ package object
|
|||||||
import "testing"
|
import "testing"
|
||||||
|
|
||||||
func TestStringHashKey(t *testing.T) {
|
func TestStringHashKey(t *testing.T) {
|
||||||
hello1 := &String{Value: "Hello World"}
|
hello1 := String{Value: "Hello World"}
|
||||||
hello2 := &String{Value: "Hello World"}
|
hello2 := String{Value: "Hello World"}
|
||||||
diff1 := &String{Value: "My name is johnny"}
|
diff1 := String{Value: "My name is johnny"}
|
||||||
diff2 := &String{Value: "My name is johnny"}
|
diff2 := String{Value: "My name is johnny"}
|
||||||
|
|
||||||
if hello1.HashKey() != hello2.HashKey() {
|
if hello1.HashKey() != hello2.HashKey() {
|
||||||
t.Errorf("string with same content have different hash keys")
|
t.Errorf("string with same content have different hash keys")
|
||||||
|
|||||||
@@ -10,59 +10,59 @@ type String struct {
|
|||||||
Value string
|
Value string
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *String) Len() int {
|
func (s String) Len() int {
|
||||||
return utf8.RuneCountInString(s.Value)
|
return utf8.RuneCountInString(s.Value)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *String) Bool() bool {
|
func (s String) Bool() bool {
|
||||||
return s.Value != ""
|
return s.Value != ""
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *String) Type() Type {
|
func (s String) Type() Type {
|
||||||
return StringType
|
return StringType
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *String) Inspect() string {
|
func (s String) Inspect() string {
|
||||||
return fmt.Sprintf("%#v", s.Value)
|
return fmt.Sprintf("%#v", s.Value)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *String) Clone() Object {
|
func (s String) Clone() Object {
|
||||||
return &String{Value: s.Value}
|
return String{Value: s.Value}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *String) String() string {
|
func (s String) String() string {
|
||||||
return s.Value
|
return s.Value
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *String) Add(other Object) (Object, error) {
|
func (s String) Add(other Object) (Object, error) {
|
||||||
if !AssertTypes(other, StringType) {
|
if !AssertTypes(other, StringType) {
|
||||||
return nil, NewBinaryOpError(s, other, "+")
|
return nil, NewBinaryOpError(s, other, "+")
|
||||||
}
|
}
|
||||||
return &String{s.Value + other.(*String).Value}, nil
|
return String{s.Value + other.(String).Value}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *String) Mul(other Object) (Object, error) {
|
func (s String) Mul(other Object) (Object, error) {
|
||||||
if !AssertTypes(other, IntegerType) {
|
if !AssertTypes(other, IntegerType) {
|
||||||
return nil, NewBinaryOpError(s, other, "*")
|
return nil, NewBinaryOpError(s, other, "*")
|
||||||
}
|
}
|
||||||
return &String{strings.Repeat(s.Value, int(other.(*Integer).Value))}, nil
|
return String{strings.Repeat(s.Value, int(other.(Integer).Value))}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *String) Get(index Object) (Object, error) {
|
func (s String) Get(index Object) (Object, error) {
|
||||||
if !AssertTypes(index, IntegerType) {
|
if !AssertTypes(index, IntegerType) {
|
||||||
return nil, fmt.Errorf("invalid type for string index, expected Integer got %s", index.Type())
|
return nil, fmt.Errorf("invalid type for string index, expected Integer got %s", index.Type())
|
||||||
}
|
}
|
||||||
|
|
||||||
i := int(index.(*Integer).Value)
|
i := int(index.(Integer).Value)
|
||||||
if i < 0 || i >= len(s.Value) {
|
if i < 0 || i >= len(s.Value) {
|
||||||
return &String{}, nil
|
return String{}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
return &String{string(s.Value[i])}, nil
|
return String{string(s.Value[i])}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *String) Compare(other Object) int {
|
func (s String) Compare(other Object) int {
|
||||||
if obj, ok := other.(*String); ok {
|
if obj, ok := other.(String); ok {
|
||||||
switch {
|
switch {
|
||||||
case s.Value < obj.Value:
|
case s.Value < obj.Value:
|
||||||
return -1
|
return -1
|
||||||
|
|||||||
@@ -4,5 +4,5 @@ import "monkey/internal/object"
|
|||||||
|
|
||||||
// Hello ...
|
// Hello ...
|
||||||
func Hello(args ...object.Object) object.Object {
|
func Hello(args ...object.Object) object.Object {
|
||||||
return &object.String{Value: "Hello World!"}
|
return object.String{Value: "Hello World!"}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -11,23 +11,39 @@ type Frame struct {
|
|||||||
basePointer int
|
basePointer int
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewFrame(cl *object.Closure, basePointer int) *Frame {
|
func NewFrame(cl *object.Closure, basePointer int) Frame {
|
||||||
return &Frame{
|
return Frame{
|
||||||
cl: cl,
|
cl: cl,
|
||||||
basePointer: basePointer,
|
basePointer: basePointer,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (f *Frame) Closure() *object.Closure {
|
||||||
|
return f.cl
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Frame) GetFree(idx uint8) object.Object {
|
||||||
|
return f.cl.Free[idx]
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Frame) SetFree(idx uint8, obj object.Object) {
|
||||||
|
f.cl.Free[idx] = obj
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Frame) SetIP(ip int) {
|
||||||
|
f.ip = ip
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f Frame) PeekNextOp() code.Opcode {
|
||||||
|
return code.Opcode(f.cl.Fn.Instructions[f.ip])
|
||||||
|
}
|
||||||
|
|
||||||
func (f *Frame) ReadNextOp() code.Opcode {
|
func (f *Frame) ReadNextOp() code.Opcode {
|
||||||
op := code.Opcode(f.cl.Fn.Instructions[f.ip])
|
op := code.Opcode(f.cl.Fn.Instructions[f.ip])
|
||||||
f.ip++
|
f.ip++
|
||||||
return op
|
return op
|
||||||
}
|
}
|
||||||
|
|
||||||
func (f *Frame) PeekNextOp() code.Opcode {
|
|
||||||
return code.Opcode(f.cl.Fn.Instructions[f.ip])
|
|
||||||
}
|
|
||||||
|
|
||||||
func (f *Frame) ReadUint8() uint8 {
|
func (f *Frame) ReadUint8() uint8 {
|
||||||
n := code.ReadUint8(f.cl.Fn.Instructions[f.ip:])
|
n := code.ReadUint8(f.cl.Fn.Instructions[f.ip:])
|
||||||
f.ip++
|
f.ip++
|
||||||
@@ -40,6 +56,6 @@ func (f *Frame) ReadUint16() uint16 {
|
|||||||
return n
|
return n
|
||||||
}
|
}
|
||||||
|
|
||||||
func (f *Frame) Instructions() code.Instructions {
|
func (f Frame) Instructions() code.Instructions {
|
||||||
return f.cl.Fn.Instructions
|
return f.cl.Fn.Instructions
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -20,17 +20,17 @@ const StackSize = 2048
|
|||||||
const GlobalsSize = 65536
|
const GlobalsSize = 65536
|
||||||
const MaxFrames = 1024
|
const MaxFrames = 1024
|
||||||
|
|
||||||
var Null = &object.Null{}
|
var Null = object.Null{}
|
||||||
var True = &object.Boolean{Value: true}
|
var True = object.Boolean{Value: true}
|
||||||
var False = &object.Boolean{Value: false}
|
var False = object.Boolean{Value: false}
|
||||||
|
|
||||||
func isTruthy(obj object.Object) bool {
|
func isTruthy(obj object.Object) bool {
|
||||||
switch obj := obj.(type) {
|
switch obj := obj.(type) {
|
||||||
|
|
||||||
case *object.Boolean:
|
case object.Boolean:
|
||||||
return obj.Value
|
return obj.Value
|
||||||
|
|
||||||
case *object.Null:
|
case object.Null:
|
||||||
return false
|
return false
|
||||||
|
|
||||||
default:
|
default:
|
||||||
@@ -38,7 +38,7 @@ func isTruthy(obj object.Object) bool {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// executeModule compiles the named module and returns a *object.Module object
|
// executeModule compiles the named module and returns a object.Module object
|
||||||
func executeModule(name string, state *VMState) (object.Object, error) {
|
func executeModule(name string, state *VMState) (object.Object, error) {
|
||||||
filename := utils.FindModule(name)
|
filename := utils.FindModule(name)
|
||||||
if filename == "" {
|
if filename == "" {
|
||||||
@@ -104,7 +104,7 @@ func (s *VMState) ExportedHash() *object.Hash {
|
|||||||
if unicode.IsUpper(rune(name[0])) {
|
if unicode.IsUpper(rune(name[0])) {
|
||||||
if symbol.Scope == compiler.GlobalScope {
|
if symbol.Scope == compiler.GlobalScope {
|
||||||
obj := s.Globals[symbol.Index]
|
obj := s.Globals[symbol.Index]
|
||||||
s := &object.String{Value: name}
|
s := object.String{Value: name}
|
||||||
pairs[s.HashKey()] = object.HashPair{Key: s, Value: obj}
|
pairs[s.HashKey()] = object.HashPair{Key: s, Value: obj}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -123,20 +123,21 @@ type VM struct {
|
|||||||
stack []object.Object
|
stack []object.Object
|
||||||
sp int // Always points to the next value. Top of stack is stack[sp-1]
|
sp int // Always points to the next value. Top of stack is stack[sp-1]
|
||||||
|
|
||||||
frames []*Frame
|
frames []Frame
|
||||||
frame *Frame // Current frame or nil
|
fp int // Always points to the current frame. Current frame is frames[fp-1]
|
||||||
fp int // Always points to the current frame. Current frame is frames[fp-1]
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (vm *VM) pushFrame(f *Frame) {
|
func (vm *VM) currentFrame() *Frame {
|
||||||
vm.frame = f
|
return &vm.frames[vm.fp-1]
|
||||||
|
}
|
||||||
|
|
||||||
|
func (vm *VM) pushFrame(f Frame) {
|
||||||
vm.frames[vm.fp] = f
|
vm.frames[vm.fp] = f
|
||||||
vm.fp++
|
vm.fp++
|
||||||
}
|
}
|
||||||
|
|
||||||
func (vm *VM) popFrame() *Frame {
|
func (vm *VM) popFrame() Frame {
|
||||||
vm.fp--
|
vm.fp--
|
||||||
vm.frame = vm.frames[vm.fp-1]
|
|
||||||
return vm.frames[vm.fp]
|
return vm.frames[vm.fp]
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -146,7 +147,7 @@ func New(fn string, bytecode *compiler.Bytecode) *VM {
|
|||||||
mainClosure := &object.Closure{Fn: mainFn}
|
mainClosure := &object.Closure{Fn: mainFn}
|
||||||
mainFrame := NewFrame(mainClosure, 0)
|
mainFrame := NewFrame(mainClosure, 0)
|
||||||
|
|
||||||
frames := make([]*Frame, MaxFrames)
|
frames := make([]Frame, MaxFrames)
|
||||||
frames[0] = mainFrame
|
frames[0] = mainFrame
|
||||||
|
|
||||||
state := NewVMState()
|
state := NewVMState()
|
||||||
@@ -159,7 +160,6 @@ func New(fn string, bytecode *compiler.Bytecode) *VM {
|
|||||||
sp: 0,
|
sp: 0,
|
||||||
|
|
||||||
frames: frames,
|
frames: frames,
|
||||||
frame: mainFrame,
|
|
||||||
fp: 1,
|
fp: 1,
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -173,14 +173,13 @@ func NewWithState(fn string, bytecode *compiler.Bytecode, state *VMState) *VM {
|
|||||||
mainClosure := &object.Closure{Fn: mainFn}
|
mainClosure := &object.Closure{Fn: mainFn}
|
||||||
mainFrame := NewFrame(mainClosure, 0)
|
mainFrame := NewFrame(mainClosure, 0)
|
||||||
|
|
||||||
frames := make([]*Frame, MaxFrames)
|
frames := make([]Frame, MaxFrames)
|
||||||
frames[0] = mainFrame
|
frames[0] = mainFrame
|
||||||
|
|
||||||
vm := &VM{
|
vm := &VM{
|
||||||
state: state,
|
state: state,
|
||||||
|
|
||||||
frames: frames,
|
frames: frames,
|
||||||
frame: mainFrame,
|
|
||||||
fp: 1,
|
fp: 1,
|
||||||
|
|
||||||
stack: make([]object.Object, StackSize),
|
stack: make([]object.Object, StackSize),
|
||||||
@@ -210,30 +209,30 @@ func (vm *VM) pop() object.Object {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (vm *VM) executeLoadBuiltin() error {
|
func (vm *VM) executeLoadBuiltin() error {
|
||||||
builtinIndex := vm.frame.ReadUint8()
|
builtinIndex := vm.currentFrame().ReadUint8()
|
||||||
builtin := builtins.BuiltinsIndex[builtinIndex]
|
builtin := builtins.BuiltinsIndex[builtinIndex]
|
||||||
return vm.push(builtin)
|
return vm.push(builtin)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (vm *VM) executeConstant() error {
|
func (vm *VM) executeConstant() error {
|
||||||
constIndex := vm.frame.ReadUint16()
|
constIndex := vm.currentFrame().ReadUint16()
|
||||||
return vm.push(vm.state.Constants[constIndex])
|
return vm.push(vm.state.Constants[constIndex])
|
||||||
}
|
}
|
||||||
|
|
||||||
func (vm *VM) executeAssignGlobal() error {
|
func (vm *VM) executeAssignGlobal() error {
|
||||||
globalIndex := vm.frame.ReadUint16()
|
globalIndex := vm.currentFrame().ReadUint16()
|
||||||
vm.state.Globals[globalIndex] = vm.pop()
|
vm.state.Globals[globalIndex] = vm.pop()
|
||||||
return vm.push(Null)
|
return vm.push(Null)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (vm *VM) executeAssignLocal() error {
|
func (vm *VM) executeAssignLocal() error {
|
||||||
localIndex := vm.frame.ReadUint8()
|
localIndex := vm.currentFrame().ReadUint8()
|
||||||
vm.stack[vm.frame.basePointer+int(localIndex)] = vm.pop()
|
vm.stack[vm.currentFrame().basePointer+int(localIndex)] = vm.pop()
|
||||||
return vm.push(Null)
|
return vm.push(Null)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (vm *VM) executeSetGlobal() error {
|
func (vm *VM) executeSetGlobal() error {
|
||||||
globalIndex := vm.frame.ReadUint16()
|
globalIndex := vm.currentFrame().ReadUint16()
|
||||||
|
|
||||||
ref := vm.pop()
|
ref := vm.pop()
|
||||||
if immutable, ok := ref.(object.Immutable); ok {
|
if immutable, ok := ref.(object.Immutable); ok {
|
||||||
@@ -246,31 +245,31 @@ func (vm *VM) executeSetGlobal() error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (vm *VM) executeGetGlobal() error {
|
func (vm *VM) executeGetGlobal() error {
|
||||||
globalIndex := vm.frame.ReadUint16()
|
globalIndex := vm.currentFrame().ReadUint16()
|
||||||
return vm.push(vm.state.Globals[globalIndex])
|
return vm.push(vm.state.Globals[globalIndex])
|
||||||
}
|
}
|
||||||
|
|
||||||
func (vm *VM) executeSetLocal() error {
|
func (vm *VM) executeSetLocal() error {
|
||||||
localIndex := vm.frame.ReadUint8()
|
localIndex := vm.currentFrame().ReadUint8()
|
||||||
|
|
||||||
ref := vm.pop()
|
ref := vm.pop()
|
||||||
if immutable, ok := ref.(object.Immutable); ok {
|
if immutable, ok := ref.(object.Immutable); ok {
|
||||||
vm.stack[vm.frame.basePointer+int(localIndex)] = immutable.Clone()
|
vm.stack[vm.currentFrame().basePointer+int(localIndex)] = immutable.Clone()
|
||||||
} else {
|
} else {
|
||||||
vm.stack[vm.frame.basePointer+int(localIndex)] = ref
|
vm.stack[vm.currentFrame().basePointer+int(localIndex)] = ref
|
||||||
}
|
}
|
||||||
|
|
||||||
return vm.push(Null)
|
return vm.push(Null)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (vm *VM) executeGetLocal() error {
|
func (vm *VM) executeGetLocal() error {
|
||||||
localIndex := vm.frame.ReadUint8()
|
localIndex := vm.currentFrame().ReadUint8()
|
||||||
return vm.push(vm.stack[vm.frame.basePointer+int(localIndex)])
|
return vm.push(vm.stack[vm.currentFrame().basePointer+int(localIndex)])
|
||||||
}
|
}
|
||||||
|
|
||||||
func (vm *VM) executeGetFree() error {
|
func (vm *VM) executeGetFree() error {
|
||||||
freeIndex := vm.frame.ReadUint8()
|
freeIndex := vm.currentFrame().ReadUint8()
|
||||||
return vm.push(vm.frame.cl.Free[freeIndex])
|
return vm.push(vm.currentFrame().GetFree(freeIndex))
|
||||||
}
|
}
|
||||||
|
|
||||||
func (vm *VM) executeLoadModule() error {
|
func (vm *VM) executeLoadModule() error {
|
||||||
@@ -279,7 +278,7 @@ func (vm *VM) executeLoadModule() error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (vm *VM) executeCurrentClosure() error {
|
func (vm *VM) executeCurrentClosure() error {
|
||||||
currentClosure := vm.frame.cl
|
currentClosure := vm.currentFrame().cl
|
||||||
return vm.push(currentClosure)
|
return vm.push(currentClosure)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -316,7 +315,7 @@ func (vm *VM) buildHash(startIndex, endIndex int) (object.Object, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (vm *VM) executeMakeHash() error {
|
func (vm *VM) executeMakeHash() error {
|
||||||
startIndex := vm.sp - int(vm.frame.ReadUint16())
|
startIndex := vm.sp - int(vm.currentFrame().ReadUint16())
|
||||||
hash, err := vm.buildHash(startIndex, vm.sp)
|
hash, err := vm.buildHash(startIndex, vm.sp)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
@@ -336,7 +335,7 @@ func (vm *VM) buildArray(startIndex, endIndex int) (object.Object, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (vm *VM) executeMakeArray() error {
|
func (vm *VM) executeMakeArray() error {
|
||||||
startIndex := vm.sp - int(vm.frame.ReadUint16())
|
startIndex := vm.sp - int(vm.currentFrame().ReadUint16())
|
||||||
hash, err := vm.buildArray(startIndex, vm.sp)
|
hash, err := vm.buildArray(startIndex, vm.sp)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
@@ -346,8 +345,8 @@ func (vm *VM) executeMakeArray() error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (vm *VM) executeClosure() error {
|
func (vm *VM) executeClosure() error {
|
||||||
constIndex := vm.frame.ReadUint16()
|
constIndex := vm.currentFrame().ReadUint16()
|
||||||
numFree := vm.frame.ReadUint8()
|
numFree := vm.currentFrame().ReadUint8()
|
||||||
return vm.pushClosure(int(constIndex), int(numFree))
|
return vm.pushClosure(int(constIndex), int(numFree))
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -659,7 +658,7 @@ func (vm *VM) executeGetItem() error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (vm *VM) executeCall() error {
|
func (vm *VM) executeCall() error {
|
||||||
args := int(vm.frame.ReadUint8())
|
args := int(vm.currentFrame().ReadUint8())
|
||||||
|
|
||||||
callee := vm.stack[vm.sp-1-args]
|
callee := vm.stack[vm.sp-1-args]
|
||||||
switch callee := callee.(type) {
|
switch callee := callee.(type) {
|
||||||
@@ -685,16 +684,16 @@ func (vm *VM) executeReturn() error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (vm *VM) executeJumpIfFalse() error {
|
func (vm *VM) executeJumpIfFalse() error {
|
||||||
pos := int(vm.frame.ReadUint16())
|
pos := int(vm.currentFrame().ReadUint16())
|
||||||
if !isTruthy(vm.pop()) {
|
if !isTruthy(vm.pop()) {
|
||||||
vm.frame.ip = pos
|
vm.currentFrame().SetIP(pos)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (vm *VM) executeJump() error {
|
func (vm *VM) executeJump() error {
|
||||||
pos := int(vm.frame.ReadUint16())
|
pos := int(vm.currentFrame().ReadUint16())
|
||||||
vm.frame.ip = pos
|
vm.currentFrame().SetIP(pos)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -709,14 +708,14 @@ func (vm *VM) callClosure(cl *object.Closure, numArgs int) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Optimize tail calls and avoid a new frame
|
// Optimize tail calls and avoid a new frame
|
||||||
if cl.Fn == vm.frame.cl.Fn {
|
if cl.Fn == vm.currentFrame().cl.Fn {
|
||||||
nextOP := vm.frame.PeekNextOp()
|
nextOP := vm.currentFrame().PeekNextOp()
|
||||||
if nextOP == code.OpReturn {
|
if nextOP == code.OpReturn {
|
||||||
for p := 0; p < numArgs; p++ {
|
for p := 0; p < numArgs; p++ {
|
||||||
vm.stack[vm.frame.basePointer+p] = vm.stack[vm.sp-numArgs+p]
|
vm.stack[vm.currentFrame().basePointer+p] = vm.stack[vm.sp-numArgs+p]
|
||||||
}
|
}
|
||||||
vm.sp -= numArgs + 1
|
vm.sp -= numArgs + 1
|
||||||
vm.frame.ip = 0 // reset IP to the beginning of the frame
|
vm.currentFrame().SetIP(0) // reset IP to the beginning of the frame
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -767,7 +766,7 @@ func (vm *VM) pushClosure(constIndex, numFree int) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (vm *VM) loadModule(name object.Object) error {
|
func (vm *VM) loadModule(name object.Object) error {
|
||||||
s, ok := name.(*object.String)
|
s, ok := name.(object.String)
|
||||||
if !ok {
|
if !ok {
|
||||||
return fmt.Errorf(
|
return fmt.Errorf(
|
||||||
"TypeError: import() expected argument #1 to be `str` got `%s`",
|
"TypeError: import() expected argument #1 to be `str` got `%s`",
|
||||||
@@ -780,7 +779,7 @@ func (vm *VM) loadModule(name object.Object) error {
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
module := &object.Module{Name: s.Value, Attrs: attrs}
|
module := object.Module{Name: s.Value, Attrs: attrs}
|
||||||
return vm.push(module)
|
return vm.push(module)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -799,16 +798,16 @@ func (vm *VM) Run() (err error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
for err == nil {
|
for err == nil {
|
||||||
op := vm.frame.ReadNextOp()
|
op := vm.currentFrame().ReadNextOp()
|
||||||
|
|
||||||
if vm.Debug {
|
if vm.Debug {
|
||||||
n++
|
n++
|
||||||
log.Printf(
|
log.Printf(
|
||||||
"%-25s %-20s\n",
|
"%-25s %-20s\n",
|
||||||
fmt.Sprintf("%04d %s", vm.frame.ip, op),
|
fmt.Sprintf("%04d %s", vm.currentFrame().ip, op),
|
||||||
fmt.Sprintf(
|
fmt.Sprintf(
|
||||||
"[ip=%02d fp=%02d, sp=%02d]",
|
"[ip=%02d fp=%02d, sp=%02d]",
|
||||||
vm.frame.ip, vm.fp-1, vm.sp,
|
vm.currentFrame().ip, vm.fp-1, vm.sp,
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
@@ -953,7 +952,7 @@ func (vm *VM) Run() (err error) {
|
|||||||
if vm.Debug {
|
if vm.Debug {
|
||||||
log.Printf(
|
log.Printf(
|
||||||
"%-25s [ip=%02d fp=%02d, sp=%02d]",
|
"%-25s [ip=%02d fp=%02d, sp=%02d]",
|
||||||
"", vm.frame.ip, vm.fp-1, vm.sp,
|
"", vm.currentFrame().ip, vm.fp-1, vm.sp,
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -42,7 +42,7 @@ func runVmTests(t *testing.T, tests []vmTestCase) {
|
|||||||
// switch constant := constant.(type) {
|
// switch constant := constant.(type) {
|
||||||
// case *object.CompiledFunction:
|
// case *object.CompiledFunction:
|
||||||
// fmt.Printf(" Instructions:\n%s", constant.Instructions)
|
// fmt.Printf(" Instructions:\n%s", constant.Instructions)
|
||||||
// case *object.Integer:
|
// case object.Integer:
|
||||||
// fmt.Printf(" Value: %d\n", constant.Value)
|
// fmt.Printf(" Value: %d\n", constant.Value)
|
||||||
// }
|
// }
|
||||||
//
|
//
|
||||||
@@ -129,13 +129,13 @@ func testExpectedObject(t *testing.T, expected interface{}, actual object.Object
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
case *object.Null:
|
case object.Null:
|
||||||
if actual != Null {
|
if actual != Null {
|
||||||
t.Errorf("object is not Null: %T (%+v)", actual, actual)
|
t.Errorf("object is not Null: %T (%+v)", actual, actual)
|
||||||
}
|
}
|
||||||
|
|
||||||
case *object.Error:
|
case object.Error:
|
||||||
errObj, ok := actual.(*object.Error)
|
errObj, ok := actual.(object.Error)
|
||||||
if !ok {
|
if !ok {
|
||||||
t.Errorf("object is not Error: %T (%+v)", actual, actual)
|
t.Errorf("object is not Error: %T (%+v)", actual, actual)
|
||||||
return
|
return
|
||||||
@@ -153,7 +153,7 @@ func parse(input string) *ast.Program {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func testIntegerObject(expected int64, actual object.Object) error {
|
func testIntegerObject(expected int64, actual object.Object) error {
|
||||||
result, ok := actual.(*object.Integer)
|
result, ok := actual.(object.Integer)
|
||||||
if !ok {
|
if !ok {
|
||||||
return fmt.Errorf("object is not Integer. got=%T (%+v", actual, actual)
|
return fmt.Errorf("object is not Integer. got=%T (%+v", actual, actual)
|
||||||
}
|
}
|
||||||
@@ -166,7 +166,7 @@ func testIntegerObject(expected int64, actual object.Object) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func testBooleanObject(expected bool, actual object.Object) error {
|
func testBooleanObject(expected bool, actual object.Object) error {
|
||||||
result, ok := actual.(*object.Boolean)
|
result, ok := actual.(object.Boolean)
|
||||||
if !ok {
|
if !ok {
|
||||||
return fmt.Errorf("object is not Boolean. got=%T (%+v", actual, actual)
|
return fmt.Errorf("object is not Boolean. got=%T (%+v", actual, actual)
|
||||||
}
|
}
|
||||||
@@ -179,7 +179,7 @@ func testBooleanObject(expected bool, actual object.Object) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func testStringObject(expected string, actual object.Object) error {
|
func testStringObject(expected string, actual object.Object) error {
|
||||||
result, ok := actual.(*object.String)
|
result, ok := actual.(object.String)
|
||||||
if !ok {
|
if !ok {
|
||||||
return fmt.Errorf("object is not String. got=%T (%+v", actual, actual)
|
return fmt.Errorf("object is not String. got=%T (%+v", actual, actual)
|
||||||
}
|
}
|
||||||
@@ -379,15 +379,15 @@ func TestHashLiterals(t *testing.T) {
|
|||||||
{
|
{
|
||||||
"{1: 2, 2: 3}",
|
"{1: 2, 2: 3}",
|
||||||
map[object.HashKey]int64{
|
map[object.HashKey]int64{
|
||||||
(&object.Integer{Value: 1}).HashKey(): 2,
|
(object.Integer{Value: 1}).HashKey(): 2,
|
||||||
(&object.Integer{Value: 2}).HashKey(): 3,
|
(object.Integer{Value: 2}).HashKey(): 3,
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
"{1 + 1: 2 * 2, 3 + 3: 4 * 4}",
|
"{1 + 1: 2 * 2, 3 + 3: 4 * 4}",
|
||||||
map[object.HashKey]int64{
|
map[object.HashKey]int64{
|
||||||
(&object.Integer{Value: 2}).HashKey(): 4,
|
(object.Integer{Value: 2}).HashKey(): 4,
|
||||||
(&object.Integer{Value: 6}).HashKey(): 16,
|
(object.Integer{Value: 6}).HashKey(): 16,
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
@@ -400,14 +400,14 @@ func TestHashMerging(t *testing.T) {
|
|||||||
{
|
{
|
||||||
`{} + {"a": 1}`,
|
`{} + {"a": 1}`,
|
||||||
map[object.HashKey]int64{
|
map[object.HashKey]int64{
|
||||||
(&object.String{Value: "a"}).HashKey(): 1,
|
(object.String{Value: "a"}).HashKey(): 1,
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
`{"a": 1} + {"b": 2}`,
|
`{"a": 1} + {"b": 2}`,
|
||||||
map[object.HashKey]int64{
|
map[object.HashKey]int64{
|
||||||
(&object.String{Value: "a"}).HashKey(): 1,
|
(object.String{Value: "a"}).HashKey(): 1,
|
||||||
(&object.String{Value: "b"}).HashKey(): 2,
|
(object.String{Value: "b"}).HashKey(): 2,
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user