Add disable and enable line wrap ANSI commands

UnkindPartition · Nov 13, 2024 · 2ac232f · 2ac232f
1 parent 52f8644
commit 2ac232f
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/ansi-terminal/src/System/Console/ANSI.hs b/ansi-terminal/src/System/Console/ANSI.hs
@@ -267,6 +267,16 @@ module System.Console.ANSI
   , clearFromCursorToLineBeginningCode
   , clearLineCode
 
+    -- * Enabling and disabling line wrap
+  , disableLineWrap
+  , enableLineWrap
+    -- ** \'h...\' variants
+  , hDisableLineWrap
+  , hEnableLineWrap
+    -- ** \'...Code\' variants
+  , disableLineWrapCode
+  , enableLineWrapCode
+
     -- * Scrolling the screen
   , scrollPageUp
   , scrollPageDown
@@ -1028,6 +1038,16 @@ clearFromCursorToLineEnd = hClearFromCursorToLineEnd stdout
 clearFromCursorToLineBeginning = hClearFromCursorToLineBeginning stdout
 clearLine = hClearLine stdout
 
+hEnableLineWrap, hDisableLineWrap ::
+     Handle
+  -> IO ()
+hEnableLineWrap h = hPutStr h enableLineWrapCode
+hDisableLineWrap h = hPutStr h disableLineWrapCode
+
+enableLineWrap, disableLineWrap :: IO ()
+enableLineWrap = hEnableLineWrap stdout
+disableLineWrap = hDisableLineWrap stdout
+
 hScrollPageUp, hScrollPageDown ::
      Handle
   -> Int -- Number of lines to scroll by

diff --git a/ansi-terminal/src/System/Console/ANSI/Codes.hs b/ansi-terminal/src/System/Console/ANSI/Codes.hs
@@ -35,6 +35,9 @@ module System.Console.ANSI.Codes
   , clearScreenCode, clearFromCursorToLineEndCode
   , clearFromCursorToLineBeginningCode, clearLineCode
 
+    -- * Enabling and disabling line wrap
+  , enableLineWrapCode, disableLineWrapCode
+
     -- * Scrolling the screen
     --
     -- | These functions yield @\"\"@ when the number is @0@ as, on some
@@ -309,6 +312,10 @@ clearFromCursorToLineEndCode = csi [0] "K"
 clearFromCursorToLineBeginningCode = csi [1] "K"
 clearLineCode = csi [2] "K"
 
+enableLineWrapCode, disableLineWrapCode :: String
+enableLineWrapCode = csi [] "?7h"
+disableLineWrapCode = csi [] "?7l"
+
 scrollPageUpCode, scrollPageDownCode ::
      Int -- ^ Number of lines to scroll by
   -> String