fuels_rs/
types.rs

1use crate::errors::Error;
2use anyhow::Result;
3use proc_macro2::TokenStream;
4use quote::quote;
5
6use fuels_core::ParamType;
7
8/// Expands a [`ParamType`] into a TokenStream.
9/// Used to expand functions when generating type-safe bindings of a JSON ABI.
10pub 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}