Skip to content

Commit

Permalink
Suggest => --> >= in conditions
Browse files Browse the repository at this point in the history
  • Loading branch information
sjwang05 committed Nov 11, 2023
1 parent ed086d8 commit 725a73c
Show file tree
Hide file tree
Showing 5 changed files with 183 additions and 0 deletions.
12 changes: 12 additions & 0 deletions compiler/rustc_parse/src/parser/diagnostics.rs
Original file line number Diff line number Diff line change
Expand Up @@ -610,6 +610,18 @@ impl<'a> Parser<'a> {
// FIXME: translation requires list formatting (for `expect`)
let mut err = self.struct_span_err(self.token.span, msg_exp);

// Look for usages of '=>' where '>=' was probably intended
if self.token == token::FatArrow
&& expected.iter().any(|tok| matches!(tok, TokenType::Token(TokenKind::Le)))
{
err.span_suggestion(
self.token.span,
"you might have meant to write a \"greater than or equal to\" comparison",
">=",
Applicability::Unspecified,
);
}

if let TokenKind::Ident(symbol, _) = &self.prev_token.kind {
if ["def", "fun", "func", "function"].contains(&symbol.as_str()) {
err.span_suggestion_short(
Expand Down
10 changes: 10 additions & 0 deletions compiler/rustc_parse/src/parser/expr.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2432,6 +2432,7 @@ impl<'a> Parser<'a> {
}
} else {
let attrs = self.parse_outer_attributes()?; // For recovery.
let maybe_fatarrow = self.token.clone();
let block = if self.check(&token::OpenDelim(Delimiter::Brace)) {
self.parse_block()?
} else {
Expand All @@ -2455,6 +2456,15 @@ impl<'a> Parser<'a> {
self.token.span.to(maybe_let.span),
"you likely meant to continue parsing the let-chain starting here",
);
}
// Look for usages of '=>' where '>=' was probably intended
if maybe_fatarrow.kind == token::FatArrow {
err.span_suggestion(
maybe_fatarrow.span,
"you probably meant to write a \"greater than or equal to\" comparison",
">=",
Applicability::MachineApplicable,
);
} else {
err.span_note(
cond_span,
Expand Down
45 changes: 45 additions & 0 deletions tests/ui/parser/eq-gt-to-gt-eq.fixed
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
// run-rustfix
// Check that we try to correct `=>` to `>=` in conditions.
#![allow(unused)]

fn main() {
let a = 0;
let b = 1;
if a >= b {} //~ERROR
}

fn foo() {
let a = 0;
if a >= 1 {} //~ERROR
}

fn a() {
let a = 0;
if 1 >= a {} //~ERROR
}

fn bar() {
let a = 0;
let b = 1;
if a >= b && a != b {} //~ERROR
}

fn qux() {
let a = 0;
let b = 1;
if a != b && a >= b {} //~ERROR
}

fn baz() {
let a = 0;
let b = 1;
let _ = a => b; //~ERROR
}

fn b() {
let a = 0;
let b = 1;
match a => b { //~ERROR
_ => todo!(),
}
}
45 changes: 45 additions & 0 deletions tests/ui/parser/eq-gt-to-gt-eq.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
// run-rustfix
// Check that we try to correct `=>` to `>=` in conditions.
#![allow(unused)]

fn main() {
let a = 0;
let b = 1;
if a => b {} //~ERROR
}

fn foo() {
let a = 0;
if a => 1 {} //~ERROR
}

fn a() {
let a = 0;
if 1 => a {} //~ERROR
}

fn bar() {
let a = 0;
let b = 1;
if a => b && a != b {} //~ERROR
}

fn qux() {
let a = 0;
let b = 1;
if a != b && a => b {} //~ERROR
}

fn baz() {
let a = 0;
let b = 1;
let _ = a => b; //~ERROR
}

fn b() {
let a = 0;
let b = 1;
match a => b { //~ERROR
_ => todo!(),
}
}
71 changes: 71 additions & 0 deletions tests/ui/parser/eq-gt-to-gt-eq.stderr
Original file line number Diff line number Diff line change
@@ -0,0 +1,71 @@
error: expected `{`, found `=>`
--> $DIR/eq-gt-to-gt-eq.rs:8:10
|
LL | if a => b {}
| ^^ expected `{`
|
help: you probably meant to write a "greater than or equal to" comparison
|
LL | if a >= b {}
| ~~

error: expected `{`, found `=>`
--> $DIR/eq-gt-to-gt-eq.rs:13:10
|
LL | if a => 1 {}
| ^^ expected `{`
|
help: you probably meant to write a "greater than or equal to" comparison
|
LL | if a >= 1 {}
| ~~

error: expected `{`, found `=>`
--> $DIR/eq-gt-to-gt-eq.rs:18:10
|
LL | if 1 => a {}
| ^^ expected `{`
|
help: you probably meant to write a "greater than or equal to" comparison
|
LL | if 1 >= a {}
| ~~

error: expected `{`, found `=>`
--> $DIR/eq-gt-to-gt-eq.rs:24:10
|
LL | if a => b && a != b {}
| ^^ expected `{`
|
help: you probably meant to write a "greater than or equal to" comparison
|
LL | if a >= b && a != b {}
| ~~

error: expected `{`, found `=>`
--> $DIR/eq-gt-to-gt-eq.rs:30:20
|
LL | if a != b && a => b {}
| ^^ expected `{`
|
help: you probably meant to write a "greater than or equal to" comparison
|
LL | if a != b && a >= b {}
| ~~

error: expected one of `!`, `.`, `::`, `;`, `?`, `else`, `{`, or an operator, found `=>`
--> $DIR/eq-gt-to-gt-eq.rs:36:15
|
LL | let _ = a => b;
| ^^ expected one of 8 possible tokens

error: expected one of `!`, `.`, `::`, `?`, `{`, or an operator, found `=>`
--> $DIR/eq-gt-to-gt-eq.rs:42:13
|
LL | match a => b {
| ----- ^^ expected one of `!`, `.`, `::`, `?`, `{`, or an operator
| |
| while parsing this `match` expression

error: aborting due to 7 previous errors

0 comments on commit 725a73c

Please sign in to comment.