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
use proc_macro::TokenStream; use syn::{parse_macro_input, Result}; mod derive; mod func; #[proc_macro_derive(CandidType, attributes(candid_path))] pub fn derive_idl_type(input: TokenStream) -> TokenStream { let input = parse_macro_input!(input as syn::DeriveInput); let custom_candid_path_result = get_custom_candid_path(&input); match custom_candid_path_result { Ok(custom_candid_path) => derive::derive_idl_type(input, &custom_candid_path).into(), Err(e) => e.to_compile_error().into(), } } #[proc_macro_attribute] pub fn candid_method(attr: TokenStream, item: TokenStream) -> TokenStream { let attrs = parse_macro_input!(attr as syn::AttributeArgs); let fun = parse_macro_input!(item as syn::ItemFn); func::candid_method(attrs, fun).map_or_else(|e| e.to_compile_error().into(), Into::into) } #[proc_macro] pub fn export_service(_: TokenStream) -> TokenStream { func::export_service().into() } #[inline] pub(crate) fn idl_hash(id: &str) -> u32 { let mut s: u32 = 0; for c in id.as_bytes().iter() { s = s.wrapping_mul(223).wrapping_add(*c as u32); } s } #[cfg(feature = "cdk")] pub(crate) fn candid_path( custom_candid_path: &Option<proc_macro2::TokenStream>, ) -> proc_macro2::TokenStream { match custom_candid_path { Some(custom_candid_path_value) => custom_candid_path_value.clone(), None => quote::quote! { ::ic_cdk::export::candid }, } } #[cfg(not(feature = "cdk"))] pub(crate) fn candid_path( custom_candid_path: &Option<proc_macro2::TokenStream>, ) -> proc_macro2::TokenStream { match custom_candid_path { Some(custom_candid_path_value) => custom_candid_path_value.clone(), None => quote::quote! { ::candid }, } } fn get_custom_candid_path(input: &syn::DeriveInput) -> Result<Option<proc_macro2::TokenStream>> { let candid_path_helper_attribute_option = input .attrs .iter() .find(|attr| attr.path.is_ident("candid_path")); match candid_path_helper_attribute_option { Some(candid_path_helper_attribute) => { let custom_candid_path_lit: syn::LitStr = candid_path_helper_attribute.parse_args()?; let custom_candid_token_stream: proc_macro2::TokenStream = custom_candid_path_lit.value().parse()?; Ok(Some(custom_candid_token_stream)) } None => Ok(None), } }