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

mod parse;

use darling::FromDeriveInput;
use proc_macro::{self, TokenStream};
use syn::parse_macro_input;

use parse::ModelInput;

/// Automatically implement `Model` for your struct.
///
/// ## Attributes
///  * `table` - for structs:
///      - `table_name: String`: optional. Overwrites the table name
///  * `column` - for struct fields:
///      - `dtype: String`: optional. Overwrites the postgres datatype
///      - `unique: bool`: optional, default: `false`. Enables the `unqiue` constraint.
///      - `auto: bool`: optional, default: `false`. This autogenerated the
///        values.
///      - `column_name: String`: optional. Overwrites the column name.
#[proc_macro_derive(Model, attributes(table, column))]
pub fn derive(input: TokenStream) -> TokenStream {
    let input = ModelInput::from_derive_input(&parse_macro_input!(input)).unwrap();

    let output = input.impl_everything();

    output.into()
}

#[cfg(test)]
mod tests {
    use darling::FromDeriveInput;
    use syn::parse_str;

    use crate::parse::ModelInput;

    #[test]
    fn test() {
        let input = r#"
            #[derive(Model)]
            struct Book {
                #[column(primary_key, auto)]
                id: i64,
                title: String
            }
        "#;
        let tokens = parse_str(input).unwrap();
        let parsed_input = ModelInput::from_derive_input(&tokens).unwrap();
        let _output = parsed_input.impl_everything();
    }
}