Add visibility for structs

1 files changed, 6 insertions, 1 deletions

diff --git a/src/main/java/org/shedlang/compiler/parser/Parser.java b/src/main/java/org/shedlang/compiler/parser/Parser.java
index b66fccb1..0151e7d6 100644
--- a/src/main/java/org/shedlang/compiler/parser/Parser.java
+++ b/src/main/java/org/shedlang/compiler/parser/Parser.java
@@ -509,6 +509,9 @@ public class Parser {
                     case KEYWORD_FUN ->
                         parseFunctionDefinition(tokens);
 
+                    case KEYWORD_STRUCT ->
+                        parseStructDefinition(tokens);
+
                     case KEYWORD_VAL ->
                         parseAssign(tokens);
 
@@ -631,6 +634,8 @@ public class Parser {
     private static UntypedNamespaceStatementNode parseStructDefinition(TokenIterator<TokenType> tokens) {
         var start = tokens.position();
 
+        var visibility = parseVisibility(tokens);
+
         tokens.skip(TokenType.KEYWORD_STRUCT);
 
         var name = tokens.nextValue(TokenType.IDENTIFIER);
@@ -649,7 +654,7 @@ public class Parser {
         var end = tokens.position();
         var source = start.to(end);
 
-        return new UntypedStructDefinitionNode(name, fields, source);
+        return new UntypedStructDefinitionNode(name, visibility, fields, source);
     }
 
     private static UntypedStructFieldDefinitionNode parseStructFieldDefinition(TokenIterator<TokenType> tokens) {