kill ghosts after EOF, vi-style

1 files changed, 9 insertions, 4 deletions

diff --git a/test4.hs b/test4.hs
index 893d120..cf5ad3e 100644
--- a/test4.hs
+++ b/test4.hs
@@ -110,11 +110,16 @@ redraw _q@State{..} = do
     -- consumes 1 screenHeight
     putStr $ "\ESC[2K" ++ flashMessage ++ " " ++ show (screenWidth, screenHeight)
 
+
+    let buf = map (take screenWidth . drop xoffset) $
+              take (screenHeight - 1) $
+              drop yoffset $
+              renderTreeView (Z.label cursor) (Z.toTree cursor)
+
     mapM_ (putStr . ("\n\ESC[2K"++)) $
-        map (take screenWidth . drop xoffset) $
-        take (screenHeight - 1) $
-        drop yoffset $
-        renderTreeView (Z.label cursor) (Z.toTree cursor)
+        buf
+        ++
+        take (screenHeight - 1 - length buf) (repeat "~")