Add evaluation of arrays to interpreter

4 files changed, 71 insertions, 14 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/ArrayNode.java b/src/main/java/org/zwobble/couscous/ast/ArrayNode.java
index ac4c284..78e6869 100644
--- a/src/main/java/org/zwobble/couscous/ast/ArrayNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/ArrayNode.java
@@ -20,6 +20,10 @@ public class ArrayNode implements ExpressionNode {
         this.elements = elements;
     }
 
+    public Type getElementType() {
+        return elementType;
+    }
+
     public List<ExpressionNode> getElements() {
         return elements;
     }
diff --git a/src/main/java/org/zwobble/couscous/interpreter/Evaluator.java b/src/main/java/org/zwobble/couscous/interpreter/Evaluator.java
index 1082ed1..8c3d355 100644
--- a/src/main/java/org/zwobble/couscous/interpreter/Evaluator.java
+++ b/src/main/java/org/zwobble/couscous/interpreter/Evaluator.java
@@ -1,18 +1,19 @@
 package org.zwobble.couscous.interpreter;
 
 import org.zwobble.couscous.ast.*;
-import org.zwobble.couscous.types.ScalarType;
 import org.zwobble.couscous.ast.visitors.AssignableExpressionNodeVisitor;
 import org.zwobble.couscous.ast.visitors.ExpressionNodeMapper;
 import org.zwobble.couscous.interpreter.errors.ConditionMustBeBoolean;
 import org.zwobble.couscous.interpreter.errors.InvalidCast;
 import org.zwobble.couscous.interpreter.values.*;
+import org.zwobble.couscous.types.ScalarType;
 import org.zwobble.couscous.types.Types;
 
 import java.util.List;
 import java.util.stream.Collectors;
 
 import static org.zwobble.couscous.types.Types.erasure;
+import static org.zwobble.couscous.util.ExtraLists.eagerMap;
 
 public class Evaluator implements ExpressionNodeMapper<InterpreterValue> {
     public static InterpreterValue eval(Environment environment, ExpressionNode expression) {
@@ -52,8 +53,10 @@ public class Evaluator implements ExpressionNodeMapper<InterpreterValue> {
     }
 
     @Override
-    public InterpreterValue visit(ArrayNode arrayNode) {
-        throw new UnsupportedOperationException();
+    public InterpreterValue visit(ArrayNode array) {
+        return new ArrayInterpreterValue(
+            array.getElementType(),
+            eagerMap(array.getElements(), this::eval));
     }
 
     @Override
diff --git a/src/main/java/org/zwobble/couscous/interpreter/types/IntrinsicInterpreterType.java b/src/main/java/org/zwobble/couscous/interpreter/types/IntrinsicInterpreterType.java
index d69c855..190908b 100644
--- a/src/main/java/org/zwobble/couscous/interpreter/types/IntrinsicInterpreterType.java
+++ b/src/main/java/org/zwobble/couscous/interpreter/types/IntrinsicInterpreterType.java
@@ -22,7 +22,7 @@ import static org.zwobble.couscous.util.ExtraLists.list;
 import static org.zwobble.couscous.util.ExtraSets.set;
 
 public class IntrinsicInterpreterType implements InterpreterType {
-    public static <T> Builder<T> builder(Class<T> interpreterValueType, ScalarType reference) {
+    public static <T> Builder<T> builder(Class<T> interpreterValueType, Type reference) {
         return new Builder<>(interpreterValueType, reference);
     }
 
@@ -40,11 +40,11 @@ public class IntrinsicInterpreterType implements InterpreterType {
         private final ImmutableMap.Builder<MethodSignature, MethodValue> methods = ImmutableMap.builder();
         private final ImmutableMap.Builder<MethodSignature, StaticMethodValue> staticMethods = ImmutableMap.builder();
         private final Class<T> interpreterValueType;
-        private final ScalarType name;
+        private final Type type;
 
-        public Builder(Class<T> interpreterValueType, ScalarType name) {
+        public Builder(Class<T> interpreterValueType, Type type) {
             this.interpreterValueType = interpreterValueType;
-            this.name = name;
+            this.type = type;
         }
 
         public Builder<T> field(String name, ScalarType type) {
@@ -88,7 +88,7 @@ public class IntrinsicInterpreterType implements InterpreterType {
 
         public InterpreterType build() {
             return new IntrinsicInterpreterType(
-                name,
+                type,
                 fields.build(),
                 constructor,
                 methods.build(),
@@ -96,20 +96,20 @@ public class IntrinsicInterpreterType implements InterpreterType {
         }
     }
 
-    private final ScalarType name;
+    private final Type type;
     private final Map<String, FieldDeclarationNode> fields;
     private final MethodValue constructor;
     private final Map<MethodSignature, MethodValue> methods;
     private final Map<MethodSignature, StaticMethodValue> staticMethods;
 
     private IntrinsicInterpreterType(
-        ScalarType name,
+        Type type,
         Map<String, FieldDeclarationNode> fields,
         MethodValue constructor,
         Map<MethodSignature, MethodValue> methods,
         Map<MethodSignature, StaticMethodValue> staticMethods)
     {
-        this.name = name;
+        this.type = type;
         this.fields = fields;
         this.constructor = constructor;
         this.methods = methods;
@@ -117,8 +117,8 @@ public class IntrinsicInterpreterType implements InterpreterType {
     }
 
     @Override
-    public ScalarType getType() {
-        return name;
+    public Type getType() {
+        return type;
     }
 
     @Override
@@ -164,6 +164,6 @@ public class IntrinsicInterpreterType implements InterpreterType {
 
     @Override
     public String toString() {
-        return "IntrinsicInterpreterType<" + name + ">";
+        return "IntrinsicInterpreterType<" + type + ">";
     }
 }
diff --git a/src/main/java/org/zwobble/couscous/interpreter/values/ArrayInterpreterValue.java b/src/main/java/org/zwobble/couscous/interpreter/values/ArrayInterpreterValue.java
new file mode 100644
index 0000000..3a1de58
--- /dev/null
+++ b/src/main/java/org/zwobble/couscous/interpreter/values/ArrayInterpreterValue.java
@@ -0,0 +1,50 @@
+package org.zwobble.couscous.interpreter.values;
+
+import org.zwobble.couscous.interpreter.errors.NoSuchField;
+import org.zwobble.couscous.interpreter.types.InterpreterType;
+import org.zwobble.couscous.interpreter.types.IntrinsicInterpreterType;
+import org.zwobble.couscous.types.Type;
+import org.zwobble.couscous.types.Types;
+import org.zwobble.couscous.values.PrimitiveValue;
+
+import java.util.List;
+import java.util.Optional;
+
+public class ArrayInterpreterValue implements InterpreterValue {
+    private final Type elementType;
+    private final List<InterpreterValue> elements;
+
+    public ArrayInterpreterValue(Type elementType, List<InterpreterValue> elements) {
+        this.elementType = elementType;
+        this.elements = elements;
+    }
+
+    public Type getElementType() {
+        return elementType;
+    }
+
+    public List<InterpreterValue> getElements() {
+        return elements;
+    }
+
+    @Override
+    public InterpreterType getType() {
+        return IntrinsicInterpreterType.builder(ArrayInterpreterValue.class, Types.array(elementType))
+            .build();
+    }
+
+    @Override
+    public Optional<PrimitiveValue> toPrimitiveValue() {
+        return Optional.empty();
+    }
+
+    @Override
+    public InterpreterValue getField(String fieldName) {
+        throw new NoSuchField(fieldName);
+    }
+
+    @Override
+    public void setField(String fieldName, InterpreterValue value) {
+        throw new NoSuchField(fieldName);
+    }
+}