Dhall/Pretty/Internal.hs

16 module Dhall.Pretty.Internal (
80 import Data.Text.Prettyprint.Doc  (Doc, Pretty, space)
95 import qualified Data.Text.Prettyprint.Doc                 as Pretty
96 import qualified Data.Text.Prettyprint.Doc.Render.String   as Pretty
98 import qualified Data.Text.Prettyprint.Doc.Render.Text     as Pretty
159 prettyExpr :: Pretty a => Expr s a -> Doc Ann
162 prettySrcExpr :: Pretty a => Expr Src a -> Doc Ann
216     Pretty.align (Pretty.concatWith f newLines <> suffix)
248                             Nothing -> Pretty.pretty l
249                             Just l' -> Pretty.pretty l'
251                 in  Pretty.pretty l0 : map perLine ls
253     f x y = x <> Pretty.hardline <> y
277 keyword  = Pretty.annotate Keyword
278 syntax   = Pretty.annotate Syntax
279 label    = Pretty.annotate Label
280 literal  = Pretty.annotate Literal
281 builtin  = Pretty.annotate Builtin
282 operator = Pretty.annotate Operator
285 comma    = syntax Pretty.comma
286 lbracket = syntax Pretty.lbracket
287 rbracket = syntax Pretty.rbracket
288 langle   = syntax Pretty.langle
289 rangle   = syntax Pretty.rangle
290 lbrace   = syntax Pretty.lbrace
291 rbrace   = syntax Pretty.rbrace
292 lparen   = syntax Pretty.lparen
293 rparen   = syntax Pretty.rparen
294 pipe     = syntax Pretty.pipe
358     Pretty.group
359         (Pretty.flatAlt
361             <> Pretty.hardline
362             <> Pretty.indent n
378     combineLong x y = x <> y <> Pretty.hardline
389 arrows characterSet docs = Pretty.group (Pretty.flatAlt long short)
391     long = Pretty.align (mconcat (Data.List.intersperse Pretty.hardline docs'))
446     Pretty.group
447         (Pretty.flatAlt
448             (Pretty.align
458     combineLong x y = x <> y <> Pretty.hardline
478     Pretty.group (Pretty.flatAlt long short)
483         Pretty.align (mconcat (Data.List.intersperse Pretty.hardline longLines))
519 prettyLabelShared b l = label (Pretty.pretty (escapeLabel b l))
528 prettyAnyLabels keys = Pretty.group (Pretty.flatAlt long short)
530     short = (mconcat . Pretty.punctuate dot . map prettyKey . toList) keys
537                   Pretty.align
539                 . Pretty.punctuate (Pretty.hardline <> ". ")
540                 $ Pretty.indent 2 doc : docs
543           Pretty.align
545         . Pretty.punctuate Pretty.hardline
558 prettyNumber = literal . Pretty.pretty
561 prettyInt = literal . Pretty.pretty
564 prettyNatural = literal . Pretty.pretty
567 prettyDouble = literal . Pretty.pretty
575 prettyVar (V x 0) = label (Pretty.unAnnotate (prettyLabel x))
576 prettyVar (V x n) = label (Pretty.unAnnotate (prettyLabel x <> "@" <> prettyInt n))
579 prettyEnvironmentVariable t = Pretty.pretty (escapeEnvironmentVariable t)
582 preserveSource (Note Src{..} (DoubleLit  {})) = Just (Pretty.pretty srcText)
583 preserveSource (Note Src{..} (IntegerLit {})) = Just (Pretty.pretty srcText)
584 preserveSource (Note Src{..} (NaturalLit {})) = Just (Pretty.pretty srcText)
623 prettyCharacterSet :: Pretty a => CharacterSet -> Expr Src a -> Doc Ann
629 prettyImportExpression :: Pretty a => Expr Src a -> Doc Ann
639 prettyPrinters :: Pretty a => CharacterSet -> PrettyPrinters a
644         Pretty.group (prettyExpression expression)
650             Pretty.group (Pretty.flatAlt long short) : docs c
653                 <>  Pretty.align
656                     <>  Pretty.hardline
659                     <>  Pretty.hardline
676         Pretty.group (Pretty.flatAlt long short)
695             Pretty.align (mconcat (Data.List.intersperse Pretty.hardline longLines))
733         enclose' "" "" space Pretty.hardline
754             Pretty.group (Pretty.flatAlt long short)
757                 <>  Pretty.align
760                     <>  equals <> Pretty.hardline <> renderSrc stripNewline src2
770             <>  Pretty.align
772                 <>  prettyLabel c <> Pretty.hardline <> renderSrc stripNewline src1
773 …          <>  colon <> space <> renderSrc stripSpaces src3 <> prettyExpression d <> Pretty.hardline
786         docs (Pi _ a   b c) = Pretty.group (Pretty.flatAlt long short) : docs c
789                 <>  Pretty.align
792                     <>  Pretty.hardline
795                     <>  Pretty.hardline
818         <>  Pretty.flatAlt long short
822         long =  Pretty.hardline
824             <>  Pretty.align (keyword "with" <> " " <> update)
829         Pretty.group (Pretty.flatAlt long short)
834             Pretty.align
836             <> Pretty.hardline <> colon <> " " <> prettyExpression a
846     prettyAnnotatedExpression :: Pretty a => Expr Src a -> Doc Ann
848         Pretty.group (Pretty.flatAlt long short)
851             Pretty.align
853                 <>  Pretty.hardline
854                 <>  Pretty.indent 2 (prettyImportExpression_ a)
855                 <>  Pretty.hardline
856                 <>  Pretty.indent 2 (prettyImportExpression_ b)
857                 <>  Pretty.hardline
869         Pretty.group (Pretty.flatAlt long short)
872             Pretty.align
874                 <>  Pretty.hardline
875                 <>  Pretty.indent 2 (prettyImportExpression_ a)
876                 <>  Pretty.hardline
913     prettyOperatorExpression :: Pretty a => Expr Src a -> Doc Ann
921             (" " <> operator (Pretty.pretty op) <> " ")
922             (operator (Pretty.pretty op) <> spacer)
929     prettyEquivalentExpression :: Pretty a => Expr Src a -> Doc Ann
949     prettyImportAltExpression :: Pretty a => Expr Src a -> Doc Ann
969     prettyOrExpression :: Pretty a => Expr Src a -> Doc Ann
989     prettyPlusExpression :: Pretty a => Expr Src a -> Doc Ann
1009     prettyTextAppendExpression :: Pretty a => Expr Src a -> Doc Ann
1029     prettyListAppendExpression :: Pretty a => Expr Src a -> Doc Ann
1049     prettyAndExpression :: Pretty a => Expr Src a -> Doc Ann
1069     prettyCombineExpression :: Pretty a => Expr Src a -> Doc Ann
1089     prettyPreferExpression :: Pretty a => Expr Src a -> Doc Ann
1109     prettyCombineTypesExpression :: Pretty a => Expr Src a -> Doc Ann
1129     prettyTimesExpression :: Pretty a => Expr Src a -> Doc Ann
1149     prettyEqualExpression :: Pretty a => Expr Src a -> Doc Ann
1169     prettyNotEqualExpression :: Pretty a => Expr Src a -> Doc Ann
1189     prettyApplicationExpression :: Pretty a => Expr Src a -> Doc Ann
1210                 : map (fmap (Pretty.indent 2) . duplicate . prettyImportExpression_) args
1213     prettyImportExpression_ :: Pretty a => Expr Src a -> Doc Ann
1215         Pretty.pretty a
1224     prettyCompletionExpression :: Pretty a => Expr Src a -> Doc Ann
1228                 Pretty.align
1244     prettySelectorExpression :: Pretty a => Expr Src a -> Doc Ann
1263     prettyPrimitiveExpression :: Pretty a => Expr Src a -> Doc Ann
1355             Pretty.group (Pretty.flatAlt long short)
1358             Pretty.align
1359                 (lparen <> space <> prettyExpression a <> Pretty.hardline <> rparen)
1364         :: Pretty a
1370         duplicate (Pretty.group (Pretty.flatAlt long short))
1387                     Just doc -> doc <> Pretty.hardline
1390         long =  Pretty.align
1398                         <>  Pretty.align
1400                                 <>  Pretty.hardline
1414                                                 Pretty.hardline
1420                                                     Pretty.hardline
1424                                         _ ->    Pretty.hardline
1434                                         _ ->    Pretty.hardline
1442                                     Pretty.hardline
1448                                         Pretty.hardline
1453                                 Pretty.group
1454                                     (   Pretty.flatAlt (Pretty.hardline <> "    ") " "
1461                         (" ", Pretty.hardline <> "    ")
1463                         (Pretty.hardline, " ")
1466     prettyRecord :: Pretty a => Map Text (RecordField Src a) -> Doc Ann
1474     prettyRecordLit :: Pretty a => Map Text (RecordField Src a) -> Doc Ann
1477     prettyCompletionLit :: Pretty a => Int -> Map Text (RecordField Src a) -> Doc Ann
1481         :: Pretty a
1506     prettyUnion :: Pretty a => Map Text (Maybe (Expr Src a)) -> Doc Ann
1510     prettyChunks :: Pretty a => Chunks Src a -> Doc Ann
1515             else Pretty.group (Pretty.flatAlt long short)
1520             Pretty.align
1521             (   literal "''" <> Pretty.hardline
1522             <>  Pretty.align
1549                     (Pretty.pretty line)
1552                 Data.List.intersperse Pretty.hardline (map prettyLine lines_)
1560         prettyText t = literal (Pretty.pretty (escapeText_ t))
1698 pretty_ :: Pretty a => a -> Text
1777 prettyToString :: Pretty a => a -> String
1779     Pretty.renderString . layout . Pretty.pretty
1782 docToStrictText = Pretty.renderStrict . layout
1784 prettyToStrictText :: Pretty a => a -> Text.Text
1785 prettyToStrictText = docToStrictText . Pretty.pretty
1793 layout :: Pretty.Doc ann -> Pretty.SimpleDocStream ann
1794 layout = Pretty.removeTrailingWhitespace . Pretty.layoutSmart layoutOpts
1797 layoutOpts :: Pretty.LayoutOptions
1799     Pretty.defaultLayoutOptions
1800         { Pretty.layoutPageWidth = Pretty.AvailablePerLine 80 1.0 }  function