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
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

use super::substitution_map::SubstitutionsMap;

pub fn substitutions() -> SubstitutionsMap {
    let mut substitutions = SubstitutionsMap::new();

    add_managed_types(&mut substitutions);
    add_storage_mappers(&mut substitutions);

    substitutions
}

#[rustfmt::skip]
fn add_managed_type(substitutions: &mut SubstitutionsMap, type_name: &proc_macro2::TokenStream) {
    substitutions.add_substitution(
        quote!(#type_name<Self::Api>), 
        quote!(#type_name<Self::Api>));
    substitutions.add_substitution(
        quote!(#type_name::),
        quote!(drt_sc::types::#type_name::<Self::Api>::),
    );
    substitutions.add_substitution(
        quote!(#type_name),
        quote!(drt_sc::types::#type_name<Self::Api>),
    );
}

fn add_managed_type_with_generics(
    substitutions: &mut SubstitutionsMap,
    type_name: &proc_macro2::TokenStream,
) {
    substitutions.add_substitution(
        quote!(#type_name<Self::Api, ),
        quote!(#type_name<Self::Api, ),
    );
    substitutions.add_substitution(quote!(#type_name<), quote!(#type_name<Self::Api, ));
}

fn add_managed_types(substitutions: &mut SubstitutionsMap) {
    // types::io
    add_managed_type(substitutions, &quote!(ManagedSCError));

    // types::interaction
    add_managed_type(substitutions, &quote!(AsyncCall));

    // types::managed::basic
    add_managed_type(substitutions, &quote!(BigFloat));
    add_managed_type(substitutions, &quote!(BigInt));
    add_managed_type(substitutions, &quote!(BigUint));
    add_managed_type(substitutions, &quote!(EllipticCurveComponents));
    add_managed_type(substitutions, &quote!(EllipticCurve));
    add_managed_type(substitutions, &quote!(ManagedBuffer));
    add_managed_type(substitutions, &quote!(ManagedMap));

    // types::managed::multi-value
    add_managed_type(substitutions, &quote!(ManagedAsyncCallError));
    add_managed_type_with_generics(substitutions, &quote!(ManagedAsyncCallResult));
    add_managed_type(substitutions, &quote!(DctTokenPaymentMultiArg));
    add_managed_type(substitutions, &quote!(DctTokenPaymentMultiValue));
    add_managed_type_with_generics(substitutions, &quote!(MultiValueEncodedIterator));
    add_managed_type_with_generics(substitutions, &quote!(MultiValueEncoded));
    add_managed_type_with_generics(substitutions, &quote!(ManagedVarArgs));
    add_managed_type_with_generics(substitutions, &quote!(ManagedMultiResultVec));
    add_managed_type_with_generics(substitutions, &quote!(MultiValueManagedVecCounted));
    add_managed_type_with_generics(substitutions, &quote!(ManagedCountedVarArgs));
    add_managed_type_with_generics(substitutions, &quote!(ManagedCountedMultiResultVec));
    add_managed_type_with_generics(substitutions, &quote!(MultiValueManagedVec));

    // types::managed::wrapped
    add_managed_type(substitutions, &quote!(RewaOrDctTokenIdentifier));
    add_managed_type(substitutions, &quote!(RewaOrDctTokenPayment));
    add_managed_type(substitutions, &quote!(DctTokenData));
    add_managed_type(substitutions, &quote!(DctTokenPayment));
    add_managed_type(substitutions, &quote!(ManagedAddress));
    add_managed_type(substitutions, &quote!(ManagedBufferBuilder));
    add_managed_type_with_generics(substitutions, &quote!(ManagedByteArray));
    add_managed_type_with_generics(substitutions, &quote!(ManagedOption));
    add_managed_type_with_generics(substitutions, &quote!(ManagedRef));
    add_managed_type_with_generics(substitutions, &quote!(ManagedVecOwnedIterator));
    add_managed_type_with_generics(substitutions, &quote!(ManagedVecRefIterator));
    add_managed_type_with_generics(substitutions, &quote!(ManagedVecRef));
    add_managed_type_with_generics(substitutions, &quote!(ManagedVec));
    add_managed_type_with_generics(substitutions, &quote!(PreloadedManagedBuffer));
    add_managed_type(substitutions, &quote!(RandomnessSource));
    add_managed_type(substitutions, &quote!(TokenIdentifier));
    add_managed_type(substitutions, &quote!(FunctionCall));
}

fn add_storage_mapper_single_generic_arg(
    substitutions: &mut SubstitutionsMap,
    mapper_name: &proc_macro2::TokenStream,
) {
    substitutions.add_substitution(
        quote!(#mapper_name<Self::Api>),
        quote!(#mapper_name<Self::Api>),
    );
    substitutions.add_substitution(quote!(#mapper_name), quote!(#mapper_name<Self::Api>));
}

fn add_storage_mapper(
    substitutions: &mut SubstitutionsMap,
    mapper_name: &proc_macro2::TokenStream,
) {
    add_managed_type_with_generics(substitutions, mapper_name);
}

fn add_storage_mappers(substitutions: &mut SubstitutionsMap) {
    add_storage_mapper_single_generic_arg(substitutions, &quote!(FungibleTokenMapper));
    add_storage_mapper_single_generic_arg(substitutions, &quote!(NonFungibleTokenMapper));
    add_storage_mapper_single_generic_arg(substitutions, &quote!(TokenAttributesMapper));
    add_storage_mapper_single_generic_arg(substitutions, &quote!(UniqueIdMapper));
    add_storage_mapper_single_generic_arg(substitutions, &quote!(UserMapper));

    add_storage_mapper(substitutions, &quote!(BiDiMapper));
    add_storage_mapper(substitutions, &quote!(LinkedListMapper));
    add_storage_mapper(substitutions, &quote!(MapMapper));
    add_storage_mapper(substitutions, &quote!(MapStorageMapper));
    add_storage_mapper(substitutions, &quote!(SetMapper));
    add_storage_mapper(substitutions, &quote!(UnorderedSetMapper));
    add_storage_mapper(substitutions, &quote!(SingleValueMapper));
    add_storage_mapper(substitutions, &quote!(VecMapper));
    add_storage_mapper(substitutions, &quote!(QueueMapper));
    add_storage_mapper(substitutions, &quote!(WhitelistMapper));
}