georm-macros 0.2.1

Macro support for Georm. Not intended to be used directly.
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
69
70
71
72
73
74
75
76
77
78
79

use quote::quote;

#[derive(deluxe::ParseMetaItem, Clone)]
pub struct M2MLink {
    pub table: String,
    pub from: String,
    pub to: String,
}

#[derive(deluxe::ParseMetaItem)]
pub struct M2MRelationship {
    pub name: String,
    pub entity: syn::Type,
    pub table: String,
    #[deluxe(default = String::from("id"))]
    pub remote_id: String,
    pub link: M2MLink,
}

pub struct Identifier {
    pub table: String,
    pub id: String,
}

pub struct M2MRelationshipComplete {
    pub name: String,
    pub entity: syn::Type,
    pub local: Identifier,
    pub remote: Identifier,
    pub link: M2MLink,
}

impl M2MRelationshipComplete {
    pub fn new(other: &M2MRelationship, local_table: &String, local_id: &String) -> Self {
        Self {
            name: other.name.clone(),
            entity: other.entity.clone(),
            link: other.link.clone(),
            local: Identifier {
                table: local_table.to_string(),
                id: local_id.to_string(),
            },
            remote: Identifier {
                table: other.table.clone(),
                id: other.remote_id.clone(),
            },
        }
    }
}

impl From<&M2MRelationshipComplete> for proc_macro2::TokenStream {
    fn from(value: &M2MRelationshipComplete) -> Self {
        let function = syn::Ident::new(
            &format!("get_{}", value.name),
            proc_macro2::Span::call_site(),
        );
        let entity = &value.entity;
        let query = format!(
            "SELECT remote.*
FROM {} local
JOIN {} link ON link.{} = local.{}
JOIN {} remote ON link.{} = remote.{}
WHERE local.{} = $1",
            value.local.table,
            value.link.table,
            value.link.from,
            value.local.id,
            value.remote.table,
            value.link.to,
            value.remote.id,
            value.local.id
        );
        quote! {
            pub async fn #function(&self, pool: &::sqlx::PgPool) -> ::sqlx::Result<Vec<#entity>> {
                ::sqlx::query_as!(#entity, #query, self.get_id()).fetch_all(pool).await
            }
        }
    }
}