1use crate::ToTokens;
2use proc_macro2::extra::DelimSpan;
3use proc_macro2::{Span, TokenStream};
4
5pub trait Spanned: private::Sealed {
7 fn __span(&self) -> Span;
8}
9
10impl Spanned for Span {
11 fn __span(&self) -> Span {
12 *self
13 }
14}
15
16impl Spanned for DelimSpan {
17 fn __span(&self) -> Span {
18 self.join()
19 }
20}
21
22impl<T: ?Sized + ToTokens> Spanned for T {
23 fn __span(&self) -> Span {
24 join_spans(self.into_token_stream())
25 }
26}
27
28fn join_spans(tokens: TokenStream) -> Span {
29 let mut iter = tokens.into_iter().map(|tt| tt.span());
30
31 let Some(first) = iter.next() else {
32 return Span::call_site();
33 };
34
35 iter.fold(None, |_prev, next| Some(next))
36 .and_then(|last| first.join(last))
37 .unwrap_or(first)
38}
39
40mod private {
41 use crate::ToTokens;
42 use proc_macro2::extra::DelimSpan;
43 use proc_macro2::Span;
44
45 pub trait Sealed {}
46 impl Sealed for Span {}
47 impl Sealed for DelimSpan {}
48 impl<T: ?Sized + ToTokens> Sealed for T {}
49}