alibi-macros 0.6.0

//! Proc macros used by the Alibi root crate.

use proc_macro::TokenStream as ProcMacroTokenStream;
use proc_macro2::TokenStream;
use quote::quote;
use syn::{Data, DeriveInput, Fields, Ident, LitStr, Type, parse_macro_input};

struct FieldInfo {
    ident: Ident,
    ty: Type,
    default_expr: TokenStream,
    skip: bool,
}

#[proc_macro_derive(AuthSchema, attributes(auth))]
pub fn derive_auth_schema(input: ProcMacroTokenStream) -> ProcMacroTokenStream {
    let input = parse_macro_input!(input as DeriveInput);
    generate_auth_schema(&input)
        .unwrap_or_else(syn::Error::into_compile_error)
        .into()
}

#[proc_macro_derive(PluginConfig, attributes(plugin, config))]
pub fn derive_plugin_config(input: ProcMacroTokenStream) -> ProcMacroTokenStream {
    let input = parse_macro_input!(input as DeriveInput);
    generate_plugin_config(&input)
        .unwrap_or_else(syn::Error::into_compile_error)
        .into()
}

fn generate_auth_schema(input: &DeriveInput) -> syn::Result<TokenStream> {
    let user = parse_type_attr(input, "user")?;
    let session = parse_type_attr(input, "session")?;
    let account = parse_type_attr(input, "account")?;
    let verification = parse_type_attr(input, "verification")?;
    let ident = &input.ident;

    Ok(quote! {
        impl ::alibi::__private_core::schema::AuthSchema for #ident {
            type User = #user;
            type Session = #session;
            type Account = #account;
            type Verification = #verification;
        }
    })
}

fn parse_type_attr(input: &DeriveInput, key: &str) -> Result<Type, syn::Error> {
    let mut parsed = None;
    for attr in &input.attrs {
        if !attr.path().is_ident("auth") {
            continue;
        }

        attr.parse_nested_meta(|meta| {
            if meta.path.is_ident(key) {
                let value = meta.value()?;
                let ty: LitStr = value.parse()?;
                parsed = Some(ty.parse()?);
            }
            Ok(())
        })?;
    }

    parsed.ok_or_else(|| {
        syn::Error::new_spanned(
            input,
            format!("missing #[auth({key} = path::to::Model)] attribute for AuthSchema"),
        )
    })
}

fn generate_plugin_config(input: &DeriveInput) -> syn::Result<TokenStream> {
    let plugin_name = parse_plugin_name(input)?;
    let config_name = &input.ident;

    let fields = match &input.data {
        Data::Struct(data) => match &data.fields {
            Fields::Named(named) => &named.named,
            Fields::Unnamed(_) | Fields::Unit => {
                return Err(syn::Error::new_spanned(
                    &input.ident,
                    "PluginConfig requires a struct with named fields",
                ));
            }
        },
        Data::Enum(_) | Data::Union(_) => {
            return Err(syn::Error::new_spanned(
                &input.ident,
                "PluginConfig requires a struct",
            ));
        }
    };

    let field_infos = fields
        .iter()
        .map(parse_field_info)
        .collect::<syn::Result<Vec<_>>>()?;

    let default_fields: Vec<_> = field_infos
        .iter()
        .map(|fi| {
            let name = &fi.ident;
            let expr = &fi.default_expr;
            quote! { #name: #expr }
        })
        .collect();

    let default_impl = quote! {
        impl Default for #config_name {
            fn default() -> Self {
                Self {
                    #(#default_fields,)*
                }
            }
        }
    };

    let builder_methods: Vec<_> = field_infos
        .iter()
        .filter(|fi| !fi.skip)
        .map(gen_builder_method)
        .collect();

    let plugin_impl = quote! {
        impl #plugin_name {
            #[must_use]
            pub fn new() -> Self {
                Self {
                    config: #config_name::default(),
                }
            }

            #[must_use]
            pub fn with_config(config: #config_name) -> Self {
                Self { config }
            }

            #(#builder_methods)*
        }
    };

    let plugin_default = quote! {
        impl Default for #plugin_name {
            fn default() -> Self {
                Self::new()
            }
        }
    };

    Ok(quote! {
        #default_impl
        #plugin_impl
        #plugin_default
    })
}

fn parse_plugin_name(input: &DeriveInput) -> Result<Ident, syn::Error> {
    for attr in &input.attrs {
        if attr.path().is_ident("plugin") {
            let name: LitStr = attr.parse_args_with(|stream: syn::parse::ParseStream<'_>| {
                let ident: Ident = stream.parse()?;
                if ident != "name" {
                    return Err(syn::Error::new(ident.span(), "expected `name`"));
                }
                let _: syn::Token![=] = stream.parse()?;
                stream.parse::<LitStr>()
            })?;
            return name.parse();
        }
    }
    Err(syn::Error::new_spanned(
        input,
        "missing #[plugin(name = ...)] attribute",
    ))
}

fn parse_field_info(field: &syn::Field) -> Result<FieldInfo, syn::Error> {
    let ident = field
        .ident
        .clone()
        .ok_or_else(|| syn::Error::new_spanned(field, "PluginConfig requires named fields"))?;
    let ty = field.ty.clone();
    let mut default_expr = None;
    let mut skip = false;

    for attr in &field.attrs {
        if attr.path().is_ident("config") {
            attr.parse_nested_meta(|meta| {
                if meta.path.is_ident("skip") {
                    skip = true;
                    Ok(())
                } else if meta.path.is_ident("default") {
                    let value = meta.value()?;
                    let expr: syn::Expr = value.parse()?;
                    default_expr = Some(quote!(#expr));
                    Ok(())
                } else {
                    Err(meta.error("expected `skip` or `default`"))
                }
            })?;
        }
    }

    let default_expr = default_expr.ok_or_else(|| {
        syn::Error::new_spanned(
            &ident,
            format!("field `{ident}` requires #[config(default = ...)]"),
        )
    })?;

    Ok(FieldInfo {
        ident,
        ty,
        default_expr,
        skip,
    })
}

fn is_string_type(ty: &Type) -> bool {
    matches!(ty, Type::Path(type_path) if type_path.path.is_ident("String"))
}

fn extract_option_inner(ty: &Type) -> Option<&Type> {
    if let Type::Path(type_path) = ty {
        let segment = type_path.path.segments.last()?;
        if segment.ident == "Option"
            && let syn::PathArguments::AngleBracketed(args) = &segment.arguments
            && let Some(syn::GenericArgument::Type(inner)) = args.args.first()
        {
            return Some(inner);
        }
    }
    None
}

fn gen_builder_method(fi: &FieldInfo) -> TokenStream {
    let field_name = &fi.ident;
    let ty = &fi.ty;

    if is_string_type(ty) {
        quote! {
            #[must_use]
            pub fn #field_name(mut self, val: impl Into<String>) -> Self {
                self.config.#field_name = val.into();
                self
            }
        }
    } else if let Some(inner) = extract_option_inner(ty) {
        quote! {
            #[must_use]
            pub fn #field_name(mut self, val: #inner) -> Self {
                self.config.#field_name = Some(val);
                self
            }
        }
    } else {
        quote! {
            #[must_use]
            pub fn #field_name(mut self, val: #ty) -> Self {
                self.config.#field_name = val;
                self
            }
        }
    }
}