use super::*;
#[derive(Debug, Clone, Eq, PartialEq)]
pub struct Attribute {
pub value: MetaItem,
pub is_sugared_doc: bool,
}
#[derive(Debug, Clone, Eq, PartialEq)]
pub enum MetaItem {
Word(Ident),
List(Ident, Vec<MetaItem>),
NameValue(Ident, Lit),
}
impl MetaItem {
pub fn name(&self) -> &str {
match *self {
MetaItem::Word(ref name) => name.as_ref(),
MetaItem::List(ref name, _) => name.as_ref(),
MetaItem::NameValue(ref name, _) => name.as_ref(),
}
}
}
#[cfg(feature = "parsing")]
pub mod parsing {
use super::*;
use ident::parsing::ident;
use lit::{Lit, StrStyle};
use lit::parsing::lit;
use nom::multispace;
named!(pub attribute -> Attribute, alt!(
do_parse!(
punct!("#") >>
punct!("[") >>
meta_item: meta_item >>
punct!("]") >>
(Attribute {
value: meta_item,
is_sugared_doc: false,
})
)
|
do_parse!(
punct!("///") >>
space: multispace >>
content: take_until!("\n") >>
(Attribute {
value: MetaItem::NameValue(
"doc".into(),
Lit::Str(
format!("///{}{}", space, content),
StrStyle::Cooked,
),
),
is_sugared_doc: true,
})
)
));
named!(meta_item -> MetaItem, alt!(
do_parse!(
id: ident >>
punct!("(") >>
inner: separated_list!(punct!(","), meta_item) >>
punct!(")") >>
(MetaItem::List(id, inner))
)
|
do_parse!(
name: ident >>
punct!("=") >>
value: lit >>
(MetaItem::NameValue(name, value))
)
|
map!(ident, MetaItem::Word)
));
}
#[cfg(feature = "printing")]
mod printing {
use super::*;
use quote::{Tokens, ToTokens};
impl ToTokens for Attribute {
fn to_tokens(&self, tokens: &mut Tokens) {
tokens.append("#");
tokens.append("[");
self.value.to_tokens(tokens);
tokens.append("]");
}
}
impl ToTokens for MetaItem {
fn to_tokens(&self, tokens: &mut Tokens) {
match *self {
MetaItem::Word(ref ident) => {
ident.to_tokens(tokens);
}
MetaItem::List(ref ident, ref inner) => {
ident.to_tokens(tokens);
tokens.append("(");
tokens.append_separated(inner, ",");
tokens.append(")");
}
MetaItem::NameValue(ref name, ref value) => {
name.to_tokens(tokens);
tokens.append("=");
value.to_tokens(tokens);
}
}
}
}
}