Extensions

Toplevel objects

grammar/src/class_members.grm

@@ -55,7 +55,11 @@ super
   ;
 
 method
-  : methodModifier* "fun" SimpleName typeParameters? functionParameters (":" type)? functionBody?
+  : methodModifier* "fun" functionRest
+  ;
+
+functionRest
+  : SimpleName typeParameters? functionParameters (":" type)? functionBody?
   ;
 
 functionParameters
@@ -72,7 +76,10 @@ functionBody
   ;
 
 property
-  : propertyModifier* ("val" | "var") SimpleName (":" type)? ("=" expression)?
+  : propertyModifier* ("val" | "var")
+
+propertyRest
+  : SimpleName (":" type)? ("=" expression)?
     "{" getter? setter? "}"
   ;
 

grammar/src/extension.grm

@@ -0,0 +1,17 @@
+extension
+  : completeExtension
+  : extensionMethod
+  : extensionProperty
+  ;
+
+completeExtension
+  : accessModifier? "extension" SimpleName? typeParameters? "for" type classBody?
+  ;
+
+extensionMethod
+  : accessModifier? "extension" "fun" type "." functionRest
+  ;
+
+extensionProperty
+  : accessModifier? "extension" propertyRest
+  ;

grammar/src/toplevel.grm

@@ -0,0 +1,14 @@
+jetlFile
+  : preamble toplevelObject*
+  ;
+
+preamble
+  : empty
+  ;
+
+toplevelObject
+  : class
+  : extension
+  : method
+  : property
+  ;