深入了解Rust过程宏-2

1extern crate proc_macro;
2use proc_macro::TokenStream;
3
4#[proc_macro]
5pub fn foo(body: TokenStream) -> TokenStream {
6   return body
7}

 1use my_proc_macro::*;
 2 foo! {
 3   fn hello() {
 4       println!("Hello, world!");
 5   }
 6 }
 7
 8fn main() {
 9   hello();
10}

 1extern crate proc_macro;
 2use proc_macro::TokenStream;
 3
 4#[proc_macro_attribute]
 5pub fn baz(
 6   attr: TokenStream,
 7   item: TokenStream
 8) -> TokenStream { 
 9    return item
10}
11…
12use my_proc_macro::*;
13#[baz]
14fn hello() {
15    println!("Hello, world!");
16}
17fn main() {
18    hello();
19}

1struct TokenStream(Vec<TokenTree>);
2enum TokenTree {
3   Ident(Ident),
4   Punct(Punct),
5   Literal(Literal),
6   ...
7}

 1#[proc_macro]
 2pub fn foo(body: TokenStream) -> TokenStream {
 3   for tt in body.into_iter() {
 4       match tt {
 5           TokenTree::Ident(_) => eprintln!("Ident"),
 6           TokenTree::Punct(_) => eprintln!("Punct"),
 7           TokenTree::Literal(_) => eprintln!("Literal"),
 8           _ => {}
 9       }
10   }
11   return TokenStream::new();
12}

1enum TokenTree {
2   Ident(Ident),
3   Punct(Punct),
4   Literal(Literal),
5   Group(Group),
6}

1#[proc_macro]
2pub fn foo(body: TokenStream) -> TokenStream {
3   return [
4       TokenTree::Ident(Ident::new("foo", Span::mixed_site())),
5       TokenTree::Group(Group::new(Delimiter::Parenthesis, body))
6   ].into_iter().collect();
7}

1#[proc_macro]
2pub fn foo(body: String) -> String {// this doesn't work!
3   format!("foo({})", body)
4}