use proc_macro2::TokenStream;
use quote::quote;
use syn::{ItemStruct, Path};
pub fn myko_report_impl(report_output_type: Path, mut input_struct: ItemStruct) -> TokenStream {
let manual_cache_key = crate::take_manual_cache_key_attr(&mut input_struct);
let non_hash_cache_key = crate::take_non_hash_cache_key_attr(&mut input_struct);
let struct_name = &input_struct.ident;
let ctx = crate::DeriveCtx::new();
let krate = &ctx.krate;
let serde_path = &ctx.serde_path;
let serde_rename_attr = ctx.serde_attr(quote!(rename_all = "camelCase"));
let is_empty = matches!(&input_struct.fields, syn::Fields::Named(f) if f.named.is_empty())
|| matches!(&input_struct.fields, syn::Fields::Unit);
let derives = if is_empty {
if non_hash_cache_key {
quote! {
#[derive(Clone, Debug, Default, #serde_path::Serialize, #serde_path::Deserialize, #krate::TS)]
#serde_rename_attr
}
} else {
quote! {
#[derive(Clone, Debug, Default, Hash, #serde_path::Serialize, #serde_path::Deserialize, #krate::TS)]
#serde_rename_attr
}
}
} else if non_hash_cache_key {
quote! {
#[derive(Clone, Debug, #serde_path::Serialize, #serde_path::Deserialize, #krate::TS)]
#serde_rename_attr
}
} else {
quote! {
#[derive(Clone, Debug, Hash, #serde_path::Serialize, #serde_path::Deserialize, #krate::TS)]
#serde_rename_attr
}
};
let report_registration = quote! {
#krate::prelude::ReportRegistration {
report_id: stringify!(#struct_name),
crate_name: module_path!(),
output_type: stringify!(#report_output_type),
output_type_crate: module_path!(),
parse: <#struct_name as #krate::report::ReportFactory>::parse,
cell_factory: <#struct_name as #krate::report::ReportFactory>::cell_factory,
}
};
let cache_key_impl = if manual_cache_key {
quote!()
} else if non_hash_cache_key {
quote! {
impl #krate::prelude::CacheKey for #struct_name {
fn cache_key(&self, state: &mut dyn std::hash::Hasher) {
#krate::cache::write_serde_cache_key(self, state);
}
}
}
} else {
quote! {
impl #krate::prelude::CacheKey for #struct_name {
fn cache_key(&self, state: &mut dyn std::hash::Hasher) {
#krate::cache::write_hash_cache_key(self, state);
}
}
}
};
let expanded = quote! {
#derives
#input_struct
#[cfg(not(target_arch = "wasm32"))]
#krate::submit! {
#report_registration
}
#krate::register_ts_export!(#struct_name);
impl #krate::prelude::ReportId for #struct_name {
fn report_id(&self) -> std::sync::Arc<str> {
stringify!(#struct_name).into()
}
}
impl #krate::prelude::ReportIdStatic for #struct_name {
fn report_id_static() -> &'static str {
stringify!(#struct_name)
}
}
impl #krate::prelude::ReportOutputType for #struct_name {
type Output = #report_output_type;
}
#cache_key_impl
};
expanded
}