From 0eead6a5b4d567284ab82d9aace68323a32de5a6 Mon Sep 17 00:00:00 2001
From: ambuj <kulshreshthaak.12@gmail.com>
Date: Tue, 2 Jul 2024 15:52:06 +0530
Subject: [PATCH] add-almalinux-version-range

Signed-off-by: ambuj <kulshreshthaak.12@gmail.com>
---
 src/univers/version_range.py | 83 ++++++++++++++++++++++++++++++++++++
 src/univers/versions.py      | 11 +++++
 2 files changed, 94 insertions(+)

diff --git a/src/univers/version_range.py b/src/univers/version_range.py
index c62d5b21..e9c8c159 100644
--- a/src/univers/version_range.py
+++ b/src/univers/version_range.py
@@ -821,6 +821,87 @@ class ComposerVersionRange(VersionRange):
     }
 
 
+class AlmaLinuxVersionRange(VersionRange):
+    scheme = "almalinux"
+    version_class = versions.AlmaLinuxVersion
+
+    vers_by_native_comparators = {
+        "=": "=",
+        "<=": "<=",
+        ">=": ">=",
+        "<": "<",
+        ">": ">",
+        # seen in RPM code but never seen in the doc or in the wild so far
+        "<>": "!=",
+        # seen in a specfile parser code
+        "!=": "!=",
+        "==": "=",
+    }
+
+    @classmethod
+    def build_constraint_from_string(cls, string):
+        """
+        Return a VersionConstraint built from a single RPM version
+        relationship ``string``.
+
+        >>> vr = RpmVersionRange.build_constraint_from_string("= 5.0")
+        >>> assert str(vr) == "5.0", str(vr)
+        >>> vr = RpmVersionRange.build_constraint_from_string("> 2.23,")
+        >>> assert str(vr) == ">2.23", str(vr)
+        >>> vr = RpmVersionRange.build_constraint_from_string("<= 2.24")
+        >>> assert str(vr) == "<=2.24", str(vr)
+        """
+        comparator, version = split_req(
+            string=string,
+            comparators=cls.vers_by_native_comparators,
+            strip=",",
+        )
+        version = cls.version_class(version)
+        return VersionConstraint(comparator=comparator, version=version)
+
+    @classmethod
+    def from_native(cls, string):
+        """
+        Return a VersionRange built from a ``string`` single RPM
+        version requirement string.
+
+        For example::
+
+        >>> vr = RpmVersionRange.from_native("= 3.5.6")
+        >>> assert str(vr) == "vers:rpm/3.5.6", str(vr)
+        """
+        return cls(constraints=[cls.build_constraint_from_string(string)])
+
+    @classmethod
+    def from_natives(cls, strings):
+        """
+        Return a VersionRange built from a ``strings`` list of RPM
+        version requirements or a single requirement string.
+
+        For example::
+
+        >>> vr = RpmVersionRange.from_natives("= 3.5.6")
+        >>> assert str(vr) == "vers:rpm/3.5.6", str(vr)
+
+        >>> reqs = [">= 2.8.16"]
+        >>> vr = RpmVersionRange.from_natives(reqs)
+        >>> assert str(vr) == "vers:rpm/>=2.8.16", str(vr)
+
+        >>> reqs = [">= 1:1.1.4", ">= 2.8.16", "<= 2.8.16-z"]
+        >>> vr = RpmVersionRange.from_natives(reqs)
+        >>> assert str(vr) == "vers:rpm/>=2.8.16|<=2.8.16-z|>=1:1.1.4", str(vr)
+
+        >>> reqs = ["= 5.0", "> 2.23,", "< 2.24"]
+        >>> vr = RpmVersionRange.from_natives(reqs)
+        >>> assert str(vr) == "vers:rpm/>2.23|<2.24|5.0", str(vr)
+        """
+
+        if isinstance(strings, str):
+            return cls.from_native(strings)
+        constraints = [cls.build_constraint_from_string(rel) for rel in strings]
+        return cls(constraints=constraints)
+    
+
 class RpmVersionRange(VersionRange):
     # http://ftp.rpm.org/api/4.4.2.2/dependencies.html
     # http://ftp.rpm.org/max-rpm/s1-rpm-depend-manual-dependencies.html
@@ -1273,6 +1354,8 @@ def build_range_from_github_advisory_constraint(scheme: str, string: str):
     "openssl": OpensslVersionRange,
     "mattermost": MattermostVersionRange,
     "conan": ConanVersionRange,
+    "almalinux:8": AlmaLinuxVersionRange,
+    "almalinux:9": AlmaLinuxVersionRange,
 }
 
 PURL_TYPE_BY_GITLAB_SCHEME = {
diff --git a/src/univers/versions.py b/src/univers/versions.py
index 768a2c86..86c19174 100644
--- a/src/univers/versions.py
+++ b/src/univers/versions.py
@@ -342,6 +342,16 @@ def is_valid(cls, string):
             return False
 
 
+class AlmaLinuxVersion(Version):
+    """
+    AlmaLinux version
+    """
+
+    @classmethod
+    def build_value(cls, string):
+        return rpm.RpmVersion.from_string(string)
+
+
 class RpmVersion(Version):
     """
     Represent an RPM version.
@@ -702,4 +712,5 @@ def bump(self, index):
     OpensslVersion,
     LegacyOpensslVersion,
     AlpineLinuxVersion,
+    AlmaLinuxVersion,
 ]