language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-08-03 19:58:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

Fix when indent in the presence of weird multiline patterns

Browse source
This commit is contained in:
Joshua Warner 2024-12-13 11:05:16 -08:00
parent 1d51f5fbec
commit 02e07f95e7
No known key found for this signature in database
GPG key ID: 89AD497003F93FDD
6 changed files with 79 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1514,7 +1514,6 @@ fn fmt_when<'a>(
buf: &mut Buf,
loc_condition: &'a Loc<Expr<'a>>,
branches: &[&'a WhenBranch<'a>],
indent: u16,
) {
buf.ensure_ends_with_newline();
@ -1604,11 +1603,14 @@ fn fmt_when<'a>(
}
buf.indent(indent + INDENT);
let line_indent = buf.cur_line_indent();
buf.push_str(" ->");
let inner_indent = line_indent + INDENT;
match expr.value {
Expr::SpaceBefore(nested, spaces) => {
fmt_spaces_no_blank_lines(buf, spaces.iter(), indent + (INDENT * 2));
fmt_spaces_no_blank_lines(buf, spaces.iter(), inner_indent);
if is_multiline_expr {
buf.ensure_ends_with_newline();
@ -1616,12 +1618,7 @@ fn fmt_when<'a>(
buf.spaces(1);
}
nested.format_with_options(
buf,
Parens::NotNeeded,
Newlines::Yes,
indent + 2 * INDENT,
);
nested.format_with_options(buf, Parens::NotNeeded, Newlines::Yes, inner_indent);
}
_ => {
if is_multiline_expr {
@ -1630,12 +1627,7 @@ fn fmt_when<'a>(
buf.spaces(1);
}
expr.format_with_options(
buf,
Parens::NotNeeded,
Newlines::Yes,
indent + 2 * INDENT,
);
expr.format_with_options(buf, Parens::NotNeeded, Newlines::Yes, inner_indent);
}
}

1
crates/compiler/fmt/src/lib.rs
View file

@ -69,6 +69,7 @@ impl<'a> Buf<'a> {
self.beginning_of_line = false;
}
#[track_caller]
pub fn cur_line_indent(&self) -> u16 {
debug_assert!(!self.beginning_of_line, "cur_line_indent before indent");
self.line_indent

5
crates/compiler/test_syntax/tests/snapshots/pass/nested_when_comment_in_pat.expr.formatted.roc Normal file
View file

@ -0,0 +1,5 @@
when 6 is
O #
B ->
when 6 is
1 -> O

62
crates/compiler/test_syntax/tests/snapshots/pass/nested_when_comment_in_pat.expr.result-ast Normal file
View file

@ -0,0 +1,62 @@
@0-31 SpaceAfter(
When(
@5-6 Num(
"6",
),
[
WhenBranch {
patterns: [
@10-15 SpaceBefore(
Apply(
@10-11 Tag(
"O",
),
[
@14-15 SpaceBefore(
Tag(
"B",
),
[
LineComment(
"",
),
],
),
],
),
[
Newline,
],
),
],
value: @17-31 When(
@22-23 Num(
"6",
),
[
WhenBranch {
patterns: [
@27-28 SpaceBefore(
NumLiteral(
"1",
),
[
Newline,
],
),
],
value: @30-31 Tag(
"O",
),
guard: None,
},
],
),
guard: None,
},
],
),
[
Newline,
],
)

4
crates/compiler/test_syntax/tests/snapshots/pass/nested_when_comment_in_pat.expr.roc Normal file
View file

@ -0,0 +1,4 @@
when 6 is
O#
B->when 6 is
1->O

1
crates/compiler/test_syntax/tests/test_snapshots.rs
View file

@ -500,6 +500,7 @@ mod test_snapshots {
pass/nested_if.expr,
pass/nested_list_comment_in_closure_arg.expr,
pass/nested_parens_in_pattern.expr,
pass/nested_when_comment_in_pat.expr,
pass/newline_after_equals.expr, // Regression test for https://github.com/roc-lang/roc/issues/51
pass/newline_after_mul.expr,
pass/newline_after_opt_field.expr,