TypePrinter.hs

module TypePrinter (Printer, monoTypePrint, polyTypePrint, plainTypePrinter, coloredTypePrinter) where

import System.Console.ANSI
import Output
import Typing (MonoType, PolyType)

type Printer a = (MarkupString Styles -> a)

typePrinter :: Monad m => (String -> m a) -> (OpenClose Styles -> m a) -> MarkupString Styles -> m a
typePrinter f g [] = f ""
typePrinter f g (Left c : xs) = f [c] >> typePrinter f g xs
typePrinter f g (Right s : xs) = g s >> typePrinter f g xs

monoTypePrint :: Monad m => OutputInfo b -> (Printer (m a)) -> MonoType b -> m a
monoTypePrint info f t = f (output info t)

polyTypePrint :: Monad m => OutputInfo b -> (Printer (m a)) -> PolyType b -> m a
polyTypePrint info f t = f (output info t)

plainTypePrinter :: Printer (IO ())
plainTypePrinter = typePrinter putStr (\x -> return ())

-- Colored Output
syntaxColor :: Styles -> Color
syntaxColor Type = Cyan
syntaxColor Variable = Yellow
syntaxColor Constant = Red
syntaxColor Keyword = Blue
syntaxColor Function = Green
syntaxColor UniqueID = Magenta
syntaxColor Comments = Black
syntaxColor QuotedString = Black

color :: OpenClose Styles -> IO ()
color (Open s) = setSGR [SetColor Foreground Vivid (syntaxColor s)]
color (Close s) = setSGR []

coloredTypePrinter :: Printer (IO ())
coloredTypePrinter = typePrinter putStr color