From ef95be4403d09ebecbaef80e668d52fc5ecd9b46 Mon Sep 17 00:00:00 2001 From: Tom Smeding Date: Tue, 25 Feb 2025 23:56:33 +0100 Subject: Pretty: Allow colouring exts (currently not) --- src/AST/Pretty.hs | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/AST/Pretty.hs b/src/AST/Pretty.hs index 35c78c1..c7b1497 100644 --- a/src/AST/Pretty.hs +++ b/src/AST/Pretty.hs @@ -298,7 +298,7 @@ ppLam args body = ppString "(" <> hang 2 (ppString "\\" <> sep (args ++ [ppStrin <> softline <> body <> ppString ")") ppX :: PrettyX x => Expr x env t -> ADoc -ppX expr = ppString $ prettyXsuffix (extOf expr) +ppX expr = annotate AExt $ ppString $ prettyXsuffix (extOf expr) data Fixity = Prefix | Infix deriving (Show) @@ -352,7 +352,7 @@ ppParen :: Bool -> Doc x -> Doc x ppParen True = parens ppParen False = id -data Annot = AKey | AWith | AHighlight | AMonoid +data Annot = AKey | AWith | AHighlight | AMonoid | AExt deriving (Show) annotToANSI :: Annot -> PT.AnsiStyle @@ -360,6 +360,7 @@ annotToANSI AKey = PT.bold annotToANSI AWith = PT.color PT.Red <> PT.underlined annotToANSI AHighlight = PT.color PT.Blue annotToANSI AMonoid = PT.color PT.Green +annotToANSI AExt = mempty type ADoc = Doc Annot -- cgit v1.2.3-70-g09d2