sqlx-models-proc-macro 0.0.3

A helper crate for `sqlx-models`
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

use crate::prelude::*; // TODO:
                       //     getters for foreign keys

use super::Model;

struct Getters<'a> {
    model: &'a Model,
    getters: Vec<Getter>,
}
enum Getter {
    Unique {
        table_name: Ident,
        column_name: Ident,
        column_type: Type,
    },

    Foreign {
        table_name: Ident,
        referred: Ident,
    },
}

impl<'a> ToTokens for Getters<'a> {
    fn to_tokens(&self, tokens: &mut TokenStream2) {
        let ident = &self.model.name;
        let getters = &self.getters;
        tokens.extend(quote! {
            impl #ident {
                #(#getters)*
            }
        })
    }
}

impl<'a> ToTokens for Getter {
    fn to_tokens(&self, tokens: &mut TokenStream2) {
        match &self {
            Self::Unique {
                table_name,
                column_name,
                column_type,
            } => {
                
                let query = format!("select * from {} where co;", table_name, column_name);
                tokens.extend(quote! {
                    fn #name(&self, val: #dtype) -> ::std::result::Result<Vec<_>, ::sqlx::Error>{
                        ::sqlx::query(#query).
                        .fetch_all(&conn)
                        .await
                    }
                });
            }
        }
    }
}

impl Getter {}