Add source to ShidNamespace

2 files changed, 4 insertions, 2 deletions

diff --git a/src/main/java/org/shedlang/compiler/shid/ShidCodeGen.java b/src/main/java/org/shedlang/compiler/shid/ShidCodeGen.java
index cf14766c..89bf275c 100644
--- a/src/main/java/org/shedlang/compiler/shid/ShidCodeGen.java
+++ b/src/main/java/org/shedlang/compiler/shid/ShidCodeGen.java
@@ -484,7 +484,7 @@ public class ShidCodeGen {
         var body = bodyResult.value;
         context = bodyResult.context;
 
-        return new ShidNamespace(node.namespaceName(), body);
+        return new ShidNamespace(node.namespaceName(), body, node.source());
     }
 
     public static Context compileNamespaceStatement(TypedNamespaceStatementNode node, Context context) {
diff --git a/src/main/java/org/shedlang/compiler/shid/ShidNamespace.java b/src/main/java/org/shedlang/compiler/shid/ShidNamespace.java
index d3fb4b0f..50b4cbb4 100644
--- a/src/main/java/org/shedlang/compiler/shid/ShidNamespace.java
+++ b/src/main/java/org/shedlang/compiler/shid/ShidNamespace.java
@@ -1,10 +1,12 @@
 package org.shedlang.compiler.shid;
 
 import org.shedlang.compiler.shid.instructions.ShidBlock;
+import org.shedlang.compiler.sources.Source;
 import org.shedlang.compiler.types.NamespaceName;
 
 public record ShidNamespace(
     NamespaceName namespaceName,
-    ShidBlock body
+    ShidBlock body,
+    Source source
 ) {
 }