cushy/gooey-macros/src/animation.rs

use manyhow::bail;
use quote::ToTokens;
use syn::{Field, ItemStruct};

use crate::*;

pub fn linear_interpolate(
    ItemStruct {
        ident,
        generics,
        fields,
        ..
    }: ItemStruct,
) -> Result<TokenStream> {
    if let Some(generic) = generics.type_params().next() {
        bail!(generic, "generics not supported");
    }

    let fields = match fields {
        syn::Fields::Unit => bail!(ident, "unit structs are not supported"),
        fields => fields
            .into_iter()
            .enumerate()
            .map(|(idx, Field {  ident, .. })| {
                let ident = ident
                    .map(ToTokens::into_token_stream)
                    .unwrap_or_else(|| proc_macro2::Literal::usize_unsuffixed(idx).into_token_stream());
                    quote!(#ident: ::gooey::animation::LinearInterpolate::lerp(&self.#ident, &__target.#ident, __percent),)
            }),
    };

    Ok(quote! {
        impl ::gooey::animation::LinearInterpolate for #ident {
            fn lerp(&self, __target: &Self, __percent: f32) -> Self {
                #ident{#(#fields)*}
            }
        }
    })
}

#[cfg(test)]
macro_rules! expansion_snapshot {
    (#[derive($fn:expr)]$($tokens:tt)*) => {{
        use insta::assert_snapshot;
        use prettyplease::unparse;
        use syn::{parse2, parse_quote};
        let input = parse_quote!($($tokens)*);
        let output = $fn(input).unwrap();
        assert_snapshot!(unparse(&parse2(output).unwrap()))
    }};
}

#[test]
fn test() {
    expansion_snapshot! {
        #[derive(linear_interpolate)]
        struct HelloWorld {
            fielda: Hello,
            fieldb: World,
        }
    };
    expansion_snapshot! {
        #[derive(linear_interpolate)]
        struct HelloWorld(Hello, World);
    };
}