1use proc_macro::TokenStream;
2use quote::quote;
3use syn::{parse_macro_input, ItemStruct};
4
5#[proc_macro_attribute]
6pub fn rootchain_contract(_attr: TokenStream, item: TokenStream) -> TokenStream {
7 let input = parse_macro_input!(item as ItemStruct);
8 let name = &input.ident;
9
10 let expanded = quote! {
11 #input
12
13 #[no_mangle]
14 pub extern "C" fn init() -> i32 {
15 use rootchain_std::traits::Contract;
16 let mut contract = #name::default();
17 match contract.init() {
18 Ok(_) => 0,
19 Err(_) => 1,
20 }
21 }
22
23 #[no_mangle]
24 pub extern "C" fn apply() -> i32 {
25 use rootchain_std::traits::Contract;
26 let mut contract = #name::default();
27 match contract.apply() {
28 Ok(_) => 0,
29 Err(_) => 1,
30 }
31 }
32 };
33
34 TokenStream::from(expanded)
35}