@@ -13,7 +13,7 @@ use std::collections::HashMap;
1313use std:: panic:: { AssertUnwindSafe , catch_unwind} ;
1414
1515use rustc_ast:: token:: { BinOpToken , Delimiter , Token , TokenKind } ;
16- use rustc_ast:: tokenstream:: { RefTokenTreeCursor , TokenStream , TokenTree } ;
16+ use rustc_ast:: tokenstream:: { TokenStream , TokenStreamIter , TokenTree } ;
1717use rustc_ast:: { ast, ptr} ;
1818use rustc_ast_pretty:: pprust;
1919use rustc_span:: {
@@ -441,7 +441,7 @@ pub(crate) fn rewrite_macro_def(
441441 }
442442
443443 let ts = def. body . tokens . clone ( ) ;
444- let mut parser = MacroParser :: new ( ts. trees ( ) ) ;
444+ let mut parser = MacroParser :: new ( ts. iter ( ) ) ;
445445 let parsed_def = match parser. parse ( ) {
446446 Some ( def) => def,
447447 None => return snippet,
@@ -792,7 +792,7 @@ impl MacroArgParser {
792792 self . buf . clear ( ) ;
793793 }
794794
795- fn add_meta_variable ( & mut self , iter : & mut RefTokenTreeCursor < ' _ > ) -> Option < ( ) > {
795+ fn add_meta_variable ( & mut self , iter : & mut TokenStreamIter < ' _ > ) -> Option < ( ) > {
796796 match iter. next ( ) {
797797 Some ( & TokenTree :: Token (
798798 Token {
@@ -824,7 +824,7 @@ impl MacroArgParser {
824824 & mut self ,
825825 inner : Vec < ParsedMacroArg > ,
826826 delim : Delimiter ,
827- iter : & mut RefTokenTreeCursor < ' _ > ,
827+ iter : & mut TokenStreamIter < ' _ > ,
828828 ) -> Option < ( ) > {
829829 let mut buffer = String :: new ( ) ;
830830 let mut first = true ;
@@ -924,7 +924,7 @@ impl MacroArgParser {
924924
925925 /// Returns a collection of parsed macro def's arguments.
926926 fn parse ( mut self , tokens : TokenStream ) -> Option < Vec < ParsedMacroArg > > {
927- let mut iter = tokens. trees ( ) ;
927+ let mut iter = tokens. iter ( ) ;
928928
929929 while let Some ( tok) = iter. next ( ) {
930930 match tok {
@@ -1061,7 +1061,7 @@ fn format_macro_args(
10611061}
10621062
10631063fn 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 ( ) )
10651065}
10661066
10671067// 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
11771177// A very simple parser that just parses a macros 2.0 definition into its branches.
11781178// Currently we do not attempt to parse any further than that.
11791179struct MacroParser < ' a > {
1180- toks : RefTokenTreeCursor < ' a > ,
1180+ iter : TokenStreamIter < ' a > ,
11811181}
11821182
11831183impl < ' 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 }
11861186 }
11871187
11881188 // (`(` ... `)` `=>` `{` ... `}`)*
11891189 fn parse ( & mut self ) -> Option < Macro > {
11901190 let mut branches = vec ! [ ] ;
1191- while self . toks . look_ahead ( 1 ) . is_some ( ) {
1191+ while self . iter . peek ( ) . is_some ( ) {
11921192 branches. push ( self . parse_branch ( ) ?) ;
11931193 }
11941194
@@ -1197,13 +1197,13 @@ impl<'a> MacroParser<'a> {
11971197
11981198 // `(` ... `)` `=>` `{` ... `}`
11991199 fn parse_branch ( & mut self ) -> Option < MacroBranch > {
1200- let tok = self . toks . next ( ) ?;
1200+ let tok = self . iter . next ( ) ?;
12011201 let ( lo, args_paren_kind) = match tok {
12021202 TokenTree :: Token ( ..) => return None ,
12031203 & TokenTree :: Delimited ( delimited_span, _, d, _) => ( delimited_span. open . lo ( ) , d) ,
12041204 } ;
12051205 let args = TokenStream :: new ( vec ! [ tok. clone( ) ] ) ;
1206- match self . toks . next ( ) ? {
1206+ match self . iter . next ( ) ? {
12071207 TokenTree :: Token (
12081208 Token {
12091209 kind : TokenKind :: FatArrow ,
@@ -1213,7 +1213,7 @@ impl<'a> MacroParser<'a> {
12131213 ) => { }
12141214 _ => return None ,
12151215 }
1216- let ( mut hi, body, whole_body) = match self . toks . next ( ) ? {
1216+ let ( mut hi, body, whole_body) = match self . iter . next ( ) ? {
12171217 TokenTree :: Token ( ..) => return None ,
12181218 TokenTree :: Delimited ( delimited_span, ..) => {
12191219 let data = delimited_span. entire ( ) . data ( ) ;
@@ -1235,10 +1235,10 @@ impl<'a> MacroParser<'a> {
12351235 span,
12361236 } ,
12371237 _,
1238- ) ) = self . toks . look_ahead ( 0 )
1238+ ) ) = self . iter . peek ( )
12391239 {
12401240 hi = span. hi ( ) ;
1241- self . toks . next ( ) ;
1241+ self . iter . next ( ) ;
12421242 }
12431243 Some ( MacroBranch {
12441244 span : mk_sp ( lo, hi) ,
0 commit comments