src/test/ui/proc-macro/auxiliary/custom-quote.rs

   1 // force-host
   2 // no-prefer-dynamic
   3 // ignore-tidy-linelength
   4
   5 #![feature(proc_macro_quote)]
   6 #![crate_type = "proc-macro"]
   7
   8 extern crate proc_macro;
   9 use std::str::FromStr;
  10 use proc_macro::*;
  11
  12 #[proc_macro]
  13 pub fn custom_quote(input: TokenStream) -> TokenStream {
  14     let mut tokens: Vec<_> = input.into_iter().collect();
  15     assert_eq!(tokens.len(), 1, "Unexpected input: {:?}", tokens);
  16     match tokens.pop() {
  17         Some(TokenTree::Ident(ident)) => {
  18             assert_eq!(ident.to_string(), "my_ident");
  19
  20             let proc_macro_crate = TokenStream::from_str("::proc_macro").unwrap();
  21             let quoted_span = proc_macro::quote_span(proc_macro_crate, ident.span());
  22             let prefix = TokenStream::from_str(r#"let mut ident = proc_macro::Ident::new("my_ident", proc_macro::Span::call_site());"#).unwrap();
  23             let set_span_method = TokenStream::from_str("ident.set_span").unwrap();
  24             let set_span_arg = TokenStream::from(TokenTree::Group(Group::new(Delimiter::Parenthesis, quoted_span)));
  25             let suffix = TokenStream::from_str(";proc_macro::TokenStream::from(proc_macro::TokenTree::Ident(ident))").unwrap();
  26             let full_stream: TokenStream = std::array::IntoIter::new([prefix, set_span_method, set_span_arg, suffix]).collect();
  27             full_stream
  28         }
  29         _ => unreachable!()
  30     }
  31 }