1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

//! Individual java modifier

use crate::java::Tokens;
use crate::{FormatTokens, Java};
use std::collections::BTreeSet;

/// A Java modifier.
#[derive(Debug, PartialEq, Eq, PartialOrd, Ord, Clone, Hash)]
pub enum Modifier {
    /// default
    Default,
    /// public
    Public,
    /// protected
    Protected,
    /// private
    Private,
    /// abstract
    Abstract,
    /// static
    Static,
    /// final
    Final,
    /// Native
    Native,
}

impl Modifier {
    /// Get the name of the modifier.
    pub fn name(&self) -> &'static str {
        use self::Modifier::*;

        match *self {
            Default => "default",
            Public => "public",
            Protected => "protected",
            Private => "private",
            Abstract => "abstract",
            Static => "static",
            Final => "final",
            Native => "native",
        }
    }
}

impl<'el> FormatTokens<'el, Java> for Modifier {
    fn format_tokens(self, tokens: &mut Tokens<'el>) {
        tokens.append(self.name());
    }
}

impl<'el> FormatTokens<'el, Java> for Vec<Modifier> {
    fn format_tokens(self, tokens: &mut Tokens<'el>) {
        let mut it = self.into_iter().collect::<BTreeSet<_>>().into_iter();

        if let Some(modifier) = it.next() {
            modifier.format_tokens(tokens);
        }

        for modifier in it {
            tokens.spacing();
            modifier.format_tokens(tokens);
        }
    }
}

#[cfg(test)]
mod tests {
    use super::Modifier;
    use crate as genco;
    use crate::{quote, Java, Tokens};

    #[test]
    fn test_vec() {
        use self::Modifier::*;
        let el: Tokens<Java> = quote!(#(vec![Public, Final, Static]));
        assert_eq!(
            Ok("public static final"),
            el.to_string().as_ref().map(|s| s.as_str())
        );
    }
}