David Tolnay | e83ef5a | 2018-01-11 15:18:36 -0800 | [diff] [blame] | 1 | /// Quasi-quotation macro that accepts input like the [`quote!`] macro but uses |
| 2 | /// type inference to figure out a return type for those tokens. |
| 3 | /// |
David Tolnay | 442cf76 | 2018-08-31 11:08:24 -0700 | [diff] [blame] | 4 | /// [`quote!`]: https://docs.rs/quote/0.6/quote/index.html |
David Tolnay | e83ef5a | 2018-01-11 15:18:36 -0800 | [diff] [blame] | 5 | /// |
David Tolnay | d85a881 | 2018-08-31 11:09:57 -0700 | [diff] [blame] | 6 | /// The return type can be any syntax tree node that implements the [`Parse`] |
David Tolnay | e83ef5a | 2018-01-11 15:18:36 -0800 | [diff] [blame] | 7 | /// trait. |
| 8 | /// |
David Tolnay | d85a881 | 2018-08-31 11:09:57 -0700 | [diff] [blame] | 9 | /// [`Parse`]: parse/trait.Parse.html |
David Tolnay | e83ef5a | 2018-01-11 15:18:36 -0800 | [diff] [blame] | 10 | /// |
David Tolnay | 95989db | 2019-01-01 15:05:57 -0500 | [diff] [blame^] | 11 | /// ```edition2018 |
David Tolnay | fd5b117 | 2018-12-31 17:54:36 -0500 | [diff] [blame] | 12 | /// # extern crate quote; |
| 13 | /// # |
| 14 | /// use quote::quote; |
| 15 | /// use syn::{parse_quote, Stmt}; |
David Tolnay | e83ef5a | 2018-01-11 15:18:36 -0800 | [diff] [blame] | 16 | /// |
| 17 | /// fn main() { |
| 18 | /// let name = quote!(v); |
| 19 | /// let ty = quote!(u8); |
| 20 | /// |
| 21 | /// let stmt: Stmt = parse_quote! { |
| 22 | /// let #name: #ty = Default::default(); |
| 23 | /// }; |
| 24 | /// |
| 25 | /// println!("{:#?}", stmt); |
| 26 | /// } |
| 27 | /// ``` |
| 28 | /// |
David Tolnay | 491680a | 2018-01-23 00:34:40 -0800 | [diff] [blame] | 29 | /// *This macro is available if Syn is built with the `"parsing"` feature, |
| 30 | /// although interpolation of syntax tree nodes into the quoted tokens is only |
| 31 | /// supported if Syn is built with the `"printing"` feature as well.* |
David Tolnay | e83ef5a | 2018-01-11 15:18:36 -0800 | [diff] [blame] | 32 | /// |
| 33 | /// # Example |
| 34 | /// |
| 35 | /// The following helper function adds a bound `T: HeapSize` to every type |
| 36 | /// parameter `T` in the input generics. |
| 37 | /// |
David Tolnay | 95989db | 2019-01-01 15:05:57 -0500 | [diff] [blame^] | 38 | /// ```edition2018 |
David Tolnay | fd5b117 | 2018-12-31 17:54:36 -0500 | [diff] [blame] | 39 | /// use syn::{parse_quote, Generics, GenericParam}; |
David Tolnay | 9b00f65 | 2018-09-01 10:31:02 -0700 | [diff] [blame] | 40 | /// |
David Tolnay | e83ef5a | 2018-01-11 15:18:36 -0800 | [diff] [blame] | 41 | /// // Add a bound `T: HeapSize` to every type parameter T. |
| 42 | /// fn add_trait_bounds(mut generics: Generics) -> Generics { |
| 43 | /// for param in &mut generics.params { |
| 44 | /// if let GenericParam::Type(ref mut type_param) = *param { |
| 45 | /// type_param.bounds.push(parse_quote!(HeapSize)); |
| 46 | /// } |
| 47 | /// } |
| 48 | /// generics |
| 49 | /// } |
David Tolnay | e83ef5a | 2018-01-11 15:18:36 -0800 | [diff] [blame] | 50 | /// ``` |
| 51 | /// |
| 52 | /// # Special cases |
| 53 | /// |
| 54 | /// This macro can parse the following additional types as a special case even |
David Tolnay | d85a881 | 2018-08-31 11:09:57 -0700 | [diff] [blame] | 55 | /// though they do not implement the `Parse` trait. |
David Tolnay | e83ef5a | 2018-01-11 15:18:36 -0800 | [diff] [blame] | 56 | /// |
| 57 | /// - [`Attribute`] — parses one attribute, allowing either outer like `#[...]` |
| 58 | /// or inner like `#![...]` |
| 59 | /// - [`Punctuated<T, P>`] — parses zero or more `T` separated by punctuation |
| 60 | /// `P` with optional trailing punctuation |
| 61 | /// |
| 62 | /// [`Attribute`]: struct.Attribute.html |
| 63 | /// [`Punctuated<T, P>`]: punctuated/struct.Punctuated.html |
| 64 | /// |
| 65 | /// # Panics |
| 66 | /// |
| 67 | /// Panics if the tokens fail to parse as the expected syntax tree type. The |
| 68 | /// caller is responsible for ensuring that the input tokens are syntactically |
| 69 | /// valid. |
David Tolnay | e0ba920 | 2018-10-06 20:09:08 -0700 | [diff] [blame] | 70 | #[macro_export(local_inner_macros)] |
David Tolnay | e83ef5a | 2018-01-11 15:18:36 -0800 | [diff] [blame] | 71 | macro_rules! parse_quote { |
| 72 | ($($tt:tt)*) => { |
David Tolnay | e0ba920 | 2018-10-06 20:09:08 -0700 | [diff] [blame] | 73 | $crate::parse_quote::parse($crate::export::From::from(quote_impl!($($tt)*))) |
| 74 | }; |
| 75 | } |
| 76 | |
| 77 | #[cfg(not(syn_can_call_macro_by_path))] |
| 78 | #[doc(hidden)] |
| 79 | #[macro_export] |
| 80 | macro_rules! quote_impl { |
| 81 | ($($tt:tt)*) => { |
| 82 | // Require caller to have their own `#[macro_use] extern crate quote`. |
| 83 | quote!($($tt)*) |
| 84 | }; |
| 85 | } |
| 86 | |
| 87 | #[cfg(syn_can_call_macro_by_path)] |
| 88 | #[doc(hidden)] |
| 89 | #[macro_export] |
| 90 | macro_rules! quote_impl { |
| 91 | ($($tt:tt)*) => { |
| 92 | $crate::export::quote::quote!($($tt)*) |
David Tolnay | e83ef5a | 2018-01-11 15:18:36 -0800 | [diff] [blame] | 93 | }; |
| 94 | } |
| 95 | |
| 96 | //////////////////////////////////////////////////////////////////////////////// |
David Tolnay | 06faaca | 2018-09-01 20:25:04 -0700 | [diff] [blame] | 97 | // Can parse any type that implements Parse. |
David Tolnay | e83ef5a | 2018-01-11 15:18:36 -0800 | [diff] [blame] | 98 | |
David Tolnay | 4ac232d | 2018-08-31 10:18:03 -0700 | [diff] [blame] | 99 | use parse::{Parse, ParseStream, Parser, Result}; |
David Tolnay | e82a2b1 | 2018-08-30 16:31:10 -0700 | [diff] [blame] | 100 | use proc_macro2::TokenStream; |
David Tolnay | e83ef5a | 2018-01-11 15:18:36 -0800 | [diff] [blame] | 101 | |
| 102 | // Not public API. |
| 103 | #[doc(hidden)] |
David Tolnay | 7d6a7b8 | 2018-01-23 00:42:22 -0800 | [diff] [blame] | 104 | pub fn parse<T: ParseQuote>(token_stream: TokenStream) -> T { |
David Tolnay | e83ef5a | 2018-01-11 15:18:36 -0800 | [diff] [blame] | 105 | let parser = T::parse; |
| 106 | match parser.parse2(token_stream) { |
| 107 | Ok(t) => t, |
David Tolnay | 40c0611 | 2018-08-30 16:14:50 -0700 | [diff] [blame] | 108 | Err(err) => panic!("{}", err), |
David Tolnay | e83ef5a | 2018-01-11 15:18:36 -0800 | [diff] [blame] | 109 | } |
| 110 | } |
| 111 | |
| 112 | // Not public API. |
| 113 | #[doc(hidden)] |
| 114 | pub trait ParseQuote: Sized { |
David Tolnay | 40c0611 | 2018-08-30 16:14:50 -0700 | [diff] [blame] | 115 | fn parse(input: ParseStream) -> Result<Self>; |
David Tolnay | e83ef5a | 2018-01-11 15:18:36 -0800 | [diff] [blame] | 116 | } |
| 117 | |
David Tolnay | 40c0611 | 2018-08-30 16:14:50 -0700 | [diff] [blame] | 118 | impl<T: Parse> ParseQuote for T { |
| 119 | fn parse(input: ParseStream) -> Result<Self> { |
| 120 | <T as Parse>::parse(input) |
David Tolnay | e83ef5a | 2018-01-11 15:18:36 -0800 | [diff] [blame] | 121 | } |
| 122 | } |
| 123 | |
| 124 | //////////////////////////////////////////////////////////////////////////////// |
| 125 | // Any other types that we want `parse_quote!` to be able to parse. |
| 126 | |
David Tolnay | 70f30e9 | 2018-09-01 02:04:17 -0700 | [diff] [blame] | 127 | use punctuated::Punctuated; |
David Tolnay | e83ef5a | 2018-01-11 15:18:36 -0800 | [diff] [blame] | 128 | #[cfg(any(feature = "full", feature = "derive"))] |
David Tolnay | 40c0611 | 2018-08-30 16:14:50 -0700 | [diff] [blame] | 129 | use {attr, Attribute}; |
David Tolnay | e83ef5a | 2018-01-11 15:18:36 -0800 | [diff] [blame] | 130 | |
David Tolnay | e83ef5a | 2018-01-11 15:18:36 -0800 | [diff] [blame] | 131 | #[cfg(any(feature = "full", feature = "derive"))] |
| 132 | impl ParseQuote for Attribute { |
David Tolnay | 40c0611 | 2018-08-30 16:14:50 -0700 | [diff] [blame] | 133 | fn parse(input: ParseStream) -> Result<Self> { |
| 134 | if input.peek(Token![#]) && input.peek2(Token![!]) { |
| 135 | attr::parsing::single_parse_inner(input) |
| 136 | } else { |
| 137 | attr::parsing::single_parse_outer(input) |
| 138 | } |
David Tolnay | e83ef5a | 2018-01-11 15:18:36 -0800 | [diff] [blame] | 139 | } |
| 140 | } |
David Tolnay | 024b576 | 2018-08-31 11:11:53 -0700 | [diff] [blame] | 141 | |
| 142 | impl<T: Parse, P: Parse> ParseQuote for Punctuated<T, P> { |
| 143 | fn parse(input: ParseStream) -> Result<Self> { |
| 144 | Self::parse_terminated(input) |
| 145 | } |
| 146 | } |