2020-02-29 20:37:32 +03:00
|
|
|
use rustc_ast::tokenstream::{TokenStream, TokenTree};
|
2019-12-29 17:23:55 +03:00
|
|
|
use rustc_expand::base::{self, ExtCtxt};
|
2020-01-01 19:30:57 +01:00
|
|
|
use rustc_span::symbol::kw;
|
2019-12-31 20:15:40 +03:00
|
|
|
use rustc_span::Span;
|
2014-05-05 18:56:44 -07:00
|
|
|
|
2019-02-04 21:49:54 +09:00
|
|
|
pub fn expand_trace_macros(
|
|
|
|
cx: &mut ExtCtxt<'_>,
|
2013-08-31 18:13:04 +02:00
|
|
|
sp: Span,
|
2019-08-31 20:08:06 +03:00
|
|
|
tt: TokenStream,
|
2018-07-12 11:58:16 +02:00
|
|
|
) -> Box<dyn base::MacResult + 'static> {
|
2019-08-31 20:08:06 +03:00
|
|
|
let mut cursor = tt.into_trees();
|
|
|
|
let mut err = false;
|
|
|
|
let value = match &cursor.next() {
|
Remove `TreeAndSpacing`.
A `TokenStream` contains a `Lrc<Vec<(TokenTree, Spacing)>>`. But this is
not quite right. `Spacing` makes sense for `TokenTree::Token`, but does
not make sense for `TokenTree::Delimited`, because a
`TokenTree::Delimited` cannot be joined with another `TokenTree`.
This commit fixes this problem, by adding `Spacing` to `TokenTree::Token`,
changing `TokenStream` to contain a `Lrc<Vec<TokenTree>>`, and removing the
`TreeAndSpacing` typedef.
The commit removes these two impls:
- `impl From<TokenTree> for TokenStream`
- `impl From<TokenTree> for TreeAndSpacing`
These were useful, but also resulted in code with many `.into()` calls
that was hard to read, particularly for anyone not highly familiar with
the relevant types. This commit makes some other changes to compensate:
- `TokenTree::token()` becomes `TokenTree::token_{alone,joint}()`.
- `TokenStream::token_{alone,joint}()` are added.
- `TokenStream::delimited` is added.
This results in things like this:
```rust
TokenTree::token(token::Semi, stmt.span).into()
```
changing to this:
```rust
TokenStream::token_alone(token::Semi, stmt.span)
```
This makes the type of the result, and its spacing, clearer.
These changes also simplifies `Cursor` and `CursorRef`, because they no longer
need to distinguish between `next` and `next_with_spacing`.
2022-07-28 10:31:04 +10:00
|
|
|
Some(TokenTree::Token(token, _)) if token.is_keyword(kw::True) => true,
|
|
|
|
Some(TokenTree::Token(token, _)) if token.is_keyword(kw::False) => false,
|
2019-08-31 20:08:06 +03:00
|
|
|
_ => {
|
|
|
|
err = true;
|
|
|
|
false
|
|
|
|
}
|
|
|
|
};
|
|
|
|
err |= cursor.next().is_some();
|
|
|
|
if err {
|
|
|
|
cx.span_err(sp, "trace_macros! accepts only `true` or `false`")
|
|
|
|
} else {
|
|
|
|
cx.set_trace_macros(value);
|
2012-08-15 10:45:10 -07:00
|
|
|
}
|
2012-11-26 22:12:31 -05:00
|
|
|
|
2018-12-20 03:57:48 +03:00
|
|
|
base::DummyResult::any_valid(sp)
|
2012-08-15 10:45:10 -07:00
|
|
|
}
|