use proc_macro2::TokenStream;
use quote::quote;
use crate::codegen::generics_utils::impl_generics_with_f;
use crate::parsing::StructInfo;
pub fn generate(info: &StructInfo) -> TokenStream {
let struct_name = &info.name;
let module_name = &info.module_name;
let (impl_generics, user_type_args) = impl_generics_with_f(&info.generics, module_name);
let (impl_generics, _, where_clause) = impl_generics.split_for_impl();
let methods = info.fields.iter().map(|field| {
let field_name = &field.name;
let field_ty = &field.ty;
let get_method = quote::format_ident!("get_{}", field_name);
let get_mut_method = quote::format_ident!("get_{}_mut", field_name);
quote! {
pub fn #get_method(&self) -> Option<&#field_ty> {
::structural_typing::access::Access::<#field_ty>::get(&self.#field_name)
}
pub fn #get_mut_method(&mut self) -> Option<&mut #field_ty> {
::structural_typing::access::Access::<#field_ty>::get_mut(&mut self.#field_name)
}
}
});
quote! {
impl #impl_generics #struct_name<#(#user_type_args,)* F> #where_clause {
#(#methods)*
}
}
}