armour-derive 0.27.3

DDL and serialization for key-value storage
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

#[macro_use]
extern crate darling;
extern crate proc_macro2;
extern crate quote;
extern crate syn;

use proc_macro::TokenStream;

use darling::util::Flag;
use syn::{Data, DeriveInput, Expr, ExprPath, LitStr, parse_macro_input};

/// Adds metrics tracking to the function.
///
/// # Example
/// ```rust
/// #[armour_metrics(prefix = "my_prefix", name = tree.raw.name.clone())]
/// fn my_function() {
///     // function body
/// }
/// ```
#[proc_macro_attribute]
pub fn armour_metrics(attr: TokenStream, item: TokenStream) -> TokenStream {
    metrics::armour_metrics_impl(attr, item)
}

mod enum_parse;
mod metrics;
mod struct_parse;

/// - `#[get_type(idx = 1)]` in fields
/// - `#[get_type(flatten)]` for unnamed structs
/// - `#[get_type(with_type(Type))]` for named struct fields
/// - `#[get_type(unimplemented)]` for named struct fields
/// - `#[get_type(custom("name", &[Typ::Bool, Typ::Str]))]` for named struct fields
#[proc_macro_derive(GetType, attributes(idx, get_type))]
pub fn derive_get_type(stream: TokenStream) -> TokenStream {
    let ast = parse_macro_input!(stream as DeriveInput);
    let name = ast.ident;
    let data = &ast.data;

    match data {
        Data::Struct(data_struct) => struct_parse::parse(data_struct, name).into(),
        Data::Enum(data_enum) => enum_parse::parse(data_enum, name),
        Data::Union(_) => {
            panic!(
                "unions not supported, but Rust enums is implemented GetType trait (use Enums instead)"
            )
        }
    }
}

#[derive(FromMeta, Debug)]
struct CustomField {
    name: LitStr,
    types: Expr,
}

#[derive(Default, FromField, Debug)]
#[darling(attributes(get_type))]
pub(crate) struct FieldAttr {
    pub(crate) idx: Option<u32>,
    pub(crate) with_type: Option<ExprPath>,
    pub(crate) unimplemented: Flag,
    pub(crate) custom: Option<CustomField>,
    pub(crate) flatten: Flag,
}