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

use darling::FromDeriveInput;
use syn::{parse_macro_input, AttributeArgs, DeriveInput, Item};
mod persistent;
use persistent::PersistentInfo;
use persistent::ProjectionInfo;
mod queries;
use queries::persistent_queries;

#[proc_macro_attribute]
pub fn queries(args: proc_macro::TokenStream, original: proc_macro::TokenStream) -> proc_macro::TokenStream {
    let parsed: Item = syn::parse(original).unwrap();
    let args: AttributeArgs = parse_macro_input!(args as AttributeArgs);
    let gen = persistent_queries(parsed, args, false);
    gen.into()
}

#[proc_macro_attribute]
pub fn embedded_queries(args: proc_macro::TokenStream, original: proc_macro::TokenStream) -> proc_macro::TokenStream {
    let parsed: Item = syn::parse(original).unwrap();
    let args: AttributeArgs = parse_macro_input!(args as AttributeArgs);
    let gen = persistent_queries(parsed, args, true);
    gen.into()
}

#[proc_macro_derive(PersistentEmbedded, attributes(index))]
pub fn persistent_embedded(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    let parsed: DeriveInput = syn::parse(input).unwrap();

    let gen = PersistentInfo::from_derive_input(&parsed).unwrap().to_embedded_tokens();
    gen.into()
}

#[proc_macro_derive(Persistent, attributes(index))]
pub fn persistent(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    let parsed: DeriveInput = syn::parse(input).unwrap();

    let gen = PersistentInfo::from_derive_input(&parsed).unwrap().to_tokens();
    gen.into()
}

#[proc_macro_derive(Projection, attributes(projection))]
pub fn projection(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    let parsed: DeriveInput = syn::parse(input).unwrap();

    let gen = ProjectionInfo::from_derive_input(&parsed).unwrap().to_tokens();
    gen.into()
}