Parse methods in extend inline node

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 dba9620d..8d79439e 100644
--- a/src/main/java/org/shedlang/compiler/parser/Parser.java
+++ b/src/main/java/org/shedlang/compiler/parser/Parser.java
@@ -413,12 +413,19 @@ public class Parser {
         tokens.skip(TokenType.KEYWORD_EXTEND);
         var extendedType = parseTopLevelTypeLevelExpression(tokens);
         tokens.skip(TokenType.SYMBOL_BRACE_OPEN);
+
+        var methods = parseMany(
+            () -> tokens.isNext(TokenType.SYMBOL_BRACE_CLOSE),
+            () -> parseFunctionDefinition(tokens),
+            () -> true
+        );
+
         tokens.skip(TokenType.SYMBOL_BRACE_CLOSE);
 
         var end = tokens.position();
         var source = start.to(end);
 
-        return new UntypedExtendInlineNode(extendedType, source);
+        return new UntypedExtendInlineNode(extendedType, methods, source);
     }
 
     private static class FieldAccessParser implements OperationParser<