baizekit-derive 0.1.22

Derive ToolKit
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68

use baizekit_seaorm::curd::derive::{CurdMacroOptions, derive_curd_impl};
use darling::FromDeriveInput;
use proc_macro::TokenStream;
use quote::quote;
use syn::{Data, DeriveInput, parse_macro_input};

use crate::derive_with::{Input, derive_with_impl};

mod derive_with;

#[proc_macro_derive(With)]
pub fn derive_with(tokens: TokenStream) -> TokenStream {
    let input = parse_macro_input!(tokens as DeriveInput);
    let input = match Input::from_derive_input(&input) {
        Ok(v) => v,
        Err(e) => return e.write_errors().into(),
    };

    match derive_with_impl(input) {
        Ok(expanded) => TokenStream::from(expanded),
        Err(e) => TokenStream::from(e.write_errors()),
    }
}

#[proc_macro_derive(Curd, attributes(curd))]
pub fn derive_curd(input: TokenStream) -> TokenStream {
    let input = parse_macro_input!(input as DeriveInput);
    let options = match CurdMacroOptions::from_derive_input(&input) {
        Ok(c) => c,
        Err(e) => return e.write_errors().into(),
    };

    derive_curd_impl(options).into()
}

#[proc_macro_derive(PaginatedFilter, attributes(paginate))]
pub fn derive_paginated_filter(input: TokenStream) -> TokenStream {
    let opts = parse_macro_input!(input as DeriveInput);

    let struct_name = &opts.ident;

    // 查找被 #[paginate] 标注的字段
    let field_name = if let Data::Struct(data_struct) = &opts.data {
        data_struct
            .fields
            .iter()
            .find(|f| f.attrs.iter().any(|attr| attr.path().is_ident("paginate")))
            .map(|f| f.ident.as_ref().unwrap())
    } else {
        None
    };

    let Some(field_name) = field_name else {
        return syn::Error::new_spanned(struct_name, "Expected one named field with #[paginate] attribute")
            .to_compile_error()
            .into();
    };

    let expanded = quote! {
        impl baizekit_seaorm::curd::PaginatedFilter for #struct_name {
            fn pagination(&self) -> Option<baizekit_seaorm::curd::Pagination> {
                self.#field_name.clone()
            }
        }
    };

    expanded.into()
}