use proc_macro2::{Delimiter, Ident, Span, TokenStream};
use quote::quote;
use syn::{ItemEnum, LitStr, Variant};
use crate::{
config::Config,
ctx::Context,
idents::config::CONFIG_DOC,
order::OrderTrait,
tokens::{
AloneDecoration, RenameField, destructure_data, destructure_types, get_delimiter,
with_delimiter,
},
};
const DEFAULT_DOC: &str = "Creates a [`{}::{}`] variant from the provided data.";
pub fn expand_from(
input: &ItemEnum,
variant: &Variant,
order: &OrderTrait,
) -> syn::Result<TokenStream> {
let mut config = Config::new(&order.config, None)?;
let doc = config.get_formatted_lit_str(
CONFIG_DOC,
LitStr::new(DEFAULT_DOC, Span::call_site()),
[&input.ident.to_string(), &variant.ident.to_string()],
)?;
config.finish()?;
let fields = &variant.fields;
let delimiter = get_delimiter(fields);
let ty = destructure_types(
fields,
TokenStream::new(),
quote! { () },
AloneDecoration::None,
);
let destruct = destructure_data(
fields,
TokenStream::new(),
with_delimiter(TokenStream::new(), delimiter),
delimiter,
AloneDecoration::DelimitedNoComma,
RenameField::Auto,
);
let ret = destructure_data(
fields,
TokenStream::new(),
quote! { () },
Delimiter::Parenthesis,
AloneDecoration::None,
RenameField::Auto,
);
let variant_ident = &variant.ident;
let trait_ident = Ident::new("From", order.ident.span());
let method_ident = Ident::new("from", order.ident.span());
let content = quote! {
#[doc = #doc]
#[inline]
fn #method_ident (#ret: #ty) -> Self {
Self::#variant_ident #destruct
}
};
Ok(input.in_impl(
quote! { ::core::convert::#trait_ident<#ty> for },
&content,
None,
))
}