Parse namespace body

2 files changed, 12 insertions, 2 deletions

diff --git a/src/main/java/org/shedlang/compiler/ast/UntypedNamespaceNode.java b/src/main/java/org/shedlang/compiler/ast/UntypedNamespaceNode.java
index 055e1a9e..c0e24f48 100644
--- a/src/main/java/org/shedlang/compiler/ast/UntypedNamespaceNode.java
+++ b/src/main/java/org/shedlang/compiler/ast/UntypedNamespaceNode.java
@@ -2,7 +2,10 @@ package org.shedlang.compiler.ast;
 
 import org.shedlang.compiler.sources.Source;
 
+import java.util.List;
+
 public record UntypedNamespaceNode(
+    List<UntypedNamespaceStatementNode> body,
     Source source
 ) implements UntypedNode {
 }
diff --git a/src/main/java/org/shedlang/compiler/parser/Parser.java b/src/main/java/org/shedlang/compiler/parser/Parser.java
index 5b17e80d..4cdcd7bf 100644
--- a/src/main/java/org/shedlang/compiler/parser/Parser.java
+++ b/src/main/java/org/shedlang/compiler/parser/Parser.java
@@ -242,10 +242,17 @@ public class Parser {
 
     public static UntypedNamespaceNode parseNamespace(TokenIterator<TokenType> tokens) {
         var start = tokens.position();
-        var end = tokens.position();
 
+        var body = parseMany(
+            () -> tokens.isNext(TokenType.END),
+            () -> parseNamespaceStatement(tokens),
+            () -> true
+        );
+
+        var end = tokens.position();
         var source = start.to(end);
-        return new UntypedNamespaceNode(source);
+
+        return new UntypedNamespaceNode(body, source);
     }
 
     private static UntypedExpressionNode parseReference(TokenIterator<TokenType> tokens) {