| 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 | /// |
| 4 | /// [`quote!`]: https://docs.rs/quote/1.0/quote/index.html |
| 5 | /// |
| 6 | /// The return type can be any syntax tree node that implements the [`Parse`] |
| 7 | /// trait. |
| 8 | /// |
| 9 | /// [`Parse`]: crate::parse::Parse |
| 10 | /// |
| 11 | /// ``` |
| 12 | /// use quote::quote; |
| 13 | /// use syn::{parse_quote, Stmt}; |
| 14 | /// |
| 15 | /// fn main() { |
| 16 | /// let name = quote!(v); |
| 17 | /// let ty = quote!(u8); |
| 18 | /// |
| 19 | /// let stmt: Stmt = parse_quote! { |
| 20 | /// let #name: #ty = Default::default(); |
| 21 | /// }; |
| 22 | /// |
| 23 | /// println!("{:#?}" , stmt); |
| 24 | /// } |
| 25 | /// ``` |
| 26 | /// |
| 27 | /// *This macro is available only if Syn is built with both the `"parsing"` and |
| 28 | /// `"printing"` features.* |
| 29 | /// |
| 30 | /// # Example |
| 31 | /// |
| 32 | /// The following helper function adds a bound `T: HeapSize` to every type |
| 33 | /// parameter `T` in the input generics. |
| 34 | /// |
| 35 | /// ``` |
| 36 | /// use syn::{parse_quote, Generics, GenericParam}; |
| 37 | /// |
| 38 | /// // Add a bound `T: HeapSize` to every type parameter T. |
| 39 | /// fn add_trait_bounds(mut generics: Generics) -> Generics { |
| 40 | /// for param in &mut generics.params { |
| 41 | /// if let GenericParam::Type(type_param) = param { |
| 42 | /// type_param.bounds.push(parse_quote!(HeapSize)); |
| 43 | /// } |
| 44 | /// } |
| 45 | /// generics |
| 46 | /// } |
| 47 | /// ``` |
| 48 | /// |
| 49 | /// # Special cases |
| 50 | /// |
| 51 | /// This macro can parse the following additional types as a special case even |
| 52 | /// though they do not implement the `Parse` trait. |
| 53 | /// |
| 54 | /// - [`Attribute`] — parses one attribute, allowing either outer like `#[...]` |
| 55 | /// or inner like `#![...]` |
| 56 | /// - [`Vec<Attribute>`] — parses multiple attributes, including mixed kinds in |
| 57 | /// any order |
| 58 | /// - [`Punctuated<T, P>`] — parses zero or more `T` separated by punctuation |
| 59 | /// `P` with optional trailing punctuation |
| 60 | /// - [`Vec<Arm>`] — parses arms separated by optional commas according to the |
| 61 | /// same grammar as the inside of a `match` expression |
| 62 | /// - [`Vec<Stmt>`] — parses the same as `Block::parse_within` |
| 63 | /// - [`Pat`], [`Box<Pat>`] — parses the same as |
| 64 | /// `Pat::parse_multi_with_leading_vert` |
| 65 | /// - [`Field`] — parses a named or unnamed struct field |
| 66 | /// |
| 67 | /// [`Vec<Attribute>`]: Attribute |
| 68 | /// [`Vec<Arm>`]: Arm |
| 69 | /// [`Vec<Stmt>`]: Block::parse_within |
| 70 | /// [`Pat`]: Pat::parse_multi_with_leading_vert |
| 71 | /// [`Box<Pat>`]: Pat::parse_multi_with_leading_vert |
| 72 | /// |
| 73 | /// # Panics |
| 74 | /// |
| 75 | /// Panics if the tokens fail to parse as the expected syntax tree type. The |
| 76 | /// caller is responsible for ensuring that the input tokens are syntactically |
| 77 | /// valid. |
| 78 | #[cfg_attr (docsrs, doc(cfg(all(feature = "parsing" , feature = "printing" ))))] |
| 79 | #[macro_export ] |
| 80 | macro_rules! parse_quote { |
| 81 | ($($tt:tt)*) => { |
| 82 | $crate::__private::parse_quote($crate::__private::quote::quote!($($tt)*)) |
| 83 | }; |
| 84 | } |
| 85 | |
| 86 | /// This macro is [`parse_quote!`] + [`quote_spanned!`][quote::quote_spanned]. |
| 87 | /// |
| 88 | /// Please refer to each of their documentation. |
| 89 | /// |
| 90 | /// # Example |
| 91 | /// |
| 92 | /// ``` |
| 93 | /// use quote::{quote, quote_spanned}; |
| 94 | /// use syn::spanned::Spanned; |
| 95 | /// use syn::{parse_quote_spanned, ReturnType, Signature}; |
| 96 | /// |
| 97 | /// // Changes `fn()` to `fn() -> Pin<Box<dyn Future<Output = ()>>>`, |
| 98 | /// // and `fn() -> T` to `fn() -> Pin<Box<dyn Future<Output = T>>>`, |
| 99 | /// // without introducing any call_site() spans. |
| 100 | /// fn make_ret_pinned_future(sig: &mut Signature) { |
| 101 | /// let ret = match &sig.output { |
| 102 | /// ReturnType::Default => quote_spanned!(sig.paren_token.span=> ()), |
| 103 | /// ReturnType::Type(_, ret) => quote!(#ret), |
| 104 | /// }; |
| 105 | /// sig.output = parse_quote_spanned! {ret.span()=> |
| 106 | /// -> ::std::pin::Pin<::std::boxed::Box<dyn ::std::future::Future<Output = #ret>>> |
| 107 | /// }; |
| 108 | /// } |
| 109 | /// ``` |
| 110 | #[cfg_attr (docsrs, doc(cfg(all(feature = "parsing" , feature = "printing" ))))] |
| 111 | #[macro_export ] |
| 112 | macro_rules! parse_quote_spanned { |
| 113 | ($span:expr=> $($tt:tt)*) => { |
| 114 | $crate::__private::parse_quote($crate::__private::quote::quote_spanned!($span=> $($tt)*)) |
| 115 | }; |
| 116 | } |
| 117 | |
| 118 | //////////////////////////////////////////////////////////////////////////////// |
| 119 | // Can parse any type that implements Parse. |
| 120 | |
| 121 | use crate::error::Result; |
| 122 | use crate::parse::{Parse, ParseStream, Parser}; |
| 123 | use proc_macro2::TokenStream; |
| 124 | |
| 125 | // Not public API. |
| 126 | #[doc (hidden)] |
| 127 | #[track_caller ] |
| 128 | pub fn parse<T: ParseQuote>(token_stream: TokenStream) -> T { |
| 129 | let parser: fn parse(&ParseBuffer<'_>) -> … = T::parse; |
| 130 | match parser.parse2(tokens:token_stream) { |
| 131 | Ok(t: T) => t, |
| 132 | Err(err: Error) => panic!("{}" , err), |
| 133 | } |
| 134 | } |
| 135 | |
| 136 | #[doc (hidden)] |
| 137 | pub trait ParseQuote: Sized { |
| 138 | fn parse(input: ParseStream) -> Result<Self>; |
| 139 | } |
| 140 | |
| 141 | impl<T: Parse> ParseQuote for T { |
| 142 | fn parse(input: ParseStream) -> Result<Self> { |
| 143 | <T as Parse>::parse(input) |
| 144 | } |
| 145 | } |
| 146 | |
| 147 | //////////////////////////////////////////////////////////////////////////////// |
| 148 | // Any other types that we want `parse_quote!` to be able to parse. |
| 149 | |
| 150 | use crate::punctuated::Punctuated; |
| 151 | #[cfg (any(feature = "full" , feature = "derive" ))] |
| 152 | use crate::{attr, Attribute, Field, FieldMutability, Ident, Type, Visibility}; |
| 153 | #[cfg (feature = "full" )] |
| 154 | use crate::{Arm, Block, Pat, Stmt}; |
| 155 | |
| 156 | #[cfg (any(feature = "full" , feature = "derive" ))] |
| 157 | impl ParseQuote for Attribute { |
| 158 | fn parse(input: ParseStream) -> Result<Self> { |
| 159 | if input.peek(Token![#]) && input.peek2(Token![!]) { |
| 160 | attr::parsing::single_parse_inner(input) |
| 161 | } else { |
| 162 | attr::parsing::single_parse_outer(input) |
| 163 | } |
| 164 | } |
| 165 | } |
| 166 | |
| 167 | #[cfg (any(feature = "full" , feature = "derive" ))] |
| 168 | impl ParseQuote for Vec<Attribute> { |
| 169 | fn parse(input: ParseStream) -> Result<Self> { |
| 170 | let mut attrs: Vec = Vec::new(); |
| 171 | while !input.is_empty() { |
| 172 | attrs.push(ParseQuote::parse(input)?); |
| 173 | } |
| 174 | Ok(attrs) |
| 175 | } |
| 176 | } |
| 177 | |
| 178 | #[cfg (any(feature = "full" , feature = "derive" ))] |
| 179 | impl ParseQuote for Field { |
| 180 | fn parse(input: ParseStream) -> Result<Self> { |
| 181 | let attrs = input.call(Attribute::parse_outer)?; |
| 182 | let vis: Visibility = input.parse()?; |
| 183 | |
| 184 | let ident: Option<Ident>; |
| 185 | let colon_token: Option<Token![:]>; |
| 186 | let is_named = input.peek(Ident) && input.peek2(Token![:]) && !input.peek2(Token![::]); |
| 187 | if is_named { |
| 188 | ident = Some(input.parse()?); |
| 189 | colon_token = Some(input.parse()?); |
| 190 | } else { |
| 191 | ident = None; |
| 192 | colon_token = None; |
| 193 | } |
| 194 | |
| 195 | let ty: Type = input.parse()?; |
| 196 | |
| 197 | Ok(Field { |
| 198 | attrs, |
| 199 | vis, |
| 200 | mutability: FieldMutability::None, |
| 201 | ident, |
| 202 | colon_token, |
| 203 | ty, |
| 204 | }) |
| 205 | } |
| 206 | } |
| 207 | |
| 208 | #[cfg (feature = "full" )] |
| 209 | impl ParseQuote for Pat { |
| 210 | fn parse(input: ParseStream) -> Result<Self> { |
| 211 | Pat::parse_multi_with_leading_vert(input) |
| 212 | } |
| 213 | } |
| 214 | |
| 215 | #[cfg (feature = "full" )] |
| 216 | impl ParseQuote for Box<Pat> { |
| 217 | fn parse(input: ParseStream) -> Result<Self> { |
| 218 | <Pat as ParseQuote>::parse(input).map(op:Box::new) |
| 219 | } |
| 220 | } |
| 221 | |
| 222 | impl<T: Parse, P: Parse> ParseQuote for Punctuated<T, P> { |
| 223 | fn parse(input: ParseStream) -> Result<Self> { |
| 224 | Self::parse_terminated(input) |
| 225 | } |
| 226 | } |
| 227 | |
| 228 | #[cfg (feature = "full" )] |
| 229 | impl ParseQuote for Vec<Stmt> { |
| 230 | fn parse(input: ParseStream) -> Result<Self> { |
| 231 | Block::parse_within(input) |
| 232 | } |
| 233 | } |
| 234 | |
| 235 | #[cfg (feature = "full" )] |
| 236 | impl ParseQuote for Vec<Arm> { |
| 237 | fn parse(input: ParseStream) -> Result<Self> { |
| 238 | Arm::parse_multiple(input) |
| 239 | } |
| 240 | } |
| 241 | |