derive_tools_meta 0.63.0

use macro_tools ::
{
  diag, generic_params, item_struct, struct_like ::StructLike, Result, qt, attr, syn, proc_macro2, return_syn_err, Spanned,
};

use super ::field_attributes :: { FieldAttributes };
use super ::item_attributes :: { ItemAttributes };

///
/// Derive macro to implement `VariadicFrom` when-ever it's possible to do automatically.
///
pub fn variadic_from(input: proc_macro ::TokenStream) -> Result< proc_macro2 ::TokenStream > 
{
  let original_input = input.clone();
  let parsed = syn ::parse :: < StructLike >(input)?;
  let has_debug = attr ::has_debug(parsed.attrs().iter())?;
  let item_attrs = ItemAttributes ::from_attrs(parsed.attrs().iter())?;
  let item_name = &parsed.ident();

  let (_generics_with_defaults, generics_impl, generics_ty, generics_where) = generic_params ::decompose(parsed.generics());

  let result =  match parsed 
  {
  StructLike ::Unit(ref _item) =>
  {
   return_syn_err!(parsed.span(), "Expects a structure with one field");
 }
  StructLike ::Struct(ref item) =>
  {
   let field_type = item_struct ::first_field_type(item)?;
   let field_name = item_struct ::first_field_name(item).ok().flatten();
   generate(
  item_name,
  &generics_impl,
  &generics_ty,
  &generics_where,
  &field_type,
  field_name.as_ref(),
 )
 }
  StructLike ::Enum(ref item) =>
  {
   let variants = item
  .variants
  .iter()
  .map(|variant| {
   variant_generate(
  item_name,
  &item_attrs,
  &generics_impl,
  &generics_ty,
  &generics_where,
  variant,
  &original_input,
 )
 })
  .collect :: < Result<Vec< proc_macro2 ::TokenStream >>>()?;

   qt! {
  #( #variants )*
 }
 }
 };

  if has_debug 
  {
  let about = format!("derive: VariadicFrom\nstructure: {item_name}");
  diag ::report_print(about, &original_input, &result);
 }

  Ok(result)
}

/// Generates `variadic_from()` inherent method for structs.
///
/// Fix(issue-4): Changed from trait implementation to inherent method.
/// Root cause: Original implementation generated `impl crate::VariadicFrom` for non-existent trait.
/// Pitfall: Variadic conversion patterns require custom traits. Must define trait before implementing it, or use inherent methods for utility constructors.
///
/// Example of generated code :
/// ```text
/// impl IsTransparent
/// {
///   pub fn variadic_from( src: bool ) -> Self
///   {
///     Self( src )
/// }
/// }
/// ```
fn generate(
  item_name: &syn ::Ident,
  generics_impl: &syn ::punctuated ::Punctuated< syn ::GenericParam, syn ::token ::Comma >,
  generics_ty: &syn ::punctuated ::Punctuated< syn ::GenericParam, syn ::token ::Comma >,
  generics_where: &syn ::punctuated ::Punctuated< syn ::WherePredicate, syn ::token ::Comma >,
  field_type: &syn ::Type,
  field_name: Option< &syn ::Ident >,
) -> proc_macro2 ::TokenStream {
  let body =  if let Some(field_name) = field_name
  {
  qt! { Self { #field_name: src } }
 } else {
  qt! { Self( src ) }
 };

  qt! {
  #[ automatically_derived ]
  impl< #generics_impl > #item_name< #generics_ty >
  where
   #generics_where
  {
   #[ inline( always ) ]
   pub fn variadic_from( src: #field_type ) -> Self
   {
  #body
 }
 }
 }
}

/// Generates `variadic_from()` inherent method for enum variants.
///
/// Fix(issue-4): Changed from trait implementation to inherent method.
/// Root cause: Original implementation generated `impl crate::VariadicFrom` for non-existent trait.
/// Pitfall: Variadic conversion patterns require custom traits. Must define trait before implementing it, or use inherent methods for utility constructors.
///
/// Example of generated code :
/// ```text
/// impl MyEnum
/// {
///   pub fn variadic_from( src: i32 ) -> Self
///   {
///     Self ::Variant( src )
/// }
/// }
/// ```
fn variant_generate(
  item_name: &syn ::Ident,
  item_attrs: &ItemAttributes,
  generics_impl: &syn ::punctuated ::Punctuated< syn ::GenericParam, syn ::token ::Comma >,
  generics_ty: &syn ::punctuated ::Punctuated< syn ::GenericParam, syn ::token ::Comma >,
  generics_where: &syn ::punctuated ::Punctuated< syn ::WherePredicate, syn ::token ::Comma >,
  variant: &syn ::Variant,
  original_input: &proc_macro ::TokenStream,
) -> Result< proc_macro2 ::TokenStream > {
  let variant_name = &variant.ident;
  let fields = &variant.fields;
  let attrs = FieldAttributes ::from_attrs(variant.attrs.iter())?;

  if !attrs.enabled.value(item_attrs.enabled.value(true)) 
  {
  return Ok(qt! {});
 }

  if fields.is_empty() 
  {
  return Ok(qt! {});
 }

  if fields.len() != 1 
  {
  return_syn_err!(fields.span(), "Expects a single field to derive VariadicFrom");
 }

  let field = fields.iter().next().expect("Expects a single field to derive VariadicFrom");
  let field_type = &field.ty;
  let field_name = &field.ident;

  let body =  if let Some(field_name) = field_name 
  {
  qt! { Self :: #variant_name { #field_name: src } }
 } else {
  qt! { Self :: #variant_name( src ) }
 };

  if attrs.debug.value(false)
  {
  let debug = format!(
   r"
#[ automatically_derived ]
impl< {} > {}< {} >
where
  {}
{{
  #[ inline ]
  pub fn variadic_from( src: {} ) -> Self
  {{
  {}
 }}
}}
   ",
   qt! { #generics_impl },
   item_name,
   qt! { #generics_ty },
   qt! { #generics_where },
   qt! { #field_type },
   body,
 );
  let about = format!(
   r"derive: VariadicFrom
item: {item_name}
field: {variant_name}",
 );
  diag ::report_print(about, original_input, debug.clone());
 }

  Ok(qt! {
  #[ automatically_derived ]
  impl< #generics_impl > #item_name< #generics_ty >
  where
   #generics_where
  {
   #[ inline ]
   pub fn variadic_from( src: #field_type ) -> Self
   {
  #body
 }
 }
 })
}