Add doc comment to sum definition nodes

4 files changed, 4 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/ast/typed/TypedSumDefinitionNode.java b/src/main/java/org/zwobble/hobgoblin/compiler/ast/typed/TypedSumDefinitionNode.java
index e04e2dd..d795430 100644
--- a/src/main/java/org/zwobble/hobgoblin/compiler/ast/typed/TypedSumDefinitionNode.java
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/ast/typed/TypedSumDefinitionNode.java
@@ -8,6 +8,7 @@ import java.util.List;
 public record TypedSumDefinitionNode(
     SumType type,
     List<TypedSumVariantDefinitionNode> variants,
+    String docComment,
     Source source
 ) implements TypedNamespaceStatementNode {
     public String name() {
diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/ast/untyped/UntypedSumDefinitionNode.java b/src/main/java/org/zwobble/hobgoblin/compiler/ast/untyped/UntypedSumDefinitionNode.java
index 66acc08..6ffcb6f 100644
--- a/src/main/java/org/zwobble/hobgoblin/compiler/ast/untyped/UntypedSumDefinitionNode.java
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/ast/untyped/UntypedSumDefinitionNode.java
@@ -7,6 +7,7 @@ import java.util.List;
 public record UntypedSumDefinitionNode(
     String name,
     List<UntypedSumVariantDefinitionNode> variants,
+    String docComment,
     Source source
 ) implements UntypedNamespaceStatementNode {
 }
diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/parser/Parser.java b/src/main/java/org/zwobble/hobgoblin/compiler/parser/Parser.java
index a4b4743..88b0c5c 100644
--- a/src/main/java/org/zwobble/hobgoblin/compiler/parser/Parser.java
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/parser/Parser.java
@@ -130,6 +130,7 @@ public class Parser {
         return new UntypedSumDefinitionNode(
             name.charSequence().toString(),
             variants,
+            "",
             source
         );
     }
diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/typechecker/TypeChecker.java b/src/main/java/org/zwobble/hobgoblin/compiler/typechecker/TypeChecker.java
index 6574b52..012fd2b 100644
--- a/src/main/java/org/zwobble/hobgoblin/compiler/typechecker/TypeChecker.java
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/typechecker/TypeChecker.java
@@ -122,6 +122,7 @@ public class TypeChecker {
         return new TypedSumDefinitionNode(
             sumType,
             typedVariants,
+            untyped.docComment(),
             untyped.source()
         );
     }