module System.Wlog.Color
( colorizer
, colorizerBuilder
) where
import Universum
import Data.Text.Lazy.Builder as B
import System.Console.ANSI (Color (Blue, Green, Magenta, Red, Yellow), ColorIntensity (Vivid),
ConsoleLayer (Foreground), SGR (Reset, SetColor), setSGRCode)
import System.Wlog.Severity (Severity (..))
table :: Severity -> (String, String)
table severity = case severity of
Error -> (setColor Red , reset)
Debug -> (setColor Green , reset)
Notice -> (setColor Magenta , reset)
Warning -> (setColor Yellow , reset)
Info -> (setColor Blue , reset)
where
setColor color = setSGRCode [SetColor Foreground Vivid color]
reset = setSGRCode [Reset]
colorizer :: Severity -> Text -> Text
colorizer pr s =
let (before, after) = table pr in toText before <> s <> toText after
colorizerBuilder :: Severity -> Builder -> Builder
colorizerBuilder pr s =
let (before, after) = table pr in B.fromText (toText before) <> s <> B.fromText (toText after)