educe/common/
where_predicates_bool.rs

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
use quote::{quote, ToTokens};
use syn::{
    parse::{Parse, ParseStream},
    punctuated::Punctuated,
    spanned::Spanned,
    token::Comma,
    Expr, GenericParam, Lit, Meta, MetaNameValue, Path, Token, Type, WherePredicate,
};

use super::path::path_to_string;

pub(crate) type WherePredicates = Punctuated<WherePredicate, Token![,]>;

pub(crate) enum WherePredicatesOrBool {
    WherePredicates(WherePredicates),
    Bool(bool),
    All,
}

impl WherePredicatesOrBool {
    fn from_lit(lit: &Lit) -> syn::Result<Self> {
        Ok(match lit {
            Lit::Bool(lit) => Self::Bool(lit.value),
            Lit::Str(lit) => match lit.parse_with(WherePredicates::parse_terminated) {
                Ok(where_predicates) => Self::WherePredicates(where_predicates),
                Err(_) if lit.value().is_empty() => Self::Bool(false),
                Err(error) => return Err(error),
            },
            other => {
                return Err(syn::Error::new(
                    other.span(),
                    "unexpected kind of literal (only boolean or string allowed)",
                ))
            },
        })
    }
}

impl Parse for WherePredicatesOrBool {
    #[inline]
    fn parse(input: ParseStream) -> syn::Result<Self> {
        if let Ok(lit) = input.parse::<Lit>() {
            return Self::from_lit(&lit);
        }

        if let Ok(_star) = input.parse::<Token![*]>() {
            return Ok(Self::All);
        }

        Ok(Self::WherePredicates(input.parse_terminated(WherePredicate::parse, Token![,])?))
    }
}

#[inline]
pub(crate) fn meta_name_value_2_where_predicates_bool(
    name_value: &MetaNameValue,
) -> syn::Result<WherePredicatesOrBool> {
    if let Expr::Lit(lit) = &name_value.value {
        return WherePredicatesOrBool::from_lit(&lit.lit);
    }

    Err(syn::Error::new(
        name_value.value.span(),
        format!(
            "expected `{path} = \"where_predicates\"` or `{path} = false`",
            path = path_to_string(&name_value.path)
        ),
    ))
}

#[inline]
pub(crate) fn meta_2_where_predicates(meta: &Meta) -> syn::Result<WherePredicatesOrBool> {
    match &meta {
        Meta::NameValue(name_value) => meta_name_value_2_where_predicates_bool(name_value),
        Meta::List(list) => list.parse_args::<WherePredicatesOrBool>(),
        Meta::Path(path) => Err(syn::Error::new(
            path.span(),
            format!(
                "expected `{path} = \"where_predicates\"`, `{path}(where_predicates)`, `{path} = \
                 false`, or `{path}(false)`",
                path = path.clone().into_token_stream()
            ),
        )),
    }
}

#[inline]
pub(crate) fn create_where_predicates_from_all_generic_parameters(
    params: &Punctuated<GenericParam, Comma>,
    bound_trait: &Path,
) -> WherePredicates {
    let mut where_predicates = Punctuated::new();

    for param in params {
        if let GenericParam::Type(ty) = param {
            let ident = &ty.ident;

            where_predicates.push(syn::parse2(quote! { #ident: #bound_trait }).unwrap());
        }
    }

    where_predicates
}

#[inline]
pub(crate) fn create_where_predicates_from_generic_parameters_check_types(
    bound_trait: &Path,
    types: &[&Type],
    supertraits: &[proc_macro2::TokenStream],
) -> WherePredicates {
    let mut where_predicates = Punctuated::new();

    for t in types {
        where_predicates.push(syn::parse2(quote! { #t: #bound_trait }).unwrap());
    }

    for supertrait in supertraits {
        where_predicates.push(syn::parse2(quote! { Self: #supertrait }).unwrap());
    }

    where_predicates
}