blob: b195fda5cbc05fa76ad1af559de1918a08f2d422 [file] [log] [blame]
David Tolnay55535012018-01-05 16:39:23 -08001// Copyright 2018 Syn Developers
2//
3// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
4// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
5// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
6// option. This file may not be copied, modified, or distributed
7// except according to those terms.
8
David Tolnaye79ae182018-01-06 19:23:37 -08009//! Tokens representing Rust punctuation, keywords, and delimiters.
Alex Crichton954046c2017-05-30 21:49:42 -070010//!
David Tolnaye79ae182018-01-06 19:23:37 -080011//! The type names in this module can be difficult to keep straight, so we
12//! prefer to use the [`Token!`] macro instead. This is a type-macro that
13//! expands to the token type of the given token.
14//!
15//! [`Token!`]: ../macro.Token.html
16//!
17//! # Example
18//!
19//! The [`ItemStatic`] syntax tree node is defined like this.
20//!
21//! [`ItemStatic`]: ../struct.ItemStatic.html
22//!
23//! ```
24//! # #[macro_use]
25//! # extern crate syn;
26//! #
David Tolnaye303b7c2018-05-20 16:46:35 -070027//! # use syn::{Attribute, Visibility, Ident, Type, Expr};
David Tolnaye79ae182018-01-06 19:23:37 -080028//! #
29//! pub struct ItemStatic {
30//! pub attrs: Vec<Attribute>,
31//! pub vis: Visibility,
32//! pub static_token: Token![static],
33//! pub mutability: Option<Token![mut]>,
34//! pub ident: Ident,
35//! pub colon_token: Token![:],
36//! pub ty: Box<Type>,
37//! pub eq_token: Token![=],
38//! pub expr: Box<Expr>,
39//! pub semi_token: Token![;],
40//! }
41//! #
42//! # fn main() {}
43//! ```
44//!
45//! # Parsing
46//!
47//! These tokens can be parsed using the [`Synom`] trait and the parser
48//! combinator macros [`punct!`], [`keyword!`], [`parens!`], [`braces!`], and
49//! [`brackets!`].
50//!
51//! [`Synom`]: ../synom/trait.Synom.html
52//! [`punct!`]: ../macro.punct.html
53//! [`keyword!`]: ../macro.keyword.html
54//! [`parens!`]: ../macro.parens.html
55//! [`braces!`]: ../macro.braces.html
56//! [`brackets!`]: ../macro.brackets.html
57//!
58//! ```
59//! #[macro_use]
60//! extern crate syn;
61//!
62//! use syn::synom::Synom;
David Tolnaye303b7c2018-05-20 16:46:35 -070063//! use syn::{Attribute, Visibility, Ident, Type, Expr};
David Tolnaye79ae182018-01-06 19:23:37 -080064//! #
65//! # struct ItemStatic;
66//! # use syn::ItemStatic as SynItemStatic;
67//!
68//! // Parse the ItemStatic struct shown above.
69//! impl Synom for ItemStatic {
70//! named!(parse -> Self, do_parse!(
71//! # (ItemStatic)
72//! # ));
73//! # }
74//! #
75//! # mod example {
76//! # use super::*;
77//! # use super::SynItemStatic as ItemStatic;
78//! #
79//! # named!(parse -> ItemStatic, do_parse!(
80//! attrs: many0!(Attribute::parse_outer) >>
81//! vis: syn!(Visibility) >>
82//! static_token: keyword!(static) >>
83//! mutability: option!(keyword!(mut)) >>
84//! ident: syn!(Ident) >>
85//! colon_token: punct!(:) >>
86//! ty: syn!(Type) >>
87//! eq_token: punct!(=) >>
88//! expr: syn!(Expr) >>
89//! semi_token: punct!(;) >>
90//! (ItemStatic {
91//! attrs, vis, static_token, mutability, ident, colon_token,
92//! ty: Box::new(ty), eq_token, expr: Box::new(expr), semi_token,
93//! })
94//! ));
95//! }
96//! #
97//! # fn main() {}
98//! ```
Alex Crichton954046c2017-05-30 21:49:42 -070099
Alex Crichtona74a1c82018-05-16 10:20:44 -0700100use proc_macro2::{Span, Ident};
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700101
102macro_rules! tokens {
103 (
David Tolnay73c98de2017-12-31 15:56:56 -0500104 punct: {
105 $($punct:tt pub struct $punct_name:ident/$len:tt #[$punct_doc:meta])*
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700106 }
David Tolnay73c98de2017-12-31 15:56:56 -0500107 delimiter: {
108 $($delimiter:tt pub struct $delimiter_name:ident #[$delimiter_doc:meta])*
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700109 }
David Tolnay73c98de2017-12-31 15:56:56 -0500110 keyword: {
111 $($keyword:tt pub struct $keyword_name:ident #[$keyword_doc:meta])*
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700112 }
113 ) => (
David Tolnaybd0bc3e2018-05-20 17:15:35 -0700114 $(token_punct_def! { #[$punct_doc] pub struct $punct_name/$len })*
Sergio Benitezd14d5362018-04-28 15:38:25 -0700115 $(token_punct_parser! { $punct pub struct $punct_name })*
David Tolnay73c98de2017-12-31 15:56:56 -0500116 $(token_delimiter! { #[$delimiter_doc] $delimiter pub struct $delimiter_name })*
117 $(token_keyword! { #[$keyword_doc] $keyword pub struct $keyword_name })*
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700118 )
119}
120
Sergio Benitezd14d5362018-04-28 15:38:25 -0700121macro_rules! token_punct_def {
David Tolnaybd0bc3e2018-05-20 17:15:35 -0700122 (#[$doc:meta] pub struct $name:ident / $len:tt) => {
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700123 #[cfg_attr(feature = "clone-impls", derive(Copy, Clone))]
David Tolnay5a20f632017-12-26 22:11:28 -0500124 #[$doc]
David Tolnay1bb760c2018-01-07 11:18:15 -0800125 ///
126 /// Don't try to remember the name of this type -- use the [`Token!`]
127 /// macro instead.
128 ///
129 /// [`Token!`]: index.html
David Tolnay5a20f632017-12-26 22:11:28 -0500130 pub struct $name(pub [Span; $len]);
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700131
David Tolnay0bdb0552017-12-27 21:31:51 -0500132 impl $name {
133 pub fn new(span: Span) -> Self {
134 $name([span; $len])
135 }
136 }
137
David Tolnay66bb8d52018-01-08 08:22:31 -0800138 impl ::std::default::Default for $name {
139 fn default() -> Self {
Alex Crichton9a4dca22018-03-28 06:32:19 -0700140 $name([Span::call_site(); $len])
David Tolnay66bb8d52018-01-08 08:22:31 -0800141 }
142 }
143
Nika Layzelld73a3652017-10-24 08:57:05 -0400144 #[cfg(feature = "extra-traits")]
145 impl ::std::fmt::Debug for $name {
146 fn fmt(&self, f: &mut ::std::fmt::Formatter) -> ::std::fmt::Result {
David Tolnay98942562017-12-26 21:24:35 -0500147 f.write_str(stringify!($name))
Nika Layzelld73a3652017-10-24 08:57:05 -0400148 }
149 }
150
David Tolnay98942562017-12-26 21:24:35 -0500151 #[cfg(feature = "extra-traits")]
152 impl ::std::cmp::Eq for $name {}
153
154 #[cfg(feature = "extra-traits")]
155 impl ::std::cmp::PartialEq for $name {
156 fn eq(&self, _other: &$name) -> bool {
157 true
158 }
159 }
160
161 #[cfg(feature = "extra-traits")]
162 impl ::std::hash::Hash for $name {
163 fn hash<H>(&self, _state: &mut H)
David Tolnay94d2b792018-04-29 12:26:10 -0700164 where
165 H: ::std::hash::Hasher,
166 {
167 }
David Tolnay98942562017-12-26 21:24:35 -0500168 }
169
Sergio Benitezd14d5362018-04-28 15:38:25 -0700170 impl From<Span> for $name {
171 fn from(span: Span) -> Self {
172 $name([span; $len])
173 }
174 }
David Tolnay94d2b792018-04-29 12:26:10 -0700175 };
Sergio Benitezd14d5362018-04-28 15:38:25 -0700176}
177
178macro_rules! token_punct_parser {
179 ($s:tt pub struct $name:ident) => {
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700180 #[cfg(feature = "printing")]
181 impl ::quote::ToTokens for $name {
Alex Crichtona74a1c82018-05-16 10:20:44 -0700182 fn to_tokens(&self, tokens: &mut ::proc_macro2::TokenStream) {
David Tolnay73c98de2017-12-31 15:56:56 -0500183 printing::punct($s, &self.0, tokens);
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700184 }
185 }
186
187 #[cfg(feature = "parsing")]
188 impl ::Synom for $name {
David Tolnaydfc886b2018-01-06 08:03:09 -0800189 fn parse(tokens: $crate::buffer::Cursor) -> $crate::synom::PResult<$name> {
David Tolnay73c98de2017-12-31 15:56:56 -0500190 parsing::punct($s, tokens, $name)
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700191 }
Sergio Benitez5680d6a2017-12-29 11:20:29 -0800192
193 fn description() -> Option<&'static str> {
194 Some(concat!("`", $s, "`"))
195 }
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700196 }
David Tolnay94d2b792018-04-29 12:26:10 -0700197 };
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700198}
199
David Tolnay73c98de2017-12-31 15:56:56 -0500200macro_rules! token_keyword {
David Tolnay5a20f632017-12-26 22:11:28 -0500201 (#[$doc:meta] $s:tt pub struct $name:ident) => {
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700202 #[cfg_attr(feature = "clone-impls", derive(Copy, Clone))]
David Tolnay5a20f632017-12-26 22:11:28 -0500203 #[$doc]
David Tolnay1bb760c2018-01-07 11:18:15 -0800204 ///
205 /// Don't try to remember the name of this type -- use the [`Token!`]
206 /// macro instead.
207 ///
208 /// [`Token!`]: index.html
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700209 pub struct $name(pub Span);
210
David Tolnay66bb8d52018-01-08 08:22:31 -0800211 impl ::std::default::Default for $name {
212 fn default() -> Self {
Alex Crichton9a4dca22018-03-28 06:32:19 -0700213 $name(Span::call_site())
David Tolnay66bb8d52018-01-08 08:22:31 -0800214 }
215 }
216
David Tolnay98942562017-12-26 21:24:35 -0500217 #[cfg(feature = "extra-traits")]
218 impl ::std::fmt::Debug for $name {
219 fn fmt(&self, f: &mut ::std::fmt::Formatter) -> ::std::fmt::Result {
220 f.write_str(stringify!($name))
221 }
222 }
223
224 #[cfg(feature = "extra-traits")]
225 impl ::std::cmp::Eq for $name {}
226
227 #[cfg(feature = "extra-traits")]
228 impl ::std::cmp::PartialEq for $name {
229 fn eq(&self, _other: &$name) -> bool {
230 true
231 }
232 }
233
234 #[cfg(feature = "extra-traits")]
235 impl ::std::hash::Hash for $name {
236 fn hash<H>(&self, _state: &mut H)
David Tolnay94d2b792018-04-29 12:26:10 -0700237 where
238 H: ::std::hash::Hasher,
239 {
240 }
David Tolnay98942562017-12-26 21:24:35 -0500241 }
242
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700243 #[cfg(feature = "printing")]
244 impl ::quote::ToTokens for $name {
Alex Crichtona74a1c82018-05-16 10:20:44 -0700245 fn to_tokens(&self, tokens: &mut ::proc_macro2::TokenStream) {
David Tolnay73c98de2017-12-31 15:56:56 -0500246 printing::keyword($s, &self.0, tokens);
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700247 }
248 }
249
250 #[cfg(feature = "parsing")]
251 impl ::Synom for $name {
David Tolnaydfc886b2018-01-06 08:03:09 -0800252 fn parse(tokens: $crate::buffer::Cursor) -> $crate::synom::PResult<$name> {
David Tolnay73c98de2017-12-31 15:56:56 -0500253 parsing::keyword($s, tokens, $name)
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700254 }
David Tolnay79777332018-01-07 10:04:42 -0800255
256 fn description() -> Option<&'static str> {
257 Some(concat!("`", $s, "`"))
258 }
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700259 }
강동윤d5229da2018-01-12 13:11:33 +0900260
261 impl From<Span> for $name {
262 fn from(span: Span) -> Self {
263 $name(span)
264 }
265 }
David Tolnay94d2b792018-04-29 12:26:10 -0700266 };
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700267}
268
David Tolnay73c98de2017-12-31 15:56:56 -0500269macro_rules! token_delimiter {
David Tolnay5a20f632017-12-26 22:11:28 -0500270 (#[$doc:meta] $s:tt pub struct $name:ident) => {
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700271 #[cfg_attr(feature = "clone-impls", derive(Copy, Clone))]
David Tolnay5a20f632017-12-26 22:11:28 -0500272 #[$doc]
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700273 pub struct $name(pub Span);
274
David Tolnay66bb8d52018-01-08 08:22:31 -0800275 impl ::std::default::Default for $name {
276 fn default() -> Self {
Alex Crichton9a4dca22018-03-28 06:32:19 -0700277 $name(Span::call_site())
David Tolnay66bb8d52018-01-08 08:22:31 -0800278 }
279 }
280
David Tolnay98942562017-12-26 21:24:35 -0500281 #[cfg(feature = "extra-traits")]
282 impl ::std::fmt::Debug for $name {
283 fn fmt(&self, f: &mut ::std::fmt::Formatter) -> ::std::fmt::Result {
284 f.write_str(stringify!($name))
285 }
286 }
287
288 #[cfg(feature = "extra-traits")]
289 impl ::std::cmp::Eq for $name {}
290
291 #[cfg(feature = "extra-traits")]
292 impl ::std::cmp::PartialEq for $name {
293 fn eq(&self, _other: &$name) -> bool {
294 true
295 }
296 }
297
298 #[cfg(feature = "extra-traits")]
299 impl ::std::hash::Hash for $name {
300 fn hash<H>(&self, _state: &mut H)
David Tolnay94d2b792018-04-29 12:26:10 -0700301 where
302 H: ::std::hash::Hasher,
303 {
304 }
David Tolnay98942562017-12-26 21:24:35 -0500305 }
306
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700307 impl $name {
308 #[cfg(feature = "printing")]
Alex Crichtona74a1c82018-05-16 10:20:44 -0700309 pub fn surround<F>(&self, tokens: &mut ::proc_macro2::TokenStream, f: F)
David Tolnay94d2b792018-04-29 12:26:10 -0700310 where
Alex Crichtona74a1c82018-05-16 10:20:44 -0700311 F: FnOnce(&mut ::proc_macro2::TokenStream),
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700312 {
313 printing::delim($s, &self.0, tokens, f);
314 }
315
316 #[cfg(feature = "parsing")]
David Tolnay94d2b792018-04-29 12:26:10 -0700317 pub fn parse<F, R>(
318 tokens: $crate::buffer::Cursor,
319 f: F,
320 ) -> $crate::synom::PResult<($name, R)>
321 where
322 F: FnOnce($crate::buffer::Cursor) -> $crate::synom::PResult<R>,
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700323 {
324 parsing::delim($s, tokens, $name, f)
325 }
326 }
강동윤d5229da2018-01-12 13:11:33 +0900327
328 impl From<Span> for $name {
329 fn from(span: Span) -> Self {
330 $name(span)
331 }
332 }
David Tolnay94d2b792018-04-29 12:26:10 -0700333 };
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700334}
335
Sergio Benitezd14d5362018-04-28 15:38:25 -0700336token_punct_def! {
337 /// `_`
David Tolnaybd0bc3e2018-05-20 17:15:35 -0700338 pub struct Underscore/1
Sergio Benitezd14d5362018-04-28 15:38:25 -0700339}
340
341#[cfg(feature = "printing")]
342impl ::quote::ToTokens for Underscore {
Alex Crichtona74a1c82018-05-16 10:20:44 -0700343 fn to_tokens(&self, tokens: &mut ::proc_macro2::TokenStream) {
344 use quote::TokenStreamExt;
345 tokens.append(::proc_macro2::Ident::new("_", self.0[0]));
Sergio Benitezd14d5362018-04-28 15:38:25 -0700346 }
347}
348
349#[cfg(feature = "parsing")]
350impl ::Synom for Underscore {
351 fn parse(input: ::buffer::Cursor) -> ::synom::PResult<Underscore> {
352 match input.term() {
Alex Crichtona74a1c82018-05-16 10:20:44 -0700353 Some((term, rest)) => {
354 if term.to_string() == "_" {
355 Ok((Underscore([term.span()]), rest))
356 } else {
357 ::parse_error()
358 }
359 }
David Tolnay94d2b792018-04-29 12:26:10 -0700360 None => parsing::punct("_", input, Underscore),
Sergio Benitezd14d5362018-04-28 15:38:25 -0700361 }
362 }
363
364 fn description() -> Option<&'static str> {
365 Some("`_`")
366 }
367}
368
Alex Crichton131308c2018-05-18 14:00:24 -0700369token_punct_def! {
370 /// `'`
David Tolnaybd0bc3e2018-05-20 17:15:35 -0700371 pub struct Apostrophe/1
Alex Crichton131308c2018-05-18 14:00:24 -0700372}
373
374#[cfg(feature = "printing")]
375impl ::quote::ToTokens for Apostrophe {
376 fn to_tokens(&self, tokens: &mut ::proc_macro2::TokenStream) {
377 use quote::TokenStreamExt;
378 let mut token = ::proc_macro2::Punct::new('\'', ::proc_macro2::Spacing::Joint);
379 token.set_span(self.0[0]);
380 tokens.append(token);
381 }
382}
383
384#[cfg(feature = "parsing")]
385impl ::Synom for Apostrophe {
386 fn parse(input: ::buffer::Cursor) -> ::synom::PResult<Apostrophe> {
387 match input.op() {
388 Some((op, rest)) => {
389 if op.as_char() == '\'' && op.spacing() == ::proc_macro2::Spacing::Joint {
390 Ok((Apostrophe([op.span()]), rest))
391 } else {
392 ::parse_error()
393 }
394 }
395 None => ::parse_error()
396 }
397 }
398
399 fn description() -> Option<&'static str> {
400 Some("`'`")
401 }
402}
403
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700404tokens! {
David Tolnay73c98de2017-12-31 15:56:56 -0500405 punct: {
David Tolnay5a20f632017-12-26 22:11:28 -0500406 "+" pub struct Add/1 /// `+`
407 "+=" pub struct AddEq/2 /// `+=`
408 "&" pub struct And/1 /// `&`
409 "&&" pub struct AndAnd/2 /// `&&`
410 "&=" pub struct AndEq/2 /// `&=`
411 "@" pub struct At/1 /// `@`
412 "!" pub struct Bang/1 /// `!`
413 "^" pub struct Caret/1 /// `^`
David Tolnay32954ef2017-12-26 22:43:16 -0500414 "^=" pub struct CaretEq/2 /// `^=`
David Tolnay5a20f632017-12-26 22:11:28 -0500415 ":" pub struct Colon/1 /// `:`
416 "::" pub struct Colon2/2 /// `::`
417 "," pub struct Comma/1 /// `,`
418 "/" pub struct Div/1 /// `/`
419 "/=" pub struct DivEq/2 /// `/=`
MSleepyPandaf5137d72018-05-05 18:57:10 +0200420 "$" pub struct Dollar/1 /// `$`
David Tolnay5a20f632017-12-26 22:11:28 -0500421 "." pub struct Dot/1 /// `.`
422 ".." pub struct Dot2/2 /// `..`
423 "..." pub struct Dot3/3 /// `...`
424 "..=" pub struct DotDotEq/3 /// `..=`
425 "=" pub struct Eq/1 /// `=`
426 "==" pub struct EqEq/2 /// `==`
427 ">=" pub struct Ge/2 /// `>=`
428 ">" pub struct Gt/1 /// `>`
429 "<=" pub struct Le/2 /// `<=`
430 "<" pub struct Lt/1 /// `<`
431 "*=" pub struct MulEq/2 /// `*=`
432 "!=" pub struct Ne/2 /// `!=`
433 "|" pub struct Or/1 /// `|`
434 "|=" pub struct OrEq/2 /// `|=`
435 "||" pub struct OrOr/2 /// `||`
436 "#" pub struct Pound/1 /// `#`
437 "?" pub struct Question/1 /// `?`
438 "->" pub struct RArrow/2 /// `->`
439 "<-" pub struct LArrow/2 /// `<-`
440 "%" pub struct Rem/1 /// `%`
441 "%=" pub struct RemEq/2 /// `%=`
David Tolnay17624152018-03-31 18:11:40 +0200442 "=>" pub struct FatArrow/2 /// `=>`
David Tolnay5a20f632017-12-26 22:11:28 -0500443 ";" pub struct Semi/1 /// `;`
444 "<<" pub struct Shl/2 /// `<<`
445 "<<=" pub struct ShlEq/3 /// `<<=`
446 ">>" pub struct Shr/2 /// `>>`
447 ">>=" pub struct ShrEq/3 /// `>>=`
448 "*" pub struct Star/1 /// `*`
449 "-" pub struct Sub/1 /// `-`
450 "-=" pub struct SubEq/2 /// `-=`
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700451 }
David Tolnay73c98de2017-12-31 15:56:56 -0500452 delimiter: {
David Tolnay5a20f632017-12-26 22:11:28 -0500453 "{" pub struct Brace /// `{...}`
454 "[" pub struct Bracket /// `[...]`
455 "(" pub struct Paren /// `(...)`
456 " " pub struct Group /// None-delimited group
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700457 }
David Tolnay73c98de2017-12-31 15:56:56 -0500458 keyword: {
David Tolnay5a20f632017-12-26 22:11:28 -0500459 "as" pub struct As /// `as`
460 "auto" pub struct Auto /// `auto`
461 "box" pub struct Box /// `box`
462 "break" pub struct Break /// `break`
463 "Self" pub struct CapSelf /// `Self`
464 "catch" pub struct Catch /// `catch`
465 "const" pub struct Const /// `const`
466 "continue" pub struct Continue /// `continue`
467 "crate" pub struct Crate /// `crate`
468 "default" pub struct Default /// `default`
469 "do" pub struct Do /// `do`
470 "dyn" pub struct Dyn /// `dyn`
471 "else" pub struct Else /// `else`
472 "enum" pub struct Enum /// `enum`
473 "extern" pub struct Extern /// `extern`
474 "fn" pub struct Fn /// `fn`
475 "for" pub struct For /// `for`
476 "if" pub struct If /// `if`
477 "impl" pub struct Impl /// `impl`
478 "in" pub struct In /// `in`
479 "let" pub struct Let /// `let`
480 "loop" pub struct Loop /// `loop`
481 "macro" pub struct Macro /// `macro`
482 "match" pub struct Match /// `match`
483 "mod" pub struct Mod /// `mod`
484 "move" pub struct Move /// `move`
485 "mut" pub struct Mut /// `mut`
486 "pub" pub struct Pub /// `pub`
487 "ref" pub struct Ref /// `ref`
488 "return" pub struct Return /// `return`
489 "self" pub struct Self_ /// `self`
490 "static" pub struct Static /// `static`
491 "struct" pub struct Struct /// `struct`
492 "super" pub struct Super /// `super`
493 "trait" pub struct Trait /// `trait`
494 "type" pub struct Type /// `type`
495 "union" pub struct Union /// `union`
496 "unsafe" pub struct Unsafe /// `unsafe`
497 "use" pub struct Use /// `use`
498 "where" pub struct Where /// `where`
499 "while" pub struct While /// `while`
500 "yield" pub struct Yield /// `yield`
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700501 }
502}
503
David Tolnayf005f962018-01-06 21:19:41 -0800504/// A type-macro that expands to the name of the Rust type representation of a
505/// given token.
506///
507/// See the [token module] documentation for details and examples.
508///
509/// [token module]: token/index.html
David Tolnayf8db7ba2017-11-11 22:52:16 -0800510// Unfortunate duplication due to a rustdoc bug.
511// https://github.com/rust-lang/rust/issues/45939
512#[macro_export]
David Tolnaya5ed2fe2018-04-29 12:23:34 -0700513#[cfg_attr(rustfmt, rustfmt_skip)]
David Tolnayf8db7ba2017-11-11 22:52:16 -0800514macro_rules! Token {
David Tolnay32954ef2017-12-26 22:43:16 -0500515 (+) => { $crate::token::Add };
516 (+=) => { $crate::token::AddEq };
517 (&) => { $crate::token::And };
518 (&&) => { $crate::token::AndAnd };
519 (&=) => { $crate::token::AndEq };
520 (@) => { $crate::token::At };
521 (!) => { $crate::token::Bang };
522 (^) => { $crate::token::Caret };
523 (^=) => { $crate::token::CaretEq };
524 (:) => { $crate::token::Colon };
525 (::) => { $crate::token::Colon2 };
526 (,) => { $crate::token::Comma };
527 (/) => { $crate::token::Div };
528 (/=) => { $crate::token::DivEq };
529 (.) => { $crate::token::Dot };
530 (..) => { $crate::token::Dot2 };
531 (...) => { $crate::token::Dot3 };
532 (..=) => { $crate::token::DotDotEq };
533 (=) => { $crate::token::Eq };
534 (==) => { $crate::token::EqEq };
535 (>=) => { $crate::token::Ge };
536 (>) => { $crate::token::Gt };
537 (<=) => { $crate::token::Le };
538 (<) => { $crate::token::Lt };
539 (*=) => { $crate::token::MulEq };
540 (!=) => { $crate::token::Ne };
541 (|) => { $crate::token::Or };
542 (|=) => { $crate::token::OrEq };
543 (||) => { $crate::token::OrOr };
544 (#) => { $crate::token::Pound };
545 (?) => { $crate::token::Question };
546 (->) => { $crate::token::RArrow };
547 (<-) => { $crate::token::LArrow };
548 (%) => { $crate::token::Rem };
549 (%=) => { $crate::token::RemEq };
David Tolnay17624152018-03-31 18:11:40 +0200550 (=>) => { $crate::token::FatArrow };
David Tolnay32954ef2017-12-26 22:43:16 -0500551 (;) => { $crate::token::Semi };
552 (<<) => { $crate::token::Shl };
553 (<<=) => { $crate::token::ShlEq };
554 (>>) => { $crate::token::Shr };
555 (>>=) => { $crate::token::ShrEq };
556 (*) => { $crate::token::Star };
557 (-) => { $crate::token::Sub };
558 (-=) => { $crate::token::SubEq };
559 (_) => { $crate::token::Underscore };
560 (as) => { $crate::token::As };
561 (auto) => { $crate::token::Auto };
562 (box) => { $crate::token::Box };
563 (break) => { $crate::token::Break };
564 (Self) => { $crate::token::CapSelf };
565 (catch) => { $crate::token::Catch };
566 (const) => { $crate::token::Const };
567 (continue) => { $crate::token::Continue };
568 (crate) => { $crate::token::Crate };
569 (default) => { $crate::token::Default };
570 (do) => { $crate::token::Do };
571 (dyn) => { $crate::token::Dyn };
572 (else) => { $crate::token::Else };
573 (enum) => { $crate::token::Enum };
574 (extern) => { $crate::token::Extern };
575 (fn) => { $crate::token::Fn };
576 (for) => { $crate::token::For };
577 (if) => { $crate::token::If };
578 (impl) => { $crate::token::Impl };
579 (in) => { $crate::token::In };
580 (let) => { $crate::token::Let };
581 (loop) => { $crate::token::Loop };
582 (macro) => { $crate::token::Macro };
583 (match) => { $crate::token::Match };
584 (mod) => { $crate::token::Mod };
585 (move) => { $crate::token::Move };
586 (mut) => { $crate::token::Mut };
587 (pub) => { $crate::token::Pub };
588 (ref) => { $crate::token::Ref };
589 (return) => { $crate::token::Return };
590 (self) => { $crate::token::Self_ };
591 (static) => { $crate::token::Static };
592 (struct) => { $crate::token::Struct };
593 (super) => { $crate::token::Super };
594 (trait) => { $crate::token::Trait };
595 (type) => { $crate::token::Type };
596 (union) => { $crate::token::Union };
597 (unsafe) => { $crate::token::Unsafe };
598 (use) => { $crate::token::Use };
599 (where) => { $crate::token::Where };
600 (while) => { $crate::token::While };
601 (yield) => { $crate::token::Yield };
David Tolnayf8db7ba2017-11-11 22:52:16 -0800602}
603
David Tolnayf005f962018-01-06 21:19:41 -0800604/// Parse a single Rust punctuation token.
605///
606/// See the [token module] documentation for details and examples.
607///
608/// [token module]: token/index.html
David Tolnay461d98e2018-01-07 11:07:19 -0800609///
610/// *This macro is available if Syn is built with the `"parsing"` feature.*
David Tolnay0fbe3282017-12-26 21:46:16 -0500611#[cfg(feature = "parsing")]
David Tolnayf8db7ba2017-11-11 22:52:16 -0800612#[macro_export]
David Tolnaya5ed2fe2018-04-29 12:23:34 -0700613#[cfg_attr(rustfmt, rustfmt_skip)]
David Tolnayf8db7ba2017-11-11 22:52:16 -0800614macro_rules! punct {
David Tolnay32954ef2017-12-26 22:43:16 -0500615 ($i:expr, +) => { call!($i, <$crate::token::Add as $crate::synom::Synom>::parse) };
616 ($i:expr, +=) => { call!($i, <$crate::token::AddEq as $crate::synom::Synom>::parse) };
617 ($i:expr, &) => { call!($i, <$crate::token::And as $crate::synom::Synom>::parse) };
618 ($i:expr, &&) => { call!($i, <$crate::token::AndAnd as $crate::synom::Synom>::parse) };
619 ($i:expr, &=) => { call!($i, <$crate::token::AndEq as $crate::synom::Synom>::parse) };
620 ($i:expr, @) => { call!($i, <$crate::token::At as $crate::synom::Synom>::parse) };
621 ($i:expr, !) => { call!($i, <$crate::token::Bang as $crate::synom::Synom>::parse) };
622 ($i:expr, ^) => { call!($i, <$crate::token::Caret as $crate::synom::Synom>::parse) };
623 ($i:expr, ^=) => { call!($i, <$crate::token::CaretEq as $crate::synom::Synom>::parse) };
624 ($i:expr, :) => { call!($i, <$crate::token::Colon as $crate::synom::Synom>::parse) };
625 ($i:expr, ::) => { call!($i, <$crate::token::Colon2 as $crate::synom::Synom>::parse) };
626 ($i:expr, ,) => { call!($i, <$crate::token::Comma as $crate::synom::Synom>::parse) };
627 ($i:expr, /) => { call!($i, <$crate::token::Div as $crate::synom::Synom>::parse) };
628 ($i:expr, /=) => { call!($i, <$crate::token::DivEq as $crate::synom::Synom>::parse) };
629 ($i:expr, .) => { call!($i, <$crate::token::Dot as $crate::synom::Synom>::parse) };
630 ($i:expr, ..) => { call!($i, <$crate::token::Dot2 as $crate::synom::Synom>::parse) };
631 ($i:expr, ...) => { call!($i, <$crate::token::Dot3 as $crate::synom::Synom>::parse) };
632 ($i:expr, ..=) => { call!($i, <$crate::token::DotDotEq as $crate::synom::Synom>::parse) };
633 ($i:expr, =) => { call!($i, <$crate::token::Eq as $crate::synom::Synom>::parse) };
634 ($i:expr, ==) => { call!($i, <$crate::token::EqEq as $crate::synom::Synom>::parse) };
635 ($i:expr, >=) => { call!($i, <$crate::token::Ge as $crate::synom::Synom>::parse) };
636 ($i:expr, >) => { call!($i, <$crate::token::Gt as $crate::synom::Synom>::parse) };
637 ($i:expr, <=) => { call!($i, <$crate::token::Le as $crate::synom::Synom>::parse) };
638 ($i:expr, <) => { call!($i, <$crate::token::Lt as $crate::synom::Synom>::parse) };
639 ($i:expr, *=) => { call!($i, <$crate::token::MulEq as $crate::synom::Synom>::parse) };
640 ($i:expr, !=) => { call!($i, <$crate::token::Ne as $crate::synom::Synom>::parse) };
641 ($i:expr, |) => { call!($i, <$crate::token::Or as $crate::synom::Synom>::parse) };
642 ($i:expr, |=) => { call!($i, <$crate::token::OrEq as $crate::synom::Synom>::parse) };
643 ($i:expr, ||) => { call!($i, <$crate::token::OrOr as $crate::synom::Synom>::parse) };
644 ($i:expr, #) => { call!($i, <$crate::token::Pound as $crate::synom::Synom>::parse) };
645 ($i:expr, ?) => { call!($i, <$crate::token::Question as $crate::synom::Synom>::parse) };
646 ($i:expr, ->) => { call!($i, <$crate::token::RArrow as $crate::synom::Synom>::parse) };
647 ($i:expr, <-) => { call!($i, <$crate::token::LArrow as $crate::synom::Synom>::parse) };
648 ($i:expr, %) => { call!($i, <$crate::token::Rem as $crate::synom::Synom>::parse) };
649 ($i:expr, %=) => { call!($i, <$crate::token::RemEq as $crate::synom::Synom>::parse) };
David Tolnay17624152018-03-31 18:11:40 +0200650 ($i:expr, =>) => { call!($i, <$crate::token::FatArrow as $crate::synom::Synom>::parse) };
David Tolnay32954ef2017-12-26 22:43:16 -0500651 ($i:expr, ;) => { call!($i, <$crate::token::Semi as $crate::synom::Synom>::parse) };
652 ($i:expr, <<) => { call!($i, <$crate::token::Shl as $crate::synom::Synom>::parse) };
653 ($i:expr, <<=) => { call!($i, <$crate::token::ShlEq as $crate::synom::Synom>::parse) };
654 ($i:expr, >>) => { call!($i, <$crate::token::Shr as $crate::synom::Synom>::parse) };
655 ($i:expr, >>=) => { call!($i, <$crate::token::ShrEq as $crate::synom::Synom>::parse) };
656 ($i:expr, *) => { call!($i, <$crate::token::Star as $crate::synom::Synom>::parse) };
657 ($i:expr, -) => { call!($i, <$crate::token::Sub as $crate::synom::Synom>::parse) };
658 ($i:expr, -=) => { call!($i, <$crate::token::SubEq as $crate::synom::Synom>::parse) };
659 ($i:expr, _) => { call!($i, <$crate::token::Underscore as $crate::synom::Synom>::parse) };
David Tolnayf8db7ba2017-11-11 22:52:16 -0800660}
661
David Tolnayf005f962018-01-06 21:19:41 -0800662/// Parse a single Rust keyword token.
663///
664/// See the [token module] documentation for details and examples.
665///
666/// [token module]: token/index.html
David Tolnay461d98e2018-01-07 11:07:19 -0800667///
668/// *This macro is available if Syn is built with the `"parsing"` feature.*
David Tolnay0fbe3282017-12-26 21:46:16 -0500669#[cfg(feature = "parsing")]
David Tolnayf8db7ba2017-11-11 22:52:16 -0800670#[macro_export]
David Tolnaya5ed2fe2018-04-29 12:23:34 -0700671#[cfg_attr(rustfmt, rustfmt_skip)]
David Tolnayf8db7ba2017-11-11 22:52:16 -0800672macro_rules! keyword {
David Tolnay32954ef2017-12-26 22:43:16 -0500673 ($i:expr, as) => { call!($i, <$crate::token::As as $crate::synom::Synom>::parse) };
674 ($i:expr, auto) => { call!($i, <$crate::token::Auto as $crate::synom::Synom>::parse) };
675 ($i:expr, box) => { call!($i, <$crate::token::Box as $crate::synom::Synom>::parse) };
676 ($i:expr, break) => { call!($i, <$crate::token::Break as $crate::synom::Synom>::parse) };
677 ($i:expr, Self) => { call!($i, <$crate::token::CapSelf as $crate::synom::Synom>::parse) };
678 ($i:expr, catch) => { call!($i, <$crate::token::Catch as $crate::synom::Synom>::parse) };
679 ($i:expr, const) => { call!($i, <$crate::token::Const as $crate::synom::Synom>::parse) };
680 ($i:expr, continue) => { call!($i, <$crate::token::Continue as $crate::synom::Synom>::parse) };
681 ($i:expr, crate) => { call!($i, <$crate::token::Crate as $crate::synom::Synom>::parse) };
682 ($i:expr, default) => { call!($i, <$crate::token::Default as $crate::synom::Synom>::parse) };
683 ($i:expr, do) => { call!($i, <$crate::token::Do as $crate::synom::Synom>::parse) };
684 ($i:expr, dyn) => { call!($i, <$crate::token::Dyn as $crate::synom::Synom>::parse) };
685 ($i:expr, else) => { call!($i, <$crate::token::Else as $crate::synom::Synom>::parse) };
686 ($i:expr, enum) => { call!($i, <$crate::token::Enum as $crate::synom::Synom>::parse) };
687 ($i:expr, extern) => { call!($i, <$crate::token::Extern as $crate::synom::Synom>::parse) };
688 ($i:expr, fn) => { call!($i, <$crate::token::Fn as $crate::synom::Synom>::parse) };
689 ($i:expr, for) => { call!($i, <$crate::token::For as $crate::synom::Synom>::parse) };
690 ($i:expr, if) => { call!($i, <$crate::token::If as $crate::synom::Synom>::parse) };
691 ($i:expr, impl) => { call!($i, <$crate::token::Impl as $crate::synom::Synom>::parse) };
692 ($i:expr, in) => { call!($i, <$crate::token::In as $crate::synom::Synom>::parse) };
693 ($i:expr, let) => { call!($i, <$crate::token::Let as $crate::synom::Synom>::parse) };
694 ($i:expr, loop) => { call!($i, <$crate::token::Loop as $crate::synom::Synom>::parse) };
695 ($i:expr, macro) => { call!($i, <$crate::token::Macro as $crate::synom::Synom>::parse) };
696 ($i:expr, match) => { call!($i, <$crate::token::Match as $crate::synom::Synom>::parse) };
697 ($i:expr, mod) => { call!($i, <$crate::token::Mod as $crate::synom::Synom>::parse) };
698 ($i:expr, move) => { call!($i, <$crate::token::Move as $crate::synom::Synom>::parse) };
699 ($i:expr, mut) => { call!($i, <$crate::token::Mut as $crate::synom::Synom>::parse) };
700 ($i:expr, pub) => { call!($i, <$crate::token::Pub as $crate::synom::Synom>::parse) };
701 ($i:expr, ref) => { call!($i, <$crate::token::Ref as $crate::synom::Synom>::parse) };
702 ($i:expr, return) => { call!($i, <$crate::token::Return as $crate::synom::Synom>::parse) };
703 ($i:expr, self) => { call!($i, <$crate::token::Self_ as $crate::synom::Synom>::parse) };
704 ($i:expr, static) => { call!($i, <$crate::token::Static as $crate::synom::Synom>::parse) };
705 ($i:expr, struct) => { call!($i, <$crate::token::Struct as $crate::synom::Synom>::parse) };
706 ($i:expr, super) => { call!($i, <$crate::token::Super as $crate::synom::Synom>::parse) };
707 ($i:expr, trait) => { call!($i, <$crate::token::Trait as $crate::synom::Synom>::parse) };
708 ($i:expr, type) => { call!($i, <$crate::token::Type as $crate::synom::Synom>::parse) };
709 ($i:expr, union) => { call!($i, <$crate::token::Union as $crate::synom::Synom>::parse) };
710 ($i:expr, unsafe) => { call!($i, <$crate::token::Unsafe as $crate::synom::Synom>::parse) };
711 ($i:expr, use) => { call!($i, <$crate::token::Use as $crate::synom::Synom>::parse) };
712 ($i:expr, where) => { call!($i, <$crate::token::Where as $crate::synom::Synom>::parse) };
713 ($i:expr, while) => { call!($i, <$crate::token::While as $crate::synom::Synom>::parse) };
714 ($i:expr, yield) => { call!($i, <$crate::token::Yield as $crate::synom::Synom>::parse) };
David Tolnayf8db7ba2017-11-11 22:52:16 -0800715}
716
David Tolnaybd0bc3e2018-05-20 17:15:35 -0700717macro_rules! ident_from_token {
718 ($token:ident) => {
719 impl From<Token![$token]> for Ident {
720 fn from(token: Token![$token]) -> Ident {
721 Ident::new(stringify!($token), token.0)
722 }
723 }
724 };
725}
726
727ident_from_token!(self);
728ident_from_token!(Self);
729ident_from_token!(super);
730ident_from_token!(crate);
731
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700732#[cfg(feature = "parsing")]
733mod parsing {
David Tolnay51382052017-12-27 13:46:21 -0500734 use proc_macro2::{Delimiter, Spacing, Span};
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700735
David Tolnaydfc886b2018-01-06 08:03:09 -0800736 use buffer::Cursor;
David Tolnay203557a2017-12-27 23:59:33 -0500737 use parse_error;
738 use synom::PResult;
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700739
740 pub trait FromSpans: Sized {
741 fn from_spans(spans: &[Span]) -> Self;
742 }
743
744 impl FromSpans for [Span; 1] {
745 fn from_spans(spans: &[Span]) -> Self {
746 [spans[0]]
747 }
748 }
749
750 impl FromSpans for [Span; 2] {
751 fn from_spans(spans: &[Span]) -> Self {
752 [spans[0], spans[1]]
753 }
754 }
755
756 impl FromSpans for [Span; 3] {
757 fn from_spans(spans: &[Span]) -> Self {
758 [spans[0], spans[1], spans[2]]
759 }
760 }
761
David Tolnay2b069542018-05-09 12:59:36 -0700762 pub fn punct<'a, T, R>(s: &str, mut tokens: Cursor<'a>, new: fn(T) -> R) -> PResult<'a, R>
David Tolnay51382052017-12-27 13:46:21 -0500763 where
764 T: FromSpans,
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700765 {
Alex Crichton9a4dca22018-03-28 06:32:19 -0700766 let mut spans = [Span::call_site(); 3];
Alex Crichton954046c2017-05-30 21:49:42 -0700767 assert!(s.len() <= spans.len());
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700768 let chars = s.chars();
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700769
Alex Crichton954046c2017-05-30 21:49:42 -0700770 for (i, (ch, slot)) in chars.zip(&mut spans).enumerate() {
Michael Layzell0a1a6632017-06-02 18:07:43 -0400771 match tokens.op() {
Alex Crichtona74a1c82018-05-16 10:20:44 -0700772 Some((op, rest)) => {
773 if op.as_char() == ch {
774 if i != s.len() - 1 {
775 match op.spacing() {
776 Spacing::Joint => {}
777 _ => return parse_error(),
778 }
Michael Layzell0a1a6632017-06-02 18:07:43 -0400779 }
Alex Crichtona74a1c82018-05-16 10:20:44 -0700780 *slot = op.span();
781 tokens = rest;
782 } else {
783 return parse_error()
Michael Layzell0a1a6632017-06-02 18:07:43 -0400784 }
Alex Crichton954046c2017-05-30 21:49:42 -0700785 }
David Tolnay51382052017-12-27 13:46:21 -0500786 _ => return parse_error(),
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700787 }
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700788 }
David Tolnay2b069542018-05-09 12:59:36 -0700789 Ok((new(T::from_spans(&spans)), tokens))
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700790 }
791
David Tolnayb57c8492018-05-05 00:32:04 -0700792 pub fn keyword<'a, T>(
793 keyword: &str,
794 tokens: Cursor<'a>,
David Tolnay2b069542018-05-09 12:59:36 -0700795 new: fn(Span) -> T,
David Tolnayb57c8492018-05-05 00:32:04 -0700796 ) -> PResult<'a, T> {
Alex Crichton9a4dca22018-03-28 06:32:19 -0700797 if let Some((term, rest)) = tokens.term() {
Alex Crichtona74a1c82018-05-16 10:20:44 -0700798 if term.to_string() == keyword {
David Tolnay2b069542018-05-09 12:59:36 -0700799 return Ok((new(term.span()), rest));
Michael Layzell0a1a6632017-06-02 18:07:43 -0400800 }
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700801 }
Michael Layzell0a1a6632017-06-02 18:07:43 -0400802 parse_error()
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700803 }
804
David Tolnay51382052017-12-27 13:46:21 -0500805 pub fn delim<'a, F, R, T>(
806 delim: &str,
807 tokens: Cursor<'a>,
David Tolnay2b069542018-05-09 12:59:36 -0700808 new: fn(Span) -> T,
David Tolnay51382052017-12-27 13:46:21 -0500809 f: F,
David Tolnay8875fca2017-12-31 13:52:37 -0500810 ) -> PResult<'a, (T, R)>
David Tolnay51382052017-12-27 13:46:21 -0500811 where
812 F: FnOnce(Cursor) -> PResult<R>,
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700813 {
Michael Layzell0a1a6632017-06-02 18:07:43 -0400814 // NOTE: We should support none-delimited sequences here.
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700815 let delim = match delim {
816 "(" => Delimiter::Parenthesis,
817 "{" => Delimiter::Brace,
818 "[" => Delimiter::Bracket,
Michael Layzell93c36282017-06-04 20:43:14 -0400819 " " => Delimiter::None,
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700820 _ => panic!("unknown delimiter: {}", delim),
821 };
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700822
David Tolnay65729482017-12-31 16:14:50 -0500823 if let Some((inside, span, rest)) = tokens.group(delim) {
824 match f(inside) {
David Tolnayf4aa6b42017-12-31 16:40:33 -0500825 Ok((ret, remaining)) => {
Michael Layzell0a1a6632017-06-02 18:07:43 -0400826 if remaining.eof() {
David Tolnay2b069542018-05-09 12:59:36 -0700827 return Ok(((new(span), ret), rest));
Michael Layzell0a1a6632017-06-02 18:07:43 -0400828 }
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700829 }
Michael Layzell0a1a6632017-06-02 18:07:43 -0400830 Err(err) => return Err(err),
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700831 }
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700832 }
Michael Layzell0a1a6632017-06-02 18:07:43 -0400833 parse_error()
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700834 }
835}
836
837#[cfg(feature = "printing")]
838mod printing {
Alex Crichtona74a1c82018-05-16 10:20:44 -0700839 use proc_macro2::{Delimiter, Group, Punct, Spacing, Span, Ident, TokenStream};
840 use quote::TokenStreamExt;
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700841
Alex Crichtona74a1c82018-05-16 10:20:44 -0700842 pub fn punct(s: &str, spans: &[Span], tokens: &mut TokenStream) {
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700843 assert_eq!(s.len(), spans.len());
844
845 let mut chars = s.chars();
846 let mut spans = spans.iter();
847 let ch = chars.next_back().unwrap();
848 let span = spans.next_back().unwrap();
849 for (ch, span) in chars.zip(spans) {
Alex Crichtona74a1c82018-05-16 10:20:44 -0700850 let mut op = Punct::new(ch, Spacing::Joint);
Alex Crichton9a4dca22018-03-28 06:32:19 -0700851 op.set_span(*span);
852 tokens.append(op);
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700853 }
854
Alex Crichtona74a1c82018-05-16 10:20:44 -0700855 let mut op = Punct::new(ch, Spacing::Alone);
Alex Crichton9a4dca22018-03-28 06:32:19 -0700856 op.set_span(*span);
857 tokens.append(op);
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700858 }
859
Alex Crichtona74a1c82018-05-16 10:20:44 -0700860 pub fn keyword(s: &str, span: &Span, tokens: &mut TokenStream) {
861 tokens.append(Ident::new(s, *span));
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700862 }
863
Alex Crichtona74a1c82018-05-16 10:20:44 -0700864 pub fn delim<F>(s: &str, span: &Span, tokens: &mut TokenStream, f: F)
David Tolnay51382052017-12-27 13:46:21 -0500865 where
Alex Crichtona74a1c82018-05-16 10:20:44 -0700866 F: FnOnce(&mut TokenStream),
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700867 {
David Tolnay00ab6982017-12-31 18:15:06 -0500868 let delim = match s {
869 "(" => Delimiter::Parenthesis,
870 "[" => Delimiter::Bracket,
871 "{" => Delimiter::Brace,
872 " " => Delimiter::None,
873 _ => panic!("unknown delimiter: {}", s),
874 };
Alex Crichtona74a1c82018-05-16 10:20:44 -0700875 let mut inner = TokenStream::empty();
David Tolnay00ab6982017-12-31 18:15:06 -0500876 f(&mut inner);
Alex Crichton9a4dca22018-03-28 06:32:19 -0700877 let mut g = Group::new(delim, inner.into());
878 g.set_span(*span);
879 tokens.append(g);
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700880 }
881}