bodega-macros 0.1.0

bodega proc macros
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

use darling::{ast::NestedMeta, FromMeta};
use proc_macro::TokenStream;
use syn::parse_macro_input;

mod db_bmc;
mod helpers;
mod insert;
mod select;
mod store_enum;
mod update;
mod uuid_id;

#[proc_macro_derive(Select, attributes(select))]
pub fn select(item: TokenStream) -> TokenStream {
    let input = parse_macro_input!(item as syn::DeriveInput);
    select::select_impl(&input).unwrap_or_else(|e| e.to_compile_error().into())
}

#[proc_macro_derive(Insert, attributes(insert))]
pub fn insert(item: TokenStream) -> TokenStream {
    let input = parse_macro_input!(item as syn::DeriveInput);
    insert::insert_impl(&input).unwrap_or_else(|e| e.to_compile_error().into())
}

#[proc_macro_derive(Update, attributes(update))]
pub fn update(item: TokenStream) -> TokenStream {
    let input = parse_macro_input!(item as syn::DeriveInput);
    update::update_impl(&input).unwrap_or_else(|e| e.to_compile_error().into())
}

#[proc_macro_derive(DbBmc, attributes(db_bmc))]
pub fn db_bmc(item: TokenStream) -> TokenStream {
    let input = parse_macro_input!(item as syn::DeriveInput);
    db_bmc::db_bmc_impl(&input).unwrap_or_else(|e| e.to_compile_error().into())
}

#[proc_macro_attribute]
pub fn uuid_id(attr_args: TokenStream, item: TokenStream) -> TokenStream {
    let attr_args = match NestedMeta::parse_meta_list(attr_args.into()) {
        Ok(v) => v,
        Err(e) => return TokenStream::from(darling::Error::from(e).write_errors()),
    };

    let args = match uuid_id::UuidArgs::from_list(&attr_args) {
        Ok(v) => v,
        Err(e) => return TokenStream::from(e.write_errors()),
    };

    let input = syn::parse_macro_input!(item as syn::ItemStruct);

    uuid_id::uuid_id_impl(args, input).unwrap_or_else(|e| e.to_compile_error().into())
}

#[proc_macro_attribute]
pub fn store_enum(attr_args: TokenStream, item: TokenStream) -> TokenStream {
    let attr_args = match NestedMeta::parse_meta_list(attr_args.into()) {
        Ok(v) => v,
        Err(e) => return TokenStream::from(darling::Error::from(e).write_errors()),
    };

    let args = match store_enum::StoreEnumArgs::from_list(&attr_args) {
        Ok(v) => v,
        Err(e) => return TokenStream::from(e.write_errors()),
    };

    let input = syn::parse_macro_input!(item as syn::ItemEnum);

    store_enum::store_enum_impl(args, input).unwrap_or_else(|e| e.to_compile_error().into())
}