Display foralls in gibiansky style

AshleyYakeley · May 29, 2015 · ebd5211 · ebd5211
1 parent 726c692
commit ebd5211
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/src/HIndent/Styles/Gibiansky.hs b/src/HIndent/Styles/Gibiansky.hs
@@ -336,7 +336,11 @@ typ :: Extend Type
 -- For contexts, check whether the context and all following function types
 -- are on the same line. If they are, print them on the same line; otherwise
 -- print the context and each argument to the function on separate lines.
-typ (TyForall _ _ (Just ctx) rest) =
+typ (TyForall _ mforall (Just ctx) rest) = do
+  forM_ mforall $ \forallVars -> do
+    write "forall "
+    spaced $ map pretty forallVars
+    write ". "
   if all (sameLine ctx) $ collectTypes rest
     then do
       pretty ctx

diff --git a/test/gibiansky/expected/45.exp b/test/gibiansky/expected/45.exp
@@ -0,0 +1,2 @@
+a :: forall b. Typeable b => b
+a = 3
diff --git a/test/gibiansky/tests/45.test b/test/gibiansky/tests/45.test
@@ -0,0 +1,2 @@
+a :: forall b. Typeable b => b
+a = 3