language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-10-22 09:31:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

Fix when in when guard

Browse source
This commit is contained in:
Joshua Warner 2025-01-10 22:36:23 -08:00
parent 586dc3486f
commit 903b792fd3
No known key found for this signature in database
GPG key ID: 89AD497003F93FDD
6 changed files with 117 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

29
crates/compiler/fmt/src/expr.rs
View file

@ -788,6 +788,7 @@ fn fmt_parens(sub_expr: &Expr<'_>, buf: &mut Buf<'_>, indent: u16) {
let should_add_newlines = match sub_expr {
Expr::Closure(..)
| Expr::SpaceBefore(..)
| Expr::When(..)
| Expr::SpaceAfter(Expr::Closure(..), ..)
| Expr::DbgStmt { .. } => false,
_ => sub_expr.is_multiline(),
@ -881,7 +882,7 @@ fn starts_with_newline(expr: &Expr) -> bool {
SpaceBefore(_, comment_or_newline) => {
matches!(comment_or_newline.first(), Some(CommentOrNewline::Newline))
}
DbgStmt { .. } => true,
DbgStmt { .. } | When(..) => true,
_ => false,
}
}
@ -1724,7 +1725,20 @@ fn fmt_when<'a>(
buf.indent(indent + INDENT);
buf.push_str(" if");
buf.spaces(1);
guard_expr.format_with_options(buf, Parens::NotNeeded, Newlines::Yes, indent + INDENT);
let guard_lifted =
expr_lift_spaces(Parens::NotNeeded, buf.text.bump(), &guard_expr.value);
if guard_needs_parens(&guard_lifted.item) {
fmt_parens(&lower(buf.text.bump(), guard_lifted), buf, indent + INDENT);
} else {
lower(buf.text.bump(), guard_lifted).format_with_options(
buf,
Parens::NotNeeded,
Newlines::Yes,
indent + INDENT,
);
}
}
buf.indent(indent + INDENT);
@ -1741,7 +1755,6 @@ fn fmt_when<'a>(
buf.spaces(1);
}
// expr.format_with_options(buf, Parens::NotNeeded, Newlines::Yes, inner_indent);
format_expr_only(
&expr.item,
buf,
@ -1760,6 +1773,16 @@ fn fmt_when<'a>(
}
}
fn guard_needs_parens(value: &Expr<'_>) -> bool {
match value {
Expr::When(..) => true,
Expr::ParensAround(expr) | Expr::SpaceBefore(expr, _) | Expr::SpaceAfter(expr, _) => {
guard_needs_parens(expr)
}
_ => false,
}
}
fn fmt_dbg_stmt<'a>(
buf: &mut Buf,
condition: &'a Loc<Expr<'a>>,