Parse effect visibility

2 files changed, 7 insertions, 5 deletions

diff --git a/src/main/java/org/shedlang/compiler/ast/untyped/UntypedEffectDefinitionNode.java b/src/main/java/org/shedlang/compiler/ast/untyped/UntypedEffectDefinitionNode.java
index 66dab2a5..740c7fcb 100644
--- a/src/main/java/org/shedlang/compiler/ast/untyped/UntypedEffectDefinitionNode.java
+++ b/src/main/java/org/shedlang/compiler/ast/untyped/UntypedEffectDefinitionNode.java
@@ -5,10 +5,7 @@ import org.shedlang.compiler.types.Visibility;
 
 public record UntypedEffectDefinitionNode(
     String name,
+    Visibility visibility,
     Source source
 ) implements UntypedNamespaceStatementNode, UntypedVariableDeclarationNode {
-    @Override
-    public Visibility visibility() {
-        return Visibility.PRIVATE;
-    }
 }
diff --git a/src/main/java/org/shedlang/compiler/parser/Parser.java b/src/main/java/org/shedlang/compiler/parser/Parser.java
index 6be82270..3efdc40b 100644
--- a/src/main/java/org/shedlang/compiler/parser/Parser.java
+++ b/src/main/java/org/shedlang/compiler/parser/Parser.java
@@ -206,6 +206,8 @@ public class Parser {
     ) {
         var start = tokens.position();
 
+        var visibility = parseVisibility(tokens);
+
         tokens.skip(TokenType.KEYWORD_EFFECT);
 
         var name = tokens.nextValue(TokenType.IDENTIFIER);
@@ -216,7 +218,7 @@ public class Parser {
         var end = tokens.position();
         var source = start.to(end);
 
-        return new UntypedEffectDefinitionNode(name, source);
+        return new UntypedEffectDefinitionNode(name, visibility, source);
     }
 
     private static final Map<TokenType, OperationParser> OPERATION_PARSERS = Stream.of(
@@ -617,6 +619,9 @@ public class Parser {
 
             case KEYWORD_PUB ->
                 switch (tokens.peek(1).tokenType()) {
+                    case KEYWORD_EFFECT ->
+                        parseEffectDefinition(tokens);
+
                     case KEYWORD_FUN ->
                         parseFunctionDefinition(tokens);