diff options
Diffstat (limited to 'vendor/derive_more/src/add_helpers.rs')
-rw-r--r-- | vendor/derive_more/src/add_helpers.rs | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/vendor/derive_more/src/add_helpers.rs b/vendor/derive_more/src/add_helpers.rs new file mode 100644 index 000000000..34f3d2ad5 --- /dev/null +++ b/vendor/derive_more/src/add_helpers.rs @@ -0,0 +1,28 @@ +use proc_macro2::TokenStream; +use quote::quote; +use syn::{Field, Ident, Index}; + +pub fn tuple_exprs(fields: &[&Field], method_ident: &Ident) -> Vec<TokenStream> { + let mut exprs = vec![]; + + for i in 0..fields.len() { + let i = Index::from(i); + // generates `self.0.add(rhs.0)` + let expr = quote!(self.#i.#method_ident(rhs.#i)); + exprs.push(expr); + } + exprs +} + +pub fn struct_exprs(fields: &[&Field], method_ident: &Ident) -> Vec<TokenStream> { + let mut exprs = vec![]; + + for field in fields { + // It's safe to unwrap because struct fields always have an identifier + let field_id = field.ident.as_ref().unwrap(); + // generates `x: self.x.add(rhs.x)` + let expr = quote!(self.#field_id.#method_ident(rhs.#field_id)); + exprs.push(expr) + } + exprs +} |