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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157

use std::ops::Deref;

use quote::ToTokens;
use syn::parse::Parse;
use syn::parse::ParseStream;
use syn::punctuated::Punctuated;
use syn::spanned::Spanned;
use syn::Attribute;
use syn::Token;

use crate::error;
use crate::value::Value;

pub mod arg;
pub mod cote;
pub mod fetch;
pub mod infer;
pub mod sub;
pub mod value;

pub use self::arg::ArgKind;
pub use self::cote::CoteKind;
pub use self::fetch::FetchKind;
pub use self::infer::InferKind;
pub use self::sub::SubKind;
pub use self::value::ValueKind;

#[derive(Debug)]
pub enum Style {
    Value,
    Flag,
    True,
}

pub trait Kind
where
    Self: Sized,
{
    fn parse(input: &mut ParseStream) -> syn::Result<(Self, Style)>;
}

#[derive(Debug, Clone)]
pub struct Config<T> {
    kind: T,

    value: Value,
}

impl<T> Config<T> {
    pub fn kind(&self) -> &T {
        &self.kind
    }

    pub fn value(&self) -> &Value {
        &self.value
    }
}

impl<T: Kind> Parse for Config<T> {
    fn parse(mut input: ParseStream) -> syn::Result<Self> {
        let (kind, style) = T::parse(&mut input)?;

        Ok(Self {
            kind,
            value: match style {
                Style::Value => input.parse()?,
                Style::Flag => Value::Null,
                Style::True => {
                    input
                        .parse()
                        .unwrap_or(Value::Literal(syn::Lit::Bool(syn::LitBool {
                            value: true,
                            span: input.span(),
                        })))
                }
            },
        })
    }
}

impl<T> ToTokens for Config<T> {
    fn to_tokens(&self, tokens: &mut proc_macro2::TokenStream) {
        ToTokens::to_tokens(&self.value, tokens)
    }
}

#[derive(Debug, Clone)]
pub struct Configs<T>(Vec<Config<T>>);

impl<T> Configs<T>
where
    T: PartialEq,
{
    pub fn has_cfg(&self, kind: T) -> bool {
        self.0.iter().any(|v| v.kind() == &kind)
    }

    pub fn find_cfg(&self, kind: T) -> Option<&Config<T>> {
        self.0.iter().find(|v| v.kind() == &kind)
    }

    // pub fn find_cfgs(&self, kind: T) -> Vec<&Config<T>> {
    //     self.0.iter().filter(|v| v.kind() == &kind).collect()
    // }

    pub fn find_value(&self, kind: T) -> Option<&Value> {
        self.0.iter().find(|v| v.kind() == &kind).map(|v| v.value())
    }

    pub fn find_values(&self, kind: T) -> Vec<&Value> {
        self.0
            .iter()
            .filter(|v| v.kind() == &kind)
            .map(|v| v.value())
            .collect()
    }
}

impl<T: Kind> Configs<T> {
    pub fn parse_attrs(name: &str, attrs: &[Attribute]) -> Self {
        let attrs = attrs.iter().filter(|v| v.path().is_ident(name));
        let cfgs = attrs.map(|attr| {
            attr.parse_args_with(Punctuated::<Config<T>, Token![,]>::parse_terminated)
                .map(|res| res.into_iter())
                .unwrap_or_else(|e| {
                    panic!(
                        "{:?}",
                        error(
                            attr.span(),
                            format!("can not parsing `{}` attributes: {:?}", name, e),
                        )
                    )
                })
        });

        Self(cfgs.flatten().collect::<Vec<Config<T>>>())
    }
}

impl<T> Deref for Configs<T> {
    type Target = Vec<Config<T>>;

    fn deref(&self) -> &Self::Target {
        &self.0
    }
}

// pub fn find_cfg_name<'a>(names: &[&'a str], attrs: &[Attribute]) -> Option<&'a str> {
//     for attr in attrs {
//         for name in names {
//             if attr.path.is_ident(*name) {
//                 return Some(*name);
//             }
//         }
//     }
//     None
// }