1use crate::errors::Error;
2use anyhow::Result;
3use proc_macro2::TokenStream;
4use quote::quote;
5
6use fuels_core::ParamType;
7
8pub fn expand_type(kind: &ParamType) -> Result<TokenStream, Error> {
11 match kind {
12 ParamType::U8 | ParamType::Byte => Ok(quote! { u8 }),
13 ParamType::U16 => Ok(quote! { u16 }),
14 ParamType::U32 => Ok(quote! { u32 }),
15 ParamType::U64 => Ok(quote! { u64 }),
16 ParamType::Bool => Ok(quote! { bool }),
17 ParamType::B256 => Ok(quote! { [u8; 32] }),
18 ParamType::String(_) => Ok(quote! { String }),
19 ParamType::Array(t, _size) => {
20 let inner = expand_type(t)?;
21 Ok(quote! { ::std::vec::Vec<#inner> })
22 }
23 ParamType::Struct(members) => {
24 if members.is_empty() {
25 return Err(Error::InvalidData);
26 }
27 let members = members
28 .iter()
29 .map(|member| expand_type(member))
30 .collect::<Result<Vec<_>, _>>()?;
31 Ok(quote! { (#(#members,)*) })
32 }
33 ParamType::Enum(members) => {
34 if members.is_empty() {
35 return Err(Error::InvalidData);
36 }
37 let members = members
38 .iter()
39 .map(|member| expand_type(member))
40 .collect::<Result<Vec<_>, _>>()?;
41 Ok(quote! { (#(#members,)*) })
42 }
43 }
44}