src/attr.rs

use super::*;

use common::word;
use helper::escaped_string;
use nom::multispace;

#[derive(Debug, Clone, Eq, PartialEq)]
pub struct Attribute {
    pub value: MetaItem,
    pub is_sugared_doc: bool,
}

/// A compile-time attribute item.
///
/// E.g. `#[test]`, `#[derive(..)]` or `#[feature = "foo"]`
#[derive(Debug, Clone, Eq, PartialEq)]
pub enum MetaItem {
    /// Word meta item.
    ///
    /// E.g. `test` as in `#[test]`
    Word(Ident),
    /// List meta item.
    ///
    /// E.g. `derive(..)` as in `#[derive(..)]`
    List(Ident, Vec<MetaItem>),
    /// Name value meta item.
    ///
    /// E.g. `feature = "foo"` as in `#[feature = "foo"]`
    NameValue(Ident, String),
}

named!(pub attribute<&str, 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_s!("\n") >>
        (Attribute {
            value: MetaItem::NameValue(
                "doc".to_string(),
                format!("///{}{}", space, content),
            ),
            is_sugared_doc: true,
        })
    )
));

named!(quoted<&str, String>, delimited!(
    punct!("\""),
    escaped_string,
    tag_s!("\"")
));

named!(meta_item<&str, MetaItem>, alt!(
    do_parse!(
        ident: word >>
        punct!("(") >>
        inner: separated_list!(punct!(","), meta_item) >>
        punct!(")") >>
        (MetaItem::List(ident, inner))
    )
    |
    do_parse!(
        ident: word >>
        punct!("=") >>
        string: quoted >>
        (MetaItem::NameValue(ident, string))
    )
    |
    map!(word, MetaItem::Word)
));