zino-derive 0.26.0

Derived traits for zino.
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

use super::parser;
use proc_macro2::TokenStream;
use quote::{format_ident, quote};
use syn::DeriveInput;

/// Integer types
const UNSIGNED_INTEGER_TYPES: [&str; 5] = ["u64", "u32", "u16", "u8", "usize"];

/// Parses the token stream for the `DecodeRow` trait derivation.
pub(super) fn parse_token_stream(input: DeriveInput) -> TokenStream {
    // Model name
    let name = input.ident;

    // Parsing field attributes
    let mut decode_model_fields = Vec::new();
    for field in parser::parse_struct_fields(input.data) {
        let type_name = parser::get_type_name(&field.ty);
        if let Some(ident) = field.ident {
            let name = ident.to_string();
            let mut ignore = false;
            'inner: for attr in field.attrs.iter() {
                let arguments = parser::parse_schema_attr(attr);
                for (key, _value) in arguments.iter() {
                    if key == "ignore" || key == "write_only" {
                        ignore = true;
                        break 'inner;
                    }
                }
            }
            if ignore {
                continue;
            }
            if type_name == "Uuid" {
                decode_model_fields.push(quote! {
                    model.#ident = orm::decode_uuid(row, #name)?;
                });
            } else if type_name == "Option<Uuid>" {
                decode_model_fields.push(quote! {
                    model.#ident = orm::decode_uuid(row, #name).ok().filter(|id| !id.is_nil());
                });
            } else if type_name == "Decimal" {
                decode_model_fields.push(quote! {
                    model.#ident = orm::decode_decimal(row, #name)?;
                });
            } else if type_name == "Map" {
                decode_model_fields.push(quote! {
                    if let Some(JsonValue::Object(map)) = orm::decode_optional(row, #name)? {
                        model.#ident = map;
                    }
                });
            } else if parser::check_vec_type(&type_name) {
                decode_model_fields.push(quote! {
                    model.#ident = orm::decode_array(row, #name)?;
                });
            } else if UNSIGNED_INTEGER_TYPES.contains(&type_name.as_str()) {
                let integer_type_ident = format_ident!("{}", type_name.replace('u', "i"));
                decode_model_fields.push(quote! {
                    if let Some(value) = orm::decode_optional::<#integer_type_ident>(row, #name)? {
                        model.#ident = value.try_into()?;
                    }
                });
            } else {
                decode_model_fields.push(quote! {
                    if let Some(value) = orm::decode_optional(row, #name)? {
                        model.#ident = value;
                    }
                });
            }
        }
    }
    quote! {
        impl zino_core::model::DecodeRow<zino_core::orm::DatabaseRow> for #name {
            type Error = zino_core::error::Error;

            fn decode_row(row: &zino_core::orm::DatabaseRow) -> Result<Self, Self::Error> {
                use zino_core::{extension::JsonValueExt, orm, JsonValue};

                let mut model = Self::default();
                #(#decode_model_fields)*
                Ok(model)
            }
        }
    }
}