prest-db-macro 0.3.0

macro that derives Table trait to work with gluesql db
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
80
81

use super::*;

pub fn from_field(field: Field) -> Column {
    let pkey = field
        .attrs
        .iter()
        .find(|a| a.path().to_token_stream().to_string() == "pkey_column")
        .is_some();

    let unique = field
        .attrs
        .iter()
        .find(|a| a.path().to_token_stream().to_string() == "unique_column")
        .is_some()
        || pkey;

    let field_name = field.ident.expect("only named structs");
    let field_name_str = field_name.to_string();
    let full_type = field.ty;
    let type_str = full_type.to_token_stream().to_string();
    let type_str = type_str.as_str();

    let (inner_type_str, optional, list) =
        if type_str.starts_with("Option < ") && type_str.ends_with(" >") {
            let inner_type_str = type_str
                .trim_start_matches("Option < ")
                .trim_end_matches(" >");
            (inner_type_str, true, false)
        } else if type_str.starts_with("Vec < ") && type_str.ends_with(" >") {
            let inner_type_str = type_str.trim_start_matches("Vec < ").trim_end_matches(" >");
            (inner_type_str, false, true)
        } else {
            (type_str, false, false)
        };

    if pkey && optional || pkey && list {
        panic!("Primary Key (first attribute by default) cannot be Option<...> or Vec<...>")
    }

    let inner_type: syn::Type = syn::parse_str(inner_type_str).unwrap();

    use SqlType::*;
    let sql_type = match inner_type_str {
        "Uuid" => Uuid,
        "NaiveDateTime" => Timestamp,
        "bool" => Boolean,
        "u128" => Uint128,
        "u64" => Uint64,
        "u32" => Uint32,
        "u16" => Uint16,
        "u8" => Uint8,
        "i128" => Int128,
        "i64" => Int,
        "i32" => Int32,
        "i16" => Int16,
        "i8" => Int8,
        "f32" => Float32,
        "f64" => Float,
        _ => Text, // string, custom types and fallback for others
    };

    let serialized = match inner_type_str {
        "Uuid" | "String" | "NaiveDateTime" | "bool" | "u128" | "u64" | "u32" | "u16" | "u8"
        | "i128" | "i64" | "i32" | "i16" | "i8" | "f64" | "f32" => false,
        _ => true,
    };

    Column {
        full_type_str: type_str.replace(' ', ""),
        sql_type,
        field_name,
        field_name_str,
        full_type,
        inner_type,
        pkey,
        optional,
        list,
        unique,
        serialized,
    }
}