]>
Commit | Line | Data |
---|---|---|
a1dfa0c6 | 1 | // force-host |
83c7162d XL |
2 | // no-prefer-dynamic |
3 | ||
dc9dc135 XL |
4 | // Proc macros commonly used by tests. |
5 | // `panic`/`print` -> `panic_bang`/`print_bang` to avoid conflicts with standard macros. | |
6 | ||
83c7162d | 7 | #![crate_type = "proc-macro"] |
83c7162d XL |
8 | |
9 | extern crate proc_macro; | |
6a06907d | 10 | use proc_macro::{TokenStream, TokenTree}; |
83c7162d | 11 | |
dc9dc135 XL |
12 | // Macro that return empty token stream. |
13 | ||
14 | #[proc_macro] | |
15 | pub fn empty(_: TokenStream) -> TokenStream { | |
16 | TokenStream::new() | |
17 | } | |
18 | ||
83c7162d | 19 | #[proc_macro_attribute] |
dc9dc135 XL |
20 | pub fn empty_attr(_: TokenStream, _: TokenStream) -> TokenStream { |
21 | TokenStream::new() | |
22 | } | |
23 | ||
24 | #[proc_macro_derive(Empty, attributes(empty_helper))] | |
25 | pub fn empty_derive(_: TokenStream) -> TokenStream { | |
26 | TokenStream::new() | |
27 | } | |
28 | ||
29 | // Macro that panics. | |
30 | ||
31 | #[proc_macro] | |
32 | pub fn panic_bang(_: TokenStream) -> TokenStream { | |
33 | panic!("panic-bang"); | |
83c7162d XL |
34 | } |
35 | ||
36 | #[proc_macro_attribute] | |
dc9dc135 XL |
37 | pub fn panic_attr(_: TokenStream, _: TokenStream) -> TokenStream { |
38 | panic!("panic-attr"); | |
39 | } | |
40 | ||
41 | #[proc_macro_derive(Panic, attributes(panic_helper))] | |
42 | pub fn panic_derive(_: TokenStream) -> TokenStream { | |
43 | panic!("panic-derive"); | |
83c7162d XL |
44 | } |
45 | ||
dc9dc135 XL |
46 | // Macros that return the input stream. |
47 | ||
83c7162d | 48 | #[proc_macro] |
dc9dc135 | 49 | pub fn identity(input: TokenStream) -> TokenStream { |
83c7162d XL |
50 | input |
51 | } | |
dc9dc135 XL |
52 | |
53 | #[proc_macro_attribute] | |
54 | pub fn identity_attr(_: TokenStream, input: TokenStream) -> TokenStream { | |
55 | input | |
56 | } | |
57 | ||
58 | #[proc_macro_derive(Identity, attributes(identity_helper))] | |
59 | pub fn identity_derive(input: TokenStream) -> TokenStream { | |
60 | input | |
61 | } | |
62 | ||
63 | // Macros that iterate and re-collect the input stream. | |
64 | ||
65 | #[proc_macro] | |
66 | pub fn recollect(input: TokenStream) -> TokenStream { | |
67 | input.into_iter().collect() | |
68 | } | |
69 | ||
70 | #[proc_macro_attribute] | |
71 | pub fn recollect_attr(_: TokenStream, input: TokenStream) -> TokenStream { | |
72 | input.into_iter().collect() | |
73 | } | |
74 | ||
75 | #[proc_macro_derive(Recollect, attributes(recollect_helper))] | |
76 | pub fn recollect_derive(input: TokenStream) -> TokenStream { | |
77 | input.into_iter().collect() | |
78 | } | |
79 | ||
80 | // Macros that print their input in the original and re-collected forms (if they differ). | |
81 | ||
82 | fn print_helper(input: TokenStream, kind: &str) -> TokenStream { | |
6a06907d XL |
83 | print_helper_ext(input, kind, true) |
84 | } | |
85 | ||
86 | fn print_helper_ext(input: TokenStream, kind: &str, debug: bool) -> TokenStream { | |
dc9dc135 XL |
87 | let input_display = format!("{}", input); |
88 | let input_debug = format!("{:#?}", input); | |
89 | let recollected = input.into_iter().collect(); | |
90 | let recollected_display = format!("{}", recollected); | |
91 | let recollected_debug = format!("{:#?}", recollected); | |
92 | println!("PRINT-{} INPUT (DISPLAY): {}", kind, input_display); | |
93 | if recollected_display != input_display { | |
94 | println!("PRINT-{} RE-COLLECTED (DISPLAY): {}", kind, recollected_display); | |
95 | } | |
6a06907d XL |
96 | if debug { |
97 | println!("PRINT-{} INPUT (DEBUG): {}", kind, input_debug); | |
98 | if recollected_debug != input_debug { | |
99 | println!("PRINT-{} RE-COLLECTED (DEBUG): {}", kind, recollected_debug); | |
100 | } | |
dc9dc135 XL |
101 | } |
102 | recollected | |
103 | } | |
104 | ||
105 | #[proc_macro] | |
106 | pub fn print_bang(input: TokenStream) -> TokenStream { | |
107 | print_helper(input, "BANG") | |
108 | } | |
109 | ||
f035d41b XL |
110 | #[proc_macro] |
111 | pub fn print_bang_consume(input: TokenStream) -> TokenStream { | |
112 | print_helper(input, "BANG"); | |
113 | TokenStream::new() | |
114 | } | |
115 | ||
dc9dc135 | 116 | #[proc_macro_attribute] |
6a06907d XL |
117 | pub fn print_attr(args: TokenStream, input: TokenStream) -> TokenStream { |
118 | let debug = match &args.into_iter().collect::<Vec<_>>()[..] { | |
119 | [TokenTree::Ident(ident)] if ident.to_string() == "nodebug" => false, | |
120 | _ => true, | |
121 | }; | |
122 | print_helper_ext(input, "ATTR", debug) | |
dc9dc135 XL |
123 | } |
124 | ||
29967ef6 XL |
125 | #[proc_macro_attribute] |
126 | pub fn print_attr_args(args: TokenStream, input: TokenStream) -> TokenStream { | |
127 | print_helper(args, "ATTR_ARGS"); | |
128 | input | |
129 | } | |
130 | ||
6a06907d XL |
131 | #[proc_macro_attribute] |
132 | pub fn print_target_and_args(args: TokenStream, input: TokenStream) -> TokenStream { | |
133 | print_helper(args, "ATTR_ARGS"); | |
134 | print_helper(input.clone(), "ATTR"); | |
135 | input | |
136 | } | |
137 | ||
138 | #[proc_macro_attribute] | |
139 | pub fn print_target_and_args_consume(args: TokenStream, input: TokenStream) -> TokenStream { | |
140 | print_helper(args, "ATTR_ARGS"); | |
141 | print_helper(input.clone(), "ATTR"); | |
142 | TokenStream::new() | |
143 | } | |
144 | ||
dc9dc135 XL |
145 | #[proc_macro_derive(Print, attributes(print_helper))] |
146 | pub fn print_derive(input: TokenStream) -> TokenStream { | |
f9f354fc XL |
147 | print_helper(input, "DERIVE"); |
148 | TokenStream::new() | |
dc9dc135 | 149 | } |