group params

semgrep · Mar 26, 2024 · ea10bc4 · ea10bc4
1 parent 68235d5
commit ea10bc4
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 16 deletions.
diff --git a/lang/semgrep-grammars/src/semgrep-kotlin/grammar.js b/lang/semgrep-grammars/src/semgrep-kotlin/grammar.js
@@ -92,15 +92,17 @@ module.exports = grammar(standard_grammar, {
   // So we just need to fix _statement.
   _statement: ($, previous) => choice(
     previous,
-    prec.left(seq(
+      $.partial_class_declaration,
+  ),
+
+        partial_class_declaration: $ =>     prec.left(seq(
       optional($.type_parameters),
       seq(optional($.modifiers), "constructor"),
       $._class_parameters,
       optional(seq(":", $._delegation_specifiers)),
       optional($.type_constraints),
       optional($.class_body)
-    ))
-  ),
+        )),
 
 	class_parameter: ($, previous) => {
 	    return choice(

diff --git a/lang/semgrep-grammars/src/semgrep-kotlin/test/corpus/semgrep.txt b/lang/semgrep-grammars/src/semgrep-kotlin/test/corpus/semgrep.txt
@@ -281,19 +281,20 @@ constructor(arg: Int) {
 (source_file
   (class_declaration
     (type_identifier))
-  (class_parameter
-    (simple_identifier)
-    (user_type
-      (type_identifier)))
-  (class_body
-    (property_declaration
-      (modifiers
-        (visibility_modifier))
-      (variable_declaration
-        (simple_identifier))
-      (string_literal
-        (interpolated_expression
-          (simple_identifier))))))
+  (partial_class_declaration
+    (class_parameter
+      (simple_identifier)
+      (user_type
+        (type_identifier)))
+    (class_body
+      (property_declaration
+        (modifiers
+          (visibility_modifier))
+        (variable_declaration
+          (simple_identifier))
+        (string_literal
+          (interpolated_expression
+            (simple_identifier)))))))
 
 =====================================
 Class in a class