fix(grainfmt): handle empty type for foreign exports (#907)

Handle printing when the type is (). Use a switch and update the test
grain-lang · Sep 19, 2021 · d3b60a4 · d3b60a4
1 parent 4acb91f
commit d3b60a4
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/compiler/grainformat/reformat.re b/compiler/grainformat/reformat.re
@@ -933,9 +933,10 @@ and print_type =
   | PTyArrow(types, parsed_type) =>
     Doc.concat([
       Doc.group(
-        if (List.length(types) == 1) {
-          print_type(List.hd(types), original_source);
-        } else {
+        switch (List.length(types)) {
+        | 0 => Doc.concat([Doc.lparen, Doc.rparen])
+        | 1 => print_type(List.hd(types), original_source)
+        | _ =>
           Doc.concat([
             Doc.lparen,
             Doc.indent(
@@ -949,7 +950,7 @@ and print_type =
             ),
             Doc.softLine,
             Doc.rparen,
-          ]);
+          ])
         },
       ),
       Doc.space,

diff --git a/compiler/test/formatter_inputs/imports.gr b/compiler/test/formatter_inputs/imports.gr
@@ -24,3 +24,5 @@ import * except {
    and
    // just because
  } from "runtime/unsafe/wasmi32"
+
+export foreign wasm promise_results_count: () -> WasmI64 as promiseResultsCount from "env"
diff --git a/compiler/test/formatter_outputs/imports.gr b/compiler/test/formatter_outputs/imports.gr
@@ -28,3 +28,5 @@ import * except {
   and
   // just because
 } from "runtime/unsafe/wasmi32"
+
+export foreign wasm promise_results_count: () -> WasmI64 as promiseResultsCount from "env"