-
Notifications
You must be signed in to change notification settings - Fork 0
/
builtins.go
109 lines (95 loc) · 2.59 KB
/
builtins.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
package evaluator
import (
"github.com/radlinskii/interpreter/object"
)
var builtins = map[string]*object.Builtin{
"len": {
Fn: func(args ...object.Object) object.Object {
if len(args) != 1 {
return newError("wrong number of arguments. got=%d want=1", len(args))
}
switch arg := args[0].(type) {
case *object.Array:
return &object.Integer{Value: int64(len(arg.Elements))}
case *object.String:
return &object.Integer{Value: int64(len(arg.Value))}
default:
return newError("argument to `len` not supported, got %s", args[0].Type())
}
},
},
"first": {
Fn: func(args ...object.Object) object.Object {
if len(args) != 1 {
return newError("wrong number of arguments. got=%d want=1", len(args))
}
arr, ok := args[0].(*object.Array)
if !ok {
return newError("argument to `first` not supported, got %s", args[0].Type())
}
if len(arr.Elements) > 0 {
return arr.Elements[0]
}
return NULL
},
},
"last": {
Fn: func(args ...object.Object) object.Object {
if len(args) != 1 {
return newError("wrong number of arguments. got=%d want=1", len(args))
}
arr, ok := args[0].(*object.Array)
if !ok {
return newError("argument to `last` not supported, got %s", args[0].Type())
}
length := len(arr.Elements)
if length > 0 {
return arr.Elements[length-1]
}
return NULL
},
},
"rest": {
Fn: func(args ...object.Object) object.Object {
if len(args) != 1 {
return newError("wrong number of arguments. got=%d want=1", len(args))
}
arr, ok := args[0].(*object.Array)
if !ok {
return newError("argument to `rest` not supported, got %s", args[0].Type())
}
length := len(arr.Elements)
if length > 0 {
newElements := make([]object.Object, length-1, length-1)
copy(newElements, arr.Elements[1:length])
return &object.Array{Elements: newElements}
}
return NULL
},
},
"push": {
Fn: func(args ...object.Object) object.Object {
if len(args) != 2 {
return newError("wrong number of arguments. got=%d want=2", len(args))
}
arr, ok := args[0].(*object.Array)
if !ok {
return newError("first argument to `push` not supported, got %s", args[0].Type())
}
length := len(arr.Elements)
newElements := make([]object.Object, length+1, length+1)
copy(newElements, arr.Elements)
newElements[length] = args[1]
return &object.Array{Elements: newElements}
},
},
"print": {
Fn: func(args ...object.Object) object.Object {
for _, arg := range args {
programOutput.WriteString(arg.Inspect() + " ")
}
programOutput.WriteString("\n")
return NULL
},
},
}