implement NetworkId UserType

1 files changed, 38 insertions, 26 deletions

diff --git a/derive/src/network/list.rs b/derive/src/network/list.rs
index 973444e..1fc8f7e 100644
--- a/derive/src/network/list.rs
+++ b/derive/src/network/list.rs
@@ -43,19 +43,31 @@ pub(crate) fn from(fields: &Vec<NetworkField>) -> Vec<TokenStream> {
         .map(|field| {
             let field_name = field.ident.as_ref().unwrap();
 
-            if field.default {
+            let field_rename = match &field.rename {
+                Some(name) => name.clone(),
+                None => format!("{}", field.ident.as_ref().unwrap()).into(),
+            };
+
+            let field_variant_type = get_field_variant_type(&field);
+
+            let extract_inner = if field.default {
                 quote! {
-                    #field_name: Default::default(),
+                    let mut i = input.iter();
+                    match i.position(|x| *x == libquassel::primitive::Variant::ByteArray(String::from(#field_rename))) {
+                        Some(_) => {
+                            match i.next().expect("failed to get next field") {
+                                libquassel::primitive::Variant::#field_variant_type(var) => var
+                                    .clone()
+                                    .try_into()
+                                    .unwrap_or(Default::default()),
+                                _ => Default::default(),
+                            }
+                        }
+                        None => Default::default(),
+                    }
                 }
             } else {
-                let field_rename = match &field.rename {
-                    Some(name) => name.clone(),
-                    None => format!("{}", field.ident.as_ref().unwrap()).into(),
-                };
-
-                let field_variant_type = get_field_variant_type(&field);
-
-                let extract_inner = quote! {
+                quote! {
                     let mut i = input.iter();
                     i.position(|x| *x == libquassel::primitive::Variant::ByteArray(String::from(#field_rename)))
                         .expect(format!("failed to get field {}", #field_rename).as_str());
@@ -64,25 +76,25 @@ pub(crate) fn from(fields: &Vec<NetworkField>) -> Vec<TokenStream> {
                         libquassel::primitive::Variant::#field_variant_type(var) => var.clone().try_into().unwrap(),
                         _ => panic!("network::list::from: wrong variant type"),
                     }
-                };
+                }
+            };
 
-                match field.network {
-                    super::NetworkRepr::List => quote! {
-                            #field_name: libquassel::message::NetworkList::from_network_list(&mut {
-                                #extract_inner
-                            }),
-                        },
-                    super::NetworkRepr::Map => quote! {
-                            #field_name: libquassel::message::NetworkMap::from_network_map(&mut {
-                                #extract_inner
-                            }),
-                        },
-                    super::NetworkRepr::None => quote! {
-                        #field_name: {
+            match field.network {
+                super::NetworkRepr::List => quote! {
+                        #field_name: libquassel::message::NetworkList::from_network_list(&mut {
                             #extract_inner
-                        },
+                        }),
                     },
-                }
+                super::NetworkRepr::Map => quote! {
+                        #field_name: libquassel::message::NetworkMap::from_network_map(&mut {
+                            #extract_inner
+                        }),
+                    },
+                super::NetworkRepr::None => quote! {
+                    #field_name: {
+                        #extract_inner
+                    },
+                },
             }
         })
         .collect()