elrond-wasm-derive 0.3.2

use super::arg_def::*;

pub fn arg_deserialize_next(arg: &MethodArg) -> proc_macro2::TokenStream {
    match &arg.ty {
        syn::Type::Path(type_path) => {
            let type_path_segment = type_path.path.segments.last().unwrap().clone();
            arg_deserialize_next_single(&type_path_segment)
        },             
        syn::Type::Reference(type_reference) => {
            if type_reference.mutability.is_some() {
                panic!("Mutable references not supported as contract method arguments");
            }
            match &*type_reference.elem {
                syn::Type::Path(type_path) => {
                    let type_path_segment = type_path.path.segments.last().unwrap().clone();
                    arg_deserialize_next_single(&type_path_segment)
                },
                _ => {
                    panic!("Unsupported reference argument type, reference does not contain type path: {:?}", type_reference)
                }
            }
        },
        other_arg => panic!("Unsupported argument type. Only path, reference, array or slice allowed. Found: {:?}", other_arg)
    }
}

fn arg_deserialize_next_single(type_path_segment: &syn::PathSegment) -> proc_macro2::TokenStream {
    let type_str = type_path_segment.ident.to_string();
    match type_str.as_str() {
        "Address" | "StorageKey" | "H256" =>
            quote!{
                match cb_data_deserializer.next_h256() {
                    elrond_wasm::DeserializerResult::NoMore => self.api.signal_error(err_msg::ARG_ASYNC_RETURN_WRONG_NUMBER),
                    elrond_wasm::DeserializerResult::Err(e) => self.api.signal_error(e),
                    elrond_wasm::DeserializerResult::Res(h256) => h256,
                }
            },
        other_stype_str => {
            panic!("Unsupported argument type {:?} for callback arg init snippet", other_stype_str)
        }
    }
}