Merge pull request #146 from nkrkv/refactor-simplify-module-expressions

Generalize module exprs and first-class modules

Author: nkrkv

grammar.js

@@ -27,6 +27,10 @@ module.exports = grammar({
     $.module_primary_expression,
   ],
 
+  inline: $ => [
+    $._module_definition,
+  ],
+
   precedences: $ => [
     // + - Operators -> precendence
     [
@@ -100,6 +104,7 @@ module.exports = grammar({
     [$._switch_value_pattern, $._parenthesized_pattern],
     [$.variant_declaration],
     [$.unit, $._function_type_parameter_list],
+    [$.functor_parameter, $.module_primary_expression, $.module_identifier_path],
   ],
 
   rules: {
@@ -159,10 +164,7 @@ module.exports = grammar({
 
     include_statement: $ => seq(
       'include',
-      choice(
-        $.module_expression,
-        seq('(', choice($.module_expression, $.functor_parameter), ')')
-      ),
+      $.module_expression,
     ),
 
     declaration: $ => choice(
@@ -177,7 +179,7 @@ module.exports = grammar({
       'module',
       optional('rec'),
       optional('type'),
-      field('name', choice($.module_identifier, $._type_identifier)),
+      field('name', choice($.module_identifier, $.type_identifier)),
       optional(seq(
         ':',
         field('signature', choice($.block, $.module_expression, $.functor)),
@@ -219,7 +221,6 @@ module.exports = grammar({
       ':',
       choice(
         $.module_expression,
-        seq('(', $.module_expression, ')'),
         $.block,
       )
     ),
@@ -740,7 +741,10 @@ module.exports = grammar({
     module_pack: $ => seq(
       'module',
       '(',
-      $._module_definition,
+      choice(
+        $.type_identifier_path,
+        $._module_definition,
+      ),
       optional($.module_type_annotation),
       ')'
     ),
@@ -1057,8 +1061,8 @@ module.exports = grammar({
       '.',
       optional(seq(
         field('module', $.module_identifier),
-        '.')
-      ),
+        '.'
+      )),
       field('property', alias($.value_identifier, $.property_identifier)),
     )),
 
@@ -1235,12 +1239,11 @@ module.exports = grammar({
       optional(alias($.variant_arguments, $.arguments)),
     )),
 
-    _type_identifier: $ =>
-      choice(
-        $.type_identifier,
-        $.type_identifier_path,
-        ".."
-      ),
+    _type_identifier: $ => choice(
+      $.type_identifier,
+      $.type_identifier_path,
+      ".."
+    ),
 
     type_identifier_path: $ => seq(
       $.module_primary_expression,
@@ -1250,18 +1253,25 @@ module.exports = grammar({
 
     module_expression: $ => choice(
       $.module_primary_expression,
-      $.type_identifier_path,
       $.module_type_of,
       $.module_type_constraint,
     ),
 
     module_primary_expression: $ => choice(
+      $.parenthesized_module_expression,
       $.module_identifier,
       $.module_identifier_path,
       $.functor_use,
       $.module_unpack,
     ),
 
+    parenthesized_module_expression: $ => seq(
+      '(',
+      $.module_expression,
+      optional($.module_type_annotation),
+      ')',
+    ),
+
     module_identifier_path: $ => path(
       $.module_primary_expression,
       $.module_identifier,

test/corpus/modules.txt

@@ -54,21 +54,23 @@ include module type of {
     (module_type_of
       (module_identifier_path (module_identifier) (module_identifier))))
   (include_statement
-    (module_type_of
-      (module_identifier_path (module_identifier) (module_identifier))))
+    (parenthesized_module_expression
+      (module_type_of
+        (module_identifier_path (module_identifier) (module_identifier)))))
 
   (include_statement
-    (functor_parameter (module_identifier)
-    (module_type_annotation
-      (module_type_constraint
-        (module_type_of (module_identifier))
-        (constrain_module
-          (module_identifier)
-          (module_identifier_path (module_identifier) (module_identifier)))
-        (constrain_module
-          (module_identifier)
-          (module_identifier_path (module_identifier)
-          (module_identifier)))))))
+    (parenthesized_module_expression
+      (module_identifier)
+      (module_type_annotation
+        (module_type_constraint
+          (module_type_of (module_identifier))
+          (constrain_module
+            (module_identifier)
+            (module_identifier_path (module_identifier) (module_identifier)))
+          (constrain_module
+            (module_identifier)
+            (module_identifier_path (module_identifier)
+            (module_identifier)))))))
 
   (include_statement
     (module_type_of
@@ -479,34 +481,38 @@ module M = (Na: N, Nb: N): (
             (type_identifier))))))
 
   (module_declaration
-    (module_identifier)
-    (functor
-      (functor_parameters)
-      (module_type_annotation
-        (module_type_constraint
-          (module_identifier)
-          (constrain_type
-            (type_identifier)
-            (type_identifier))))
-      (block)))
+    name: (module_identifier)
+    definition: (functor
+      parameters: (functor_parameters)
+      return_module_type: (module_type_annotation
+        (parenthesized_module_expression
+          (module_type_constraint
+            (module_identifier)
+            (constrain_type
+              (type_identifier)
+              (type_identifier)))))
+      body: (block)))
 
   (module_declaration
-    (module_identifier)
-    (functor
-      (functor_parameters
+    name: (module_identifier)
+    definition: (functor
+      parameters: (functor_parameters
         (functor_parameter
           (module_identifier)
           (module_type_annotation (module_identifier)))
         (functor_parameter
           (module_identifier)
           (module_type_annotation (module_identifier))))
-      (module_type_annotation
-        (module_type_constraint
-          (module_identifier)
-          (constrain_type
-            (type_identifier)
-            (type_identifier))
-          (constrain_type
-            (type_identifier)
-            (type_identifier))))
-      (block))))
+      return_module_type: (module_type_annotation
+        (parenthesized_module_expression
+          (module_type_constraint
+            (parenthesized_module_expression
+              (module_type_constraint
+                (module_identifier)
+                (constrain_type
+                  (type_identifier)
+                  (type_identifier))))
+            (constrain_type
+              (type_identifier)
+              (type_identifier)))))
+      body: (block))))