Add Attribute::meta_kind
This commit is contained in:
parent
f8abed9ed4
commit
047275a682
4 changed files with 32 additions and 11 deletions
|
@ -136,15 +136,15 @@ impl Attribute {
|
||||||
|
|
||||||
pub fn value_str(&self) -> Option<Symbol> {
|
pub fn value_str(&self) -> Option<Symbol> {
|
||||||
match self.kind {
|
match self.kind {
|
||||||
AttrKind::Normal(ref item, _) => item.meta(self.span).and_then(|meta| meta.value_str()),
|
AttrKind::Normal(ref item, _) => item.meta_kind().and_then(|kind| kind.value_str()),
|
||||||
AttrKind::DocComment(..) => None,
|
AttrKind::DocComment(..) => None,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn meta_item_list(&self) -> Option<Vec<NestedMetaItem>> {
|
pub fn meta_item_list(&self) -> Option<Vec<NestedMetaItem>> {
|
||||||
match self.kind {
|
match self.kind {
|
||||||
AttrKind::Normal(ref item, _) => match item.meta(self.span) {
|
AttrKind::Normal(ref item, _) => match item.meta_kind() {
|
||||||
Some(MetaItem { kind: MetaItemKind::List(list), .. }) => Some(list),
|
Some(MetaItemKind::List(list)) => Some(list),
|
||||||
_ => None,
|
_ => None,
|
||||||
},
|
},
|
||||||
AttrKind::DocComment(..) => None,
|
AttrKind::DocComment(..) => None,
|
||||||
|
@ -228,6 +228,10 @@ impl AttrItem {
|
||||||
span,
|
span,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn meta_kind(&self) -> Option<MetaItemKind> {
|
||||||
|
Some(MetaItemKind::from_mac_args(&self.args)?)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Attribute {
|
impl Attribute {
|
||||||
|
@ -242,7 +246,7 @@ impl Attribute {
|
||||||
match self.kind {
|
match self.kind {
|
||||||
AttrKind::DocComment(.., data) => Some(data),
|
AttrKind::DocComment(.., data) => Some(data),
|
||||||
AttrKind::Normal(ref item, _) if item.path == sym::doc => {
|
AttrKind::Normal(ref item, _) if item.path == sym::doc => {
|
||||||
item.meta(self.span).and_then(|meta| meta.value_str())
|
item.meta_kind().and_then(|kind| kind.value_str())
|
||||||
}
|
}
|
||||||
_ => None,
|
_ => None,
|
||||||
}
|
}
|
||||||
|
@ -270,6 +274,13 @@ impl Attribute {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn meta_kind(&self) -> Option<MetaItemKind> {
|
||||||
|
match self.kind {
|
||||||
|
AttrKind::Normal(ref item, _) => item.meta_kind(),
|
||||||
|
AttrKind::DocComment(..) => None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
pub fn tokens(&self) -> AttrAnnotatedTokenStream {
|
pub fn tokens(&self) -> AttrAnnotatedTokenStream {
|
||||||
match self.kind {
|
match self.kind {
|
||||||
AttrKind::Normal(_, ref tokens) => tokens
|
AttrKind::Normal(_, ref tokens) => tokens
|
||||||
|
@ -436,6 +447,16 @@ impl MetaItem {
|
||||||
}
|
}
|
||||||
|
|
||||||
impl MetaItemKind {
|
impl MetaItemKind {
|
||||||
|
pub fn value_str(&self) -> Option<Symbol> {
|
||||||
|
match self {
|
||||||
|
MetaItemKind::NameValue(ref v) => match v.kind {
|
||||||
|
LitKind::Str(ref s, _) => Some(*s),
|
||||||
|
_ => None,
|
||||||
|
},
|
||||||
|
_ => None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
pub fn mac_args(&self, span: Span) -> MacArgs {
|
pub fn mac_args(&self, span: Span) -> MacArgs {
|
||||||
match self {
|
match self {
|
||||||
MetaItemKind::Word => MacArgs::Empty,
|
MetaItemKind::Word => MacArgs::Empty,
|
||||||
|
|
|
@ -484,7 +484,7 @@ pub(crate) fn check_attr_crate_type(
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
if let ast::MetaItemKind::NameValue(spanned) = a.meta().unwrap().kind {
|
if let ast::MetaItemKind::NameValue(spanned) = a.meta_kind().unwrap() {
|
||||||
let span = spanned.span;
|
let span = spanned.span;
|
||||||
let lev_candidate = find_best_match_for_name(
|
let lev_candidate = find_best_match_for_name(
|
||||||
&CRATE_TYPES.iter().map(|(k, _)| *k).collect::<Vec<_>>(),
|
&CRATE_TYPES.iter().map(|(k, _)| *k).collect::<Vec<_>>(),
|
||||||
|
|
|
@ -4,7 +4,7 @@
|
||||||
// and `#[unstable (..)]`), but are not declared in one single location
|
// and `#[unstable (..)]`), but are not declared in one single location
|
||||||
// (unlike lang features), which means we need to collect them instead.
|
// (unlike lang features), which means we need to collect them instead.
|
||||||
|
|
||||||
use rustc_ast::{Attribute, MetaItem, MetaItemKind};
|
use rustc_ast::{Attribute, MetaItemKind};
|
||||||
use rustc_errors::struct_span_err;
|
use rustc_errors::struct_span_err;
|
||||||
use rustc_hir::intravisit::{NestedVisitorMap, Visitor};
|
use rustc_hir::intravisit::{NestedVisitorMap, Visitor};
|
||||||
use rustc_middle::hir::map::Map;
|
use rustc_middle::hir::map::Map;
|
||||||
|
@ -34,8 +34,8 @@ impl<'tcx> LibFeatureCollector<'tcx> {
|
||||||
// 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| attr.has_name(**stab_attr)) {
|
if let Some(stab_attr) = stab_attrs.iter().find(|stab_attr| attr.has_name(**stab_attr)) {
|
||||||
let meta_item = attr.meta();
|
let meta_kind = attr.meta_kind();
|
||||||
if let Some(MetaItem { kind: MetaItemKind::List(ref metas), .. }) = meta_item {
|
if let Some(MetaItemKind::List(ref metas)) = meta_kind {
|
||||||
let mut feature = None;
|
let mut feature = None;
|
||||||
let mut since = None;
|
let mut since = None;
|
||||||
for meta in metas {
|
for meta in metas {
|
||||||
|
|
|
@ -2894,7 +2894,7 @@ fn codegen_fn_attrs(tcx: TyCtxt<'_>, id: DefId) -> CodegenFnAttrs {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} else if attr.has_name(sym::instruction_set) {
|
} else if attr.has_name(sym::instruction_set) {
|
||||||
codegen_fn_attrs.instruction_set = match attr.meta().map(|i| i.kind) {
|
codegen_fn_attrs.instruction_set = match attr.meta_kind() {
|
||||||
Some(MetaItemKind::List(ref items)) => match items.as_slice() {
|
Some(MetaItemKind::List(ref items)) => match items.as_slice() {
|
||||||
[NestedMetaItem::MetaItem(set)] => {
|
[NestedMetaItem::MetaItem(set)] => {
|
||||||
let segments =
|
let segments =
|
||||||
|
@ -2999,7 +2999,7 @@ fn codegen_fn_attrs(tcx: TyCtxt<'_>, id: DefId) -> CodegenFnAttrs {
|
||||||
if !attr.has_name(sym::inline) {
|
if !attr.has_name(sym::inline) {
|
||||||
return ia;
|
return ia;
|
||||||
}
|
}
|
||||||
match attr.meta().map(|i| i.kind) {
|
match attr.meta_kind() {
|
||||||
Some(MetaItemKind::Word) => InlineAttr::Hint,
|
Some(MetaItemKind::Word) => InlineAttr::Hint,
|
||||||
Some(MetaItemKind::List(ref items)) => {
|
Some(MetaItemKind::List(ref items)) => {
|
||||||
inline_span = Some(attr.span);
|
inline_span = Some(attr.span);
|
||||||
|
@ -3038,7 +3038,7 @@ fn codegen_fn_attrs(tcx: TyCtxt<'_>, id: DefId) -> CodegenFnAttrs {
|
||||||
return ia;
|
return ia;
|
||||||
}
|
}
|
||||||
let err = |sp, s| struct_span_err!(tcx.sess.diagnostic(), sp, E0722, "{}", s).emit();
|
let err = |sp, s| struct_span_err!(tcx.sess.diagnostic(), sp, E0722, "{}", s).emit();
|
||||||
match attr.meta().map(|i| i.kind) {
|
match attr.meta_kind() {
|
||||||
Some(MetaItemKind::Word) => {
|
Some(MetaItemKind::Word) => {
|
||||||
err(attr.span, "expected one argument");
|
err(attr.span, "expected one argument");
|
||||||
ia
|
ia
|
||||||
|
|
Loading…
Add table
Reference in a new issue