Force token collection to run when parsing nonterminals
Fixes #81007 Previously, we would fail to collect tokens in the proper place when only builtin attributes were present. As a result, we would end up with attribute tokens in the collected `TokenStream`, leading to duplication when we attempted to prepend the attributes from the AST node. We now explicitly track when token collection must be performed due to nomterminal parsing.
This commit is contained in:
parent
a4cbb44ae2
commit
11b1e37016
13 changed files with 233 additions and 72 deletions
|
@ -54,6 +54,13 @@ enum BlockMode {
|
|||
Ignore,
|
||||
}
|
||||
|
||||
/// Whether or not we should force collection of tokens for an AST node,
|
||||
/// regardless of whether or not it has attributes
|
||||
pub enum ForceCollect {
|
||||
Yes,
|
||||
No,
|
||||
}
|
||||
|
||||
/// Like `maybe_whole_expr`, but for things other than expressions.
|
||||
#[macro_export]
|
||||
macro_rules! maybe_whole {
|
||||
|
@ -1413,3 +1420,16 @@ fn make_token_stream(
|
|||
assert!(stack.is_empty(), "Stack should be empty: final_buf={:?} stack={:?}", final_buf, stack);
|
||||
TokenStream::new(final_buf.inner)
|
||||
}
|
||||
|
||||
#[macro_export]
|
||||
macro_rules! maybe_collect_tokens {
|
||||
($self:ident, $force_collect:expr, $attrs:expr, $f:expr) => {
|
||||
if matches!($force_collect, ForceCollect::Yes)
|
||||
|| $crate::parser::attr::maybe_needs_tokens($attrs)
|
||||
{
|
||||
$self.collect_tokens($f)
|
||||
} else {
|
||||
$f($self)
|
||||
}
|
||||
};
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue