0
0
mirror of https://github.com/rust-lang/rust.git synced 2024-11-30 20:01:19 +01:00
rust/tests/incremental/issue-85197-invalid-span/auxiliary/respan.rs
2024-11-27 07:18:25 -08:00

15 lines
471 B
Rust

extern crate proc_macro;
use proc_macro::TokenStream;
/// Copies the resolution information (the `SyntaxContext`) of the first
/// token to all other tokens in the stream. Does not recurse into groups.
#[proc_macro]
pub fn respan(input: TokenStream) -> TokenStream {
let first_span = input.clone().into_iter().next().unwrap().span();
input.into_iter().map(|mut tree| {
tree.set_span(tree.span().resolved_at(first_span));
tree
}).collect()
}