pubky-homeserver 0.5.3

use std::str::FromStr;

use serde::{Deserialize, Serialize};

use super::{webdav_path_pub::WebDavPathPub, WebDavPath};

/// A webdav path /pub/ that can be used with axum.
///
/// When using `.route("/{*path}", your_handler)` in axum, the path is passed without the leading slash.
/// This struct adds the leading slash back and therefore allows direct validation of the path.
///
/// Usage in handler:
///
/// `Path(path): Path<WebDavPathPubAxum>`
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub struct WebDavPathPubAxum(pub WebDavPathPub);

impl WebDavPathPubAxum {
    pub fn inner(&self) -> &WebDavPath {
        self.0.inner()
    }
}

impl std::fmt::Display for WebDavPathPubAxum {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}", self.0 .0.as_str())
    }
}

impl FromStr for WebDavPathPubAxum {
    type Err = anyhow::Error;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        let with_slash = format!("/{}", s);
        let inner = WebDavPathPub::from_str(&with_slash)?;
        Ok(Self(inner))
    }
}

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

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

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_webdav_path_axum() {
        let path = WebDavPathPubAxum::from_str("pub/foo/bar").unwrap();
        assert_eq!(path.0 .0.as_str(), "/pub/foo/bar");
    }

    #[test]
    fn test_webdav_pub_required() {
        WebDavPathPubAxum::from_str("pub/file.txt").expect("Should be valid");
        WebDavPathPubAxum::from_str("file.txt").expect_err("Should not be valid. /pub/ required.");
    }
}