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

feat: if expression formatting

Browse Source
This commit is contained in:
rvcas
2022-11-05 17:53:11 -04:00
parent 809d5ea5c5
commit 7092673c45
2 changed files with 48 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
crates/lang/src/format.rs
View File

@@ -601,7 +601,48 @@ impl<'comments> Formatter<'comments> {
let document = match expr {
UntypedExpr::ByteArray { .. } => todo!(),
UntypedExpr::If { .. } => todo!(),
UntypedExpr::If {
branches,
final_else,
..
} => {
let first = branches.first();
break_("if", "if ")
.append(self.wrap_expr(&first.condition))
.nest(INDENT)
.append(break_("", " "))
.append("{")
.group()
.append(line())
.nest(INDENT)
.append(self.expr(&first.body))
.append(line())
.append("} ")
.append(join(
branches[1..].iter().map(|branch| {
break_("else if", "else if ")
.append(self.wrap_expr(&branch.condition))
.nest(INDENT)
.append(break_("", " "))
.append("{")
.group()
.append(line())
.nest(INDENT)
.append(self.expr(&branch.body))
.append(line())
.append("} ")
}),
nil(),
))
.append("else {")
.group()
.append(line().nest(INDENT))
.append(self.expr(final_else))
.append(line())
.append("}")
.force_break()
}
UntypedExpr::Todo { label: None, .. } => "todo".to_doc(),
UntypedExpr::Todo { label: Some(l), .. } => docvec!["todo(\"", l, "\")"],