wheel-rs 1.8.0

A Rust utility library providing file operations, time utilities and Duration serialization support.
Documentation
use serde::{Deserialize, Serialize};
use thiserror::Error;

#[derive(Debug, Error)]
pub enum AddrError {
    #[error("Fail to parse Addr string: {0}")]
    Parse(String),
}

#[derive(Debug)]
pub struct Addr {
    pub host: String,
    pub port: Option<u16>,
}

impl Addr {
    pub fn new(host: String, port: Option<u16>) -> Addr {
        Addr { host, port }
    }

    pub fn from_str(s: &str) -> Result<Addr, AddrError> {
        let parts: Vec<_> = s.split(':').collect();
        let len = parts.len();
        let mut port = None;
        if len < 1 || len > 2 {
            return Err(AddrError::Parse(format!("Invalid address: {s}")));
        }
        let host = parts[0].to_string();
        if len == 2 {
            port = Some(
                parts[1]
                    .parse::<u16>()
                    .map_err(|_| AddrError::Parse(format!("Invalid port: {s}")))?,
            );
        }
        Ok(Addr { host, port })
    }

    pub fn to_string(&self) -> String {
        if let Some(port) = self.port {
            format!("{}:{}", self.host, port)
        } else {
            self.host.clone()
        }
    }
}

impl Serialize for Addr {
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        serializer.serialize_str(&self.to_string())
    }
}

impl<'de> Deserialize<'de> for Addr {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: serde::Deserializer<'de>,
    {
        let s = String::deserialize(deserializer)?;
        Addr::from_str(&s).map_err(|e| serde::de::Error::custom(format!("{:?}", e)))
    }
}