    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

    use crate::ToTokens;
    use proc_macro2::{Span, TokenStream};

    pub trait Spanned {
        fn __span(&self) -> Span;
    }

    impl Spanned for Span {
        fn __span(&self) -> Span {
            *self
        }
    }

    impl<T: ?Sized + ToTokens> Spanned for T {
        fn __span(&self) -> Span {
            join_spans(self.into_token_stream())
        }
    }

    fn join_spans(tokens: TokenStream) -> Span {
        #[cfg(not(needs_invalid_span_workaround))]
        let mut iter = tokens.into_iter().map(|tt| tt.span());

        #[cfg(needs_invalid_span_workaround)]
        let mut iter = tokens.into_iter().filter_map(|tt| {
            let span = tt.span();
            let debug = format!("{:?}", span);
            if debug.ends_with("bytes(0..0)") {
                None
            } else {
                Some(span)
            }
        });

        let first = match iter.next() {
            Some(span) => span,
            None => return Span::call_site(),
        };

        iter.fold(None, |_prev, next| Some(next))
            .and_then(|last| first.join(last))
            .unwrap_or(first)
    }