ts-sql-helper-derive 0.6.0

Derives for ts-sql-helper-lib
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

use proc_macro2::TokenStream;
use quote::{format_ident, quote};
use syn::Ident;

pub fn create_main_struct(name: &Ident, query: &str, parameters: &[syn::Type]) -> TokenStream {
    let parameter_fields: Vec<_> = parameters
        .iter()
        .enumerate()
        .map(|(index, field_type)| {
            let name = format_ident!("p{}", index + 1);
            quote! {
                #name: #field_type
            }
        })
        .collect();

    let parameter_names: Vec<_> = parameters
        .iter()
        .enumerate()
        .map(|(index, _)| format_ident!("p{}", index + 1))
        .collect();

    let self_parameter_names = parameter_names.iter().map(|param| quote!(&self.#param));

    let parameter_count = parameters.len();

    quote! {
        struct #name<'a> {
            #( #parameter_fields , )*
            pub phantom_data: core::marker::PhantomData<&'a ()>,
        }
        impl<'a> #name<'a> {
            pub const QUERY: &'static str = #query;
            pub fn params(#( #parameter_fields ),*) -> Self {
                Self {
                    #( #parameter_names , )*
                    phantom_data: core::marker::PhantomData,
                }
            }

            pub fn as_array(&'a self) -> [&'a (dyn ts_sql_helper_lib::postgres::types::ToSql + Sync); #parameter_count] {
                [
                    #( #self_parameter_names , )*
                ]
            }
        }
    }
}