errore_impl/
prop.rs

1use proc_macro2::Span;
2use syn::Member;
3
4use crate::ast::{Enum, Field, Struct, Variant};
5use crate::span::MemberSpan;
6
7impl Enum<'_> {
8    pub(crate) fn has_source(&self) -> bool {
9        self.variants
10            .iter()
11            .any(|variant| variant.source_field().is_some() || variant.attrs.transparent.is_some())
12    }
13
14    pub(crate) fn has_display(&self) -> bool {
15        self.attrs.display.is_some()
16            || self.attrs.transparent.is_some()
17            || self
18                .variants
19                .iter()
20                .any(|variant| variant.attrs.display.is_some())
21            || self
22                .variants
23                .iter()
24                .all(|variant| variant.attrs.transparent.is_some())
25    }
26}
27
28impl Struct<'_> {
29    pub(crate) fn from_field(&self) -> Option<&Field> {
30        from_field(&self.fields)
31    }
32
33    pub(crate) fn source_field(&self) -> Option<&Field> {
34        source_field(&self.fields)
35    }
36}
37
38impl Variant<'_> {
39    pub(crate) fn from_field(&self) -> Option<&Field> {
40        from_field(&self.fields)
41    }
42
43    pub(crate) fn source_field(&self) -> Option<&Field> {
44        source_field(&self.fields)
45    }
46}
47
48impl Field<'_> {
49    pub(crate) fn source_span(&self) -> Span {
50        if let Some(source_attr) = &self.attrs.source {
51            source_attr.path().get_ident().unwrap().span()
52        } else if let Some(from_attr) = &self.attrs.from {
53            from_attr.path().get_ident().unwrap().span()
54        } else {
55            self.member.member_span()
56        }
57    }
58}
59
60fn from_field<'a, 'b>(fields: &'a [Field<'b>]) -> Option<&'a Field<'b>> {
61    for field in fields {
62        if field.attrs.from.is_some() {
63            return Some(field);
64        }
65    }
66    None
67}
68
69fn source_field<'a, 'b>(fields: &'a [Field<'b>]) -> Option<&'a Field<'b>> {
70    for field in fields {
71        if field.attrs.from.is_some() || field.attrs.source.is_some() {
72            return Some(field);
73        }
74    }
75    for field in fields {
76        match &field.member {
77            Member::Named(ident) if ident == "source" => return Some(field),
78            _ => {}
79        }
80    }
81    None
82}
errore_impl/prop.rs

errore_impl/
prop.rs
