Fix module printing (#6963)

* Handle parens when mod type is a module signature

* Refactor

* Update CHANGELOG

* Add tests

* Handle more edge cases

* Add more test cases from jscomp/test/coercion_module_alias_test.res

Author: shulhi

CHANGELOG.md

@@ -14,6 +14,7 @@
 
 - Fix issue where long layout break added a trailing comma in partial application `...`. https://github.com/rescript-lang/rescript-compiler/pull/6949
 - Fix incorrect format of function under unary operator. https://github.com/rescript-lang/rescript-compiler/pull/6953
+- Fix incorrect incorrect printing of module binding with signature. https://github.com/rescript-lang/rescript-compiler/pull/6963
 
 # 11.1.3
 

jscomp/syntax/src/res_parens.ml

@@ -450,6 +450,23 @@ let includeModExpr modExpr =
   | Parsetree.Pmod_constraint _ -> true
   | _ -> false
 
+let modExprParens modExpr =
+  match modExpr with
+  | {
+   Parsetree.pmod_desc =
+     Pmod_constraint
+       ( {Parsetree.pmod_desc = Pmod_structure _},
+         {Parsetree.pmty_desc = Pmty_signature [{psig_desc = Psig_module _}]} );
+  } ->
+    false
+  | {
+   Parsetree.pmod_desc =
+     Pmod_constraint
+       (_, {Parsetree.pmty_desc = Pmty_signature [{psig_desc = Psig_module _}]});
+  } ->
+    true
+  | _ -> false
+
 let arrowReturnTypExpr typExpr =
   match typExpr.Parsetree.ptyp_desc with
   | Parsetree.Ptyp_arrow _ -> true

jscomp/syntax/src/res_parens.mli

@@ -32,6 +32,8 @@ val callExpr : Parsetree.expression -> kind
 
 val includeModExpr : Parsetree.module_expr -> bool
 
+val modExprParens : Parsetree.module_expr -> bool
+
 val arrowReturnTypExpr : Parsetree.core_type -> bool
 
 val patternRecordRowRhs : Parsetree.pattern -> bool

jscomp/syntax/src/res_printer.ml

@@ -719,6 +719,11 @@ and printModuleBinding ~state ~isRec moduleBinding cmtTbl i =
         Doc.concat [Doc.text ": "; printModType ~state modType cmtTbl] )
     | modExpr -> (printModExpr ~state modExpr cmtTbl, Doc.nil)
   in
+  let modExprDocParens =
+    if Parens.modExprParens moduleBinding.pmb_expr then
+      Doc.concat [Doc.lparen; modExprDoc; Doc.rparen]
+    else modExprDoc
+  in
   let modName =
     let doc = Doc.text moduleBinding.pmb_name.Location.txt in
     printComments doc cmtTbl moduleBinding.pmb_name.loc
@@ -732,7 +737,7 @@ and printModuleBinding ~state ~isRec moduleBinding cmtTbl i =
         modName;
         modConstraintDoc;
         Doc.text " = ";
-        modExprDoc;
+        modExprDocParens;
       ]
   in
   printComments doc cmtTbl moduleBinding.pmb_loc

jscomp/syntax/tests/printer/modExpr/expected/structure.res.txt

@@ -22,3 +22,49 @@ let g = {
   module M: T = {}
   0
 }
+
+module M7: {
+  module N': {
+    let x: int
+  }
+} = (M6: {
+  module N: {
+    let x: int
+  }
+  module N' = N
+})
+
+module M8 = M7
+
+module M5 = G0()
+
+module M7: {
+  let x: int
+} = {
+  let x = 8
+}
+
+module M3: {
+  module N': {
+    let x: int
+  }
+} = {
+  include M'
+}
+
+module G0: (X: {}) =>
+{
+  module N': {
+    let x: int
+  }
+} = F0
+
+module M6 = {
+  module D = {
+    let y = 3
+  }
+  module N = {
+    let x = 1
+  }
+  module N' = N
+}

jscomp/syntax/tests/printer/modExpr/structure.res

@@ -21,4 +21,51 @@ module type T = {}
 let g = {
   module M: T = {}
   0
-}
+}
+
+module M7: {
+  module N': {
+    let x: int
+  }
+} = (M6: {
+  module N: {
+    let x: int
+  }
+  module N' = N
+})
+
+
+module M8 = M7
+
+module M5 = G0()
+
+module M7: {
+  let x: int
+} = {
+  let x = 8
+}
+
+module M3: {
+  module N': {
+    let x: int
+  }
+} = {
+  include M'
+}
+
+module G0: (X: {}) =>
+{
+  module N': {
+    let x: int
+  }
+} = F0
+
+module M6 = {
+  module D = {
+    let y = 3
+  }
+  module N = {
+    let x = 1
+  }
+  module N' = N
+}