Don't forget the other applications of commitSpaces

No tests though!
sjakobi · Nov 5, 2019 · 6c525e3 · 6c525e3
1 parent b825fbf
commit 6c525e3
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/prettyprinter/src/Data/Text/Prettyprint/Doc/Internal.hs b/prettyprinter/src/Data/Text/Prettyprint/Doc/Internal.hs
@@ -1465,11 +1465,13 @@ removeTrailingWhitespace = go (RecordedWhitespace [] 0)
         -> Int -- Withheld spaces
         -> SimpleDocStream ann
         -> SimpleDocStream ann
-    commitSpaces [] 0 = id
-    commitSpaces [] 1 = SChar ' '
-    commitSpaces [] n = SText n (T.replicate n " ")
-    commitSpaces [i] n = SLine i . commitSpaces [] n
-    commitSpaces (_:is) n = SLine 0 . commitSpaces is n
+    commitSpaces is0 n0 = commitSpaces' (reverse is0) n0
+      where
+        commitSpaces' [] 0 = id
+        commitSpaces' [] 1 = SChar ' '
+        commitSpaces' [] n = SText n (T.replicate n " ")
+        commitSpaces' [i] n = SLine i . commitSpaces' [] n
+        commitSpaces' (_:is) n = SLine 0 . commitSpaces' is n
 
     go :: WhitespaceStrippingState -> SimpleDocStream ann -> SimpleDocStream ann
     -- We do not strip whitespace inside annotated documents, since it might
@@ -1494,7 +1496,7 @@ removeTrailingWhitespace = go (RecordedWhitespace [] 0)
         SChar c rest
             | c == ' ' -> go (RecordedWhitespace withheldLines (withheldSpaces+1)) rest
             | otherwise -> commitSpaces
-                               (reverse withheldLines)
+                               withheldLines
                                withheldSpaces
                                (SChar c (go (RecordedWhitespace [] 0) rest))
         SText textLength text rest ->