Merge pull request #157 from aspeddro/decorated-call-function

feat: support decorated expressions

Author: nkrkv

grammar.js

@@ -143,7 +143,10 @@ module.exports = grammar({
 
     _decorated_statement: $ => seq(
       repeat1($.decorator),
-      $.declaration,
+      choice(
+        $.declaration,
+        $.expression_statement
+      )
     ),
 
     decorator_statement: $ => seq(

test/corpus/decorators.txt

@@ -33,6 +33,9 @@ let foo = @doesNotRaise String.make(12, ' ')
 
 let onResult = () => @doesNotRaise Belt.Array.getExn([], 0)
 
+@local
+call()
+
 ---
 
 (source_file
@@ -72,7 +75,13 @@ let onResult = () => @doesNotRaise Belt.Array.getExn([], 0)
           (value_identifier))
         (arguments
           (array)
-          (number))))))
+          (number)))))
+
+  (decorated
+    (decorator (decorator_identifier))
+    (expression_statement
+      (call_expression
+        (value_identifier) (arguments)))))
 
 ============================================
 Decorator before type `and`