waalge
/
aiken
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: everything lines up

This commit is contained in:
rvcas 2022-06-18 17:45:50 -04:00 committed by Lucas
parent d18ac475d1
commit 6a39d4349a
1 changed files with 37 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
crates/uplc/src/pretty.rs
View File

@ -5,9 +5,8 @@ use crate::ast::{Constant, Name, Program, Term};
impl Program<Name> { impl Program<Name> {
pub fn to_pretty(&self) -> String { pub fn to_pretty(&self) -> String {
let mut w = Vec::new(); let mut w = Vec::new();
println!("here");
self.to_doc().render(100, &mut w).unwrap(); self.to_doc().render(80, &mut w).unwrap();
String::from_utf8(w).unwrap() String::from_utf8(w).unwrap()
} }
@ -36,8 +35,9 @@ impl Term<Name> {
RcDoc::text("delay") RcDoc::text("delay")
.append(RcDoc::line()) .append(RcDoc::line())
.append(term.to_doc()) .append(term.to_doc())
.append(RcDoc::line_()), .nest(2),
) )
.append(RcDoc::line_())
.append(RcDoc::text(")")), .append(RcDoc::text(")")),
Term::Lambda { Term::Lambda {
parameter_name, parameter_name,
@ -51,40 +51,58 @@ impl Term<Name> {
parameter_name.text, parameter_name.unique parameter_name.text, parameter_name.unique
))) )))
.append(RcDoc::line()) .append(RcDoc::line())
.append(body.to_doc()), .append(body.to_doc())
.nest(2),
) )
.append(RcDoc::line_()) .append(RcDoc::line_())
.append(RcDoc::text(")")), .append(RcDoc::text(")")),
Term::Apply { function, argument } => RcDoc::text("[") Term::Apply { function, argument } => RcDoc::text("[")
.append(RcDoc::line()) .append(
.append(function.to_doc()) RcDoc::line()
.append(
function
.to_doc()
.append(RcDoc::line()) .append(RcDoc::line())
.append(argument.to_doc()) .append(argument.to_doc())
.group(),
)
.nest(2),
)
.append(RcDoc::line()) .append(RcDoc::line())
.append(RcDoc::text("]")), .append(RcDoc::text("]")),
Term::Constant(constant) => RcDoc::text("(") Term::Constant(constant) => RcDoc::text("(")
.append(RcDoc::text("con")) .append(
RcDoc::text("con")
.append(RcDoc::line()) .append(RcDoc::line())
.append(constant.to_doc()) .append(constant.to_doc())
.nest(2),
)
.append(RcDoc::line_()) .append(RcDoc::line_())
.append(RcDoc::text(")")), .append(RcDoc::text(")")),
Term::Force(term) => RcDoc::text("(") Term::Force(term) => RcDoc::text("(")
.append(RcDoc::text("force")) .append(
RcDoc::text("force")
.append(RcDoc::line()) .append(RcDoc::line())
.append(term.to_doc()) .append(term.to_doc())
.nest(2),
)
.append(RcDoc::line_()) .append(RcDoc::line_())
.append(RcDoc::text(")")), .append(RcDoc::text(")")),
Term::Error => RcDoc::text("(") Term::Error => RcDoc::text("(")
.append(RcDoc::text("error")) .append(RcDoc::text("error").nest(2))
.append(RcDoc::line())
.append(RcDoc::line_())
.append(RcDoc::text(")")), .append(RcDoc::text(")")),
Term::Builtin(builtin) => RcDoc::text("(") Term::Builtin(builtin) => RcDoc::text("(")
.append(RcDoc::text("builtin")) .append(
RcDoc::text("builtin")
.append(RcDoc::line()) .append(RcDoc::line())
.append(RcDoc::text(builtin.to_string())) .append(RcDoc::text(builtin.to_string()))
.nest(2),
)
.append(RcDoc::line_()) .append(RcDoc::line_())
.append(RcDoc::text(")")), .append(RcDoc::text(")")),
} }
.nest(2)
.group() .group()
} }
} }