-
Notifications
You must be signed in to change notification settings - Fork 13.4k
Fix range literals borrowing suggestions #54734
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Changes from 1 commit
0626afb
f7d35cc
87bf9e2
611e5c4
882e373
1f7dafb
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -317,7 +317,11 @@ impl<'a, 'gcx, 'tcx> FnCtxt<'a, 'gcx, 'tcx> { | |
_ if self.is_range_literal(expr) => true, | ||
_ => false, | ||
}; | ||
let sugg_expr = if needs_parens { format!("({})", src) } else { src }; | ||
let sugg_expr = if needs_parens { | ||
format!("({})", src) | ||
} else { | ||
src | ||
}; | ||
|
||
if let Some(sugg) = self.can_use_as_ref(expr) { | ||
return Some(sugg); | ||
|
@@ -379,67 +383,56 @@ impl<'a, 'gcx, 'tcx> FnCtxt<'a, 'gcx, 'tcx> { | |
None | ||
} | ||
|
||
// This function checks if the specified expression is a built-in range literal | ||
// (See: librustc/hir/lowering.rs::LoweringContext::lower_expr() ) | ||
/// This function checks if the specified expression is a built-in range literal. | ||
/// (See: ``librustc::hir::lowering::LoweringContext::lower_expr()``). | ||
fn is_range_literal(&self, expr: &hir::Expr) -> bool { | ||
use hir::{Path, QPath, ExprKind, TyKind}; | ||
|
||
// we support `::std::ops::Range` and `::std::core::Range` prefixes | ||
// (via split on "|") | ||
let ops_path = ["{{root}}", "std|core", "ops"]; | ||
|
||
// We support `::std::ops::Range` and `::core::ops::Range` prefixes | ||
let is_range_path = |path: &Path| { | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. I might be alone in this (cc @estebank), but I think for a short path like this, a more explicit approach (like below) is more readable. (I specifically don't like splitting on "|" because it feels a little too ad hoc.) let is_range_path = |path: &Path| {
let segs = path.segments.iter().map(|seg| seg.ident.as_str());
if let (Some("{{root}}"), std_core, "ops", range, None) = (segs.next(), segs.next(), segs.next(), segs.next(), segs.next()) {
(std_core == "std" || std_core == "ops") && range.starts_with("Range")
} else {
false
}
}; There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. That's a great suggestion! I was looking for something concise like this syntax-wise, it didn't occur to me that I could use |
||
let ident_names: Vec<_> = path.segments | ||
.iter() | ||
.map(|seg| seg.ident.as_str()) | ||
.collect(); | ||
|
||
if let Some((last, preceding)) = ident_names.split_last() { | ||
last.starts_with("Range") && | ||
preceding.len() == 3 && | ||
preceding.iter() | ||
.zip(ops_path.iter()) | ||
.all(|(seg, match_seg)| { | ||
match_seg.split("|") | ||
.into_iter() | ||
.any(|ref spl_seg| seg == spl_seg) | ||
}) | ||
let mut segs = path.segments.iter() | ||
.map(|seg| seg.ident.as_str()); | ||
|
||
if let (Some(root), Some(std_core), Some(ops), Some(range), None) = | ||
(segs.next(), segs.next(), segs.next(), segs.next(), segs.next()) | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more.
|
||
{ | ||
// "{{root}}" is the equivalent of `::` prefix in Path | ||
root == "{{root}}" && (std_core == "std" || std_core == "core") | ||
&& ops == "ops" && range.starts_with("Range") | ||
} else { | ||
false | ||
} | ||
}; | ||
|
||
let is_range_struct_snippet = |span: &Span| { | ||
// Tell if expression span snippet looks like an explicit | ||
// Range struct or new() call. This is to allow rejecting | ||
// Ranges constructed with non-literals. | ||
let is_range_literal = |span: &Span| { | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. As you pointed out, it would be clearer to give this a different name than the function it's in. Maybe |
||
// Tell if expression span snippet doesn't look like an explicit | ||
// Range struct or `new()` call. This is to allow inferring | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. "Inferring" has a very specific meaning, so I'd tend to avoid it in favour of something like: // Check whether a span corresponding to a range is a range literal,
// rather than an explicit struct or `new()` call. |
||
// that this is a range literal. | ||
let source_map = self.tcx.sess.source_map(); | ||
let end_point = source_map.end_point(*span); | ||
|
||
if let Ok(end_string) = source_map.span_to_snippet(end_point) { | ||
end_string.ends_with("}") || end_string.ends_with(")") | ||
!(end_string.ends_with("}") || end_string.ends_with(")")) | ||
} else { | ||
false | ||
} | ||
|
||
}; | ||
|
||
match expr.node { | ||
// all built-in range literals but `..=` and `..` | ||
// desugar to Structs, `..` desugars to its struct path | ||
// All built-in range literals but `..=` and `..` desugar to Structs | ||
ExprKind::Struct(QPath::Resolved(None, ref path), _, _) | | ||
// `..` desugars to its struct path | ||
ExprKind::Path(QPath::Resolved(None, ref path)) => { | ||
pawroman marked this conversation as resolved.
Show resolved
Hide resolved
|
||
return is_range_path(&path) && !is_range_struct_snippet(&expr.span); | ||
return is_range_path(&path) && is_range_literal(&expr.span); | ||
} | ||
|
||
// `..=` desugars into RangeInclusive::new(...) | ||
// `..=` desugars into `::std::ops::RangeInclusive::new(...)` | ||
ExprKind::Call(ref func, _) => { | ||
if let ExprKind::Path(QPath::TypeRelative(ref ty, ref segment)) = func.node { | ||
if let TyKind::Path(QPath::Resolved(None, ref path)) = ty.node { | ||
let calls_new = segment.ident.as_str() == "new"; | ||
let call_to_new = segment.ident.as_str() == "new"; | ||
|
||
return is_range_path(&path) && calls_new && | ||
!is_range_struct_snippet(&expr.span); | ||
return is_range_path(&path) && is_range_literal(&expr.span) && call_to_new; | ||
} | ||
} | ||
} | ||
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Sorry, I meant single rather than double backticks; I should have clarified. On second thoughts, though, it might be more convenient to include the path:
/// (See `LoweringContext::lower_expr()` in `src/librustc/hir/lowering.rs`.)
as then someone can jump straight to the path. (Like in the following.)
https://github.com/rust-lang/rust/blob/f89622a7548f69c76e5c9857e34be568ba322d02/src/librustc_privacy/lib.rs#L1388