-- | This module handles the complexities of writing information to the
-- terminal, including modifying text in place.

module Test.HUnit.Terminal (
        terminalAppearance
    ) where

import Data.Char (isPrint)


-- | Simplifies the input string by interpreting @\\r@ and @\\b@ characters
-- specially so that the result string has the same final (or /terminal/,
-- pun intended) appearance as would the input string when written to a
-- terminal that overwrites character positions following carriage
-- returns and backspaces.

terminalAppearance :: String -> String
terminalAppearance str = ta id "" "" str

-- | The helper function @ta@ takes an accumulating @ShowS@-style function
-- that holds /committed/ lines of text, a (reversed) list of characters
-- on the current line /before/ the cursor, a (normal) list of characters
-- on the current line /after/ the cursor, and the remaining input.

ta 
    :: ([Char] -> t) -- ^ An accumulating @ShowS@-style function
                     -- that holds /committed/ lines of text
    -> [Char] -- ^ A (reversed) list of characters
              -- on the current line /before/ the cursor
    -> [Char] -- ^ A (normal) list of characters
              -- on the current line /after/ the cursor
    -> [Char] -- ^ The remaining input
    -> t
ta f    bs  as ('\n':cs) = ta (\t -> f (reverse bs ++ as ++ '\n' : t)) "" "" cs
ta f    bs  as ('\r':cs) = ta f "" (reverse bs ++ as) cs
ta f (b:bs) as ('\b':cs) = ta f bs (b:as) cs
ta _    ""   _ ('\b': _) = error "'\\b' at beginning of line"
ta f    bs  as (c:cs) 
    | not (isPrint c)    = error "invalid nonprinting character"
    | null as            = ta f (c:bs) ""        cs
    | otherwise          = ta f (c:bs) (tail as) cs
ta f    bs  as       ""  = f (reverse bs ++ as)