diesel_derives/
query_id.rs

1use proc_macro2::TokenStream;
2use quote::quote;
3use syn::parse_quote;
4use syn::DeriveInput;
5
6use crate::util::wrap_in_dummy_mod;
7
8pub fn derive(mut item: DeriveInput) -> TokenStream {
9    for ty_param in item.generics.type_params_mut() {
10        ty_param
11            .bounds
12            .push(parse_quote!(diesel::query_builder::QueryId));
13    }
14    let (impl_generics, ty_generics, where_clause) = item.generics.split_for_impl();
15
16    let struct_name = &item.ident;
17    let lifetimes = item.generics.lifetimes();
18
19    let ty_params = item
20        .generics
21        .type_params()
22        .map(|ty_param| &ty_param.ident)
23        .collect::<Vec<_>>();
24
25    let query_id_ty_params = ty_params
26        .iter()
27        .map(|ty_param| quote!(<#ty_param as diesel::query_builder::QueryId>::QueryId));
28    let has_static_query_id = ty_params
29        .iter()
30        .map(|ty_param| quote!(<#ty_param as diesel::query_builder::QueryId>::HAS_STATIC_QUERY_ID));
31
32    wrap_in_dummy_mod(quote! {
33        #[allow(non_camel_case_types)]
34        impl #impl_generics diesel::query_builder::QueryId for #struct_name #ty_generics
35        #where_clause
36        {
37            type QueryId = #struct_name<#(#lifetimes,)* #(#query_id_ty_params,)*>;
38
39            const HAS_STATIC_QUERY_ID: bool = #(#has_static_query_id &&)* true;
40        }
41    })
42}