fix(grainfmt): Group infix functions when passed as params

Added a group around infix functions when passed as args to keep the break outside the function applications
spotandjake · Oct 6, 2021 · bb9666d · bb9666d
1 parent 4e739f1
commit bb9666d
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 8 deletions.
diff --git a/compiler/grainformat/reformat.re b/compiler/grainformat/reformat.re
@@ -1200,14 +1200,15 @@ and print_application =
       } else {
         right_expr;
       };
-
-    Doc.concat([
-      Doc.group(wrapped_left),
-      Doc.space,
-      Doc.text(function_name),
-      Doc.line,
-      Doc.group(wrapped_right),
-    ]);
+    Doc.group(
+      Doc.concat([
+        Doc.group(wrapped_left),
+        Doc.space,
+        Doc.text(function_name),
+        Doc.line,
+        Doc.group(wrapped_right),
+      ]),
+    );
 
   | _ when prefixop(function_name) || infixop(function_name) =>
     raise(Error(Illegal_parse("Formatter error, wrong number of args ")))

diff --git a/compiler/test/formatter_inputs/application.gr b/compiler/test/formatter_inputs/application.gr
@@ -32,3 +32,10 @@ export let unique = array => {
 export let batchActionCreateAccount = ({index}) => {
   Native.promiseBatchActionCreateAccount(Conv.fromInt64(index))
 }
+
+
+let ret = filteri(
+  dst + _VALUE_OFFSET + dstIndex,
+  src + _VALUE_OFFSET + srcIndex,
+  length,
+)
diff --git a/compiler/test/formatter_outputs/application.gr b/compiler/test/formatter_outputs/application.gr
@@ -46,3 +46,9 @@ export let unique = array => {
 export let batchActionCreateAccount = ({ index }) => {
   Native.promiseBatchActionCreateAccount(Conv.fromInt64(index))
 }
+
+let ret = filteri(
+  dst + _VALUE_OFFSET + dstIndex,
+  src + _VALUE_OFFSET + srcIndex,
+  length,
+)