tests/ui/proc-macro/auxiliary/span-api-tests.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

#![feature(proc_macro_span)]

extern crate proc_macro;

use proc_macro::*;

// Re-emits the input tokens by parsing them from strings
#[proc_macro]
pub fn reemit(input: TokenStream) -> TokenStream {
    input.to_string().parse().unwrap()
}

#[proc_macro]
pub fn assert_local_file(input: TokenStream) -> TokenStream {
    for tk in input {
        assert!(tk.span().local_file().is_some(), "No local file for span: {:?}", tk.span());
    }

    "".parse().unwrap()
}

#[proc_macro]
pub fn macro_stringify(input: TokenStream) -> TokenStream {
    let mut tokens = input.into_iter();
    let first_span = tokens.next().expect("first token").span();
    let last_span = tokens.last().map(|x| x.span()).unwrap_or(first_span);
    let span = first_span.join(last_span).expect("joined span");
    let src = span.source_text().expect("source_text");
    TokenTree::Literal(Literal::string(&src)).into()
}