aboutsummaryrefslogtreecommitdiff
path: root/gen/lib/src/error.rs
blob: 79a27bd911d1eb7bd224d1882430d289a7d1b54d (plain)
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
// We can expose more detail on the error as the need arises, but start with an
// opaque error type for now.

use std::error::Error as StdError;
use std::fmt::{self, Debug, Display};
use std::iter;

#[allow(missing_docs)]
pub struct Error {
    pub(crate) err: crate::gen::Error,
}

impl Error {
    /// Returns the span of the error, if available.
    pub fn span(&self) -> Option<proc_macro2::Span> {
        match &self.err {
            crate::gen::Error::Syn(err) => Some(err.span()),
            _ => None,
        }
    }
}

impl From<crate::gen::Error> for Error {
    fn from(err: crate::gen::Error) -> Self {
        Error { err }
    }
}

impl Display for Error {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        Display::fmt(&self.err, f)
    }
}

impl Debug for Error {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        Debug::fmt(&self.err, f)
    }
}

impl StdError for Error {
    fn source(&self) -> Option<&(dyn StdError + 'static)> {
        self.err.source()
    }
}

impl IntoIterator for Error {
    type Item = Error;
    type IntoIter = IntoIter;

    fn into_iter(self) -> Self::IntoIter {
        match self.err {
            crate::gen::Error::Syn(err) => IntoIter::Syn(err.into_iter()),
            _ => IntoIter::Other(iter::once(self)),
        }
    }
}

pub enum IntoIter {
    Syn(<syn::Error as IntoIterator>::IntoIter),
    Other(iter::Once<Error>),
}

impl Iterator for IntoIter {
    type Item = Error;

    fn next(&mut self) -> Option<Self::Item> {
        match self {
            IntoIter::Syn(iter) => iter
                .next()
                .map(|syn_err| Error::from(crate::gen::Error::Syn(syn_err))),
            IntoIter::Other(iter) => iter.next(),
        }
    }
}