Refactor WriteBriDoc FlushCommentsPrior code slightly

source/library/Language/Haskell/Brittany/Internal/S4_WriteBriDoc.hs

@@ -277,24 +277,22 @@ layoutBriDocM = \case
     comms <- takeBefore loc
     printComments comms
     mModify (\s -> s + CommentCounter (length comms))
-    do
+    mModify $ \s -> s
-      state <- mGet
+      { _lstate_markerForDelta = Nothing
-      mModify $ \s -> s { _lstate_markerForDelta = Nothing }
+      , _lstate_plannedSpace = case _lstate_markerForDelta s of
-      case _lstate_markerForDelta state of
+          Nothing -> _lstate_plannedSpace s
-        Just m -> do
+          Just m ->
             let p1 = (srcLocLine m, srcLocCol m)
-          let p2 = (srcLocLine loc, srcLocCol loc)
+                p2 = (srcLocLine loc, srcLocCol loc)
-          let newlinePlanned = case _lstate_plannedSpace state of
-                PlannedNone       -> False
-                PlannedSameline{} -> False
-                PlannedNewline{}  -> True
-                PlannedDelta{}    -> True
             -- traceShow (m, ExactPrint.pos2delta p1 p2) $ pure ()
-          case ExactPrint.pos2delta p1 p2 of
+            in case ExactPrint.pos2delta p1 p2 of
-            SameLine{} -> pure ()
+              SameLine{} -> _lstate_plannedSpace s
-            DifferentLine n _ | newlinePlanned -> layoutWriteNewlines n
+              DifferentLine n _ -> case _lstate_plannedSpace s of
-                              | otherwise      -> pure ()
+                PlannedNone       -> PlannedNone
-        _ -> pure ()
+                PlannedSameline i -> PlannedSameline i
+                PlannedNewline{}  -> PlannedNewline n
+                PlannedDelta{}    -> PlannedNewline n
+      }
     layoutBriDocM bd
   BDFlushCommentsPost loc shouldMark bd -> do
     layoutBriDocM bd