support more of the class syntax

Author: resolritter

common/corpus/declarations.txt

@@ -832,3 +832,122 @@ let a!: b;
       (identifier)
       (type_annotation
       (type_identifier)))))
+
+=======================================
+Classes with generic parameters
+=======================================
+
+class A<
+  B,
+  C,
+> {}
+
+class D extends A<
+  X,
+  Y,
+> {}
+
+---
+
+(program
+  (class_declaration
+    (type_identifier)
+    (type_parameters (type_parameter (type_identifier)) (type_parameter (type_identifier)))
+    (class_body))
+  (class_declaration
+    (type_identifier)
+    (class_heritage (extends_clause (generic_type
+      (type_identifier)
+      (type_arguments (type_identifier) (type_identifier)))))
+    (class_body)))
+
+=======================================
+Classes with extensions
+=======================================
+
+class A extends B<C>(D) implements C {}
+export class A extends B<C>(D) implements C {}
+
+class A extends B<C>(D)<E> implements C {}
+export class A extends B<C>(D)<E> implements C {}
+
+export class A extends B<C, D> implements C {}
+
+---
+
+(program
+  (class_declaration
+    (type_identifier)
+    (class_heritage
+      (extends_clause
+        (functional_extension
+          (generic_type
+            (type_identifier)
+            (type_arguments
+              (type_identifier)))
+          (arguments
+            (identifier))))
+      (implements_clause
+        (type_identifier)))
+    (class_body))
+  (export_statement
+    (class_declaration
+      (type_identifier)
+      (class_heritage
+        (extends_clause
+          (functional_extension
+            (generic_type
+              (type_identifier)
+              (type_arguments
+                (type_identifier)))
+            (arguments
+              (identifier))))
+        (implements_clause
+          (type_identifier)))
+      (class_body)))
+  (class_declaration
+    (type_identifier)
+    (class_heritage
+      (extends_clause
+        (functional_extension
+          (generic_type
+            (type_identifier)
+            (type_arguments
+              (type_identifier)))
+          (arguments
+            (identifier))
+          (type_arguments
+            (type_identifier))))
+      (implements_clause
+        (type_identifier)))
+    (class_body))
+  (export_statement
+    (class_declaration
+      (type_identifier)
+      (class_heritage
+        (extends_clause
+          (functional_extension
+            (generic_type
+              (type_identifier)
+              (type_arguments
+                (type_identifier)))
+            (arguments
+              (identifier))
+            (type_arguments
+              (type_identifier))))
+        (implements_clause
+          (type_identifier)))
+      (class_body)))
+  (export_statement
+    (class_declaration
+      (type_identifier)
+      (class_heritage
+        (extends_clause
+          (generic_type
+            (type_identifier)
+            (type_arguments
+              (type_identifier)
+              (type_identifier))))
+        (implements_clause
+          (type_identifier)))
+      (class_body))))

common/corpus/types.txt

@@ -460,42 +460,14 @@ interface Enum extends Bar, Baz, funThatEvalsToInterface() {
     (extends_clause
       (type_identifier)
       (type_identifier)
-      (call_expression (identifier) (arguments)))
+      (functional_extension (type_identifier) (arguments)))
       (object_type
         (method_signature
           (accessibility_modifier)
           (property_identifier)
           (formal_parameters)
           (type_annotation (type_identifier))))))
 
-=======================================
-Generic types
-=======================================
-
-class A<
-  B,
-  C,
-> {}
-
-class D extends A<
-  X,
-  Y,
-> {}
-
----
-
-(program
-  (class_declaration
-    (type_identifier)
-    (type_parameters (type_parameter (type_identifier)) (type_parameter (type_identifier)))
-    (class_body))
-  (class_declaration
-    (type_identifier)
-    (class_heritage (extends_clause (generic_type
-      (type_identifier)
-      (type_arguments (type_identifier) (type_identifier)))))
-    (class_body)))
-
 =======================================
 Existential types
 =======================================

common/define-grammar.js

@@ -36,7 +36,7 @@ module.exports = function defineGrammar(dialect) {
       [$.mapped_type_clause, $.primary_expression],
       [$.accessibility_modifier, $.primary_expression],
       ['unary_void', $.expression],
-      ['extends_type', $.primary_expression],
+      [$._extends_type, $.extends_clause, $.primary_expression],
       ['unary', 'assign'],
       ['declaration', $.expression],
       [$.predefined_type, $.unary_expression],
@@ -437,17 +437,20 @@ module.exports = function defineGrammar(dialect) {
         field('body', $.object_type)
       ),
 
-      extends_clause: $ => prec('extends_type', seq(
+      _extends_type: $ => choice(
+        $._type_identifier,
+        $.nested_type_identifier,
+        $.generic_type,
+      ),
+      functional_extension: $ => seq($._extends_type, $.arguments, optional($.type_arguments)),
+
+      extends_clause: $ => seq(
         'extends',
         commaSep1(choice(
-          prec('extends_type', choice(
-            $._type_identifier,
-            $.nested_type_identifier,
-            $.generic_type
-          )),
+          choice($._extends_type, $.functional_extension),
           $.expression
         ))
-      )),
+      ),
 
       enum_declaration: $ => seq(
         optional('const'),