lux_lib/package/

version.rs

use std::{
    cmp::{self, Ordering},
    fmt::Display,
    str::FromStr,
};

use html_escape::decode_html_entities;
use itertools::Itertools;

use miette::Diagnostic;
use nonempty::NonEmpty;
use semver::{Comparator, Error, Op, Prerelease};
use serde::{de, Deserialize, Deserializer, Serialize};
use thiserror::Error;

mod upgrade;

#[derive(Debug, Error, Diagnostic)]
pub enum ReqToVersionError {
    #[error("cannot parse version from non-exact version requirement '{0}'")]
    NonExactVersionReq(semver::VersionReq),
    #[error("cannot parse version from version requirement '*' (any version)")]
    Any,
}

#[derive(Clone, Eq, PartialEq, Hash, Debug)]
pub enum Version {
    /// **SemVer version** as defined by <https://semver.org>,
    /// but a bit more lenient for compatibility with luarocks
    SemVer(SemVer),
    /// A known **Dev** version
    DevVer(DevVer),
    /// An arbitrary string version.
    /// Yes, luarocks-site allows arbitrary string versions in the root manifest
    /// ┻━┻ ︵ヽ(`Д´)ﾉ︵ ┻━┻
    StringVer(StringVer),
}

impl HasModRev for Version {
    fn to_modrev_string(&self) -> String {
        match self {
            Self::SemVer(ver) => ver.to_modrev_string(),
            Self::DevVer(ver) => ver.to_modrev_string(),
            Self::StringVer(ver) => ver.to_modrev_string(),
        }
    }
}

impl Version {
    pub fn parse(text: &str) -> Result<Self, ParseError> {
        Version::from_str(text)
    }
    /// The package version without its spec revision.
    pub fn to_version_string(&self) -> String {
        self.to_modrev_string()
    }
    /// Note that this loses the specrev information.
    pub fn into_version_req(&self) -> VersionReq {
        match self {
            Version::DevVer(DevVer { modrev, .. }) => VersionReq::DevVer(modrev.to_owned()),
            Version::StringVer(StringVer { modrev, .. }) => {
                VersionReq::StringVer(modrev.to_owned())
            }
            Version::SemVer(SemVer { version, .. }) => {
                let version = version.to_owned();
                VersionReq::SemVer(semver::VersionReq {
                    comparators: vec![Comparator {
                        op: Op::Exact,
                        major: version.major,
                        minor: Some(version.minor),
                        patch: Some(version.patch),
                        pre: version.pre,
                    }],
                })
            }
        }
    }

    /// A `~>` version requirement that allows non-breaking updates, or `None` if this is
    /// not a SemVer version.
    /// Following the SemVer conventions around zero versions, the most significant non-zero
    /// component is kept, so that only the components to its right may change.
    pub(crate) fn to_pessimistic_version_req(&self) -> Option<String> {
        let Version::SemVer(sem_ver) = self else {
            return None;
        };
        let version = &sem_ver.version;
        Some(match (version.major, version.minor) {
            (0, 0) => format!("~> 0.0.{}", version.patch),
            (0, minor) => format!("~> 0.{minor}"),
            (major, _) => format!("~> {major}"),
        })
    }

    pub(crate) fn is_semver(&self) -> bool {
        matches!(self, Version::SemVer(_))
    }

    pub(crate) fn default_dev_version() -> Self {
        Self::DevVer(DevVer::default())
    }

    pub(crate) fn default_dev_version_with_specrev(specrev: SpecRev) -> Self {
        Self::DevVer(DevVer::default_with_specrev(specrev))
    }

    pub(crate) fn with_specrev(mut self, specrev: SpecRev) -> Self {
        match &mut self {
            Self::SemVer(version) => version.specrev = specrev,
            Self::DevVer(version) => version.specrev = specrev,
            Self::StringVer(version) => version.specrev = specrev,
        }
        self
    }
}

impl TryFrom<VersionReq> for Version {
    type Error = ReqToVersionError;

    fn try_from(req: VersionReq) -> Result<Self, Self::Error> {
        match req {
            VersionReq::SemVer(version_req) => {
                match NonEmpty::try_from(version_req.comparators.clone()) {
                    Ok(comparators)
                        if comparators
                            .iter()
                            .any(|comparator| comparator.op != semver::Op::Exact) =>
                    {
                        Err(ReqToVersionError::NonExactVersionReq(version_req))
                    }
                    Ok(comparators) => {
                        let comparator = comparators.first();
                        let version = semver::Version {
                            major: comparator.major,
                            minor: comparator.minor.unwrap_or(0),
                            patch: comparator.patch.unwrap_or(0),
                            pre: comparator.pre.clone(),
                            build: semver::BuildMetadata::EMPTY,
                        };
                        Ok(Version::SemVer(SemVer {
                            version,
                            component_count: comparator_component_count(comparator),
                            specrev: 1.into(),
                        }))
                    }
                    Err(_) => Err(ReqToVersionError::NonExactVersionReq(version_req)),
                }
            }
            VersionReq::DevVer(modrev) => Ok(Version::DevVer(DevVer {
                modrev,
                specrev: 1.into(),
            })),
            VersionReq::StringVer(modrev) => Ok(Version::StringVer(StringVer {
                modrev,
                specrev: 1.into(),
            })),
            VersionReq::Any => Err(ReqToVersionError::Any),
        }
    }
}

#[derive(Error, Debug, Diagnostic)]
#[non_exhaustive]
pub enum ParseError {
    #[error(transparent)]
    #[diagnostic(transparent)]
    Specrev(#[from] SpecrevParseError),
    #[error("failed to parse version")]
    Version(#[from] Error),
}

impl Serialize for Version {
    fn serialize<S>(&self, serializer: S) -> std::result::Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        match self {
            Version::SemVer(version) => version.serialize(serializer),
            Version::DevVer(version) => version.serialize(serializer),
            Version::StringVer(version) => version.serialize(serializer),
        }
    }
}

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

impl Display for Version {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Version::SemVer(version) => version.fmt(f),
            Version::DevVer(version) => version.fmt(f),
            Version::StringVer(version) => version.fmt(f),
        }
    }
}

impl PartialOrd for Version {
    fn partial_cmp(&self, other: &Self) -> Option<std::cmp::Ordering> {
        Some(self.cmp(other))
    }
}

impl Ord for Version {
    fn cmp(&self, other: &Self) -> Ordering {
        match (self, other) {
            (Version::SemVer(a), Version::SemVer(b)) => a.cmp(b),
            (Version::SemVer(..), Version::DevVer(..)) => Ordering::Less,
            (Version::SemVer(..), Version::StringVer(..)) => Ordering::Greater,
            (Version::DevVer(..), Version::SemVer(..)) => Ordering::Greater,
            (Version::DevVer(a), Version::DevVer(b)) => a.cmp(b),
            (Version::DevVer(..), Version::StringVer(..)) => Ordering::Greater,
            (Version::StringVer(a), Version::StringVer(b)) => a.cmp(b),
            (Version::StringVer(..), Version::SemVer(..)) => Ordering::Less,
            (Version::StringVer(..), Version::DevVer(..)) => Ordering::Less,
        }
    }
}

impl FromStr for Version {
    type Err = ParseError;

    fn from_str(text: &str) -> Result<Self, Self::Err> {
        let (modrev, specrev) = split_specrev(text)?;
        match modrev {
            "scm" => Ok(Version::DevVer(DevVer {
                modrev: DevVersion::Scm,
                specrev,
            })),
            "dev" => Ok(Version::DevVer(DevVer {
                modrev: DevVersion::Dev,
                specrev,
            })),
            modrev => match parse_version(modrev) {
                Ok(version) => Ok(Version::SemVer(SemVer {
                    component_count: cmp::min(text.chars().filter(|c| *c == '.').count() + 1, 3),
                    version,
                    specrev,
                })),
                Err(_) => Ok(Version::StringVer(StringVer {
                    modrev: modrev.into(),
                    specrev,
                })),
            },
        }
    }
}

/// The revision of a rov
#[derive(Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug)]
pub struct SpecRev(u16);

impl From<u16> for SpecRev {
    fn from(value: u16) -> Self {
        Self(value)
    }
}

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

impl Default for SpecRev {
    fn default() -> Self {
        Self(1)
    }
}

/// Iterates `SpecRev`s upwards, starting from `1`.
pub(crate) struct SpecRevIterator {
    current: u16,
}

impl SpecRevIterator {
    pub fn new() -> Self {
        SpecRevIterator {
            current: SpecRev::default().0,
        }
    }
}

impl Iterator for SpecRevIterator {
    type Item = SpecRev;

    fn next(&mut self) -> Option<Self::Item> {
        let current = self.current;
        self.current += 1;
        if self.current == 0 {
            None // overflow
        } else {
            Some(SpecRev(current))
        }
    }
}

// TODO: Stop deriving Eq here
#[derive(Clone, Eq, PartialEq, Hash, Debug)]
pub struct SemVer {
    version: semver::Version,
    component_count: usize,
    specrev: SpecRev,
}

impl HasModRev for SemVer {
    fn to_modrev_string(&self) -> String {
        self.version.to_string()
    }
}

impl Display for SemVer {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        if self
            .version
            .pre
            .as_str()
            .chars()
            .any(|c| c.is_ascii_alphabetic())
        {
            // pre-release instead of specrev
            return self.version.fmt(f);
        }
        let (version_str, remainder) = split_semver_version(&self.version.to_string());
        let mut luarocks_version_str = version_str.split('.').take(self.component_count).join(".");
        if let Some(remainder) = remainder {
            // luarocks allows and arbitrary number of '.' separators
            // We treat anything after the third '.' as a semver prerelease/build version,
            // so we have to convert it back for luarocks.
            luarocks_version_str.push_str(&format!(".{remainder}"));
        }
        let str = format!("{}-{}", luarocks_version_str, self.specrev);
        str.fmt(f)
    }
}

fn split_semver_version(version_str: &str) -> (String, Option<String>) {
    if let Some(pos) = version_str.rfind('-') {
        if let Some(pre_build_str) = version_str.get(pos + 1..) {
            (version_str[..pos].into(), Some(pre_build_str.into()))
        } else {
            (version_str[..pos].into(), None)
        }
    } else {
        (version_str.into(), None)
    }
}

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

impl Ord for SemVer {
    fn cmp(&self, other: &Self) -> Ordering {
        match cmp_versions(&self.version, &other.version) {
            Ordering::Equal => self.specrev.cmp(&other.specrev),
            ord => ord,
        }
    }
}

/// Compare two LuaRocks versions.
/// ┻━┻ ︵╰(°□°╰) Luarocks allows for an arbitrary number of version digits (e.g. `2.1.0.10`),
/// which Lux treats as a SemVer prerelease.
fn cmp_versions(a: &semver::Version, b: &semver::Version) -> Ordering {
    match (a.major, a.minor, a.patch).cmp(&(b.major, b.minor, b.patch)) {
        Ordering::Equal => cmp_prerelease(&a.pre, &b.pre),
        ord => ord,
    }
}

fn cmp_prerelease(a: &Prerelease, b: &Prerelease) -> Ordering {
    match (a.is_empty(), b.is_empty()) {
        (true, true) => Ordering::Equal,
        (false, true) => Ordering::Greater,
        (true, false) => Ordering::Less,
        (false, false) => a.cmp(b),
    }
}

fn comparator_component_count(comparator: &Comparator) -> usize {
    if comparator.patch.is_some() {
        3
    } else if comparator.minor.is_some() {
        2
    } else {
        1
    }
}

/// Based on `semver`'s matching, but with the prerelease compared as an extension
/// for compatibility with LuaRocks.
/// The "prereleases only match prerelease comparators" rule is not applied.
fn comparator_matches(comparator: &Comparator, version: &semver::Version) -> bool {
    match comparator.op {
        Op::Exact | Op::Wildcard => matches_exact(comparator, version),
        Op::Greater => matches_greater(comparator, version),
        Op::GreaterEq => matches_greater(comparator, version) || matches_exact(comparator, version),
        Op::Less => matches_less(comparator, version),
        Op::LessEq => matches_less(comparator, version) || matches_exact(comparator, version),
        Op::Tilde => matches_tilde(comparator, version),
        Op::Caret => matches_caret(comparator, version),
        _ => false,
    }
}

fn matches_exact(comparator: &Comparator, version: &semver::Version) -> bool {
    if version.major != comparator.major {
        return false;
    }
    if let Some(minor) = comparator.minor {
        if version.minor != minor {
            return false;
        }
    }
    if let Some(patch) = comparator.patch {
        if version.patch != patch {
            return false;
        }
    }
    version.pre == comparator.pre
}

fn matches_greater(comparator: &Comparator, version: &semver::Version) -> bool {
    if version.major != comparator.major {
        return version.major > comparator.major;
    }
    match comparator.minor {
        None => return false,
        Some(minor) => {
            if version.minor != minor {
                return version.minor > minor;
            }
        }
    }
    match comparator.patch {
        None => return false,
        Some(patch) => {
            if version.patch != patch {
                return version.patch > patch;
            }
        }
    }
    cmp_prerelease(&version.pre, &comparator.pre) == Ordering::Greater
}

fn matches_less(comparator: &Comparator, version: &semver::Version) -> bool {
    if version.major != comparator.major {
        return version.major < comparator.major;
    }
    match comparator.minor {
        None => return false,
        Some(minor) => {
            if version.minor != minor {
                return version.minor < minor;
            }
        }
    }
    match comparator.patch {
        None => return false,
        Some(patch) => {
            if version.patch != patch {
                return version.patch < patch;
            }
        }
    }
    cmp_prerelease(&version.pre, &comparator.pre) == Ordering::Less
}

fn matches_tilde(comparator: &Comparator, version: &semver::Version) -> bool {
    if version.major != comparator.major {
        return false;
    }
    if let Some(minor) = comparator.minor {
        if version.minor != minor {
            return false;
        }
    }
    if let Some(patch) = comparator.patch {
        if version.patch != patch {
            return version.patch > patch;
        }
    }
    cmp_prerelease(&version.pre, &comparator.pre) != Ordering::Less
}

fn matches_caret(comparator: &Comparator, version: &semver::Version) -> bool {
    if version.major != comparator.major {
        return false;
    }
    let minor = match comparator.minor {
        None => return true,
        Some(minor) => minor,
    };
    let patch = match comparator.patch {
        None => {
            if comparator.major > 0 {
                return version.minor >= minor;
            } else {
                return version.minor == minor;
            }
        }
        Some(patch) => patch,
    };
    if comparator.major > 0 {
        if version.minor != minor {
            return version.minor > minor;
        } else if version.patch != patch {
            return version.patch > patch;
        }
    } else if minor > 0 {
        if version.minor != minor {
            return false;
        } else if version.patch != patch {
            return version.patch > patch;
        }
    } else if version.minor != minor || version.patch != patch {
        return false;
    }
    cmp_prerelease(&version.pre, &comparator.pre) != Ordering::Less
}

impl PartialOrd for SemVer {
    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
        Some(self.cmp(other))
    }
}

#[derive(Clone, Eq, PartialEq, PartialOrd, Ord, Hash, Debug, Serialize, Deserialize, Default)]
#[serde(rename_all = "lowercase")]
pub enum DevVersion {
    #[default]
    Dev,
    Scm,
}

impl Display for DevVersion {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Self::Dev => "dev".fmt(f),
            Self::Scm => "scm".fmt(f),
        }
    }
}

#[derive(Clone, Eq, PartialEq, Hash, Debug, Default)]
pub struct DevVer {
    modrev: DevVersion,
    specrev: SpecRev,
}

impl DevVer {
    fn default_with_specrev(specrev: SpecRev) -> Self {
        Self {
            modrev: DevVersion::default(),
            specrev,
        }
    }
}

impl HasModRev for DevVer {
    fn to_modrev_string(&self) -> String {
        self.modrev.to_string().to_lowercase()
    }
}

impl Display for DevVer {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        let str = format!("{}-{}", self.modrev, self.specrev);
        str.fmt(f)
    }
}

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

impl Ord for DevVer {
    fn cmp(&self, other: &Self) -> Ordering {
        // NOTE: We compare specrevs first for dev versions
        let result = self.specrev.cmp(&other.specrev);
        if result == Ordering::Equal {
            return self.modrev.cmp(&other.modrev);
        }
        result
    }
}

impl PartialOrd for DevVer {
    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
        Some(self.cmp(other))
    }
}

#[derive(Clone, Eq, PartialEq, Hash, Debug)]
pub struct StringVer {
    modrev: String,
    specrev: SpecRev,
}

impl HasModRev for StringVer {
    fn to_modrev_string(&self) -> String {
        self.modrev.to_string()
    }
}

impl Display for StringVer {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        let str = format!("{}-{}", self.modrev, self.specrev);
        str.fmt(f)
    }
}

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

impl Ord for StringVer {
    fn cmp(&self, other: &Self) -> Ordering {
        // NOTE: We compare specrevs first for dev versions
        let result = self.specrev.cmp(&other.specrev);
        if result == Ordering::Equal {
            return self.modrev.cmp(&other.modrev);
        }
        result
    }
}

impl PartialOrd for StringVer {
    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
        Some(self.cmp(other))
    }
}

pub(crate) trait HasModRev {
    /// If a version has a modrev (and possibly a specrev),
    /// this is equivalent to `to_string()`, but includes only the modrev.
    fn to_modrev_string(&self) -> String;
}

#[derive(Error, Debug, Diagnostic)]
#[non_exhaustive]
#[error(transparent)]
pub struct ReqError(#[from] Error);

/// **SemVer version** requirement as defined by <https://semver.org>.
/// or a **Dev** version requirement, which can be one of "dev", "scm", or "git"
#[derive(Clone, Eq, PartialEq, Hash, Debug)]
pub enum VersionReq {
    /// A VersionReq that matches a SemVer version.
    SemVer(semver::VersionReq),
    /// A VersionReq that matches only known dev versions.
    DevVer(DevVersion),
    /// A VersionReq that matches a arbitrary string version.
    StringVer(String),
    /// A VersionReq that has no version constraint.
    Any,
}

impl VersionReq {
    /// Returns a `VersionReq` that matches any version.
    pub fn any() -> Self {
        VersionReq::Any
    }

    pub fn parse(text: &str) -> Result<Self, ReqError> {
        VersionReq::from_str(text)
    }

    pub fn matches(&self, version: &Version) -> bool {
        match (self, version) {
            (VersionReq::SemVer(req), Version::SemVer(ver)) => req
                .comparators
                .iter()
                .all(|comparator| comparator_matches(comparator, &ver.version)),
            (VersionReq::DevVer(req), Version::DevVer(ver)) => req == &ver.modrev,
            (VersionReq::StringVer(req), Version::StringVer(ver)) => req == &ver.modrev,
            (VersionReq::Any, _) => true,
            _ => false,
        }
    }

    pub fn is_any(&self) -> bool {
        matches!(self, VersionReq::Any)
    }

    /// Returns true if this requirement pins the Luau dialect (`lua = "luau"`).
    pub fn is_luau(&self) -> bool {
        matches!(self, VersionReq::StringVer(req) if req == "luau")
    }
}

impl Display for VersionReq {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            VersionReq::SemVer(version_req) => {
                let mut str = version_req.to_string();
                if str.starts_with("=") {
                    str = str.replacen("=", "==", 1);
                } else if str.starts_with("^") {
                    str = str.replacen("^", "~>", 1);
                }
                str.fmt(f)
            }
            VersionReq::DevVer(name_req) => write!(f, "=={}", name_req),
            VersionReq::StringVer(name_req) => write!(f, "=={}", name_req),
            VersionReq::Any => f.write_str("any"),
        }
    }
}

impl<'de> Deserialize<'de> for VersionReq {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: Deserializer<'de>,
    {
        String::deserialize(deserializer)?
            .parse()
            .map_err(serde::de::Error::custom)
    }
}

impl FromStr for VersionReq {
    type Err = ReqError;

    fn from_str(text: &str) -> Result<Self, Self::Err> {
        let text = correct_version_req_str(text);

        let trimmed = text.trim_start_matches('=').trim_start_matches('@').trim();

        match parse_version_req(&text) {
            Ok(ver) if ver.comparators.is_empty() => Ok(VersionReq::Any),
            Ok(ver) => Ok(VersionReq::SemVer(ver)),
            Err(_) => match trimmed {
                "scm" => Ok(VersionReq::DevVer(DevVersion::Scm)),
                "dev" => Ok(VersionReq::DevVer(DevVersion::Dev)),
                ver => Ok(VersionReq::StringVer(ver.to_string())),
            },
        }
    }
}

fn correct_version_req_str(text: &str) -> String {
    text.chars()
        .chunk_by(|t| t.is_alphanumeric() || matches!(t, '-' | '_' | '.'))
        .into_iter()
        .map(|(is_version_str, chars)| (is_version_str, chars.collect::<String>()))
        .map(|(is_version_str, chunk)| {
            if is_version_str && !is_known_dev_version_str(&chunk) {
                let version_str = trim_specrev(&chunk);
                correct_prerelease_version_string(version_str)
            } else {
                chunk
            }
        })
        .collect::<String>()
}

fn trim_specrev(version_str: &str) -> &str {
    if let Some(pos) = version_str.rfind('-') {
        let suffix = &version_str[pos + 1..];
        if suffix.chars().all(|c| c.is_ascii_digit()) {
            &version_str[..pos]
        } else {
            version_str
        }
    } else {
        version_str
    }
}

#[derive(Error, Debug, Diagnostic)]
#[non_exhaustive]
pub enum SpecrevParseError {
    #[error("specrev {specrev} in version {full_version} contains non-numeric characters")]
    InvalidSpecrev {
        specrev: String,
        full_version: String,
    },
    #[error("could not parse specrev in version {0}")]
    InvalidVersion(String),
}

fn split_specrev(version_str: &str) -> Result<(&str, SpecRev), SpecrevParseError> {
    if let Some(pos) = version_str.rfind('-') {
        if let Some(specrev_str) = version_str.get(pos + 1..) {
            if specrev_str.chars().all(|c| c.is_ascii_digit()) {
                let specrev =
                    specrev_str
                        .parse::<u16>()
                        .map_err(|_| SpecrevParseError::InvalidSpecrev {
                            specrev: specrev_str.into(),
                            full_version: version_str.into(),
                        })?;
                Ok((&version_str[..pos], specrev.into()))
            } else {
                // A non-numeric suffix is a semver prerelease, not a specrev.
                Ok((version_str, 1.into()))
            }
        } else {
            Err(SpecrevParseError::InvalidVersion(version_str.into()))
        }
    } else {
        // We assume a specrev of 1 if none can be found.
        Ok((version_str, 1.into()))
    }
}

fn is_known_dev_version_str(text: &str) -> bool {
    matches!(text, "dev" | "scm")
}

/// Parses a Version from a string, automatically supplying any missing details (i.e. missing
/// minor/patch sections).
fn parse_version(s: &str) -> Result<semver::Version, Error> {
    let version_str = correct_version_string(s);
    semver::Version::parse(&version_str)
}

/// Transform LuaRocks constraints into constraints that can be parsed by the semver crate.
fn parse_version_req(version_constraints: &str) -> Result<semver::VersionReq, Error> {
    let unescaped = decode_html_entities(version_constraints)
        .to_string()
        .as_str()
        .to_owned();
    let transformed = match unescaped {
        s if s.starts_with("~>") => parse_pessimistic_version_constraint(s)?,
        s if s.starts_with("@") => format!("={}", &s[1..]),
        // The semver crate only understands "= version", unlike luarocks which understands "== version".
        s if s.starts_with("==") => s[1..].to_string(),
        s if s // semver parses no constraint prefix as ^, unlike luarocks which reads it as "=="
            .find(|c: char| c.is_alphanumeric())
            .is_some_and(|idx| idx == 0) =>
        {
            format!("={}", s)
        }
        s => s,
    };

    let version_req = semver::VersionReq::parse(&transformed)?;
    Ok(version_req)
}

fn parse_pessimistic_version_constraint(version_constraint: String) -> Result<String, Error> {
    // pessimistic operator
    let min_version_str = &version_constraint[2..].trim();
    let min_version = semver::Version::parse(&correct_version_string(min_version_str))?;

    let max_version = match min_version_str.matches('.').count() {
        0 => semver::Version {
            major: &min_version.major + 1,
            ..min_version.clone()
        },
        1 => semver::Version {
            minor: &min_version.minor + 1,
            ..min_version.clone()
        },
        _ => semver::Version {
            patch: &min_version.patch + 1,
            ..min_version.clone()
        },
    };

    Ok(format!(">= {min_version}, < {max_version}"))
}

/// ┻━┻ ︵╰(°□°╰) Luarocks allows for an arbitrary number of version digits
/// This function attempts to correct a non-semver compliant version string,
/// by swapping the third '.' out with a '-', converting the non-semver
/// compliant digits to a pre-release identifier.
fn correct_version_string(version: &str) -> String {
    let version = append_minor_patch_if_missing(version);
    correct_prerelease_version_string(&version)
}

fn correct_prerelease_version_string(version: &str) -> String {
    if version.contains('-') {
        return version.to_string();
    }
    let parts: Vec<&str> = version.split('.').collect();
    if parts.len() > 3 {
        format!(
            "{}.{}.{}-{}",
            parts[0],
            parts[1],
            parts[2],
            parts[3..].join(".")
        )
    } else {
        version.to_string()
    }
}

/// Recursively append .0 until the version string has a minor or patch version
fn append_minor_patch_if_missing(version: &str) -> String {
    if version.matches('.').count() < 2 {
        append_minor_patch_if_missing(&format!("{version}.0"))
    } else {
        version.to_string()
    }
}

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

    #[test]
    fn parse_semver_version() {
        assert_eq!(
            Version::parse("1-1").unwrap(),
            Version::SemVer(SemVer {
                version: "1.0.0".parse().unwrap(),
                component_count: 1,
                specrev: 1.into(),
            })
        );
        assert_eq!(
            Version::parse("1.0-1").unwrap(),
            Version::SemVer(SemVer {
                version: "1.0.0".parse().unwrap(),
                component_count: 2,
                specrev: 1.into(),
            })
        );
        assert_eq!(
            Version::parse("1.0.0-1").unwrap(),
            Version::SemVer(SemVer {
                version: "1.0.0".parse().unwrap(),
                component_count: 3,
                specrev: 1.into()
            })
        );
        assert_eq!(
            Version::parse("1.0.0-1").unwrap(),
            Version::SemVer(SemVer {
                version: "1.0.0".parse().unwrap(),
                component_count: 3,
                specrev: 1.into()
            })
        );
        assert_eq!(
            Version::parse("1.0.0-10-1").unwrap(),
            Version::SemVer(SemVer {
                version: "1.0.0-10".parse().unwrap(),
                component_count: 3,
                specrev: 1.into()
            })
        );
        assert_eq!(
            Version::parse("1.0.0.10-1").unwrap(),
            Version::SemVer(SemVer {
                version: "1.0.0-10".parse().unwrap(),
                component_count: 3,
                specrev: 1.into()
            })
        );
        assert_eq!(
            Version::parse("1.0.0.10.0-1").unwrap(),
            Version::SemVer(SemVer {
                version: "1.0.0-10.0".parse().unwrap(),
                component_count: 3,
                specrev: 1.into()
            })
        );
    }

    #[test]
    fn to_pessimistic_version_req() {
        for (version, req, expected) in [
            ("1.5.6-2", "~> 1", ">=1.0.0, <2.0.0"),
            ("0.5.6-2", "~> 0.5", ">=0.5.0, <0.6.0"),
            ("0.0.6-2", "~> 0.0.6", ">=0.0.6, <0.0.7"),
        ] {
            let pessimistic = Version::parse(version)
                .unwrap()
                .to_pessimistic_version_req()
                .unwrap();
            assert_eq!(pessimistic, req);
            let req = VersionReq::parse(&pessimistic).unwrap();
            assert_eq!(req.to_string(), expected);
        }

        let req = VersionReq::parse("~> 1").unwrap();
        assert!(req.matches(&Version::parse("1.9.0-2").unwrap()));
        assert!(!req.matches(&Version::parse("2.0.0-1").unwrap()));

        let req = VersionReq::parse("~> 0.5").unwrap();
        assert!(req.matches(&Version::parse("0.5.9-1").unwrap()));
        assert!(!req.matches(&Version::parse("0.6.0-1").unwrap()));

        assert_eq!(
            Version::parse("scm").unwrap().to_pessimistic_version_req(),
            None
        );
    }

    #[test]
    fn four_component_versions_sort_after_base_version() {
        let base = Version::parse("2.1.0-1").unwrap();
        let extended = Version::parse("2.1.0.10-1").unwrap();
        assert!(extended > base);
        assert!(base < extended);
        assert_eq!(
            Version::parse("2.1.0.10-1").unwrap(),
            Version::parse("2.1.0.10-1").unwrap()
        );
    }

    #[test]
    fn four_component_version_satisfies_greater_eq_req() {
        let req = VersionReq::parse(">= 2.1.0").unwrap();
        assert!(req.matches(&Version::parse("2.1.0.10-1").unwrap()));
        assert!(req.matches(&Version::parse("2.1.0-1").unwrap()));
        assert!(!req.matches(&Version::parse("2.0.9-1").unwrap()));
    }

    #[test]
    fn parse_semver_prerelease() {
        let prerelease = Version::parse("1.2.3-alpha.1").unwrap();
        assert_eq!(
            prerelease,
            Version::SemVer(SemVer {
                version: "1.2.3-alpha.1".parse().unwrap(),
                component_count: 3,
                specrev: 1.into()
            })
        );
        assert_eq!(prerelease.to_string(), "1.2.3-alpha.1");
        assert_eq!(Version::parse("1.2.3-1").unwrap().to_string(), "1.2.3-1");
        assert_eq!(Version::parse(&prerelease.to_string()).unwrap(), prerelease);
    }

    #[test]
    fn prerelease_req_matches() {
        let req = VersionReq::parse("=1.2.3-alpha.1").unwrap();
        assert!(req.matches(&Version::parse("1.2.3-alpha.1").unwrap()));
        assert!(!req.matches(&Version::parse("1.2.3-alpha.2").unwrap()));
    }

    #[test]
    fn parse_dev_version() {
        assert_eq!(
            Version::parse("dev-1").unwrap(),
            Version::DevVer(DevVer {
                modrev: DevVersion::Dev,
                specrev: 1.into()
            })
        );
        assert_eq!(
            Version::parse("scm-1").unwrap(),
            Version::DevVer(DevVer {
                modrev: DevVersion::Scm,
                specrev: 1.into()
            })
        );
        assert_eq!(
            Version::parse("git-1").unwrap(),
            Version::StringVer(StringVer {
                modrev: "git".into(),
                specrev: 1.into()
            })
        );
        assert_eq!(
            Version::parse("scm-1").unwrap(),
            Version::DevVer(DevVer {
                modrev: DevVersion::Scm,
                specrev: 1.into()
            })
        );
    }

    #[test]
    fn parse_dev_version_req() {
        assert_eq!(
            VersionReq::parse("dev").unwrap(),
            VersionReq::DevVer(DevVersion::Dev)
        );
        assert_eq!(
            VersionReq::parse("scm").unwrap(),
            VersionReq::DevVer(DevVersion::Scm)
        );
        assert_eq!(
            VersionReq::parse("git").unwrap(),
            VersionReq::StringVer("git".into())
        );
        assert_eq!(
            VersionReq::parse("==dev").unwrap(),
            VersionReq::DevVer(DevVersion::Dev)
        );
        assert_eq!(
            VersionReq::parse("==git").unwrap(),
            VersionReq::StringVer("git".into())
        );
        assert_eq!(
            VersionReq::parse("== dev").unwrap(),
            VersionReq::DevVer(DevVersion::Dev)
        );
        assert_eq!(
            VersionReq::parse("== scm").unwrap(),
            VersionReq::DevVer(DevVersion::Scm)
        );
        assert_eq!(
            VersionReq::parse("@dev").unwrap(),
            VersionReq::DevVer(DevVersion::Dev)
        );
        assert_eq!(
            VersionReq::parse("@git").unwrap(),
            VersionReq::StringVer("git".into())
        );
        assert_eq!(
            VersionReq::parse("@ dev").unwrap(),
            VersionReq::DevVer(DevVersion::Dev)
        );
        assert_eq!(
            VersionReq::parse("@ scm").unwrap(),
            VersionReq::DevVer(DevVersion::Scm)
        );
        assert_eq!(
            VersionReq::parse(">1-1,<1.2-2").unwrap(),
            VersionReq::SemVer(">1,<1.2".parse().unwrap())
        );
        assert_eq!(
            VersionReq::parse("> 1-1, < 1.2-2").unwrap(),
            VersionReq::SemVer("> 1, < 1.2".parse().unwrap())
        );
        assert_eq!(
            VersionReq::parse("> 2.1.0.10, < 2.1.1").unwrap(),
            VersionReq::SemVer("> 2.1.0-10, < 2.1.1".parse().unwrap())
        );
    }

    #[test]
    fn parse_any_version_req() {
        assert_eq!(VersionReq::parse("*").unwrap(), VersionReq::Any);
    }

    #[test]
    fn package_version_req_semver_roundtrips() {
        let req = VersionReq::parse("==0.7.1").unwrap();
        assert_eq!(req.to_string(), "==0.7.1");

        let req = VersionReq::parse("0.7.1").unwrap();
        assert_eq!(req.to_string(), "==0.7.1");

        let req = VersionReq::parse(">=0.7.1").unwrap();
        assert_eq!(req.to_string(), ">=0.7.1");

        let req = VersionReq::parse(">0.7.1").unwrap();
        assert_eq!(req.to_string(), ">0.7.1");

        let req = VersionReq::parse("<0.7.1").unwrap();
        assert_eq!(req.to_string(), "<0.7.1");

        let req = VersionReq::parse("~> 0.7.1").unwrap();
        assert_eq!(req.to_string(), ">=0.7.1, <0.7.2");
    }

    #[test]
    fn package_version_req_devver_roundtrips() {
        let req = VersionReq::parse("==scm").unwrap();
        assert_eq!(req.to_string(), "==scm");

        let req = VersionReq::parse("@scm").unwrap();
        assert_eq!(req.to_string(), "==scm");

        let req = VersionReq::parse("scm").unwrap();
        assert_eq!(req.to_string(), "==scm");

        let req = VersionReq::parse("==a144124839f027a2d0a95791936c478d047126fc").unwrap();
        assert_eq!(
            req.to_string(),
            "==a144124839f027a2d0a95791936c478d047126fc"
        );
    }
}