Refactor PR #119397 & improve pat range bound paren recovery

Co-authored-by: León Orell Valerian Liehr <[email protected]>

Author: ShE3py

compiler/rustc_parse/src/errors.rs

@@ -3,7 +3,7 @@ use std::borrow::Cow;
 use rustc_ast::token::Token;
 use rustc_ast::{Path, Visibility};
 use rustc_errors::{
-    AddToDiagnostic, Applicability, DiagCtxt, DiagnosticBuilder, IntoDiagnostic, Level, MultiSpan,
+    AddToDiagnostic, Applicability, DiagCtxt, DiagnosticBuilder, IntoDiagnostic, Level,
     SubdiagnosticMessage,
 };
 use rustc_macros::{Diagnostic, Subdiagnostic};
@@ -2382,7 +2382,7 @@ pub(crate) struct ExpectedCommaAfterPatternField {
 #[diag(parse_unexpected_paren_in_range_pat)]
 pub(crate) struct UnexpectedParenInRangePat {
     #[primary_span]
-    pub span: MultiSpan,
+    pub span: Vec<Span>,
     #[subdiagnostic]
     pub sugg: UnexpectedParenInRangePatSugg,
 }

compiler/rustc_parse/src/parser/pat.rs

@@ -1,14 +1,14 @@
 use super::{ForceCollect, Parser, PathStyle, TrailingToken};
 use crate::errors::{
-    AmbiguousRangePattern, BoxNotPat, DotDotDotForRemainingFields,
-    DotDotDotRangeToPatternNotAllowed, DotDotDotRestPattern, EnumPatternInsteadOfIdentifier,
-    ExpectedBindingLeftOfAt, ExpectedCommaAfterPatternField,
-    GenericArgsInPatRequireTurbofishSyntax, InclusiveRangeExtraEquals, InclusiveRangeMatchArrow,
-    InclusiveRangeNoEnd, InvalidMutInPattern, PatternOnWrongSideOfAt, RefMutOrderIncorrect,
-    RemoveLet, RepeatedMutInPattern, SwitchRefBoxOrder, TopLevelOrPatternNotAllowed,
-    TopLevelOrPatternNotAllowedSugg, TrailingVertNotAllowed, UnexpectedLifetimeInPattern,
-    UnexpectedParenInRangePat, UnexpectedParenInRangePatSugg,
-    UnexpectedVertVertBeforeFunctionParam, UnexpectedVertVertInPattern,
+    self, AmbiguousRangePattern, DotDotDotForRemainingFields, DotDotDotRangeToPatternNotAllowed,
+    DotDotDotRestPattern, EnumPatternInsteadOfIdentifier, ExpectedBindingLeftOfAt,
+    ExpectedCommaAfterPatternField, GenericArgsInPatRequireTurbofishSyntax,
+    InclusiveRangeExtraEquals, InclusiveRangeMatchArrow, InclusiveRangeNoEnd, InvalidMutInPattern,
+    PatternOnWrongSideOfAt, RefMutOrderIncorrect, RemoveLet, RepeatedMutInPattern,
+    SwitchRefBoxOrder, TopLevelOrPatternNotAllowed, TopLevelOrPatternNotAllowedSugg,
+    TrailingVertNotAllowed, UnexpectedLifetimeInPattern, UnexpectedParenInRangePat,
+    UnexpectedParenInRangePatSugg, UnexpectedVertVertBeforeFunctionParam,
+    UnexpectedVertVertInPattern,
 };
 use crate::{maybe_recover_from_interpolated_ty_qpath, maybe_whole};
 use rustc_ast::mut_visit::{noop_visit_pat, MutVisitor};
@@ -19,7 +19,7 @@ use rustc_ast::{
     PatField, PatFieldsRest, PatKind, Path, QSelf, RangeEnd, RangeSyntax,
 };
 use rustc_ast_pretty::pprust;
-use rustc_errors::{Applicability, DiagnosticBuilder, MultiSpan, PResult};
+use rustc_errors::{Applicability, DiagnosticBuilder, PResult};
 use rustc_session::errors::ExprParenthesesNeeded;
 use rustc_span::source_map::{respan, Spanned};
 use rustc_span::symbol::{kw, sym, Ident};
@@ -604,7 +604,7 @@ impl<'a> Parser<'a> {
                         && let Some(form) = self.parse_range_end() =>
                 {
                     self.dcx().emit_err(UnexpectedParenInRangePat {
-                        span: MultiSpan::from_spans(vec![open_paren, close_paren]),
+                        span: vec![open_paren, close_paren],
                         sugg: UnexpectedParenInRangePatSugg {
                             start_span: open_paren,
                             end_span: close_paren,
@@ -752,14 +752,6 @@ impl<'a> Parser<'a> {
         begin: P<Expr>,
         re: Spanned<RangeEnd>,
     ) -> PResult<'a, PatKind> {
-        // recover from `(`
-        let open_paren = (self.may_recover()
-            && self.token.kind == token::OpenDelim(Delimiter::Parenthesis))
-        .then(|| {
-            self.bump();
-            self.prev_token.span
-        });
-
         let end = if self.is_pat_range_end_start(0) {
             // Parsing e.g. `X..=Y`.
             Some(self.parse_pat_range_end()?)
@@ -771,19 +763,6 @@ impl<'a> Parser<'a> {
             }
             None
         };
-
-        if let Some(span) = open_paren {
-            self.expect(&token::CloseDelim(Delimiter::Parenthesis))?;
-
-            self.dcx().emit_err(UnexpectedParenInRangePat {
-                span: MultiSpan::from_spans(vec![span, self.prev_token.span]),
-                sugg: UnexpectedParenInRangePatSugg {
-                    start_span: span,
-                    end_span: self.prev_token.span,
-                },
-            });
-        }
-
         Ok(PatKind::Range(Some(begin), end, re))
     }
 
@@ -823,32 +802,11 @@ impl<'a> Parser<'a> {
     /// The form `...X` is prohibited to reduce confusion with the potential
     /// expression syntax `...expr` for splatting in expressions.
     fn parse_pat_range_to(&mut self, mut re: Spanned<RangeEnd>) -> PResult<'a, PatKind> {
-        // recover from `(`
-        let open_paren = (self.may_recover()
-            && self.token.kind == token::OpenDelim(Delimiter::Parenthesis))
-        .then(|| {
-            self.bump();
-            self.prev_token.span
-        });
-
         let end = self.parse_pat_range_end()?;
         if let RangeEnd::Included(syn @ RangeSyntax::DotDotDot) = &mut re.node {
             *syn = RangeSyntax::DotDotEq;
             self.dcx().emit_err(DotDotDotRangeToPatternNotAllowed { span: re.span });
         }
-
-        if let Some(span) = open_paren {
-            self.expect(&token::CloseDelim(Delimiter::Parenthesis))?;
-
-            self.dcx().emit_err(UnexpectedParenInRangePat {
-                span: MultiSpan::from_spans(vec![span, self.prev_token.span]),
-                sugg: UnexpectedParenInRangePatSugg {
-                    start_span: span,
-                    end_span: self.prev_token.span,
-                },
-            });
-        }
-
         Ok(PatKind::Range(None, Some(end), re))
     }
 
@@ -868,8 +826,14 @@ impl<'a> Parser<'a> {
             })
     }
 
+    /// Parse a range pattern end bound
     fn parse_pat_range_end(&mut self) -> PResult<'a, P<Expr>> {
-        if self.check_inline_const(0) {
+        // recover leading `(`
+        let open_paren = (self.may_recover()
+            && self.eat_noexpect(&token::OpenDelim(Delimiter::Parenthesis)))
+        .then_some(self.prev_token.span);
+
+        let bound = if self.check_inline_const(0) {
             self.parse_const_block(self.token.span, true)
         } else if self.check_path() {
             let lo = self.token.span;
@@ -885,6 +849,32 @@ impl<'a> Parser<'a> {
             Ok(self.mk_expr(lo.to(hi), ExprKind::Path(qself, path)))
         } else {
             self.parse_literal_maybe_minus()
+        };
+
+        // recover trailing `)`
+        if let Some(open_paren) = open_paren {
+            let bound = bound.map_err(|err| {
+                // we couldn't parse a bound after the `(`, so cancel the recovery
+                err.cancel();
+                self.dcx().struct_span_err(open_paren, "unexpected token: `(`")
+            })?;
+
+            if self.eat_noexpect(&token::CloseDelim(Delimiter::Parenthesis)) {
+                self.dcx().emit_err(UnexpectedParenInRangePat {
+                    span: vec![open_paren, self.prev_token.span],
+                    sugg: UnexpectedParenInRangePatSugg {
+                        start_span: open_paren,
+                        end_span: self.prev_token.span,
+                    },
+                });
+
+                Ok(bound)
+            } else {
+                // the `(bound` is missing its closing `)`, so cancel the recovery
+                Err(self.dcx().struct_span_err(open_paren, "unexpected token: `(`"))
+            }
+        } else {
+            bound
         }
     }
 
@@ -1013,7 +1003,7 @@ impl<'a> Parser<'a> {
 
         if self.isnt_pattern_start() {
             let descr = super::token_descr(&self.token);
-            self.dcx().emit_err(BoxNotPat {
+            self.dcx().emit_err(errors::BoxNotPat {
                 span: self.token.span,
                 kw: box_span,
                 lo: box_span.shrink_to_lo(),

tests/ui/half-open-range-patterns/range_pat_interactions2.rs

@@ -8,7 +8,7 @@ fn main() {
     for x in -9 + 1..=(9 - 2) {
         match x as i32 {
             0..=(5+1) => errors_only.push(x),
-            //~^ error: expected `)`, found `+`
+            //~^ error: unexpected token: `(`
             1 | -3..0 => first_or.push(x),
             y @ (0..5 | 6) => or_two.push(y),
             y @ 0..const { 5 + 1 } => assert_eq!(y, 5),

tests/ui/half-open-range-patterns/range_pat_interactions2.stderr

@@ -1,8 +1,8 @@
-error: expected `)`, found `+`
-  --> $DIR/range_pat_interactions2.rs:10:19
+error: unexpected token: `(`
+  --> $DIR/range_pat_interactions2.rs:10:17
    |
 LL |             0..=(5+1) => errors_only.push(x),
-   |                   ^ expected `)`
+   |                 ^
 
 error: aborting due to 1 previous error