cote-derive 0.12.2

Quickly build your command line utils
Documentation 
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
83
84
85
86
87
88
89

use quote::ToTokens;
use syn::Path;

use super::Kind;
use super::Style;

#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord)]
pub enum CoteKind {
    Policy,

    Name,

    Help,

    HelpOpt,

    Head,

    Foot,

    HelpWidth,

    UsageWidth,

    AbortHelp,

    NotExit,

    On,

    Fallback,

    Then,

    Strict,

    Combine,

    EmbeddedPlus,

    Flag,

    Overload,

    PrePolicy,

    ShellCompletion,

    MethodCall(String),
}

impl Kind for CoteKind {
    fn parse(input: &mut syn::parse::ParseStream) -> syn::Result<(Self, Style)> {
        let path: Path = input.parse()?;

        if let Some(ident) = path.get_ident() {
            let kind_str = ident.to_string();

            Ok(match kind_str.as_str() {
                "policy" => (Self::Policy, Style::Value),
                "name" => (Self::Name, Style::Value),
                "help" => (Self::Help, Style::Flag),
                "helpopt" => (Self::HelpOpt, Style::Value),
                "head" => (Self::Head, Style::Value),
                "foot" => (Self::Foot, Style::Value),
                "width" => (Self::HelpWidth, Style::Value),
                "usagew" => (Self::UsageWidth, Style::Value),
                "aborthelp" => (Self::AbortHelp, Style::Flag),
                "notexit" => (Self::NotExit, Style::Flag),
                "on" => (Self::On, Style::Value),
                "fallback" => (Self::Fallback, Style::Value),
                "then" => (Self::Then, Style::Value),
                "strict" => (Self::Strict, Style::True),
                "combine" => (Self::Combine, Style::Flag),
                "embedded" => (Self::EmbeddedPlus, Style::Flag),
                "flag" => (Self::Flag, Style::Flag),
                "overload" => (Self::Overload, Style::True),
                "prepolicy" => (Self::PrePolicy, Style::True),
                "shellcomp" => (Self::ShellCompletion, Style::True),
                method => (Self::MethodCall(method.to_owned()), Style::Value),
            })
        } else {
            let method = path.to_token_stream().to_string();
            let method = method.replace(char::is_whitespace, "");

            Ok((Self::MethodCall(method), Style::Value))
        }
    }
}