support decorator - take2

Author: aspeddro

grammar.js

@@ -91,6 +91,7 @@ module.exports = grammar({
     [$.parameter, $._pattern],
     [$.parameter, $._parenthesized_pattern],
     [$._switch_value_pattern, $._parenthesized_pattern],
+    [$.variant_declaration]
   ],
 
   rules: {
@@ -290,7 +291,7 @@ module.exports = grammar({
     )),
 
     variant_declaration: $ => prec.right(seq(
-      optional($.decorator),
+      repeat($.decorator),
       $.variant_identifier,
       optional($.variant_parameters),
       optional($.type_annotation),
@@ -498,6 +499,7 @@ module.exports = grammar({
         $._definition_signature
       ),
       '=>',
+      repeat($.decorator),
       field('body', $.expression),
     )),
 

test/corpus/decorators.txt

@@ -31,6 +31,8 @@ let foo = (@doesNotRaise String.make)(12, ' ')
 
 let foo = @doesNotRaise String.make(12, ' ')
 
+let onResult = () => @doesNotRaise Belt.Array.getExn([], 0)
+
 ---
 
 (source_file
@@ -52,4 +54,16 @@ let foo = @doesNotRaise String.make(12, ' ')
       (value_identifier_path (module_identifier) (value_identifier))
       (arguments
         (number)
-        (character)))))
+        (character))))
+
+  (let_binding
+    (value_identifier)
+    (function
+      (formal_parameters)
+      (decorator (decorator_identifier))
+      (call_expression
+        (value_identifier_path (module_identifier) (module_identifier) (value_identifier))
+        (arguments
+          (array)
+          (number))))))
+