kosame_macro 0.2.0

Macro-based Rust ORM focused on developer ergonomics
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100

use crate::{
    attribute::{CustomMeta, MetaLocation},
    data_type::DataType,
    path_ext::PathExt,
    quote_option::QuoteOption,
};

use super::column_constraint::ColumnConstraints;
use convert_case::{Case, Casing};
use proc_macro2::TokenStream;
use quote::{ToTokens, quote};
use syn::{
    Attribute, Ident,
    parse::{Parse, ParseStream},
};

pub struct Column {
    pub attrs: Vec<Attribute>,
    pub name: Ident,
    pub data_type: DataType,
    pub constraints: ColumnConstraints,
}

impl Column {
    pub fn rust_name(&self) -> Ident {
        let meta = CustomMeta::parse_attrs(&self.attrs, MetaLocation::Column)
            .expect("custom meta should be checked earlier");
        match meta.rename {
            Some(rename) => rename.value,
            None => Ident::new(
                &self.name.to_string().to_case(Case::Snake),
                self.name.span(),
            ),
        }
    }
}

impl Parse for Column {
    fn parse(input: ParseStream) -> syn::Result<Self> {
        let attrs = Attribute::parse_outer(input)?;
        CustomMeta::parse_attrs(&attrs, MetaLocation::Column)?;
        let name = input.parse()?;
        let data_type = input.parse()?;

        Ok(Self {
            attrs,
            name,
            data_type,
            constraints: input.parse()?,
        })
    }
}

impl ToTokens for Column {
    fn to_tokens(&self, tokens: &mut TokenStream) {
        let meta = CustomMeta::parse_attrs(&self.attrs, MetaLocation::Column)
            .expect("custom meta should be checked earlier");

        let name = self.name.to_string();
        let rust_name = self.rust_name();

        let data_type = &self.data_type;
        let data_type_string = data_type.name.to_string();
        let rust_type_not_null = match meta.type_override {
            Some(type_override) => type_override.value.to_call_site(3).to_token_stream(),
            None => quote! { #data_type },
        };
        let rust_type_nullable = quote! { Option<#data_type> };
        let rust_type_auto =
            if self.constraints.not_null().is_none() && self.constraints.primary_key().is_none() {
                rust_type_nullable.clone()
            } else {
                rust_type_not_null.clone()
            };

        let not_null = self.constraints.not_null().is_some();
        let primary_key = self.constraints.primary_key().is_some();
        let default = QuoteOption(self.constraints.default().map(|default| {
            let expr = default.expr();
            quote! { &#expr }
        }));

        quote! {
            pub mod #rust_name {
                pub const COLUMN_NAME: &str = #name;
                pub const COLUMN: ::kosame::repr::schema::Column<'_> = ::kosame::repr::schema::Column {
                    name: #name,
                    data_type: #data_type_string,
                    not_null: #not_null,
                    primary_key: #primary_key,
                    default: #default,
                };
                pub type TypeNotNull = #rust_type_not_null;
                pub type TypeNullable = #rust_type_nullable;
                pub type Type = #rust_type_auto;
            }
        }
        .to_tokens(tokens);
    }
}