casco/
proc_macro2.rs

1//! Implementation of the token stream and related traits for `proc_macro2`.
2
3use crate::stream;
4
5/// Wrapper around the `proc_macro2` `TokenStream`.
6pub struct TokenStream(proc_macro2::TokenStream);
7
8/// Wrapper around the `proc_macro2` `IntoIter`.
9pub struct IntoIter(proc_macro2::token_stream::IntoIter);
10
11impl From<proc_macro2::TokenStream> for TokenStream {
12    fn from(value: proc_macro2::TokenStream) -> Self {
13        TokenStream(value)
14    }
15}
16
17impl Into<proc_macro2::TokenStream> for TokenStream {
18    fn into(self) -> proc_macro2::TokenStream {
19        self.0
20    }
21}
22
23impl stream::IntoTokenStream for proc_macro2::TokenStream {
24    type TokenStream = TokenStream;
25
26    fn into_token_stream(self) -> Self::TokenStream {
27        TokenStream(self)
28    }
29}
30
31impl stream::TokenStream for TokenStream {
32    type Group = proc_macro2::Group;
33    type Ident = proc_macro2::Ident;
34    type Literal = proc_macro2::Literal;
35    type Punct = proc_macro2::Punct;
36    type Span = proc_macro2::Span;
37}
38
39impl IntoIterator for TokenStream {
40    type IntoIter = IntoIter;
41    type Item = stream::TokenTree<TokenStream>;
42
43    fn into_iter(self) -> Self::IntoIter {
44        IntoIter(self.0.into_iter())
45    }
46}
47
48impl Iterator for IntoIter {
49    type Item = stream::TokenTree<TokenStream>;
50
51    fn next(&mut self) -> Option<Self::Item> {
52        Some(match self.0.next()? {
53            proc_macro2::TokenTree::Group(group) => stream::TokenTree::Group(group),
54            proc_macro2::TokenTree::Ident(ident) => stream::TokenTree::Ident(ident),
55            proc_macro2::TokenTree::Punct(punct) => stream::TokenTree::Punct(punct),
56            proc_macro2::TokenTree::Literal(literal) => stream::TokenTree::Literal(literal),
57        })
58    }
59}
60
61impl stream::Spanned<proc_macro2::Span> for proc_macro2::Group {
62    fn span(&self) -> proc_macro2::Span {
63        proc_macro2::Group::span(self)
64    }
65}
66
67impl stream::Group<TokenStream> for proc_macro2::Group {
68    fn delimiter(&self) -> stream::Delimiter {
69        match proc_macro2::Group::delimiter(self) {
70            proc_macro2::Delimiter::Brace => stream::Delimiter::Brace,
71            proc_macro2::Delimiter::Bracket => stream::Delimiter::Bracket,
72            proc_macro2::Delimiter::Parenthesis => stream::Delimiter::Parenthesis,
73            proc_macro2::Delimiter::None => stream::Delimiter::None,
74        }
75    }
76
77    fn stream(&self) -> TokenStream {
78        TokenStream(proc_macro2::Group::stream(self))
79    }
80}
81
82impl stream::Ident for proc_macro2::Ident {}
83
84impl stream::Spanned<proc_macro2::Span> for proc_macro2::Ident {
85    fn span(&self) -> proc_macro2::Span {
86        proc_macro2::Ident::span(self)
87    }
88}
89
90impl stream::Literal for proc_macro2::Literal {}
91
92impl stream::Spanned<proc_macro2::Span> for proc_macro2::Literal {
93    fn span(&self) -> proc_macro2::Span {
94        proc_macro2::Literal::span(self)
95    }
96}
97
98impl stream::Punct for proc_macro2::Punct {
99    fn as_char(&self) -> char {
100        proc_macro2::Punct::as_char(self)
101    }
102
103    fn spacing(&self) -> stream::Spacing {
104        match proc_macro2::Punct::spacing(self) {
105            proc_macro2::Spacing::Alone => stream::Spacing::Alone,
106            proc_macro2::Spacing::Joint => stream::Spacing::Joint,
107        }
108    }
109}
110
111impl stream::Spanned<proc_macro2::Span> for proc_macro2::Punct {
112    fn span(&self) -> proc_macro2::Span {
113        proc_macro2::Punct::span(self)
114    }
115}
casco/proc_macro2.rs

casco/
proc_macro2.rs
