From 7c2120788ae3574992b2babcae5c3ee0d0aaea3b Mon Sep 17 00:00:00 2001
From: Konrad Kleine <kkleine@redhat.com>
Date: Thu, 12 Sep 2024 23:58:17 +0200
Subject: [PATCH] Add support for centos-stream-10 chroots (#718)

---
 .github/workflows/check-snapshots.yml                  | 2 +-
 .github/workflows/fedora-copr-build.yml                | 2 +-
 scripts/functions-pgo.sh                               | 6 ++++++
 snapshot_manager/snapshot_manager/testing_farm_util.py | 8 ++++++++
 snapshot_manager/snapshot_manager/util.py              | 5 ++++-
 5 files changed, 20 insertions(+), 3 deletions(-)
 create mode 100644 scripts/functions-pgo.sh

diff --git a/.github/workflows/check-snapshots.yml b/.github/workflows/check-snapshots.yml
index 3e0f8804..d14db449 100644
--- a/.github/workflows/check-snapshots.yml
+++ b/.github/workflows/check-snapshots.yml
@@ -48,7 +48,7 @@ jobs:
             copr_ownername: "@fedora-llvm-team"
             copr_project_tpl: "llvm-snapshots-pgo-YYYYMMDD"
             copr_monitor_tpl: "https://copr.fedorainfracloud.org/coprs/g/fedora-llvm-team/llvm-snapshots-pgo-YYYYMMDD/monitor/"
-            chroot_pattern: '^(fedora-(rawhide|[0-9]+)|rhel-[8,9]-)'
+            chroot_pattern: '^(fedora-(rawhide|[0-9]+)|rhel-[8,9]-|centos-stream-10)'
             packages: "llvm"
           # - name: bootstrap
           #   maintainer_handle: "kwk"
diff --git a/.github/workflows/fedora-copr-build.yml b/.github/workflows/fedora-copr-build.yml
index 461934cc..4cfc0bf2 100644
--- a/.github/workflows/fedora-copr-build.yml
+++ b/.github/workflows/fedora-copr-build.yml
@@ -23,7 +23,7 @@ jobs:
           - name: pgo
             copr_project_tpl: "@fedora-llvm-team/llvm-snapshots-pgo-YYYYMMDD"
             copr_target_project: "@fedora-llvm-team/llvm-snapshots-pgo"
-            # extra_script_file: "scripts/functions-pgo.sh"
+            extra_script_file: "scripts/functions-pgo.sh"
             clone_url_tpl: "https://src.fedoraproject.org/forks/kkleine/rpms/PKG.git"
             clone_ref: pgo
     runs-on: ubuntu-latest
diff --git a/scripts/functions-pgo.sh b/scripts/functions-pgo.sh
new file mode 100644
index 00000000..31a5c997
--- /dev/null
+++ b/scripts/functions-pgo.sh
@@ -0,0 +1,6 @@
+set -x
+
+# Prints the chroots we care about.
+function get_chroots() {
+  copr list-chroots | grep -P '^(fedora-(rawhide|[0-9]+)|rhel-[8,9]-|centos-stream-10)' | sort |  tr '\n' ' '
+}
diff --git a/snapshot_manager/snapshot_manager/testing_farm_util.py b/snapshot_manager/snapshot_manager/testing_farm_util.py
index a4c2ecc6..cf32cc0f 100644
--- a/snapshot_manager/snapshot_manager/testing_farm_util.py
+++ b/snapshot_manager/snapshot_manager/testing_farm_util.py
@@ -301,6 +301,12 @@ def select_ranch(cls, chroot: str) -> str:
 
         >>> TestingFarmRequest.select_ranch("fedora-rawhide-i386")
         'redhat'
+
+        >>> TestingFarmRequest.select_ranch("centos-stream-x86_64")
+        'public'
+
+        >>> TestingFarmRequest.select_ranch("centos-stream-ppc64le")
+        'redhat'
         """
         util.expect_chroot(chroot)
         ranch = None
@@ -405,6 +411,8 @@ def get_compose(cls, chroot: str) -> str:
         'RHEL-9-Nightly'
         >>> TestingFarmRequest.get_compose("rhel-8-x86_64")
         'RHEL-8-Nightly'
+        >>> TestingFarmRequest.get_compose("centos-stream-10-s390x")
+        CentOS-Stream-10
         """
         util.expect_chroot(chroot)
 
diff --git a/snapshot_manager/snapshot_manager/util.py b/snapshot_manager/snapshot_manager/util.py
index 8859d46a..900ccc16 100644
--- a/snapshot_manager/snapshot_manager/util.py
+++ b/snapshot_manager/snapshot_manager/util.py
@@ -229,12 +229,15 @@ def expect_chroot(chroot: str) -> str:
     >>> expect_chroot("fedora-rawhide-x86_64")
     'fedora-rawhide-x86_64'
 
+    >>> expect_chroot("centos-stream-10-x86_64")
+    'centos-stream-10-x86_64'
+
     >>> expect_chroot("fedora-rawhide-")
     Traceback (most recent call last):
       ...
     ValueError: invalid chroot fedora-rawhide-
     """
-    if not re.search(pattern=r"^[^-]+-[^-]+-[^-]+$", string=chroot):
+    if not re.search(pattern=r"^[^-]+-[^-]+-[^-]+(-[^-]+)?$", string=chroot):
         raise ValueError(f"invalid chroot {chroot}")
     return chroot