moxy_token/bridge/
proc_macro.rs

1use crate::span::fallback;
2use crate::{Delim, Group, Ident, Keyword, Lit, Spacing, Span, ToTokens, TokenStream, TokenTree};
3
4// --- Span (fallback) ---
5
6impl From<proc_macro::Span> for fallback::Span {
7    fn from(#[allow(unused)] value: proc_macro::Span) -> Self {
8        #[cfg(nightly)]
9        {
10            let r = value.byte_range();
11            Self::new(r.start as u32, r.end as u32)
12        }
13
14        #[cfg(not(nightly))]
15        {
16            Self::default()
17        }
18    }
19}
20
21impl From<fallback::Span> for proc_macro::Span {
22    fn from(_value: fallback::Span) -> Self {
23        Self::call_site()
24    }
25}
26
27// --- Span ---
28
29impl From<proc_macro::Span> for Span {
30    #[inline]
31    fn from(value: proc_macro::Span) -> Self {
32        Self::Compiler(value)
33    }
34}
35
36impl From<Span> for proc_macro::Span {
37    fn from(value: Span) -> Self {
38        match value {
39            Span::Compiler(s) => s,
40            Span::Fallback(_) => Self::call_site(),
41        }
42    }
43}
44
45#[cfg(nightly)]
46impl proc_macro::MultiSpan for Span {
47    fn into_spans(self) -> Vec<proc_macro::Span> {
48        match self {
49            Self::Compiler(s) => vec![s],
50            Self::Fallback(_) => vec![proc_macro::Span::call_site()],
51        }
52    }
53}
54
55// --- Delim ---
56
57impl From<proc_macro::Delimiter> for Delim {
58    #[inline]
59    fn from(value: proc_macro::Delimiter) -> Self {
60        match value {
61            proc_macro::Delimiter::Parenthesis => Self::Paren,
62            proc_macro::Delimiter::Brace => Self::Brace,
63            proc_macro::Delimiter::Bracket => Self::Bracket,
64            proc_macro::Delimiter::None => Self::None,
65        }
66    }
67}
68
69impl From<Delim> for proc_macro::Delimiter {
70    #[inline]
71    fn from(value: Delim) -> Self {
72        match value {
73            Delim::Paren => Self::Parenthesis,
74            Delim::Brace => Self::Brace,
75            Delim::Bracket => Self::Bracket,
76            Delim::None => Self::None,
77        }
78    }
79}
80
81// --- Spacing ---
82
83impl From<proc_macro::Spacing> for Spacing {
84    #[inline]
85    fn from(value: proc_macro::Spacing) -> Self {
86        match value {
87            proc_macro::Spacing::Alone => Self::Alone,
88            proc_macro::Spacing::Joint => Self::Joint,
89        }
90    }
91}
92
93impl From<Spacing> for proc_macro::Spacing {
94    #[inline]
95    fn from(value: Spacing) -> Self {
96        match value {
97            Spacing::Alone => Self::Alone,
98            Spacing::Joint => Self::Joint,
99        }
100    }
101}
102
103// --- Ident ---
104
105impl From<proc_macro::Ident> for Ident {
106    #[inline]
107    fn from(value: proc_macro::Ident) -> Self {
108        Self::new(value.to_string()).with_span(value.span().into())
109    }
110}
111
112impl From<Ident> for proc_macro::Ident {
113    fn from(value: Ident) -> Self {
114        let span: proc_macro::Span = value.span().into();
115        let name = value.text();
116
117        match name.strip_prefix("r#") {
118            Some(raw) => Self::new_raw(raw, span),
119            None => Self::new(name, span),
120        }
121    }
122}
123
124// --- Literal ---
125
126impl From<proc_macro::Literal> for Lit {
127    fn from(value: proc_macro::Literal) -> Self {
128        Self::from_repr(&value.to_string(), value.span().into())
129    }
130}
131
132impl From<Lit> for proc_macro::Literal {
133    fn from(value: Lit) -> Self {
134        let repr = value.repr();
135        let mut lit = repr.parse().unwrap_or_else(|_| Self::string(repr));
136
137        lit.set_span(value.span().into());
138        lit
139    }
140}
141
142// --- Group ---
143
144impl From<proc_macro::Group> for Group {
145    #[inline]
146    fn from(value: proc_macro::Group) -> Self {
147        use crate::span::DelimSpan;
148        let mut group = Self::new(value.delimiter().into(), value.stream().into());
149        group.set_span(DelimSpan::new(value.span_open().into(), value.span_close().into()));
150        group
151    }
152}
153
154impl From<Group> for proc_macro::Group {
155    #[inline]
156    fn from(value: Group) -> Self {
157        let span = value.span().span().into();
158        let mut group = Self::new(value.delim.into(), value.tokens.into());
159        group.set_span(span);
160        group
161    }
162}
163
164// --- TokenStream ---
165
166impl ToTokens<TokenStream> for proc_macro::TokenTree {
167    fn to_tokens(&self, tokens: &mut TokenStream) {
168        match self {
169            Self::Ident(v) => {
170                let tt = match Keyword::from_str(&v.to_string(), v.span().into()) {
171                    Some(kw) => TokenTree::Keyword(kw),
172                    None => TokenTree::Ident(v.clone().into()),
173                };
174                tokens.extend_one(tt)
175            }
176            Self::Literal(v) => tokens.extend_one(TokenTree::Literal(v.clone().into())),
177            Self::Group(v) => tokens.extend_one(TokenTree::Group(v.clone().into())),
178            Self::Punct(p) => crate::scan_puncts_spanned(&[(p.as_char(), p.span().into(), p.spacing().into())], tokens),
179        }
180    }
181}
182
183impl ToTokens<TokenStream> for proc_macro::TokenStream {
184    fn to_tokens(&self, tokens: &mut TokenStream) {
185        let mut punct_run: Vec<(char, Span, Spacing)> = Vec::new();
186
187        for tt in self.clone() {
188            match tt {
189                proc_macro::TokenTree::Punct(p) => punct_run.push((p.as_char(), p.span().into(), p.spacing().into())),
190                other => {
191                    if !punct_run.is_empty() {
192                        crate::scan_puncts_spanned(&punct_run, tokens);
193                        punct_run.clear();
194                    }
195                    other.to_tokens(tokens);
196                }
197            }
198        }
199
200        if !punct_run.is_empty() {
201            crate::scan_puncts_spanned(&punct_run, tokens);
202        }
203    }
204}
205
206impl ToTokens<proc_macro::TokenStream> for TokenTree {
207    fn to_tokens(&self, out: &mut proc_macro::TokenStream) {
208        match self {
209            Self::Group(g) => out.extend(vec![proc_macro::TokenTree::Group(g.clone().into())]),
210            Self::Ident(v) => out.extend(vec![proc_macro::TokenTree::Ident(v.clone().into())]),
211            Self::Keyword(kw) => {
212                let id = proc_macro::Ident::new(kw.as_str(), kw.span().into());
213                out.extend(vec![proc_macro::TokenTree::Ident(id)])
214            }
215            Self::Literal(v) => out.extend(vec![proc_macro::TokenTree::Literal(v.clone().into())]),
216            Self::Punct(op) => {
217                let text = op.as_str();
218                let span: proc_macro::Span = op.span().into();
219                let last = text.chars().count() - 1;
220
221                for (i, ch) in text.chars().enumerate() {
222                    let spacing = if i == last {
223                        op.spacing().into()
224                    } else {
225                        proc_macro::Spacing::Joint
226                    };
227
228                    let mut p = proc_macro::Punct::new(ch, spacing);
229                    p.set_span(span);
230                    out.extend(vec![proc_macro::TokenTree::Punct(p)]);
231                }
232            }
233        }
234    }
235}
236
237impl ToTokens<proc_macro::TokenStream> for TokenStream {
238    fn to_tokens(&self, out: &mut proc_macro::TokenStream) {
239        for t in self.iter() {
240            t.to_tokens(out);
241        }
242    }
243}
244
245impl From<proc_macro::TokenStream> for TokenStream {
246    fn from(value: proc_macro::TokenStream) -> Self {
247        let mut out = Self::new();
248        value.to_tokens(&mut out);
249        out
250    }
251}
252
253impl From<TokenStream> for proc_macro::TokenStream {
254    fn from(value: TokenStream) -> Self {
255        let mut out = Self::new();
256        for t in value.iter() {
257            t.to_tokens(&mut out);
258        }
259        out
260    }
261}
moxy_token/bridge/proc_macro.rs

moxy_token/bridge/
proc_macro.rs
