bomboni_request_derive 0.3.0

use std::collections::BTreeSet;

use proc_macro2::TokenStream;
use quote::{ToTokens, quote};
use syn::Path;

use crate::parse::{
    oneof::utility::get_variant_extract,
    oneof::utility::get_variant_source_ident,
    options::{ParseDerive, ParseOptions, ParseTaggedUnion, ParseVariant},
    parse_utility::{expand_field_extract, expand_parse_field_type, make_field_error_path},
};

pub fn expand(options: &ParseOptions, variants: &[ParseVariant]) -> syn::Result<TokenStream> {
    options.tagged_union.as_ref().map_or_else(
        || expand_parse(options, variants),
        |tagged_union| expand_tagged_union(options, variants, tagged_union),
    )
}

fn expand_parse(options: &ParseOptions, variants: &[ParseVariant]) -> syn::Result<TokenStream> {
    let (parse_derived_borrowed, parse_variants) =
        expand_parse_variants(options, variants, &options.source)?;

    let source = &options.source;
    let ident = &options.ident;

    let (impl_generics, type_generics, where_clause) = options.generics.split_for_impl();

    Ok(quote! {
        #[automatically_derived]
        impl #impl_generics RequestParse<#source> for #ident #type_generics #where_clause {
            fn parse(source: #source) -> RequestResult<Self> {
                let variant_name = source.get_variant_name();
                #parse_derived_borrowed
                Ok(match source {
                    #parse_variants
                    _ => {
                        return Err(RequestError::generic(CommonError::UnknownOneofVariant));
                    }
                })
            }
        }
    })
}

fn expand_tagged_union(
    options: &ParseOptions,
    variants: &[ParseVariant],
    tagged_union: &ParseTaggedUnion,
) -> syn::Result<TokenStream> {
    let (parse_derived_borrowed, parse_variants) =
        expand_parse_variants(options, variants, &tagged_union.oneof)?;

    let source = &options.source;
    let ident = &options.ident;

    let field_ident = &tagged_union.field;
    let field_literal = tagged_union.field.to_string();

    let (impl_generics, type_generics, where_clause) = options.generics.split_for_impl();

    Ok(quote! {
        #[automatically_derived]
        impl #impl_generics RequestParse<#source> for #ident #type_generics #where_clause {
            fn parse(source: #source) -> RequestResult<Self> {
                let source = source.#field_ident
                    .ok_or_else(|| RequestError::field(#field_literal, CommonError::RequiredFieldMissing))?;
                let variant_name = source.get_variant_name();
                #parse_derived_borrowed
                Ok(match source {
                    #parse_variants
                    _ => {
                        return Err(RequestError::generic(CommonError::UnknownOneofVariant));
                    }
                })
            }
        }
    })
}

fn expand_parse_variants(
    options: &ParseOptions,
    variants: &[ParseVariant],
    source: &Path,
) -> syn::Result<(TokenStream, TokenStream)> {
    let ident = &options.ident;

    let mut parse_derived_borrowed = quote!();
    let mut parse_variants = quote!();

    for variant in variants.iter().filter(|variant| !variant.options.skip) {
        let target_variant_ident = &variant.ident;
        let source_variant_ident = get_variant_source_ident(variant)?;

        if let Some(ParseDerive { source_borrow, .. }) = variant.options.derive.as_ref() {
            let parse_variant = expand_parse_variant(variant)?;
            if *source_borrow {
                parse_derived_borrowed.extend(parse_variant);
            } else {
                parse_variants.extend(quote! {
                    #source::#source_variant_ident(source) => {
                        #ident::#target_variant_ident({
                            #parse_variant
                        })
                    }
                });
            }
            continue;
        }

        if variant.fields.is_empty() {
            parse_variants.extend(if variant.source_unit {
                quote! {
                    #source::#source_variant_ident(_) => {
                        #ident::#target_variant_ident
                    }
                }
            } else {
                quote! {
                    #source::#source_variant_ident => {
                        #ident::#target_variant_ident
                    }
                }
            });
        } else {
            let parse_variant = expand_parse_variant(variant)?;
            parse_variants.extend(quote! {
                #source::#source_variant_ident(source) => {
                    #ident::#target_variant_ident({
                        #parse_variant
                    })
                }
            });
        }
    }

    Ok((parse_derived_borrowed, parse_variants))
}

fn expand_parse_variant(variant: &ParseVariant) -> syn::Result<TokenStream> {
    if variant.options.keep {
        return Ok(quote! {
            source.clone()
        });
    }

    let extract = get_variant_extract(variant)?;
    let field_error_path_wrapper = quote! {
        PathErrorStep::Field(variant_name.into())
    };

    if let Some(ParseDerive {
        parse,
        module,
        source_borrow,
        ..
    }) = variant.options.derive.as_ref()
    {
        let parse_impl = parse
            .as_ref()
            .map(ToTokens::to_token_stream)
            .or_else(|| module.as_ref().map(|module| quote!(#module::parse)))
            .unwrap();

        if variant.options.source.is_some() || variant.options.extract.is_some() {
            let (extract_impl, _get_impl, field_path) = expand_field_extract(
                &extract,
                &BTreeSet::new(),
                None,
                Some(&field_error_path_wrapper),
                *source_borrow,
            );
            let field_error_path =
                make_field_error_path(&field_path, Some(&field_error_path_wrapper));

            return Ok(if *source_borrow {
                quote! {
                    if let Some(result) = #parse_impl(&source) {
                        return result.map_err(|err: RequestError| err.wrap_field(variant_name));
                    }
                }
            } else {
                quote! {
                    #extract_impl
                    let target = #parse_impl(target)
                        .map_err(|err: RequestError| err.wrap_path(#field_error_path))?;
                    target
                }
            });
        }

        if *source_borrow {
            return Ok(quote! {
                if let Some(result) = #parse_impl(&source) {
                    return result.map_err(|err: RequestError| err.wrap_field(variant_name));
                }
            });
        }

        return Ok(quote! {
            #parse_impl(source).map_err(|err: RequestError| err.wrap_field(variant_name))?
        });
    }

    let field_type_info = variant.type_info.as_ref().unwrap();
    let (extract_impl, get_impl, field_path) = expand_field_extract(
        &extract,
        &BTreeSet::new(),
        Some(field_type_info),
        Some(&field_error_path_wrapper),
        false,
    );
    let field_error_path = make_field_error_path(&field_path, Some(&field_error_path_wrapper));
    let parse_field_impl = expand_parse_field_type(
        &variant.options,
        field_type_info,
        field_error_path,
        get_impl,
    );

    Ok(quote! {
        #extract_impl
        #parse_field_impl
        target
    })
}