@@ -13,7 +13,7 @@ use std::collections::HashMap;
13
13
use std:: panic:: { AssertUnwindSafe , catch_unwind} ;
14
14
15
15
use rustc_ast:: token:: { BinOpToken , Delimiter , Token , TokenKind } ;
16
- use rustc_ast:: tokenstream:: { RefTokenTreeCursor , TokenStream , TokenTree } ;
16
+ use rustc_ast:: tokenstream:: { TokenStream , TokenStreamIter , TokenTree } ;
17
17
use rustc_ast:: { ast, ptr} ;
18
18
use rustc_ast_pretty:: pprust;
19
19
use rustc_span:: {
@@ -441,7 +441,7 @@ pub(crate) fn rewrite_macro_def(
441
441
}
442
442
443
443
let ts = def. body . tokens . clone ( ) ;
444
- let mut parser = MacroParser :: new ( ts. trees ( ) ) ;
444
+ let mut parser = MacroParser :: new ( ts. iter ( ) ) ;
445
445
let parsed_def = match parser. parse ( ) {
446
446
Some ( def) => def,
447
447
None => return snippet,
@@ -792,7 +792,7 @@ impl MacroArgParser {
792
792
self . buf . clear ( ) ;
793
793
}
794
794
795
- fn add_meta_variable ( & mut self , iter : & mut RefTokenTreeCursor < ' _ > ) -> Option < ( ) > {
795
+ fn add_meta_variable ( & mut self , iter : & mut TokenStreamIter < ' _ > ) -> Option < ( ) > {
796
796
match iter. next ( ) {
797
797
Some ( & TokenTree :: Token (
798
798
Token {
@@ -824,7 +824,7 @@ impl MacroArgParser {
824
824
& mut self ,
825
825
inner : Vec < ParsedMacroArg > ,
826
826
delim : Delimiter ,
827
- iter : & mut RefTokenTreeCursor < ' _ > ,
827
+ iter : & mut TokenStreamIter < ' _ > ,
828
828
) -> Option < ( ) > {
829
829
let mut buffer = String :: new ( ) ;
830
830
let mut first = true ;
@@ -924,7 +924,7 @@ impl MacroArgParser {
924
924
925
925
/// Returns a collection of parsed macro def's arguments.
926
926
fn parse ( mut self , tokens : TokenStream ) -> Option < Vec < ParsedMacroArg > > {
927
- let mut iter = tokens. trees ( ) ;
927
+ let mut iter = tokens. iter ( ) ;
928
928
929
929
while let Some ( tok) = iter. next ( ) {
930
930
match tok {
@@ -1061,7 +1061,7 @@ fn format_macro_args(
1061
1061
}
1062
1062
1063
1063
fn span_for_token_stream ( token_stream : & TokenStream ) -> Option < Span > {
1064
- token_stream. trees ( ) . next ( ) . map ( |tt| tt. span ( ) )
1064
+ token_stream. iter ( ) . next ( ) . map ( |tt| tt. span ( ) )
1065
1065
}
1066
1066
1067
1067
// We should insert a space if the next token is a:
@@ -1177,18 +1177,18 @@ pub(crate) fn macro_style(mac: &ast::MacCall, context: &RewriteContext<'_>) -> D
1177
1177
// A very simple parser that just parses a macros 2.0 definition into its branches.
1178
1178
// Currently we do not attempt to parse any further than that.
1179
1179
struct MacroParser < ' a > {
1180
- toks : RefTokenTreeCursor < ' a > ,
1180
+ iter : TokenStreamIter < ' a > ,
1181
1181
}
1182
1182
1183
1183
impl < ' a > MacroParser < ' a > {
1184
- const fn new ( toks : RefTokenTreeCursor < ' a > ) -> Self {
1185
- Self { toks }
1184
+ const fn new ( iter : TokenStreamIter < ' a > ) -> Self {
1185
+ Self { iter }
1186
1186
}
1187
1187
1188
1188
// (`(` ... `)` `=>` `{` ... `}`)*
1189
1189
fn parse ( & mut self ) -> Option < Macro > {
1190
1190
let mut branches = vec ! [ ] ;
1191
- while self . toks . look_ahead ( 1 ) . is_some ( ) {
1191
+ while self . iter . peek ( ) . is_some ( ) {
1192
1192
branches. push ( self . parse_branch ( ) ?) ;
1193
1193
}
1194
1194
@@ -1197,13 +1197,13 @@ impl<'a> MacroParser<'a> {
1197
1197
1198
1198
// `(` ... `)` `=>` `{` ... `}`
1199
1199
fn parse_branch ( & mut self ) -> Option < MacroBranch > {
1200
- let tok = self . toks . next ( ) ?;
1200
+ let tok = self . iter . next ( ) ?;
1201
1201
let ( lo, args_paren_kind) = match tok {
1202
1202
TokenTree :: Token ( ..) => return None ,
1203
1203
& TokenTree :: Delimited ( delimited_span, _, d, _) => ( delimited_span. open . lo ( ) , d) ,
1204
1204
} ;
1205
1205
let args = TokenStream :: new ( vec ! [ tok. clone( ) ] ) ;
1206
- match self . toks . next ( ) ? {
1206
+ match self . iter . next ( ) ? {
1207
1207
TokenTree :: Token (
1208
1208
Token {
1209
1209
kind : TokenKind :: FatArrow ,
@@ -1213,7 +1213,7 @@ impl<'a> MacroParser<'a> {
1213
1213
) => { }
1214
1214
_ => return None ,
1215
1215
}
1216
- let ( mut hi, body, whole_body) = match self . toks . next ( ) ? {
1216
+ let ( mut hi, body, whole_body) = match self . iter . next ( ) ? {
1217
1217
TokenTree :: Token ( ..) => return None ,
1218
1218
TokenTree :: Delimited ( delimited_span, ..) => {
1219
1219
let data = delimited_span. entire ( ) . data ( ) ;
@@ -1235,10 +1235,10 @@ impl<'a> MacroParser<'a> {
1235
1235
span,
1236
1236
} ,
1237
1237
_,
1238
- ) ) = self . toks . look_ahead ( 0 )
1238
+ ) ) = self . iter . peek ( )
1239
1239
{
1240
1240
hi = span. hi ( ) ;
1241
- self . toks . next ( ) ;
1241
+ self . iter . next ( ) ;
1242
1242
}
1243
1243
Some ( MacroBranch {
1244
1244
span : mk_sp ( lo, hi) ,
0 commit comments