blob: 6b14b762d81dcb1ed8745b107c73f91595f60856 [file] [log] [blame]
Alex Crichton954046c2017-05-30 21:49:42 -07001//! Discrete tokens that can be parsed out by synom.
2//!
3//! This module contains a number of useful tokens like `+=` and `/` along with
4//! keywords like `crate` and such. These structures are used to track the spans
5//! of these tokens and all implment the `ToTokens` and `Synom` traits when the
6//! corresponding feature is activated.
7
Alex Crichton7b9e02f2017-05-30 15:54:33 -07008use span::Span;
9
10macro_rules! tokens {
11 (
12 ops: {
13 $(($($op:tt)*),)*
14 }
15 delim: {
16 $(($($delim:tt)*),)*
17 }
18 syms: {
19 $(($($sym:tt)*),)*
20 }
21 ) => (
22 $(op! { $($op)* })*
23 $(delim! { $($delim)* })*
24 $(sym! { $($sym)* })*
25 )
26}
27
28macro_rules! op {
29 (pub struct $name:ident($($contents:tt)*) => $s:expr) => {
30 #[cfg_attr(feature = "clone-impls", derive(Copy, Clone))]
Nika Layzelld73a3652017-10-24 08:57:05 -040031 #[cfg_attr(feature = "extra-traits", derive(Eq, PartialEq, Hash))]
Alex Crichton7b9e02f2017-05-30 15:54:33 -070032 #[derive(Default)]
33 pub struct $name(pub $($contents)*);
34
Nika Layzelld73a3652017-10-24 08:57:05 -040035 #[cfg(feature = "extra-traits")]
36 impl ::std::fmt::Debug for $name {
37 fn fmt(&self, f: &mut ::std::fmt::Formatter) -> ::std::fmt::Result {
38 let mut t = f.debug_tuple(stringify!($name));
39 for span in &self.0 {
40 t.field(span);
41 }
42 t.finish()
43 }
44 }
45
Alex Crichton7b9e02f2017-05-30 15:54:33 -070046 #[cfg(feature = "printing")]
47 impl ::quote::ToTokens for $name {
48 fn to_tokens(&self, tokens: &mut ::quote::Tokens) {
49 printing::op($s, &self.0, tokens);
50 }
51 }
52
53 #[cfg(feature = "parsing")]
54 impl ::Synom for $name {
Michael Layzell760fd662017-05-31 22:46:05 -040055 fn parse(tokens: $crate::Cursor) -> $crate::PResult<$name> {
Alex Crichton7b9e02f2017-05-30 15:54:33 -070056 parsing::op($s, tokens, $name)
57 }
58 }
59 }
60}
61
62macro_rules! sym {
63 (pub struct $name:ident => $s:expr) => {
64 #[cfg_attr(feature = "clone-impls", derive(Copy, Clone))]
65 #[cfg_attr(feature = "extra-traits", derive(Debug, Eq, PartialEq, Hash))]
66 #[derive(Default)]
67 pub struct $name(pub Span);
68
69 #[cfg(feature = "printing")]
70 impl ::quote::ToTokens for $name {
71 fn to_tokens(&self, tokens: &mut ::quote::Tokens) {
72 printing::sym($s, &self.0, tokens);
73 }
74 }
75
76 #[cfg(feature = "parsing")]
77 impl ::Synom for $name {
Michael Layzell760fd662017-05-31 22:46:05 -040078 fn parse(tokens: $crate::Cursor) -> $crate::PResult<$name> {
Alex Crichton7b9e02f2017-05-30 15:54:33 -070079 parsing::sym($s, tokens, $name)
80 }
81 }
82 }
83}
84
85macro_rules! delim {
86 (pub struct $name:ident => $s:expr) => {
87 #[cfg_attr(feature = "clone-impls", derive(Copy, Clone))]
88 #[cfg_attr(feature = "extra-traits", derive(Debug, Eq, PartialEq, Hash))]
89 #[derive(Default)]
90 pub struct $name(pub Span);
91
92 impl $name {
93 #[cfg(feature = "printing")]
94 pub fn surround<F>(&self,
95 tokens: &mut ::quote::Tokens,
96 f: F)
97 where F: FnOnce(&mut ::quote::Tokens)
98 {
99 printing::delim($s, &self.0, tokens, f);
100 }
101
102 #[cfg(feature = "parsing")]
Michael Layzell760fd662017-05-31 22:46:05 -0400103 pub fn parse<F, R>(tokens: $crate::Cursor, f: F) -> $crate::PResult<(R, $name)>
104 where F: FnOnce($crate::Cursor) -> $crate::PResult<R>
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700105 {
106 parsing::delim($s, tokens, $name, f)
107 }
108 }
109 }
110}
111
112tokens! {
113 ops: {
114 (pub struct Add([Span; 1]) => "+"),
115 (pub struct AddEq([Span; 2]) => "+="),
116 (pub struct And([Span; 1]) => "&"),
117 (pub struct AndAnd([Span; 2]) => "&&"),
118 (pub struct AndEq([Span; 2]) => "&="),
119 (pub struct At([Span; 1]) => "@"),
120 (pub struct Bang([Span; 1]) => "!"),
121 (pub struct Caret([Span; 1]) => "^"),
122 (pub struct CaretEq([Span; 2]) => "^="),
123 (pub struct Colon([Span; 1]) => ":"),
124 (pub struct Colon2([Span; 2]) => "::"),
125 (pub struct Comma([Span; 1]) => ","),
126 (pub struct Div([Span; 1]) => "/"),
127 (pub struct DivEq([Span; 2]) => "/="),
128 (pub struct Dot([Span; 1]) => "."),
129 (pub struct Dot2([Span; 2]) => ".."),
130 (pub struct Dot3([Span; 3]) => "..."),
131 (pub struct Eq([Span; 1]) => "="),
132 (pub struct EqEq([Span; 2]) => "=="),
133 (pub struct Ge([Span; 2]) => ">="),
134 (pub struct Gt([Span; 1]) => ">"),
135 (pub struct Le([Span; 2]) => "<="),
136 (pub struct Lt([Span; 1]) => "<"),
137 (pub struct MulEq([Span; 2]) => "*="),
138 (pub struct Ne([Span; 2]) => "!="),
139 (pub struct Or([Span; 1]) => "|"),
140 (pub struct OrEq([Span; 2]) => "|="),
141 (pub struct OrOr([Span; 2]) => "||"),
142 (pub struct Pound([Span; 1]) => "#"),
143 (pub struct Question([Span; 1]) => "?"),
144 (pub struct RArrow([Span; 2]) => "->"),
Michael Layzellb78f3b52017-06-04 19:03:03 -0400145 (pub struct LArrow([Span; 2]) => "<-"),
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700146 (pub struct Rem([Span; 1]) => "%"),
147 (pub struct RemEq([Span; 2]) => "%="),
148 (pub struct Rocket([Span; 2]) => "=>"),
149 (pub struct Semi([Span; 1]) => ";"),
150 (pub struct Shl([Span; 2]) => "<<"),
151 (pub struct ShlEq([Span; 3]) => "<<="),
152 (pub struct Shr([Span; 2]) => ">>"),
153 (pub struct ShrEq([Span; 3]) => ">>="),
154 (pub struct Star([Span; 1]) => "*"),
155 (pub struct Sub([Span; 1]) => "-"),
156 (pub struct SubEq([Span; 2]) => "-="),
157 (pub struct Underscore([Span; 1]) => "_"),
158 }
159 delim: {
160 (pub struct Brace => "{"),
161 (pub struct Bracket => "["),
162 (pub struct Paren => "("),
Michael Layzell93c36282017-06-04 20:43:14 -0400163 (pub struct Group => " "),
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700164 }
165 syms: {
166 (pub struct As => "as"),
Nika Layzell0dc6e632017-11-18 12:55:25 -0500167 (pub struct Auto => "auto"),
David Tolnayf8db7ba2017-11-11 22:52:16 -0800168 (pub struct Box => "box"),
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700169 (pub struct Break => "break"),
Alex Crichton954046c2017-05-30 21:49:42 -0700170 (pub struct CapSelf => "Self"),
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700171 (pub struct Catch => "catch"),
172 (pub struct Const => "const"),
173 (pub struct Continue => "continue"),
174 (pub struct Crate => "crate"),
David Tolnayf8db7ba2017-11-11 22:52:16 -0800175 (pub struct Default => "default"),
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700176 (pub struct Do => "do"),
177 (pub struct Else => "else"),
178 (pub struct Enum => "enum"),
179 (pub struct Extern => "extern"),
David Tolnayf8db7ba2017-11-11 22:52:16 -0800180 (pub struct Fn => "fn"),
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700181 (pub struct For => "for"),
182 (pub struct If => "if"),
183 (pub struct Impl => "impl"),
184 (pub struct In => "in"),
185 (pub struct Let => "let"),
186 (pub struct Loop => "loop"),
187 (pub struct Match => "match"),
188 (pub struct Mod => "mod"),
189 (pub struct Move => "move"),
190 (pub struct Mut => "mut"),
191 (pub struct Pub => "pub"),
192 (pub struct Ref => "ref"),
193 (pub struct Return => "return"),
194 (pub struct Self_ => "self"),
195 (pub struct Static => "static"),
196 (pub struct Struct => "struct"),
Alex Crichton954046c2017-05-30 21:49:42 -0700197 (pub struct Super => "super"),
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700198 (pub struct Trait => "trait"),
199 (pub struct Type => "type"),
200 (pub struct Union => "union"),
201 (pub struct Unsafe => "unsafe"),
202 (pub struct Use => "use"),
203 (pub struct Where => "where"),
204 (pub struct While => "while"),
Alex Crichtonfe110462017-06-01 12:49:27 -0700205 (pub struct Yield => "yield"),
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700206 }
207}
208
David Tolnayf8db7ba2017-11-11 22:52:16 -0800209// Unfortunate duplication due to a rustdoc bug.
210// https://github.com/rust-lang/rust/issues/45939
211#[macro_export]
212macro_rules! Token {
213 (+) => { $crate::tokens::Add };
214 (+=) => { $crate::tokens::AddEq };
215 (&) => { $crate::tokens::And };
216 (&&) => { $crate::tokens::AndAnd };
217 (&=) => { $crate::tokens::AndEq };
218 (@) => { $crate::tokens::At };
219 (!) => { $crate::tokens::Bang };
220 (^) => { $crate::tokens::Caret };
221 (^=) => { $crate::tokens::CaretEq };
222 (:) => { $crate::tokens::Colon };
223 (::) => { $crate::tokens::Colon2 };
224 (,) => { $crate::tokens::Comma };
225 (/) => { $crate::tokens::Div };
226 (/=) => { $crate::tokens::DivEq };
227 (.) => { $crate::tokens::Dot };
228 (..) => { $crate::tokens::Dot2 };
229 (...) => { $crate::tokens::Dot3 };
230 (=) => { $crate::tokens::Eq };
231 (==) => { $crate::tokens::EqEq };
232 (>=) => { $crate::tokens::Ge };
233 (>) => { $crate::tokens::Gt };
234 (<=) => { $crate::tokens::Le };
235 (<) => { $crate::tokens::Lt };
236 (*=) => { $crate::tokens::MulEq };
237 (!=) => { $crate::tokens::Ne };
238 (|) => { $crate::tokens::Or };
239 (|=) => { $crate::tokens::OrEq };
240 (||) => { $crate::tokens::OrOr };
241 (#) => { $crate::tokens::Pound };
242 (?) => { $crate::tokens::Question };
243 (->) => { $crate::tokens::RArrow };
244 (<-) => { $crate::tokens::LArrow };
245 (%) => { $crate::tokens::Rem };
246 (%=) => { $crate::tokens::RemEq };
247 (=>) => { $crate::tokens::Rocket };
248 (;) => { $crate::tokens::Semi };
249 (<<) => { $crate::tokens::Shl };
250 (<<=) => { $crate::tokens::ShlEq };
251 (>>) => { $crate::tokens::Shr };
252 (>>=) => { $crate::tokens::ShrEq };
253 (*) => { $crate::tokens::Star };
254 (-) => { $crate::tokens::Sub };
255 (-=) => { $crate::tokens::SubEq };
256 (_) => { $crate::tokens::Underscore };
257 (as) => { $crate::tokens::As };
Nika Layzell0dc6e632017-11-18 12:55:25 -0500258 (auto) => { $crate::tokens::Auto };
David Tolnayf8db7ba2017-11-11 22:52:16 -0800259 (box) => { $crate::tokens::Box };
260 (break) => { $crate::tokens::Break };
261 (Self) => { $crate::tokens::CapSelf };
262 (catch) => { $crate::tokens::Catch };
263 (const) => { $crate::tokens::Const };
264 (continue) => { $crate::tokens::Continue };
265 (crate) => { $crate::tokens::Crate };
266 (default) => { $crate::tokens::Default };
267 (do) => { $crate::tokens::Do };
268 (else) => { $crate::tokens::Else };
269 (enum) => { $crate::tokens::Enum };
270 (extern) => { $crate::tokens::Extern };
271 (fn) => { $crate::tokens::Fn };
272 (for) => { $crate::tokens::For };
273 (if) => { $crate::tokens::If };
274 (impl) => { $crate::tokens::Impl };
275 (in) => { $crate::tokens::In };
276 (let) => { $crate::tokens::Let };
277 (loop) => { $crate::tokens::Loop };
278 (match) => { $crate::tokens::Match };
279 (mod) => { $crate::tokens::Mod };
280 (move) => { $crate::tokens::Move };
281 (mut) => { $crate::tokens::Mut };
282 (pub) => { $crate::tokens::Pub };
283 (ref) => { $crate::tokens::Ref };
284 (return) => { $crate::tokens::Return };
285 (self) => { $crate::tokens::Self_ };
286 (static) => { $crate::tokens::Static };
287 (struct) => { $crate::tokens::Struct };
288 (super) => { $crate::tokens::Super };
289 (trait) => { $crate::tokens::Trait };
290 (type) => { $crate::tokens::Type };
291 (union) => { $crate::tokens::Union };
292 (unsafe) => { $crate::tokens::Unsafe };
293 (use) => { $crate::tokens::Use };
294 (where) => { $crate::tokens::Where };
295 (while) => { $crate::tokens::While };
296 (yield) => { $crate::tokens::Yield };
297}
298
299#[macro_export]
300macro_rules! punct {
301 ($i:expr, +) => { call!($i, <$crate::tokens::Add as $crate::Synom>::parse) };
302 ($i:expr, +=) => { call!($i, <$crate::tokens::AddEq as $crate::Synom>::parse) };
303 ($i:expr, &) => { call!($i, <$crate::tokens::And as $crate::Synom>::parse) };
304 ($i:expr, &&) => { call!($i, <$crate::tokens::AndAnd as $crate::Synom>::parse) };
305 ($i:expr, &=) => { call!($i, <$crate::tokens::AndEq as $crate::Synom>::parse) };
306 ($i:expr, @) => { call!($i, <$crate::tokens::At as $crate::Synom>::parse) };
307 ($i:expr, !) => { call!($i, <$crate::tokens::Bang as $crate::Synom>::parse) };
308 ($i:expr, ^) => { call!($i, <$crate::tokens::Caret as $crate::Synom>::parse) };
309 ($i:expr, ^=) => { call!($i, <$crate::tokens::CaretEq as $crate::Synom>::parse) };
310 ($i:expr, :) => { call!($i, <$crate::tokens::Colon as $crate::Synom>::parse) };
311 ($i:expr, ::) => { call!($i, <$crate::tokens::Colon2 as $crate::Synom>::parse) };
312 ($i:expr, ,) => { call!($i, <$crate::tokens::Comma as $crate::Synom>::parse) };
313 ($i:expr, /) => { call!($i, <$crate::tokens::Div as $crate::Synom>::parse) };
314 ($i:expr, /=) => { call!($i, <$crate::tokens::DivEq as $crate::Synom>::parse) };
315 ($i:expr, .) => { call!($i, <$crate::tokens::Dot as $crate::Synom>::parse) };
316 ($i:expr, ..) => { call!($i, <$crate::tokens::Dot2 as $crate::Synom>::parse) };
317 ($i:expr, ...) => { call!($i, <$crate::tokens::Dot3 as $crate::Synom>::parse) };
318 ($i:expr, =) => { call!($i, <$crate::tokens::Eq as $crate::Synom>::parse) };
319 ($i:expr, ==) => { call!($i, <$crate::tokens::EqEq as $crate::Synom>::parse) };
320 ($i:expr, >=) => { call!($i, <$crate::tokens::Ge as $crate::Synom>::parse) };
321 ($i:expr, >) => { call!($i, <$crate::tokens::Gt as $crate::Synom>::parse) };
322 ($i:expr, <=) => { call!($i, <$crate::tokens::Le as $crate::Synom>::parse) };
323 ($i:expr, <) => { call!($i, <$crate::tokens::Lt as $crate::Synom>::parse) };
324 ($i:expr, *=) => { call!($i, <$crate::tokens::MulEq as $crate::Synom>::parse) };
325 ($i:expr, !=) => { call!($i, <$crate::tokens::Ne as $crate::Synom>::parse) };
326 ($i:expr, |) => { call!($i, <$crate::tokens::Or as $crate::Synom>::parse) };
327 ($i:expr, |=) => { call!($i, <$crate::tokens::OrEq as $crate::Synom>::parse) };
328 ($i:expr, ||) => { call!($i, <$crate::tokens::OrOr as $crate::Synom>::parse) };
329 ($i:expr, #) => { call!($i, <$crate::tokens::Pound as $crate::Synom>::parse) };
330 ($i:expr, ?) => { call!($i, <$crate::tokens::Question as $crate::Synom>::parse) };
331 ($i:expr, ->) => { call!($i, <$crate::tokens::RArrow as $crate::Synom>::parse) };
332 ($i:expr, <-) => { call!($i, <$crate::tokens::LArrow as $crate::Synom>::parse) };
333 ($i:expr, %) => { call!($i, <$crate::tokens::Rem as $crate::Synom>::parse) };
334 ($i:expr, %=) => { call!($i, <$crate::tokens::RemEq as $crate::Synom>::parse) };
335 ($i:expr, =>) => { call!($i, <$crate::tokens::Rocket as $crate::Synom>::parse) };
336 ($i:expr, ;) => { call!($i, <$crate::tokens::Semi as $crate::Synom>::parse) };
337 ($i:expr, <<) => { call!($i, <$crate::tokens::Shl as $crate::Synom>::parse) };
338 ($i:expr, <<=) => { call!($i, <$crate::tokens::ShlEq as $crate::Synom>::parse) };
339 ($i:expr, >>) => { call!($i, <$crate::tokens::Shr as $crate::Synom>::parse) };
340 ($i:expr, >>=) => { call!($i, <$crate::tokens::ShrEq as $crate::Synom>::parse) };
341 ($i:expr, *) => { call!($i, <$crate::tokens::Star as $crate::Synom>::parse) };
342 ($i:expr, -) => { call!($i, <$crate::tokens::Sub as $crate::Synom>::parse) };
343 ($i:expr, -=) => { call!($i, <$crate::tokens::SubEq as $crate::Synom>::parse) };
344 ($i:expr, _) => { call!($i, <$crate::tokens::Underscore as $crate::Synom>::parse) };
345}
346
347#[macro_export]
348macro_rules! keyword {
349 ($i:expr, as) => { call!($i, <$crate::tokens::As as $crate::Synom>::parse) };
Nika Layzell0dc6e632017-11-18 12:55:25 -0500350 ($i:expr, auto) => { call!($i, <$crate::tokens::Auto as $crate::Synom>::parse) };
David Tolnayf8db7ba2017-11-11 22:52:16 -0800351 ($i:expr, box) => { call!($i, <$crate::tokens::Box as $crate::Synom>::parse) };
352 ($i:expr, break) => { call!($i, <$crate::tokens::Break as $crate::Synom>::parse) };
353 ($i:expr, Self) => { call!($i, <$crate::tokens::CapSelf as $crate::Synom>::parse) };
354 ($i:expr, catch) => { call!($i, <$crate::tokens::Catch as $crate::Synom>::parse) };
355 ($i:expr, const) => { call!($i, <$crate::tokens::Const as $crate::Synom>::parse) };
356 ($i:expr, continue) => { call!($i, <$crate::tokens::Continue as $crate::Synom>::parse) };
357 ($i:expr, crate) => { call!($i, <$crate::tokens::Crate as $crate::Synom>::parse) };
358 ($i:expr, default) => { call!($i, <$crate::tokens::Default as $crate::Synom>::parse) };
359 ($i:expr, do) => { call!($i, <$crate::tokens::Do as $crate::Synom>::parse) };
360 ($i:expr, else) => { call!($i, <$crate::tokens::Else as $crate::Synom>::parse) };
361 ($i:expr, enum) => { call!($i, <$crate::tokens::Enum as $crate::Synom>::parse) };
362 ($i:expr, extern) => { call!($i, <$crate::tokens::Extern as $crate::Synom>::parse) };
363 ($i:expr, fn) => { call!($i, <$crate::tokens::Fn as $crate::Synom>::parse) };
364 ($i:expr, for) => { call!($i, <$crate::tokens::For as $crate::Synom>::parse) };
365 ($i:expr, if) => { call!($i, <$crate::tokens::If as $crate::Synom>::parse) };
366 ($i:expr, impl) => { call!($i, <$crate::tokens::Impl as $crate::Synom>::parse) };
367 ($i:expr, in) => { call!($i, <$crate::tokens::In as $crate::Synom>::parse) };
368 ($i:expr, let) => { call!($i, <$crate::tokens::Let as $crate::Synom>::parse) };
369 ($i:expr, loop) => { call!($i, <$crate::tokens::Loop as $crate::Synom>::parse) };
370 ($i:expr, match) => { call!($i, <$crate::tokens::Match as $crate::Synom>::parse) };
371 ($i:expr, mod) => { call!($i, <$crate::tokens::Mod as $crate::Synom>::parse) };
372 ($i:expr, move) => { call!($i, <$crate::tokens::Move as $crate::Synom>::parse) };
373 ($i:expr, mut) => { call!($i, <$crate::tokens::Mut as $crate::Synom>::parse) };
374 ($i:expr, pub) => { call!($i, <$crate::tokens::Pub as $crate::Synom>::parse) };
375 ($i:expr, ref) => { call!($i, <$crate::tokens::Ref as $crate::Synom>::parse) };
376 ($i:expr, return) => { call!($i, <$crate::tokens::Return as $crate::Synom>::parse) };
377 ($i:expr, self) => { call!($i, <$crate::tokens::Self_ as $crate::Synom>::parse) };
378 ($i:expr, static) => { call!($i, <$crate::tokens::Static as $crate::Synom>::parse) };
379 ($i:expr, struct) => { call!($i, <$crate::tokens::Struct as $crate::Synom>::parse) };
380 ($i:expr, super) => { call!($i, <$crate::tokens::Super as $crate::Synom>::parse) };
381 ($i:expr, trait) => { call!($i, <$crate::tokens::Trait as $crate::Synom>::parse) };
382 ($i:expr, type) => { call!($i, <$crate::tokens::Type as $crate::Synom>::parse) };
383 ($i:expr, union) => { call!($i, <$crate::tokens::Union as $crate::Synom>::parse) };
384 ($i:expr, unsafe) => { call!($i, <$crate::tokens::Unsafe as $crate::Synom>::parse) };
385 ($i:expr, use) => { call!($i, <$crate::tokens::Use as $crate::Synom>::parse) };
386 ($i:expr, where) => { call!($i, <$crate::tokens::Where as $crate::Synom>::parse) };
387 ($i:expr, while) => { call!($i, <$crate::tokens::While as $crate::Synom>::parse) };
388 ($i:expr, yield) => { call!($i, <$crate::tokens::Yield as $crate::Synom>::parse) };
389}
390
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700391#[cfg(feature = "parsing")]
392mod parsing {
Alex Crichtonf9e8f1a2017-07-05 18:20:44 -0700393 use proc_macro2::{Delimiter, Spacing};
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700394
Michael Layzell760fd662017-05-31 22:46:05 -0400395 use {PResult, Cursor, parse_error};
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700396 use span::Span;
397
398 pub trait FromSpans: Sized {
399 fn from_spans(spans: &[Span]) -> Self;
400 }
401
402 impl FromSpans for [Span; 1] {
403 fn from_spans(spans: &[Span]) -> Self {
404 [spans[0]]
405 }
406 }
407
408 impl FromSpans for [Span; 2] {
409 fn from_spans(spans: &[Span]) -> Self {
410 [spans[0], spans[1]]
411 }
412 }
413
414 impl FromSpans for [Span; 3] {
415 fn from_spans(spans: &[Span]) -> Self {
416 [spans[0], spans[1], spans[2]]
417 }
418 }
419
420 pub fn op<'a, T, R>(s: &str,
Michael Layzell0a1a6632017-06-02 18:07:43 -0400421 mut tokens: Cursor<'a>,
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700422 new: fn(T) -> R)
Michael Layzell760fd662017-05-31 22:46:05 -0400423 -> PResult<'a, R>
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700424 where T: FromSpans,
425 {
426 let mut spans = [Span::default(); 3];
Alex Crichton954046c2017-05-30 21:49:42 -0700427 assert!(s.len() <= spans.len());
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700428 let chars = s.chars();
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700429
Alex Crichton954046c2017-05-30 21:49:42 -0700430 for (i, (ch, slot)) in chars.zip(&mut spans).enumerate() {
Michael Layzell0a1a6632017-06-02 18:07:43 -0400431 match tokens.op() {
432 Some((rest, span, c, kind)) if c == ch => {
433 if i != s.len() - 1 {
Alex Crichtonf9e8f1a2017-07-05 18:20:44 -0700434 match kind {
435 Spacing::Joint => {}
436 _ => return parse_error(),
Michael Layzell0a1a6632017-06-02 18:07:43 -0400437 }
438 }
439 *slot = Span(span);
440 tokens = rest;
Alex Crichton954046c2017-05-30 21:49:42 -0700441 }
Michael Layzell0a1a6632017-06-02 18:07:43 -0400442 _ => return parse_error()
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700443 }
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700444 }
Michael Layzell0a1a6632017-06-02 18:07:43 -0400445 Ok((tokens, new(T::from_spans(&spans))))
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700446 }
447
448 pub fn sym<'a, T>(sym: &str,
Michael Layzell760fd662017-05-31 22:46:05 -0400449 tokens: Cursor<'a>,
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700450 new: fn(Span) -> T)
Michael Layzell760fd662017-05-31 22:46:05 -0400451 -> PResult<'a, T>
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700452 {
Michael Layzell0a1a6632017-06-02 18:07:43 -0400453 if let Some((rest, span, s)) = tokens.word() {
454 if s.as_str() == sym {
455 return Ok((rest, new(Span(span))));
456 }
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700457 }
Michael Layzell0a1a6632017-06-02 18:07:43 -0400458 parse_error()
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700459 }
460
461 pub fn delim<'a, F, R, T>(delim: &str,
Michael Layzell760fd662017-05-31 22:46:05 -0400462 tokens: Cursor<'a>,
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700463 new: fn(Span) -> T,
464 f: F)
Michael Layzell760fd662017-05-31 22:46:05 -0400465 -> PResult<'a, (R, T)>
466 where F: FnOnce(Cursor) -> PResult<R>
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700467 {
Michael Layzell0a1a6632017-06-02 18:07:43 -0400468 // NOTE: We should support none-delimited sequences here.
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700469 let delim = match delim {
470 "(" => Delimiter::Parenthesis,
471 "{" => Delimiter::Brace,
472 "[" => Delimiter::Bracket,
Michael Layzell93c36282017-06-04 20:43:14 -0400473 " " => Delimiter::None,
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700474 _ => panic!("unknown delimiter: {}", delim),
475 };
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700476
Michael Layzell0a1a6632017-06-02 18:07:43 -0400477 if let Some(seqinfo) = tokens.seq(delim) {
478 match f(seqinfo.inside) {
479 Ok((remaining, ret)) => {
480 if remaining.eof() {
481 return Ok((seqinfo.outside, (ret, new(Span(seqinfo.span)))));
482 }
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700483 }
Michael Layzell0a1a6632017-06-02 18:07:43 -0400484 Err(err) => return Err(err),
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700485 }
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700486 }
Michael Layzell0a1a6632017-06-02 18:07:43 -0400487 parse_error()
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700488 }
489}
490
491#[cfg(feature = "printing")]
492mod printing {
Alex Crichtonf9e8f1a2017-07-05 18:20:44 -0700493 use proc_macro2::{TokenTree, TokenNode, Spacing, Term};
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700494 use quote::Tokens;
495
496 use span::Span;
497
498 pub fn op(s: &str, spans: &[Span], tokens: &mut Tokens) {
499 assert_eq!(s.len(), spans.len());
500
501 let mut chars = s.chars();
502 let mut spans = spans.iter();
503 let ch = chars.next_back().unwrap();
504 let span = spans.next_back().unwrap();
505 for (ch, span) in chars.zip(spans) {
506 tokens.append(TokenTree {
507 span: span.0,
Alex Crichtonf9e8f1a2017-07-05 18:20:44 -0700508 kind: TokenNode::Op(ch, Spacing::Joint),
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700509 });
510 }
511
512 tokens.append(TokenTree {
513 span: span.0,
Alex Crichtonf9e8f1a2017-07-05 18:20:44 -0700514 kind: TokenNode::Op(ch, Spacing::Alone),
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700515 });
516 }
517
518 pub fn sym(s: &str, span: &Span, tokens: &mut Tokens) {
519 tokens.append(TokenTree {
520 span: span.0,
Alex Crichtonf9e8f1a2017-07-05 18:20:44 -0700521 kind: TokenNode::Term(Term::intern(s)),
Alex Crichton7b9e02f2017-05-30 15:54:33 -0700522 });
523 }
524
525 pub fn delim<F>(s: &str, span: &Span, tokens: &mut Tokens, f: F)
526 where F: FnOnce(&mut Tokens)
527 {
528 tokens.append_delimited(s, span.0, f)
529 }
530}