loadsmith-core 0.3.1

use std::{fmt::Display, str::FromStr};

use camino::Utf8PathBuf;
use serde::{Deserialize, Serialize};
use url::Url;

use crate::{Error, Result};

/// A URL or local file path that points to a package source.
///
/// The parser accepts both proper URLs (`"https://..."`) and `file://`
/// prefixed local paths.
///
/// ```rust
/// # use loadsmith_core::FileUrl;
/// let url = FileUrl::try_from_url(
///     "https://thunderstore.io/package/download/denikson/BepInExPack_Valheim/5.4.2202/",
/// ).unwrap();
/// assert!(url.to_string().starts_with("https://"));
///
/// let local: FileUrl = "file:///home/user/.config/loadsmith/cache/BepInExPack_Valheim-5.4.2202.zip"
///     .parse().unwrap();
/// assert_eq!(
///     local.to_string(),
///     "file:///home/user/.config/loadsmith/cache/BepInExPack_Valheim-5.4.2202.zip",
/// );
/// ```
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
#[serde(into = "String", try_from = "String")]
pub enum FileUrl {
    Url(Url),
    Path(Utf8PathBuf),
}

impl FileUrl {
    /// Parse a string as a proper URL.
    ///
    /// This does not recognise `file://` prefixes. To handle both forms,
    /// use [`FromStr`](str::parse) instead.
    ///
    /// ```rust
    /// # use loadsmith_core::FileUrl;
    /// let url = FileUrl::try_from_url("https://thunderstore.io/api/v1/package/").unwrap();
    /// assert!(url.to_string().starts_with("https://"));
    /// ```
    pub fn try_from_url(s: &str) -> Result<Self> {
        let url = Url::parse(s)?;
        Ok(Self::Url(url))
    }
}

impl Display for FileUrl {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            FileUrl::Url(url) => write!(f, "{url}"),
            FileUrl::Path(path) => write!(f, "file://{path}"),
        }
    }
}

impl From<FileUrl> for String {
    fn from(source: FileUrl) -> Self {
        source.to_string()
    }
}

impl FromStr for FileUrl {
    type Err = Error;

    fn from_str(s: &str) -> Result<Self> {
        if let Some(path) = s.strip_prefix("file://") {
            Ok(FileUrl::Path(Utf8PathBuf::from(path)))
        } else {
            FileUrl::try_from_url(s)
        }
    }
}

impl TryFrom<String> for FileUrl {
    type Error = Error;

    fn try_from(s: String) -> Result<Self> {
        s.parse()
    }
}

impl From<Url> for FileUrl {
    fn from(url: Url) -> Self {
        Self::Url(url)
    }
}

impl From<Utf8PathBuf> for FileUrl {
    fn from(path: Utf8PathBuf) -> Self {
        Self::Path(path)
    }
}