nordic-dev.net/rust
nordic-dev.net/rust
2
0
Fork 0
mirror of https://github.com/rust-lang/rust.git synced 2025-02-25 13:24:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Visit all attributes for feature collection

Browse Source
This commit is contained in:
varkor 2018-08-15 17:34:56 +01:00
parent a5733050de
commit 23d10ec4cc
1 changed files with 39 additions and 55 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

94
src/librustc/middle/lib_features.rs
View File

@ -18,8 +18,7 @@ use ty::TyCtxt;
use syntax::symbol::Symbol; use syntax::symbol::Symbol;
use syntax::ast::{Attribute, MetaItem, MetaItemKind}; use syntax::ast::{Attribute, MetaItem, MetaItemKind};
use syntax_pos::{Span, DUMMY_SP}; use syntax_pos::{Span, DUMMY_SP};
use hir; use hir::intravisit::{self, NestedVisitorMap, Visitor};
use hir::itemlikevisit::ItemLikeVisitor;
use rustc_data_structures::fx::{FxHashSet, FxHashMap}; use rustc_data_structures::fx::{FxHashSet, FxHashMap};
use errors::DiagnosticId; use errors::DiagnosticId;
@ -59,47 +58,44 @@ impl<'a, 'tcx> LibFeatureCollector<'a, 'tcx> {
} }
} }
fn extract(&self, attrs: &[Attribute]) -> Vec<(Symbol, Option<Symbol>, Span)> { fn extract(&self, attr: &Attribute) -> Option<(Symbol, Option<Symbol>, Span)> {
let stab_attrs = vec!["stable", "unstable", "rustc_const_unstable"]; let stab_attrs = vec!["stable", "unstable", "rustc_const_unstable"];
let mut features = vec![];
for attr in attrs { // Find a stability attribute (i.e. `#[stable (..)]`, `#[unstable (..)]`,
// Find a stability attribute (i.e. `#[stable (..)]`, `#[unstable (..)]`, // `#[rustc_const_unstable (..)]`).
// `#[rustc_const_unstable (..)]`). if let Some(stab_attr) = stab_attrs.iter().find(|stab_attr| {
if let Some(stab_attr) = stab_attrs.iter().find(|stab_attr| { attr.check_name(stab_attr)
attr.check_name(stab_attr) }) {
}) { let meta_item = attr.meta();
let meta_item = attr.meta(); if let Some(MetaItem { node: MetaItemKind::List(ref metas), .. }) = meta_item {
if let Some(MetaItem { node: MetaItemKind::List(ref metas), .. }) = meta_item { let mut feature = None;
let mut feature = None; let mut since = None;
let mut since = None; for meta in metas {
for meta in metas { if let Some(mi) = meta.meta_item() {
if let Some(mi) = meta.meta_item() { // Find the `feature = ".."` meta-item.
// Find the `feature = ".."` meta-item. match (&*mi.name().as_str(), mi.value_str()) {
match (&*mi.name().as_str(), mi.value_str()) { ("feature", val) => feature = val,
("feature", val) => feature = val, ("since", val) => since = val,
("since", val) => since = val, _ => {}
_ => {}
}
} }
} }
if let Some(feature) = feature {
// This additional check for stability is to make sure we
// don't emit additional, irrelevant errors for malformed
// attributes.
if *stab_attr != "stable" || since.is_some() {
features.push((feature, since, attr.span));
}
}
// We need to iterate over the other attributes, because
// `rustc_const_unstable` is not mutually exclusive with
// the other stability attributes, so we can't just `break`
// here.
} }
if let Some(feature) = feature {
// This additional check for stability is to make sure we
// don't emit additional, irrelevant errors for malformed
// attributes.
if *stab_attr != "stable" || since.is_some() {
return Some((feature, since, attr.span));
}
}
// We need to iterate over the other attributes, because
// `rustc_const_unstable` is not mutually exclusive with
// the other stability attributes, so we can't just `break`
// here.
} }
} }
features None
} }
fn collect_feature(&mut self, feature: Symbol, since: Option<Symbol>, span: Span) { fn collect_feature(&mut self, feature: Symbol, since: Option<Symbol>, span: Span) {
@ -140,25 +136,17 @@ impl<'a, 'tcx> LibFeatureCollector<'a, 'tcx> {
} }
} }
} }
fn collect_from_attrs(&mut self, attrs: &[Attribute]) {
for (feature, stable, span) in self.extract(attrs) {
self.collect_feature(feature, stable, span);
}
}
} }
impl<'a, 'v, 'tcx> ItemLikeVisitor<'v> for LibFeatureCollector<'a, 'tcx> { impl<'a, 'tcx> Visitor<'tcx> for LibFeatureCollector<'a, 'tcx> {
fn visit_item(&mut self, item: &hir::Item) { fn nested_visit_map<'this>(&'this mut self) -> NestedVisitorMap<'this, 'tcx> {
self.collect_from_attrs(&item.attrs); NestedVisitorMap::All(&self.tcx.hir)
} }
fn visit_trait_item(&mut self, trait_item: &hir::TraitItem) { fn visit_attribute(&mut self, attr: &'tcx Attribute) {
self.collect_from_attrs(&trait_item.attrs); if let Some((feature, stable, span)) = self.extract(attr) {
} self.collect_feature(feature, stable, span);
}
fn visit_impl_item(&mut self, impl_item: &hir::ImplItem) {
self.collect_from_attrs(&impl_item.attrs);
} }
} }
@ -169,10 +157,6 @@ pub fn collect<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>) -> LibFeatures {
collector.collect_feature(feature, since, DUMMY_SP); collector.collect_feature(feature, since, DUMMY_SP);
} }
} }
collector.collect_from_attrs(&tcx.hir.krate().attrs); intravisit::walk_crate(&mut collector, tcx.hir.krate());
tcx.hir.krate().visit_all_item_likes(&mut collector);
for exported_macro in &tcx.hir.krate().exported_macros {
collector.collect_from_attrs(&exported_macro.attrs);
}
collector.lib_features collector.lib_features
} }