soroban_builtin_sdk_macros/
lib.rs

1mod derive_fn;
2mod derive_type;
3
4extern crate proc_macro;
5
6use crate::derive_fn::derive_contract_function_set;
7use crate::derive_type::{derive_type_enum, derive_type_struct};
8use proc_macro::TokenStream;
9use quote::quote;
10use syn::{
11    parse_macro_input, spanned::Spanned, DeriveInput, Error, ImplItem, ImplItemFn, ItemImpl,
12};
13
14#[proc_macro_attribute]
15pub fn contracttype(_metadata: TokenStream, input: TokenStream) -> TokenStream {
16    let input = parse_macro_input!(input as DeriveInput);
17    quote! {
18        #[derive(soroban_builtin_sdk_macros::ContractType)]
19        #input
20    }
21    .into()
22}
23
24#[doc(hidden)]
25#[proc_macro_derive(ContractType)]
26pub fn derive_contract_type(input: TokenStream) -> TokenStream {
27    let input = parse_macro_input!(input as DeriveInput);
28    let ident = &input.ident;
29    let derived = match &input.data {
30        syn::Data::Struct(s) => derive_type_struct(ident, s),
31        syn::Data::Enum(e) => derive_type_enum(ident, e),
32        syn::Data::Union(u) => Error::new(
33            u.union_token.span(),
34            "unions are unsupported as contract types",
35        )
36        .to_compile_error(),
37    };
38    quote! { #derived }.into()
39}
40
41fn get_methods(imp: &ItemImpl) -> impl Iterator<Item = &ImplItemFn> {
42    imp.items.iter().filter_map(|i| match i {
43        ImplItem::Fn(m) => Some(m),
44        _ => None,
45    })
46}
47
48#[proc_macro_attribute]
49pub fn contractimpl(_metadata: TokenStream, input: TokenStream) -> TokenStream {
50    let imp = parse_macro_input!(input as ItemImpl);
51    let ty = &imp.self_ty;
52    let all_methods: Vec<_> = get_methods(&imp).collect();
53
54    let cfs = derive_contract_function_set(ty, all_methods.into_iter());
55    quote! {
56        #imp
57        #cfs
58    }
59    .into()
60}
soroban_builtin_sdk_macros/lib.rs

soroban_builtin_sdk_macros/
lib.rs
