#[macro_use]
pub(crate) mod sanitization;
mod read_options;
use crate::parser::{Input, ParseResult};
use proc_macro2::TokenStream;
use quote::quote;
#[allow(clippy::wildcard_imports)]
use sanitization::*;
pub(crate) fn generate_impl(
derive_input: &syn::DeriveInput,
binread_input: &ParseResult<Input>,
) -> TokenStream {
let (arg_type, read_opt_impl) = match binread_input {
ParseResult::Ok(binread_input) => (
binread_input.imports().types(),
read_options::generate(&derive_input.ident, &binread_input),
),
ParseResult::Partial(binread_input, error) => {
(binread_input.imports().types(), error.to_compile_error())
}
ParseResult::Err(error) => (quote! { () }, error.to_compile_error()),
};
let name = &derive_input.ident;
let (impl_generics, ty_generics, where_clause) = derive_input.generics.split_for_impl();
quote! {
#[allow(non_snake_case)]
impl #impl_generics #TRAIT_NAME for #name #ty_generics #where_clause {
type Args = #arg_type;
fn read_options<R: #READ_TRAIT + #SEEK_TRAIT>
(#READER: &mut R, #OPT: &#OPTIONS, #ARGS: Self::Args)
-> #BIN_RESULT<Self>
{
#read_opt_impl
}
}
}
}