feat: support parenthesized expressions in decorators

tree-sitter · Jul 20, 2024 · e0ffcd0 · e0ffcd0
1 parent 043fc26
commit e0ffcd0
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 0 deletions.
diff --git a/common/define-grammar.js b/common/define-grammar.js
@@ -387,6 +387,16 @@ module.exports = function defineGrammar(dialect) {
         choice($._semicolon, $._function_signature_automatic_semicolon),
       ),
 
+      decorator: $ => seq(
+        '@',
+        choice(
+          $.identifier,
+          alias($.decorator_member_expression, $.member_expression),
+          alias($.decorator_call_expression, $.call_expression),
+          alias($.decorator_parenthesized_expression, $.parenthesized_expression),
+        ),
+      ),
+
       decorator_call_expression: $ => prec('call', seq(
         field('function', choice(
           $.identifier,
@@ -396,6 +406,16 @@ module.exports = function defineGrammar(dialect) {
         field('arguments', $.arguments),
       )),
 
+      decorator_parenthesized_expression: $ => seq(
+        '(',
+        choice(
+          $.identifier,
+          alias($.decorator_member_expression, $.member_expression),
+          alias($.decorator_call_expression, $.call_expression),
+        ),
+        ')',
+      ),
+
       class_body: $ => seq(
         '{',
         repeat(choice(

diff --git a/test/corpus/declarations.txt b/test/corpus/declarations.txt
@@ -1189,6 +1189,31 @@ abstract class Foo {
     name: (type_identifier)
     body: (class_body)))
 
+=======================================
+Decorator with parenthesized expression
+=======================================
+
+class C {
+    @(super.decorate)
+    method2() { }
+}
+
+---
+
+(program
+  (class_declaration
+    (type_identifier)
+    (class_body
+      (decorator
+        (parenthesized_expression
+          (member_expression
+            (identifier)
+            (property_identifier))))
+      (method_definition
+        (property_identifier)
+        (formal_parameters)
+        (statement_block)))))
+
 ==================================
 Index type queries
 ==================================