drt_sc_derive/
lib.rs

1// ensure we don't run out of macro stack
2#![recursion_limit = "1024"]
3
4#[macro_use]
5extern crate syn;
6
7#[macro_use]
8extern crate quote;
9
10mod contract_impl;
11mod format;
12mod generate;
13mod macro_contract;
14mod macro_module;
15mod macro_proxy;
16mod managed_vec_item_derive;
17mod model;
18mod parse;
19mod preprocessing;
20mod type_abi_derive;
21mod validate;
22
23#[proc_macro_attribute]
24pub fn contract(
25    args: proc_macro::TokenStream,
26    input: proc_macro::TokenStream,
27) -> proc_macro::TokenStream {
28    macro_contract::process_contract(args, input)
29}
30
31#[proc_macro_attribute]
32pub fn module(
33    args: proc_macro::TokenStream,
34    input: proc_macro::TokenStream,
35) -> proc_macro::TokenStream {
36    macro_module::process_module(args, input)
37}
38
39#[proc_macro_attribute]
40pub fn proxy(
41    args: proc_macro::TokenStream,
42    input: proc_macro::TokenStream,
43) -> proc_macro::TokenStream {
44    macro_proxy::process_proxy(args, input)
45}
46
47#[proc_macro_derive(TypeAbi)]
48pub fn type_abi_derive(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
49    type_abi_derive::type_abi_derive(input).into()
50}
51
52#[proc_macro_attribute]
53pub fn type_abi(
54    args: proc_macro::TokenStream,
55    input: proc_macro::TokenStream,
56) -> proc_macro::TokenStream {
57    assert!(args.is_empty(), "#[type_abi] attribute takes no args");
58    type_abi_derive::type_abi_full(input).into()
59}
60
61#[proc_macro_derive(ManagedVecItem)]
62pub fn managed_vec_item_derive(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
63    let ast = syn::parse(input).unwrap();
64    managed_vec_item_derive::managed_vec_item_derive(&ast)
65}
66
67#[proc_macro]
68pub fn format_receiver_args(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
69    format::format_receiver_args_macro(input.into()).into()
70}
71
72#[proc_macro]
73pub fn semver_tuple(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
74    format::semver_tuple(input.into()).into()
75}
76
77#[proc_macro]
78pub fn const_managed_decimal(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
79    let input = parse_macro_input!(input as syn::LitStr);
80    let (raw_int, decimals) = format::extract_number_data(input);
81
82    let expanded = quote! {
83        drt_sc::types::ManagedDecimal::<<Self as ContractBase>::Api, drt_sc::types::ConstDecimals<#decimals>>::const_decimals_from_raw(drt_sc::types::BigUint::from(#raw_int))
84    };
85
86    proc_macro::TokenStream::from(expanded)
87}
88
89#[proc_macro]
90pub fn managed_decimal(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
91    let input = parse_macro_input!(input as syn::LitStr);
92    let (raw_int, decimals) = format::extract_number_data(input);
93
94    let expanded = quote! {
95        drt_sc::types::ManagedDecimal::<<Self as ContractBase>::Api, usize>::from_raw_units(drt_sc::types::BigUint::from(#raw_int), #decimals)
96    };
97
98    proc_macro::TokenStream::from(expanded)
99}
100
101#[proc_macro]
102pub fn debug_const_managed_decimal(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
103    let input = parse_macro_input!(input as syn::LitStr);
104    let (raw_int, decimals) = format::extract_number_data(input);
105
106    let expanded = quote! {
107        drt_sc::types::ManagedDecimal::<drt_sc_scenario::imports::StaticApi, drt_sc::types::ConstDecimals<#decimals>>::const_decimals_from_raw(drt_sc::types::BigUint::from(#raw_int))
108    };
109
110    proc_macro::TokenStream::from(expanded)
111}
112
113#[proc_macro]
114pub fn debug_managed_decimal(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
115    let input = parse_macro_input!(input as syn::LitStr);
116    let (raw_int, decimals) = format::extract_number_data(input);
117
118    let expanded = quote! {
119        drt_sc::types::ManagedDecimal::<drt_sc_scenario::imports::StaticApi, usize>::from_raw_units(drt_sc::types::BigUint::from(#raw_int), #decimals)
120    };
121
122    proc_macro::TokenStream::from(expanded)
123}