Only allow declared fields to be set

2 files changed, 2 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/ClassNode.java b/src/main/java/org/zwobble/couscous/ast/ClassNode.java
index 674e347..d0598d8 100644
--- a/src/main/java/org/zwobble/couscous/ast/ClassNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/ClassNode.java
@@ -13,6 +13,7 @@ public class ClassNode implements Node {
     }
     
     TypeName name;
+    List<FieldDeclarationNode> fields;
     ConstructorNode constructor;
     List<MethodNode> methods;
     
diff --git a/src/main/java/org/zwobble/couscous/ast/ClassNodeBuilder.java b/src/main/java/org/zwobble/couscous/ast/ClassNodeBuilder.java
index 2a0d967..b4ddad9 100644
--- a/src/main/java/org/zwobble/couscous/ast/ClassNodeBuilder.java
+++ b/src/main/java/org/zwobble/couscous/ast/ClassNodeBuilder.java
@@ -65,6 +65,7 @@ public class ClassNodeBuilder {
     public ClassNode build() {
         return new ClassNode(
             name,
+            fields.build(),
             constructor.orElse(constructorBuilder().build()),
             methods.build());
     }