drt-sc-derive 0.0.1

DharitriOne smart contract API procedural 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

use super::{
    method_call_gen::{generate_call_method, generate_promises_callback_call_method},
    method_gen,
};
use crate::model::{ContractTrait, MethodImpl, PublicRole};

pub fn extract_method_impls(contract_trait: &ContractTrait) -> Vec<proc_macro2::TokenStream> {
    contract_trait
        .methods
        .iter()
        .filter_map(|m| {
            if let MethodImpl::Explicit(body) = &m.implementation {
                let msig = method_gen::generate_sig_with_attributes(m);
                Some(quote! {
                    #msig
                    #body
                })
            } else {
                None
            }
        })
        .collect()
}

pub fn generate_call_methods(contract_trait: &ContractTrait) -> Vec<proc_macro2::TokenStream> {
    contract_trait
        .methods
        .iter()
        .filter_map(|m| match &m.public_role {
            PublicRole::Init(_init_metadata) => Some(generate_call_method(m)),
            PublicRole::Upgrade(_upgrade_metadata) => Some(generate_call_method(m)),
            PublicRole::Endpoint(_endpoint_metadata) => Some(generate_call_method(m)),
            PublicRole::CallbackPromise(_callback_metadata) => {
                Some(generate_promises_callback_call_method(m))
            },
            _ => None,
        })
        .collect()
}

/// Definitions for methods that get auto-generated implementations: events, getters, setters
pub fn generate_auto_impl_defs(contract_trait: &ContractTrait) -> Vec<proc_macro2::TokenStream> {
    contract_trait
        .methods
        .iter()
        .filter_map(|m| {
            if let MethodImpl::Generated(_) = &m.implementation {
                let sig = method_gen::generate_sig_with_attributes(m);
                Some(quote! { #sig ; })
            } else {
                None
            }
        })
        .collect()
}