diff options
| author | Max Audron <audron@cocaine.farm> | 2025-02-26 22:52:16 +0100 |
|---|---|---|
| committer | Max Audron <audron@cocaine.farm> | 2025-02-26 22:58:03 +0100 |
| commit | a130eee7a71996f9ae50502c477a2c6494b78a83 (patch) | |
| tree | 39d5e659ad1d537728757aa542968cb49b06bd35 /derive/src/network/map.rs | |
| parent | implement IdentityId (diff) | |
implement NetworkId UserType
Diffstat (limited to 'derive/src/network/map.rs')
| -rw-r--r-- | derive/src/network/map.rs | 42 |
1 files changed, 21 insertions, 21 deletions
diff --git a/derive/src/network/map.rs b/derive/src/network/map.rs index 0550ccb..4efc3bf 100644 --- a/derive/src/network/map.rs +++ b/derive/src/network/map.rs @@ -45,29 +45,29 @@ pub(crate) fn from(fields: &Vec<NetworkField>) -> Vec<TokenStream> { .map(|field| { let field_name = field.ident.as_ref().unwrap(); - if field.default { - quote! { - #field_name: Default::default(), - } + let unwrap = if field.default { + quote! { unwrap_or_default() } } else { - let field_rename = match &field.rename { - Some(name) => name.clone(), - None => format!("{}", field.ident.as_ref().unwrap()).into(), - }; + quote! { unwrap() } + }; - match field.network { - super::NetworkRepr::List => quote! { - #field_name: libquassel::message::NetworkList::from_network_list( - &mut std::convert::TryInto::try_into(input.remove(#field_rename).unwrap()).unwrap()), - }, - super::NetworkRepr::Map => quote! { - #field_name: libquassel::message::NetworkMap::from_network_map( - &mut std::convert::TryInto::try_into(input.remove(#field_rename).unwrap()).unwrap()), - }, - super::NetworkRepr::None => quote! { - #field_name: std::convert::TryInto::try_into(input.remove(#field_rename).unwrap()).unwrap(), - }, - } + let field_rename = match &field.rename { + Some(name) => name.clone(), + None => format!("{}", field.ident.as_ref().unwrap()).into(), + }; + + match field.network { + super::NetworkRepr::List => quote! { + #field_name: libquassel::message::NetworkList::from_network_list( + &mut std::convert::TryInto::try_into(input.remove(#field_rename).unwrap()).#unwrap), + }, + super::NetworkRepr::Map => quote! { + #field_name: libquassel::message::NetworkMap::from_network_map( + &mut std::convert::TryInto::try_into(input.remove(#field_rename).unwrap()).#unwrap), + }, + super::NetworkRepr::None => quote! { + #field_name: std::convert::TryInto::try_into(input.remove(#field_rename).unwrap()).#unwrap, + }, } }) .collect() |