David Tolnay | f60f426 | 2017-12-28 19:17:58 -0500 | [diff] [blame] | 1 | #[cfg(feature = "fold")] |
| 2 | pub mod fold { |
David Tolnay | 4b4c4b6 | 2018-01-06 13:48:05 -0800 | [diff] [blame] | 3 | use fold::Fold; |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 4 | use proc_macro2::Span; |
David Tolnay | 94d2b79 | 2018-04-29 12:26:10 -0700 | [diff] [blame] | 5 | use punctuated::{Pair, Punctuated}; |
David Tolnay | f60f426 | 2017-12-28 19:17:58 -0500 | [diff] [blame] | 6 | |
| 7 | pub trait FoldHelper { |
| 8 | type Item; |
David Tolnay | 61037c6 | 2018-01-05 16:21:03 -0800 | [diff] [blame] | 9 | fn lift<F>(self, f: F) -> Self |
| 10 | where |
| 11 | F: FnMut(Self::Item) -> Self::Item; |
David Tolnay | f60f426 | 2017-12-28 19:17:58 -0500 | [diff] [blame] | 12 | } |
| 13 | |
| 14 | impl<T> FoldHelper for Vec<T> { |
| 15 | type Item = T; |
David Tolnay | 61037c6 | 2018-01-05 16:21:03 -0800 | [diff] [blame] | 16 | fn lift<F>(self, f: F) -> Self |
| 17 | where |
| 18 | F: FnMut(Self::Item) -> Self::Item, |
| 19 | { |
David Tolnay | f60f426 | 2017-12-28 19:17:58 -0500 | [diff] [blame] | 20 | self.into_iter().map(f).collect() |
| 21 | } |
| 22 | } |
| 23 | |
David Tolnay | f2cfd72 | 2017-12-31 18:02:51 -0500 | [diff] [blame] | 24 | impl<T, U> FoldHelper for Punctuated<T, U> { |
David Tolnay | f60f426 | 2017-12-28 19:17:58 -0500 | [diff] [blame] | 25 | type Item = T; |
David Tolnay | 61037c6 | 2018-01-05 16:21:03 -0800 | [diff] [blame] | 26 | fn lift<F>(self, mut f: F) -> Self |
| 27 | where |
| 28 | F: FnMut(Self::Item) -> Self::Item, |
| 29 | { |
David Tolnay | 5608068 | 2018-01-06 14:01:52 -0800 | [diff] [blame] | 30 | self.into_pairs() |
| 31 | .map(Pair::into_tuple) |
| 32 | .map(|(t, u)| Pair::new(f(t), u)) |
David Tolnay | 660fd1f | 2017-12-31 01:52:57 -0500 | [diff] [blame] | 33 | .collect() |
David Tolnay | f60f426 | 2017-12-28 19:17:58 -0500 | [diff] [blame] | 34 | } |
| 35 | } |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 36 | |
David Tolnay | 4b4c4b6 | 2018-01-06 13:48:05 -0800 | [diff] [blame] | 37 | pub fn tokens_helper<F: Fold + ?Sized, S: Spans>(folder: &mut F, spans: &S) -> S { |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 38 | spans.fold(folder) |
| 39 | } |
| 40 | |
| 41 | pub trait Spans { |
David Tolnay | 4b4c4b6 | 2018-01-06 13:48:05 -0800 | [diff] [blame] | 42 | fn fold<F: Fold + ?Sized>(&self, folder: &mut F) -> Self; |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 43 | } |
| 44 | |
| 45 | impl Spans for Span { |
David Tolnay | 4b4c4b6 | 2018-01-06 13:48:05 -0800 | [diff] [blame] | 46 | fn fold<F: Fold + ?Sized>(&self, folder: &mut F) -> Self { |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 47 | folder.fold_span(*self) |
| 48 | } |
| 49 | } |
| 50 | |
| 51 | impl Spans for [Span; 1] { |
David Tolnay | 4b4c4b6 | 2018-01-06 13:48:05 -0800 | [diff] [blame] | 52 | fn fold<F: Fold + ?Sized>(&self, folder: &mut F) -> Self { |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 53 | [folder.fold_span(self[0])] |
| 54 | } |
| 55 | } |
| 56 | |
| 57 | impl Spans for [Span; 2] { |
David Tolnay | 4b4c4b6 | 2018-01-06 13:48:05 -0800 | [diff] [blame] | 58 | fn fold<F: Fold + ?Sized>(&self, folder: &mut F) -> Self { |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 59 | [folder.fold_span(self[0]), folder.fold_span(self[1])] |
| 60 | } |
| 61 | } |
| 62 | |
| 63 | impl Spans for [Span; 3] { |
David Tolnay | 4b4c4b6 | 2018-01-06 13:48:05 -0800 | [diff] [blame] | 64 | fn fold<F: Fold + ?Sized>(&self, folder: &mut F) -> Self { |
David Tolnay | 61037c6 | 2018-01-05 16:21:03 -0800 | [diff] [blame] | 65 | [ |
| 66 | folder.fold_span(self[0]), |
| 67 | folder.fold_span(self[1]), |
| 68 | folder.fold_span(self[2]), |
| 69 | ] |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 70 | } |
| 71 | } |
| 72 | } |
| 73 | |
| 74 | #[cfg(feature = "visit")] |
| 75 | pub mod visit { |
| 76 | use proc_macro2::Span; |
David Tolnay | 4b4c4b6 | 2018-01-06 13:48:05 -0800 | [diff] [blame] | 77 | use visit::Visit; |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 78 | |
David Tolnay | 94d2b79 | 2018-04-29 12:26:10 -0700 | [diff] [blame] | 79 | pub fn tokens_helper<'ast, V: Visit<'ast> + ?Sized, S: Spans>(visitor: &mut V, spans: &'ast S) { |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 80 | spans.visit(visitor); |
| 81 | } |
| 82 | |
| 83 | pub trait Spans { |
David Tolnay | 4b4c4b6 | 2018-01-06 13:48:05 -0800 | [diff] [blame] | 84 | fn visit<'ast, V: Visit<'ast> + ?Sized>(&'ast self, visitor: &mut V); |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 85 | } |
| 86 | |
| 87 | impl Spans for Span { |
David Tolnay | 4b4c4b6 | 2018-01-06 13:48:05 -0800 | [diff] [blame] | 88 | fn visit<'ast, V: Visit<'ast> + ?Sized>(&'ast self, visitor: &mut V) { |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 89 | visitor.visit_span(self); |
| 90 | } |
| 91 | } |
| 92 | |
| 93 | impl Spans for [Span; 1] { |
David Tolnay | 4b4c4b6 | 2018-01-06 13:48:05 -0800 | [diff] [blame] | 94 | fn visit<'ast, V: Visit<'ast> + ?Sized>(&'ast self, visitor: &mut V) { |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 95 | visitor.visit_span(&self[0]); |
| 96 | } |
| 97 | } |
| 98 | |
| 99 | impl Spans for [Span; 2] { |
David Tolnay | 4b4c4b6 | 2018-01-06 13:48:05 -0800 | [diff] [blame] | 100 | fn visit<'ast, V: Visit<'ast> + ?Sized>(&'ast self, visitor: &mut V) { |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 101 | visitor.visit_span(&self[0]); |
| 102 | visitor.visit_span(&self[1]); |
| 103 | } |
| 104 | } |
| 105 | |
| 106 | impl Spans for [Span; 3] { |
David Tolnay | 4b4c4b6 | 2018-01-06 13:48:05 -0800 | [diff] [blame] | 107 | fn visit<'ast, V: Visit<'ast> + ?Sized>(&'ast self, visitor: &mut V) { |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 108 | visitor.visit_span(&self[0]); |
| 109 | visitor.visit_span(&self[1]); |
| 110 | visitor.visit_span(&self[2]); |
| 111 | } |
| 112 | } |
| 113 | } |
| 114 | |
David Tolnay | 9df02c4 | 2018-01-06 13:52:48 -0800 | [diff] [blame] | 115 | #[cfg(feature = "visit-mut")] |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 116 | pub mod visit_mut { |
| 117 | use proc_macro2::Span; |
David Tolnay | 4b4c4b6 | 2018-01-06 13:48:05 -0800 | [diff] [blame] | 118 | use visit_mut::VisitMut; |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 119 | |
David Tolnay | 4b4c4b6 | 2018-01-06 13:48:05 -0800 | [diff] [blame] | 120 | pub fn tokens_helper<V: VisitMut + ?Sized, S: Spans>(visitor: &mut V, spans: &mut S) { |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 121 | spans.visit_mut(visitor); |
| 122 | } |
| 123 | |
| 124 | pub trait Spans { |
David Tolnay | 4b4c4b6 | 2018-01-06 13:48:05 -0800 | [diff] [blame] | 125 | fn visit_mut<V: VisitMut + ?Sized>(&mut self, visitor: &mut V); |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 126 | } |
| 127 | |
| 128 | impl Spans for Span { |
David Tolnay | 4b4c4b6 | 2018-01-06 13:48:05 -0800 | [diff] [blame] | 129 | fn visit_mut<V: VisitMut + ?Sized>(&mut self, visitor: &mut V) { |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 130 | visitor.visit_span_mut(self); |
| 131 | } |
| 132 | } |
| 133 | |
| 134 | impl Spans for [Span; 1] { |
David Tolnay | 4b4c4b6 | 2018-01-06 13:48:05 -0800 | [diff] [blame] | 135 | fn visit_mut<V: VisitMut + ?Sized>(&mut self, visitor: &mut V) { |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 136 | visitor.visit_span_mut(&mut self[0]); |
| 137 | } |
| 138 | } |
| 139 | |
| 140 | impl Spans for [Span; 2] { |
David Tolnay | 4b4c4b6 | 2018-01-06 13:48:05 -0800 | [diff] [blame] | 141 | fn visit_mut<V: VisitMut + ?Sized>(&mut self, visitor: &mut V) { |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 142 | visitor.visit_span_mut(&mut self[0]); |
| 143 | visitor.visit_span_mut(&mut self[1]); |
| 144 | } |
| 145 | } |
| 146 | |
| 147 | impl Spans for [Span; 3] { |
David Tolnay | 4b4c4b6 | 2018-01-06 13:48:05 -0800 | [diff] [blame] | 148 | fn visit_mut<V: VisitMut + ?Sized>(&mut self, visitor: &mut V) { |
David Tolnay | cc0f037 | 2017-12-28 19:11:04 -0500 | [diff] [blame] | 149 | visitor.visit_span_mut(&mut self[0]); |
| 150 | visitor.visit_span_mut(&mut self[1]); |
| 151 | visitor.visit_span_mut(&mut self[2]); |
| 152 | } |
| 153 | } |
David Tolnay | f60f426 | 2017-12-28 19:17:58 -0500 | [diff] [blame] | 154 | } |