picnoir
/
hnix
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Use pretty util functions for filling paragraphs

This commit is contained in:
Luca Bruno 2014-07-07 17:29:55 +02:00
parent a12c576bcc
commit 610fc3c780
1 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
Nix/Pretty.hs
View File

@ -12,8 +12,6 @@ prettySetArg :: (Text, Maybe NExpr) -> Doc
prettySetArg (n, Nothing) = text (unpack n)
prettySetArg (n, Just v) = text (unpack n) <+> text "?" <+> prettyNix v
prettyFold = foldr ($$) empty
infixOper :: NExpr -> String -> NExpr -> Doc
infixOper l op r = prettyNix l <+> text op <+> prettyNix r
@ -44,19 +42,19 @@ prettyNix :: NExpr -> Doc
prettyNix (Fix expr) = go expr where
go (NConstant atom) = text $ unpack $ atomText atom
go (NOper oper) = prettyOper oper
go (NList list) = lbrack <+> (prettyFold $ map prettyNix list) <+> rbrack
go (NList list) = lbrack <+> (fsep $ map prettyNix list) <+> rbrack
go (NArgSet args) = lbrace <+> (prettyFold $ map prettySetArg $ toList args) <+> rbrace
go (NArgSet args) = lbrace <+> (vcat $ map prettySetArg $ toList args) <+> rbrace
go (NSet rec list) =
(case rec of Rec -> "rec"; NonRec -> empty)
<+> lbrace <+> (prettyFold $ map prettyBind list) <+> rbrace
<+> lbrace <+> (vcat $ map prettyBind list) <+> rbrace
go (NLet binds body) = text "let"
go (NIf cond trueBody falseBody) =
text "if" <+> prettyNix cond
<+> text "then" <+> prettyNix trueBody
<+> text "else" <+> prettyNix falseBody
(text "if" <+> prettyNix cond)
$$ (text "then" <+> prettyNix trueBody)
$$ (text "else" <+> prettyNix falseBody)
go (NWith scope body) = text "with" <+> prettyNix scope <> semi <+> prettyNix body
go (NAssert cond body) = text "assert" <+> prettyNix cond <> semi <+> prettyNix body