| 1 | //! Utility functions. |
| 2 | |
| 3 | use proc_macro2::Span; |
| 4 | use syn::{ |
| 5 | punctuated::Punctuated, spanned::Spanned, Ident, Meta, MetaList, Path, PathArguments, |
| 6 | PathSegment, Result, Token, |
| 7 | }; |
| 8 | |
| 9 | use crate::error::Error; |
| 10 | |
| 11 | /// Convenience type to return two possible values. |
| 12 | pub enum Either<L, R> { |
| 13 | /// `L` return value. |
| 14 | Left(L), |
| 15 | /// `R` return value. |
| 16 | Right(R), |
| 17 | } |
| 18 | |
| 19 | /// Create [`PathSegment`] from [`str`]. |
| 20 | pub fn path_segment(ident: &str) -> PathSegment { |
| 21 | PathSegment { |
| 22 | ident: Ident::new(string:ident, Span::call_site()), |
| 23 | arguments: PathArguments::None, |
| 24 | } |
| 25 | } |
| 26 | |
| 27 | /// Create [`Path`] from `[&str]`s. |
| 28 | pub fn path_from_strs(segments: &[&str]) -> Path { |
| 29 | Path { |
| 30 | leading_colon: Some(<Token![::]>::default()), |
| 31 | segments: Punctuated::from_iter(segments.iter().map(|segment: &&str| path_segment(ident:segment))), |
| 32 | } |
| 33 | } |
| 34 | |
| 35 | /// Create [`Path`] from `[&Ident]`s. |
| 36 | pub fn path_from_idents(segments: &[&Ident]) -> Path { |
| 37 | Path { |
| 38 | leading_colon: None, |
| 39 | segments: Punctuated::from_iter(segments.iter().map(|ident: &&Ident| PathSegment { |
| 40 | ident: (*ident).clone(), |
| 41 | arguments: PathArguments::None, |
| 42 | })), |
| 43 | } |
| 44 | } |
| 45 | |
| 46 | /// Create [`Path`] from a root [`Path`] and `[&str]`s. |
| 47 | pub fn path_from_root_and_strs(root: Path, segments: &[&str]) -> Path { |
| 48 | Path { |
| 49 | leading_colon: root.leading_colon, |
| 50 | segments: rootimpl Iterator |
| 51 | .segments |
| 52 | .into_iter() |
| 53 | .chain(segments.iter().map(|segment: &&str| path_segment(ident:segment))) |
| 54 | .collect(), |
| 55 | } |
| 56 | } |
| 57 | |
| 58 | /// Extension for [`MetaList`]. |
| 59 | pub trait MetaListExt { |
| 60 | /// Shorthand for parsing a [`MetaList`] into a list of [`Meta`]s. |
| 61 | fn parse_non_empty_nested_metas(&self) -> Result<Punctuated<Meta, Token![,]>>; |
| 62 | } |
| 63 | |
| 64 | impl MetaListExt for MetaList { |
| 65 | fn parse_non_empty_nested_metas(&self) -> Result<Punctuated<Meta, Token![,]>> { |
| 66 | let list: Punctuated = self.parse_args_with(parser:Punctuated::<Meta, Token![,]>::parse_terminated)?; |
| 67 | |
| 68 | if list.is_empty() { |
| 69 | return Err(Error::option_empty(self.span())); |
| 70 | } |
| 71 | |
| 72 | Ok(list) |
| 73 | } |
| 74 | } |
| 75 | |