cypher-dto-macros 0.3.0

The macros for cypher-dto.
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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130

use super::FieldType;
use quote::{__private::TokenStream, quote};

/// The code for extracting a field from a [neo4rs::Row] or other BoltMap impl.
///
/// It uses `value.get()` where value is a BoltMap such as a Row, Node, Relation, or UnboundedRelation.
pub fn field_from_boltmap(name: &str, typ: &FieldType) -> TokenStream {
    match typ {
        FieldType::DateTimeUtc(_ty) => {
            quote!(
                value.get::<::chrono::DateTime<::chrono::FixedOffset>>(#name)
                    .map(|dt| dt.into())
                    .map_err(|e| ::cypher_dto::Error::MissingField(#name.to_owned()))?
            )
        }
        FieldType::OptionDateTimeUtc(_ty) => {
            quote!(
                value.get::<::chrono::DateTime<::chrono::FixedOffset>>(#name)
                    .map(|dt| dt.into()).ok()
            )
        }
        FieldType::Num(_ty, num) => {
            // Build from the inside out.
            // Example: cypher-dto/lib/src/entity.rs#L425
            let mut tokens = quote!(
                value.get(#name).map_err(|e| ::cypher_dto::Error::MissingField(#name.to_owned()))?
            );

            // Handle `value.get::<type>()`
            if let Some(type_arg) = num.map_getter_type_arg() {
                let cast_type = type_arg.to_type();
                tokens = quote!(
                    value.get::<#cast_type>(#name).map_err(|e| ::cypher_dto::Error::MissingField(#name.to_owned()))?
                );
            }

            // Handle `some_type::try_from(value.get())`
            if num.map_uses_try_from() {
                let try_from_type = num.to_type();
                tokens = quote!(
                    #try_from_type::try_from(#tokens)
                    .map_err(|_| ::cypher_dto::Error::TypeMismatch(#name.to_owned()))?
                );
            }

            // Handle `value.get() as type`
            if let Some(type_arg) = num.map_cast() {
                let cast_type = type_arg.to_type();
                tokens = quote!(#tokens as #cast_type);
            }

            tokens
        }
        FieldType::OptionNum(_ty, num) => {
            // Build from the inside out.
            // Example: cypher-dto/lib/src/entity.rs#L595
            //
            // i32_opt: match value.get::<i64>("i32_opt") {
            //     Some(v) => Some(
            //         v.try_into()
            //             .map_err(|_| Error::TypeMismatch("i32_opt".to_owned()))?,
            //     ),
            //     None => None,
            // },
            // i64_opt: match value.get("i64_opt") {
            //     Some(v) => Some(v),
            //     None => None,
            // },
            // i128_opt: match value.get::<i64>("i128_opt") {
            //     Some(v) => Some(
            //         v.try_into()
            //             .map_err(|_| Error::TypeMismatch("i128_opt".to_owned()))?,
            //     ),
            //     None => None,
            // },
            // f32_opt: match value.get::<f64>("f32_opt") {
            //     Some(v) => Some(v as f32),
            //     None => None,
            // },
            // f64_opt: match value.get("f64_opt") {
            //     Some(v) => Some(v),
            //     None => None,
            // },
            let mut get_call = quote!(
                value.get(#name)
            );
            let mut some_inner = quote!(v);

            // Handle `value.get::<type>()`
            if let Some(type_arg) = num.map_getter_type_arg() {
                let cast_type = type_arg.to_type();
                get_call = quote!(
                    value.get::<#cast_type>(#name)
                );
            }

            // Handle `some_type::try_from(value.get())`
            if num.map_uses_try_from() {
                // let try_from_type = num.to_type();
                some_inner = quote!(
                    v.try_into()
                    .map_err(|_| ::cypher_dto::Error::TypeMismatch(#name.to_owned()))?
                );
            }

            // Handle `value.get() as type`
            if let Some(type_arg) = num.map_cast() {
                let cast_type = type_arg.to_type();
                some_inner = quote!(#some_inner as #cast_type);
            }

            quote!(
              match #get_call {
                  Ok(v) => Some(#some_inner),
                  Err(_) => None,
              }
            )
        }
        FieldType::OptionOther(_ty) => {
            quote!(
                value.get(#name)
            )
        }
        FieldType::Other(_ty) => {
            quote!(
                value.get(#name).map_err(|e| ::cypher_dto::Error::MissingField(#name.to_owned()))?
            )
        }
    }
}