Parse doc comment for sum definitions

1 files changed, 4 insertions, 3 deletions

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 49464b6..c50c4d6 100644
--- a/src/main/java/org/zwobble/hobgoblin/compiler/parser/Parser.java
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/parser/Parser.java
@@ -54,7 +54,7 @@ public class Parser {
         if (tokens.isNext(TokenType.KEYWORD_STRUCT)) {
             return parseStructDefinition(tokens, docComment);
         } else if (tokens.isNext(TokenType.KEYWORD_SUM)) {
-            return parseSumDefinition(tokens);
+            return parseSumDefinition(tokens, docComment);
         } else {
             throw new UnsupportedOperationException("TODO");
         }
@@ -130,7 +130,8 @@ public class Parser {
     }
 
     private static UntypedSumDefinitionNode parseSumDefinition(
-        TokenIterator tokens
+        TokenIterator tokens,
+        String docComment
     ) {
         var start = tokens.startPosition();
 
@@ -152,7 +153,7 @@ public class Parser {
         return new UntypedSumDefinitionNode(
             name.charSequence().toString(),
             variants,
-            "",
+            docComment,
             source
         );
     }