use ipnet::IpNet;
use serde::de::{self, Deserializer, SeqAccess, Visitor};
use std::fmt;
pub fn serialize<S>(vec: &Vec<IpNet>, serializer: S) -> Result<S::Ok, S::Error>
where
S: serde::Serializer,
{
let string = vec
.iter()
.map(|ipnet| ipnet.to_string())
.collect::<Vec<_>>()
.join(",");
serializer.serialize_str(&string)
}
pub fn deserialize<'de, D>(deserializer: D) -> Result<Vec<IpNet>, D::Error>
where
D: Deserializer<'de>,
{
deserializer.deserialize_any(IpNetVecVisitor)
}
struct IpNetVecVisitor;
impl<'de> Visitor<'de> for IpNetVecVisitor {
type Value = Vec<IpNet>;
fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
formatter.write_str("a string, comma-separated string, or array of IPNet strings")
}
fn visit_str<E>(self, value: &str) -> Result<Self::Value, E>
where
E: de::Error,
{
self.visit_string(value.to_string())
}
fn visit_string<E>(self, value: String) -> Result<Self::Value, E>
where
E: de::Error,
{
value
.split(',')
.map(|s| s.trim().to_string())
.filter(|s| !s.is_empty())
.map(|s| s.parse::<IpNet>().map_err(de::Error::custom))
.collect()
}
fn visit_seq<A>(self, mut seq: A) -> Result<Self::Value, A::Error>
where
A: SeqAccess<'de>,
{
let mut vec = Vec::new();
while let Some(element) = seq.next_element::<String>()? {
let element = element.trim().to_string();
if element.is_empty() {
continue;
}
let ipnet = element.parse::<IpNet>().map_err(de::Error::custom)?;
vec.push(ipnet);
}
Ok(vec)
}
}