microtype-macro 0.2.6

Proc macro for generating microtypes
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

use proc_macro2::TokenStream;
use quote::quote;
use syn::{Ident, Type};

use super::HAS_DIESEL;

pub fn diesel_impl_not_secret(sql_type: &Type, inner: &Type, name: &Ident) -> TokenStream {
    let from_sql = from_sql_not_secret(sql_type, inner, name);
    let to_sql = to_sql_not_secret(sql_type, inner, name);

    if HAS_DIESEL {
        quote! {
            #from_sql
            #to_sql
        }
    } else {
        quote! {}
    }
}

pub fn diesel_impl_secret(sql_type: &Type, inner: &Type, name: &Ident) -> TokenStream {
    let from_sql = from_sql_secret(sql_type, inner, name);
    let to_sql = to_sql_secret(sql_type, inner, name);

    if HAS_DIESEL {
        quote! {
            #from_sql
            #to_sql
        }
    } else {
        quote! {}
    }
}

fn from_sql_not_secret(sql_type: &Type, inner: &Type, name: &Ident) -> TokenStream {
    quote! {
        impl<B: ::diesel::backend::Backend> ::diesel::deserialize::FromSql<#sql_type, B> for #name
        where
            #inner: ::diesel::deserialize::FromSql<#sql_type, B>,
        {
            fn from_sql(
                bytes: ::diesel::backend::RawValue<'_, B>,
            ) -> ::diesel::deserialize::Result<Self> {
                <#inner as ::diesel::deserialize::FromSql<#sql_type, B>>::from_sql(bytes).map(#name)
            }
        }
    }
}

fn to_sql_not_secret(sql_type: &Type, inner: &Type, name: &Ident) -> TokenStream {
    quote! {
        impl<B: ::diesel::backend::Backend> ::diesel::serialize::ToSql<#sql_type, B> for #name
        where
            #inner: ::diesel::serialize::ToSql<#sql_type, B>,
        {
            fn to_sql<'b>(
                &'b self,
                out: &mut diesel::serialize::Output<'b, '_, B>,
            ) -> diesel::serialize::Result {
                <#inner as ::diesel::serialize::ToSql<#sql_type, B>>::to_sql(&self.0, out)
            }
        }

    }
}

fn from_sql_secret(sql_type: &Type, inner: &Type, name: &Ident) -> TokenStream {
    quote! {
        impl<B: ::diesel::backend::Backend> ::diesel::deserialize::FromSql<#sql_type, B> for #name
        where
            #inner: ::diesel::deserialize::FromSql<#sql_type, B>,
        {
            fn from_sql(bytes: ::diesel::backend::RawValue<'_, B>) -> ::diesel::deserialize::Result<Self> {
                <#inner as ::diesel::deserialize::FromSql<#sql_type, B>>::from_sql(bytes)
                    .map(<Self as ::microtype::SecretMicrotype>::new)
            }
        }

    }
}

fn to_sql_secret(sql_type: &Type, inner: &Type, name: &Ident) -> TokenStream {
    quote! {
        impl<B: ::diesel::backend::Backend> ::diesel::serialize::ToSql<#sql_type, B> for #name
        where
            #inner: ::diesel::serialize::ToSql<#sql_type, B>,
        {
            fn to_sql<'b>(
                &'b self,
                out: &mut diesel::serialize::Output<'b, '_, B>,
            ) -> diesel::serialize::Result {
                <#inner as ::diesel::serialize::ToSql<#sql_type, B>>::to_sql(
                    <Self as ::microtype::secrecy::ExposeSecret<#inner>>::expose_secret(&self),
                    out,
                )
            }
        }
    }
}