fix(grainfmt): Properly format comments within a record (#1435)

Author: marcusroberts

compiler/src/formatting/format.re

@@ -1453,7 +1453,19 @@ and print_record =
   };
 
   let after_brace_comments =
-    Comment_utils.get_after_brace_comments(~loc=recloc, comments);
+    switch (fields) {
+    | [field, ..._] =>
+      let (ident, expr) = field;
+
+      Comment_utils.get_after_brace_comments(
+        ~loc=recloc,
+        ~first=ident.loc,
+        comments,
+      );
+
+    | _ => Comment_utils.get_after_brace_comments(~loc=recloc, comments) // let s = {}  is not legal syntax, but we can use all the comments
+    };
+
   let cleaned_comments =
     remove_used_comments(~remove_comments=after_brace_comments, comments);
 

compiler/test/formatter_inputs/records.gr

@@ -60,3 +60,16 @@ let { // a comment 3
   l2,
   l3
 } = x
+
+ let s=  { num: 1, var: A, // end line comment
+    str: "" }
+
+
+ let s=  { num: 1, 
+   var: A, // end line comment
+    str: "" }
+
+ let s=  {  // comment 1
+    // comment 2
+    num: 1, var: A, // end line comment
+    str: "" }

compiler/test/formatter_outputs/records.gr

@@ -75,3 +75,22 @@ let { // a comment 3
   l2,
   l3,
 } = x
+
+let s = {
+  num: 1,
+  var: A, // end line comment
+  str: "",
+}
+
+let s = {
+  num: 1,
+  var: A, // end line comment
+  str: "",
+}
+
+let s = { // comment 1
+  // comment 2
+  num: 1,
+  var: A, // end line comment
+  str: "",
+}