@@ -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:: {
@@ -443,7 +443,7 @@ pub(crate) fn rewrite_macro_def(
443
443
}
444
444
445
445
let ts = def. body . tokens . clone ( ) ;
446
- let mut parser = MacroParser :: new ( ts. trees ( ) ) ;
446
+ let mut parser = MacroParser :: new ( ts. iter ( ) ) ;
447
447
let parsed_def = match parser. parse ( ) {
448
448
Some ( def) => def,
449
449
None => return snippet,
@@ -794,7 +794,7 @@ impl MacroArgParser {
794
794
self . buf . clear ( ) ;
795
795
}
796
796
797
- fn add_meta_variable ( & mut self , iter : & mut RefTokenTreeCursor < ' _ > ) -> Option < ( ) > {
797
+ fn add_meta_variable ( & mut self , iter : & mut TokenStreamIter < ' _ > ) -> Option < ( ) > {
798
798
match iter. next ( ) {
799
799
Some ( & TokenTree :: Token (
800
800
Token {
@@ -826,7 +826,7 @@ impl MacroArgParser {
826
826
& mut self ,
827
827
inner : Vec < ParsedMacroArg > ,
828
828
delim : Delimiter ,
829
- iter : & mut RefTokenTreeCursor < ' _ > ,
829
+ iter : & mut TokenStreamIter < ' _ > ,
830
830
) -> Option < ( ) > {
831
831
let mut buffer = String :: new ( ) ;
832
832
let mut first = true ;
@@ -926,7 +926,7 @@ impl MacroArgParser {
926
926
927
927
/// Returns a collection of parsed macro def's arguments.
928
928
fn parse ( mut self , tokens : TokenStream ) -> Option < Vec < ParsedMacroArg > > {
929
- let mut iter = tokens. trees ( ) ;
929
+ let mut iter = tokens. iter ( ) ;
930
930
931
931
while let Some ( tok) = iter. next ( ) {
932
932
match tok {
@@ -1063,7 +1063,7 @@ fn format_macro_args(
1063
1063
}
1064
1064
1065
1065
fn span_for_token_stream ( token_stream : & TokenStream ) -> Option < Span > {
1066
- token_stream. trees ( ) . next ( ) . map ( |tt| tt. span ( ) )
1066
+ token_stream. iter ( ) . next ( ) . map ( |tt| tt. span ( ) )
1067
1067
}
1068
1068
1069
1069
// We should insert a space if the next token is a:
@@ -1179,18 +1179,18 @@ pub(crate) fn macro_style(mac: &ast::MacCall, context: &RewriteContext<'_>) -> D
1179
1179
// A very simple parser that just parses a macros 2.0 definition into its branches.
1180
1180
// Currently we do not attempt to parse any further than that.
1181
1181
struct MacroParser < ' a > {
1182
- toks : RefTokenTreeCursor < ' a > ,
1182
+ iter : TokenStreamIter < ' a > ,
1183
1183
}
1184
1184
1185
1185
impl < ' a > MacroParser < ' a > {
1186
- const fn new ( toks : RefTokenTreeCursor < ' a > ) -> Self {
1187
- Self { toks }
1186
+ const fn new ( iter : TokenStreamIter < ' a > ) -> Self {
1187
+ Self { iter }
1188
1188
}
1189
1189
1190
1190
// (`(` ... `)` `=>` `{` ... `}`)*
1191
1191
fn parse ( & mut self ) -> Option < Macro > {
1192
1192
let mut branches = vec ! [ ] ;
1193
- while self . toks . peek ( ) . is_some ( ) {
1193
+ while self . iter . peek ( ) . is_some ( ) {
1194
1194
branches. push ( self . parse_branch ( ) ?) ;
1195
1195
}
1196
1196
@@ -1199,13 +1199,13 @@ impl<'a> MacroParser<'a> {
1199
1199
1200
1200
// `(` ... `)` `=>` `{` ... `}`
1201
1201
fn parse_branch ( & mut self ) -> Option < MacroBranch > {
1202
- let tok = self . toks . next ( ) ?;
1202
+ let tok = self . iter . next ( ) ?;
1203
1203
let ( lo, args_paren_kind) = match tok {
1204
1204
TokenTree :: Token ( ..) => return None ,
1205
1205
& TokenTree :: Delimited ( delimited_span, _, d, _) => ( delimited_span. open . lo ( ) , d) ,
1206
1206
} ;
1207
1207
let args = TokenStream :: new ( vec ! [ tok. clone( ) ] ) ;
1208
- match self . toks . next ( ) ? {
1208
+ match self . iter . next ( ) ? {
1209
1209
TokenTree :: Token (
1210
1210
Token {
1211
1211
kind : TokenKind :: FatArrow ,
@@ -1215,7 +1215,7 @@ impl<'a> MacroParser<'a> {
1215
1215
) => { }
1216
1216
_ => return None ,
1217
1217
}
1218
- let ( mut hi, body, whole_body) = match self . toks . next ( ) ? {
1218
+ let ( mut hi, body, whole_body) = match self . iter . next ( ) ? {
1219
1219
TokenTree :: Token ( ..) => return None ,
1220
1220
TokenTree :: Delimited ( delimited_span, ..) => {
1221
1221
let data = delimited_span. entire ( ) . data ( ) ;
@@ -1237,10 +1237,10 @@ impl<'a> MacroParser<'a> {
1237
1237
span,
1238
1238
} ,
1239
1239
_,
1240
- ) ) = self . toks . peek ( )
1240
+ ) ) = self . iter . peek ( )
1241
1241
{
1242
1242
hi = span. hi ( ) ;
1243
- self . toks . next ( ) ;
1243
+ self . iter . next ( ) ;
1244
1244
}
1245
1245
Some ( MacroBranch {
1246
1246
span : mk_sp ( lo, hi) ,
0 commit comments