rust-clippy/src/utils.rs

43 lines
1.5 KiB
Rust
Raw Normal View History

2015-06-01 20:30:34 +00:00
use rustc::lint::Context;
use syntax::ast::{DefId, Name, Path};
use syntax::codemap::{ExpnInfo, Span};
use rustc::middle::ty;
2015-06-01 20:30:34 +00:00
/// returns true if the macro that expanded the crate was outside of
/// the current crate or was a compiler plugin
2015-06-01 20:36:56 +00:00
pub fn in_macro(cx: &Context, opt_info: Option<&ExpnInfo>) -> bool {
// no ExpnInfo = no macro
2015-06-01 20:30:34 +00:00
opt_info.map_or(false, |info| {
// no span for the callee = external macro
2015-06-01 20:30:34 +00:00
info.callee.span.map_or(true, |span| {
// no snippet = external macro or compiler-builtin expansion
2015-06-01 20:30:34 +00:00
cx.sess().codemap().span_to_snippet(span).ok().map_or(true, |code|
// macro doesn't start with "macro_rules"
// = compiler plugin
2015-06-01 20:30:34 +00:00
!code.starts_with("macro_rules")
)
})
})
}
/// invokes in_macro with the expansion info of the given span
pub fn in_external_macro(cx: &Context, span: Span) -> bool {
cx.sess().codemap().with_expn_info(span.expn_id,
|info| in_macro(cx, info))
}
/// check if a DefId's path matches the given absolute type path
/// usage e.g. with
/// `match_def_path(cx, id, &["core", "option", "Option"])`
pub fn match_def_path(cx: &Context, def_id: DefId, path: &[&str]) -> bool {
cx.tcx.with_path(def_id, |iter| iter.map(|elem| elem.name())
.zip(path.iter()).all(|(nm, p)| &nm.as_str() == p))
}
/// match a Path against a slice of segment string literals, e.g.
/// `match_path(path, &["std", "rt", "begin_unwind"])`
pub fn match_path(path: &Path, segments: &[&str]) -> bool {
path.segments.iter().rev().zip(segments.iter().rev()).all(
|(a,b)| a.identifier.as_str() == *b)
}