Parse effect in function definition

1 files changed, 8 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 f76dfd78..1972163a 100644
--- a/src/main/java/org/shedlang/compiler/parser/Parser.java
+++ b/src/main/java/org/shedlang/compiler/parser/Parser.java
@@ -91,6 +91,13 @@ public class Parser {
         var params = parseParams(tokens);
         tokens.skip(TokenType.SYMBOL_PAREN_CLOSE);
 
+        Optional<UntypedTypeLevelExpressionNode> effect;
+        if (tokens.trySkip(TokenType.SYMBOL_BANG)) {
+            effect = Optional.of(parseTopLevelTypeLevelExpression(tokens));
+        } else {
+            effect = Optional.empty();
+        }
+
         tokens.skip(TokenType.SYMBOL_THIN_ARROW);
         var returnType = parseTopLevelTypeLevelExpression(tokens);
 
@@ -104,7 +111,7 @@ public class Parser {
         return new UntypedFunctionDefinitionNode(
             name,
             params,
-            Optional.empty(),
+            effect,
             returnType,
             body,
             source