441: fix missing comments in switch (#510)

Author: cknitt

src/res_printer.ml

@@ -4434,12 +4434,7 @@ and printCases (cases: Parsetree.case list) cmtTbl =
         Doc.concat [
           Doc.line;
           printList
-            ~getLoc:(fun n -> {n.Parsetree.pc_lhs.ppat_loc with
-              loc_end =
-                match ParsetreeViewer.processBracesAttr n.Parsetree.pc_rhs with
-                | (None, _) -> n.pc_rhs.pexp_loc.loc_end
-                | (Some ({loc}, _), _) -> loc.Location.loc_end
-            })
+            ~getLoc:(fun n -> {n.Parsetree.pc_lhs.ppat_loc with loc_end = n.pc_rhs.pexp_loc.loc_end})
             ~print:printCase
             ~nodes:cases
             cmtTbl

tests/printer/expr/expected/switch.res.txt

@@ -22,6 +22,21 @@ let x =
   | Universe => ()
   }
 
+// #441 rescript format sometimes deletes comments
 switch count {
+// String in braces
 | 1 => "one"
+// Int in braces
+| 2 => 2
+// Float in braces
+| 3 => 3.
+// Bool in braces
+| 4 => true
+// Expression in braces
+| 5 => expr
+// Block
+| 6 => {
+    let _ = 123
+    Js.Console.log("Must be block")
+  }
 }

tests/printer/expr/switch.res

@@ -20,6 +20,21 @@ let x = @attr switch x {
 | Universe => ()
 }
 
+// #441 rescript format sometimes deletes comments
 switch count {
+// String in braces
 | 1 => { "one" }
+// Int in braces
+| 2 => {2}
+// Float in braces
+| 3 => {3.}
+// Bool in braces
+| 4 => { true }
+// Expression in braces
+| 5 => { expr }
+// Block
+| 6 => {
+    let _ = 123
+    Js.Console.log("Must be block")
+  }
 }