Skip to content

Commit

Permalink
feat: evaluate array expression
Browse files Browse the repository at this point in the history
  • Loading branch information
Aden-Q committed May 13, 2024
1 parent c5c979b commit 20b9bf8
Show file tree
Hide file tree
Showing 2 changed files with 36 additions and 0 deletions.
17 changes: 17 additions & 0 deletions internal/evaluator/evaluator.go
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,8 @@ func (e *evaluator) Eval(node ast.Node) (object.Object, error) {
return booleanConv(node.Value), nil
case *ast.StringExpression:
return object.NewString(node.Value), nil
case *ast.ArrayExpression:
return e.evalArrayExpression(node)
case *ast.IfExpression:
return e.evalIfExpression(node)
case *ast.FuncExpression:
Expand Down Expand Up @@ -124,6 +126,21 @@ func (e *evaluator) evalIdentifierExpression(ie *ast.IdentifierExpression) (obje
return object.NIL, ErrIdentifierNotFound
}

func (e *evaluator) evalArrayExpression(ae *ast.ArrayExpression) (object.Object, error) {
elements := make([]object.Object, 0, len(ae.Elements))

for _, exp := range ae.Elements {
val, err := e.Eval(exp)
if err != nil {
return object.NIL, err
}

elements = append(elements, val)
}

return object.NewArray(elements...), nil
}

func (e *evaluator) evalIfExpression(ie *ast.IfExpression) (object.Object, error) {
condition, err := e.Eval(ie.Condition)
if err != nil {
Expand Down
19 changes: 19 additions & 0 deletions internal/evaluator/evaluator_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -594,6 +594,25 @@ var _ = Describe("Evaluator", func() {
})
})

Context("array object", func() {
It("array expression", func() {
text = `
[1, 2 * 2, true];
`
expectedObject := object.NewArray(object.NewInteger(1), object.NewInteger(4), object.NewBoolean(true))
expectedErrors := []error{}

// parse the program
program, errs = p.ParseProgram(text)
Expect(errs).To(Equal(expectedErrors))

// evaluate the AST tree
obj, err := e.Eval(program)
Expect(err).To(BeNil())
Expect(obj).To(Equal(expectedObject))
})
})

Context("if conditionals", func() {
It("if condition is truthy", func() {
text = `
Expand Down

0 comments on commit 20b9bf8

Please sign in to comment.