Update connector search in ControlFlow::rewrite_pat_expr for for loops

Resolves 5009

For loops represented by a ControlFlow object use " in" as their connector.
rustfmt searches for the first uncommented occurrence of the word "in" within the
current span and adjusts it's starting point to look for comments right after that.
visually this looks like this:

    rustfmt starts looking for comments here
            |
            V
    for x in /* ... */ 0..1 {}

This works well in most cases, however when the pattern also contains
the word "in", this leads to issues.

    rustfmt starts looking for comments here
          |
          V
    for in_here in /* ... */ 0..1 {}
        -------
        pattern

In order to correctly identify the connector, the new approach first
updates the span to start after the pattern and then searches for the
first uncommented occurrence of "in".
This commit is contained in:
Yacin Tmimi 2021-10-06 13:09:24 -04:00 committed by Caleb Cartwright
parent f7c4a44149
commit f2fb3c9659
7 changed files with 73 additions and 1 deletions

View File

@ -822,7 +822,7 @@ impl<'a> ControlFlow<'a> {
let pat_string = pat.rewrite(context, pat_shape)?;
let comments_lo = context
.snippet_provider
.span_after(self.span, self.connector.trim());
.span_after(self.span.with_lo(pat.span.hi()), self.connector.trim());
let comments_span = mk_sp(comments_lo, expr.span.lo());
return rewrite_assign_rhs_with_comments(
context,

View File

@ -0,0 +1,4 @@
fn main() {
// the "in" inside the pattern produced invalid syntax
for variable_in_here /* ... */ in 0..1 {}
}

View File

@ -0,0 +1,3 @@
fn main() {
for in_in_in_in_in_in_in_in /* ... */ in 0..1 {}
}

View File

@ -0,0 +1,5 @@
fn main() {
for variable_in_x /* ... */ in 0..1 {
for variable_in_y /* ... */ in 0..1 {}
}
}

View File

@ -0,0 +1,13 @@
fn main() {
for variable_in_x /* ... */ in 0..1 {
for variable_in_y /* ... */ in 0..1 {
if false {
} else if false {
} else {
}
}
}
}

View File

@ -0,0 +1,15 @@
fn main() {
let in_ = false;
for variable_in_x /* ... */ in 0..1 {
for variable_in_y /* ... */ in 0..1 {
if in_ {
} else if in_ {
} else {
}
}
}
}

View File

@ -0,0 +1,32 @@
fn main() {
for variable_in_a /* ... */ in 0..1 {
for variable_in_b /* ... */ in 0..1 {
for variable_in_c /* ... */ in 0..1 {
for variable_in_d /* ... */ in 0..1 {
for variable_in_e /* ... */ in 0..1 {
for variable_in_f /* ... */ in 0..1 {
for variable_in_g /* ... */ in 0..1 {
for variable_in_h /* ... */ in 0..1 {
for variable_in_i /* ... */ in 0..1 {
for variable_in_j /* ... */ in 0..1 {
for variable_in_k /* ... */ in 0..1 {
for variable_in_l /* ... */ in 0..1 {
for variable_in_m /* ... */ in 0..1 {
for variable_in_n /* ... */ in 0..1 {
for variable_in_o /* ... */ in 0..1 {
}
}
}
}
}
}
}
}
}
}
}
}
}
}
}
}