Merge pull request #132 from aspeddro/module-type

support first class modules

Author: nkrkv

grammar.js

@@ -167,7 +167,7 @@ module.exports = grammar({
       'module',
       optional('rec'),
       optional('type'),
-      field('name', $.module_identifier),
+      field('name', choice($.module_identifier, $._type_identifier)),
       optional(seq(
         ':',
         field('signature', choice($.block, $.module_expression, $.functor)),
@@ -185,6 +185,8 @@ module.exports = grammar({
       $.extension_expression,
     ),
 
+    module_unpack: $ => seq('unpack', $.call_arguments),
+
     functor: $ => seq(
       field('parameters', $.functor_parameters),
       optional(field('return_module_type', $.module_type_annotation)),
@@ -277,6 +279,17 @@ module.exports = grammar({
       $.object_type,
       $.generic_type,
       $.unit_type,
+      $.module_pack_type,
+    ),
+
+    module_pack_type: $ => seq(
+      'module',
+      '(',
+      choice($.module_identifier, $._type_identifier),
+      optional(
+        seq('with', sep1('and', seq('type', $._type_identifier, '=', $._type_identifier)))
+      ),
+      ')'
     ),
 
     tuple_type: $ => prec.dynamic(-1, seq(
@@ -429,6 +442,7 @@ module.exports = grammar({
       $.record_pattern,
       $.array_pattern,
       $.list_pattern,
+      $.module_unpack_pattern,
       $.unit
     ),
 
@@ -475,7 +489,7 @@ module.exports = grammar({
       $.pipe_expression,
       $.subscript_expression,
       $.member_expression,
-      $.module_destructuring_expression,
+      $.module_pack_expression,
       $.extension_expression,
     ),
 
@@ -702,8 +716,12 @@ module.exports = grammar({
       ),
     )),
 
-    module_destructuring_expression: $ => seq(
-      'module', '(', choice($.module_identifier, $.module_identifier_path), ')'
+    module_pack_expression: $ => seq(
+      'module',
+      '(',
+      choice($.module_expression, $.block),
+      optional(seq(':', $.module_expression)),
+      ')'
     ),
 
     call_arguments: $ => seq(
@@ -772,7 +790,7 @@ module.exports = grammar({
 
     type_parameter: $ => seq(
       'type',
-      $.type_identifier,
+      repeat1($.type_identifier),
     ),
 
     _labeled_parameter_default_value: $ => seq(
@@ -886,6 +904,10 @@ module.exports = grammar({
       choice($.value_identifier, $.list_pattern, $.array_pattern),
     ),
 
+    module_unpack_pattern: $ => seq(
+      'module', '(', $.module_identifier, ')',
+    ),
+
     _jsx_element: $ => choice($.jsx_element, $.jsx_self_closing_element),
 
     jsx_element: $ => seq(
@@ -1207,7 +1229,8 @@ module.exports = grammar({
       $.module_identifier_path,
       $.module_type_of,
       $.functor_use,
-      $.module_type_constraint
+      $.module_type_constraint,
+      $.module_unpack,
     ),
 
     module_identifier_path: $ => prec.left(seq(

queries/highlights.scm

@@ -113,6 +113,7 @@
   "async"
   "await"
   "with"
+  "unpack"
 ] @keyword
 
 [

test/corpus/let_bindings.txt

@@ -247,3 +247,55 @@ let test = (. ~attr) => ()
           (labeled_parameter
             (value_identifier))))
       (unit))))
+
+===========================================
+Destructuring module
+===========================================
+
+
+let {foo, bar} = module(User)
+let {baz, _} = module(User.Inner)
+
+---
+
+(source_file
+  (let_binding
+    (record_pattern
+      (value_identifier)
+      (value_identifier))
+    (module_pack_expression (module_identifier)))
+
+  (let_binding
+    (record_pattern
+      (value_identifier)
+      (value_identifier))
+    (module_pack_expression
+      (module_identifier_path
+        (module_identifier)
+        (module_identifier)))))
+
+===========================================
+Packing module
+===========================================
+
+let foo = module(Bar)
+
+---
+
+(source_file
+  (let_binding
+    (value_identifier)
+    (module_pack_expression (module_identifier))))
+
+===========================================
+Unpacking module
+===========================================
+
+let module(Bar) = foo
+
+---
+
+(source_file
+  (let_binding
+    (module_unpack_pattern (module_identifier))
+    (value_identifier)))

test/corpus/modules.txt

@@ -12,32 +12,6 @@ open! Foo.Bar
   (open_statement
     (module_identifier_path (module_identifier) (module_identifier))))
 
-===========================================
-Module destructuring
-===========================================
-
-
-let {foo, bar} = module(User)
-let {baz, _} = module(User.Inner)
-
----
-
-(source_file
-  (let_binding
-    (record_pattern
-      (value_identifier)
-      (value_identifier))
-    (module_destructuring_expression (module_identifier)))
-
-  (let_binding
-    (record_pattern
-      (value_identifier)
-      (value_identifier))
-    (module_destructuring_expression
-      (module_identifier_path
-        (module_identifier)
-        (module_identifier)))))
-
 ===========================================
 Include
 ===========================================
@@ -156,6 +130,7 @@ Module types
 
 module type S1 = { type t }
 module type S2 = module type of MyModule.Submod
+module type t
 
 ---
 
@@ -165,7 +140,46 @@ module type S2 = module type of MyModule.Submod
     (block (type_declaration (type_identifier))))
   (module_declaration
     (module_identifier)
-    (module_type_of (module_identifier_path (module_identifier) (module_identifier)))))
+    (module_type_of (module_identifier_path (module_identifier) (module_identifier))))
+  (module_declaration (type_identifier)))
+
+===========================================
+First Class module
+===========================================
+
+module(LightTheme)
+module(A: A)
+module(
+  {
+    type t
+    let foo = "Hello"
+  }: X
+)
+module(SomeFunctor(unpack(x)))
+
+---
+
+(source_file
+  (expression_statement
+    (module_pack_expression (module_identifier)))
+  (expression_statement
+    (module_pack_expression (module_identifier) (module_identifier)))
+  (expression_statement
+    (module_pack_expression
+      (block
+        (type_declaration (type_identifier))
+        (let_binding
+          (value_identifier)
+          (string (string_fragment))))
+      (module_identifier)))
+  (expression_statement
+    (module_pack_expression
+      (functor_use
+        (module_identifier)
+        (arguments
+          (module_unpack
+            (call_arguments (value_identifier))))))))
+
 
 ===========================================
 Functor definition

test/corpus/type_declarations.txt

@@ -136,6 +136,8 @@ type t =
   | @live("t.D") D
   | B(anotherType)
   | C({foo: int, bar: string})
+  | D(module(Foo))
+  | D(module(Bar.t))
 
 ---
 
@@ -156,8 +158,16 @@ type t =
             (record_type_field
               (property_identifier) (type_annotation (type_identifier)))
             (record_type_field
-              (property_identifier) (type_annotation (type_identifier)))
-            ))))))
+              (property_identifier) (type_annotation (type_identifier))))))
+      (variant_declaration
+        (variant_identifier)
+        (variant_parameters
+          (module_pack_type (module_identifier))))
+      (variant_declaration
+        (variant_identifier)
+        (variant_parameters
+          (module_pack_type
+            (type_identifier_path (module_identifier) (type_identifier))))))))
 
 ===========================================
 Annotated variant

test/highlight/modules.res

@@ -22,3 +22,29 @@ module Belt = {
   //                                  ^ keyword
   //                                                                                ^ operator
 }
+
+let a = module(
+//      ^ keyword
+  {
+    type t
+    let hello = "Hello"
+  }: X
+//   ^ namespace
+)
+
+module B = unpack(a)
+//         ^ keyword
+
+module type A = {
+  type t = int
+  let value: t
+}
+
+module A: A = {
+  type t = int
+  let value: t = 42
+}
+
+let packedA = module(A: A)
+//                   ^ namespace
+//                      ^ namespace

test/highlight/type_declarations.res

@@ -8,6 +8,8 @@ type rec t =
 //       ^ type
   | Terminal
 //  ^ constant
+  | Component(module(Foo.t))
+//             ^ keyword
 
 type obj = {
   "x": int,