add to and from network derive

1 files changed, 127 insertions, 0 deletions

diff --git a/derive/src/from_network_impl.rs b/derive/src/from_network_impl.rs
new file mode 100644
index 0000000..c3a3be2
--- /dev/null
+++ b/derive/src/from_network_impl.rs
@@ -0,0 +1,127 @@
+use proc_macro2::TokenStream;
+use quote::quote;
+
+use crate::{get_field_type, NetworkField};
+
+pub(crate) fn map(fields: &Vec<NetworkField>) -> Vec<TokenStream> {
+    fields
+        .iter()
+        .map(|field| {
+            let field_rename = match &field.rename {
+                Some(name) => name.clone(),
+                None => format!("{}", field.ident.as_ref().unwrap()).into(),
+            };
+
+            let field_name = field.ident.as_ref().unwrap();
+
+            let field_type = get_field_type(&field);
+
+            if field.from_map.is_some() {
+                let field_map: syn::ExprClosure =
+                    syn::parse_str(&field.from_map.as_ref().unwrap()).unwrap();
+
+                quote! {
+                    #field_name: match &input.get(#field_rename).unwrap() {
+                            crate::primitive::Variant::#field_type(input) => input.iter().map(#field_map).collect(),
+                            _ => unimplemented!()
+                        },
+                }
+            } else {
+                quote! {
+                    #field_name: match &input.get(#field_rename).unwrap() {
+                            crate::primitive::Variant::#field_type(input) => input.clone(),
+                            _ => unimplemented!()
+                        },
+                }
+            }
+        })
+        .collect()
+}
+
+pub(crate) fn map_list(fields: &Vec<NetworkField>) -> Vec<TokenStream> {
+    fields
+        .iter()
+        .map(|field| {
+            let field_rename = match &field.rename {
+                Some(name) => name.clone(),
+                None => format!("{}", field.ident.as_ref().unwrap()).into(),
+            };
+
+            let field_name = field.ident.as_ref().unwrap();
+
+            let field_type = get_field_type(&field);
+
+            if field.from_map.is_some() {
+                let field_map: syn::ExprClosure =
+                    syn::parse_str(&field.from_map.as_ref().unwrap()).unwrap();
+
+                quote! {
+                    #field_name: match input.get(#field_rename).unwrap() {
+                        crate::primitive::Variant::VariantList(input) => match &input.remove(0) {
+                            crate::primitive::Variant::#field_type(input) => input.iter().map(#field_map).collect(),
+                            _ => unimplemented!()
+                        },
+                        _ => unimplemented!()
+                    },
+                }
+            } else {
+                quote! {
+                    #field_name: match input.get(#field_rename).unwrap() {
+                        crate::primitive::Variant::VariantList(input) => match &input.remove(0) {
+                            crate::primitive::Variant::#field_type(input) => input.clone(),
+                            _ => unimplemented!()
+                        },
+                        _ => unimplemented!()
+                    },
+                }
+            }
+        })
+        .collect()
+}
+
+pub(crate) fn list(fields: &Vec<NetworkField>) -> Vec<TokenStream> {
+    fields
+        .iter()
+        .map(|field| {
+            let field_rename = match &field.rename {
+                Some(name) => name.clone(),
+                None => format!("{}", field.ident.as_ref().unwrap()).into(),
+            };
+
+            let field_name = field.ident.as_ref().unwrap();
+
+            let field_type = get_field_type(&field);
+
+            if field.from_map.is_some() {
+                let field_map: syn::ExprClosure =
+                    syn::parse_str(&field.from_map.as_ref().unwrap()).unwrap();
+
+                quote! {
+                    #field_name: match_variant!(
+                        input
+                            .iter()
+                            .nth(input
+                                 .iter()
+                                 .position(|x| *x == crate::primitive::Variant::ByteArray(#field_rename.to_string()))
+                                 .unwrap())
+                            .unwrap(),
+                        crate::primitive::Variant::#field_type
+                    ).iter().map(#field_map).collect(),
+                }
+            } else {
+                quote! {
+                    #field_name: match_variant!(
+                        input
+                            .iter()
+                            .nth(input
+                                 .iter()
+                                 .position(|x| *x == Variant::ByteArray(#field_rename.to_string()))
+                                 .unwrap() + 1)
+                            .unwrap(),
+                        crate::primitive::Variant::#field_type
+                    ),
+                }
+            }
+        })
+        .collect()
+}