From d217c839a9e13c783a5bf1a83598c0e1a6ba5496 Mon Sep 17 00:00:00 2001
From: Alfred Wingate <parona@protonmail.com>
Date: Sat, 2 Mar 2024 10:21:24 +0200
Subject: [PATCH] Add GitLab handler

Signed-off-by: Alfred Wingate <parona@protonmail.com>
---
 src/euscan/handlers/gitlab.py | 82 +++++++++++++++++++++++++++++++++++
 1 file changed, 82 insertions(+)
 create mode 100644 src/euscan/handlers/gitlab.py

diff --git a/src/euscan/handlers/gitlab.py b/src/euscan/handlers/gitlab.py
new file mode 100644
index 0000000..ab6a0c1
--- /dev/null
+++ b/src/euscan/handlers/gitlab.py
@@ -0,0 +1,82 @@
+# Copyright 2020-2024 src_prepare group
+# Distributed under the terms of the GNU General Public License v2
+
+import json
+import re
+
+import portage
+
+from euscan import helpers, mangling, output
+
+HANDLER_NAME = "gitlab"
+CONFIDENCE = 100
+PRIORITY = 90
+
+_gitlab_instances = [
+    "gitlab.com",
+    "gitlab.freedesktop.org",
+    "invent.kde.org/",
+    "gitlab.gnome.org",
+    "gitlab.kitware.com",
+    "gitlab.xfce.org",
+    "code.videolan.org",
+    "gitlab.xiph.org",
+]
+
+gitlab_patterns = [
+    # Regular expression adapted from pkgcheck
+    # https://docs.gitlab.com/ee/user/reserved_names.html
+    re.compile(
+        rf"https://(?P<domain>{domain})/(?P<repository>((?!api/)\w[^/]*/)+(?!raw/)\w[^/]*)"
+    )
+    for domain in _gitlab_instances
+]
+
+
+def can_handle(pkg, url=None):
+    return url and any([re.search(pattern, url) for pattern in gitlab_patterns])
+
+
+def scan_url(pkg, url, options):
+    "https://docs.gitlab.com/ee/api/releases/index.html"
+
+    match = [
+        re.search(pattern, url)
+        for pattern in gitlab_patterns
+        if re.search(pattern, url) is not None
+    ][0]
+
+    domain = match.group("domain")
+    repository = match.group("repository")
+
+    output.einfo(f"Using GitLab REST API in {domain}: {repository}")
+
+    request = helpers.urlopen(
+        f"https://{domain}/api/v4/projects/{repository.replace('/', '%2F')}/releases"
+    )
+
+    data = json.load(request)
+
+    versions = [release["tag_name"] for release in data]
+
+    cp, ver, rev = portage.pkgsplit(pkg.cpv)
+
+    ret = []
+    for up_pv in versions:
+        pv = mangling.mangle_version(up_pv, options)
+        if helpers.version_filtered(cp, ver, pv):
+            continue
+        urls = " ".join(
+            [
+                mangling.mangle_url(source["url"], options)
+                for source in [
+                    release["assets"]["sources"]
+                    for release in data
+                    if release["tag_name"] == up_pv
+                ][0]
+                # prefer tar.bz2
+                if source["format"] == "tar.bz2"
+            ]
+        )
+        ret.append((urls, pv, HANDLER_NAME, CONFIDENCE))
+    return ret