From a5244a3568ad70da0768f11cebeb5337ebe043d7 Mon Sep 17 00:00:00 2001
From: John Mulligan <jmulligan@redhat.com>
Date: Tue, 17 Sep 2024 11:06:58 -0700
Subject: [PATCH] images/server: make it easier to do custom package builds

Creating a custom "devbuilds" container with custom ceph packages and/or
custom samba packages was somewhat annoying. Tweak the
install-packages.sh to try and make it slightly easier to do one of
these builds.

Signed-off-by: John Mulligan <jmulligan@redhat.com>
---
 images/server/install-packages.sh | 45 +++++++++++++++++++++++++------
 1 file changed, 37 insertions(+), 8 deletions(-)

diff --git a/images/server/install-packages.sh b/images/server/install-packages.sh
index 323d6dd..d7abf7d 100755
--- a/images/server/install-packages.sh
+++ b/images/server/install-packages.sh
@@ -40,6 +40,31 @@ EOF
     rm -rf "${tmpfile}"
 }
 
+get_samba_nightly_repo() {
+    get_custom_repo "https://artifacts.ci.centos.org/samba/pkgs/master/${OS_BASE}/samba-nightly-master.repo"
+}
+
+get_distro_ceph_repo() {
+    if [[ "${OS_BASE}" = centos ]]; then
+        dnf install --setopt=install_weak_deps=False -y \
+            epel-release centos-release-ceph-reef
+    fi
+}
+
+get_epel_repo_if_needed() {
+    if [[ "${OS_BASE}" = centos ]]; then
+        dnf install --setopt=install_weak_deps=False -y epel-release
+    fi
+}
+
+get_ceph_shaman_repo() {
+    ceph_ref="${CEPH_REPO_REF:-main}"
+    ceph_sha="${CEPH_REPO_SHA:-latest}"
+    url="https://shaman.ceph.com/api/search/?project=ceph&distros=${OS_BASE}/9/x86_64&flavor=default&ref=${ceph_ref}&sha1=${ceph_sha}"
+    generate_repo_from_shaman "${url}" "ceph-${ceph_ref}.repo"
+    cat "/etc/yum.repos.d/ceph-${ceph_ref}.repo"
+}
+
 install_packages_from="$1"
 samba_version_suffix="$2"
 install_custom_repo="$3"
@@ -50,22 +75,26 @@ OS_BASE="$(. /etc/os-release && echo "${ID}")"
 
 case "${install_packages_from}" in
     samba-nightly)
-        get_custom_repo "https://artifacts.ci.centos.org/samba/pkgs/master/${OS_BASE}/samba-nightly-master.repo"
-        if [[ "${OS_BASE}" = centos ]]; then
-            dnf install --setopt=install_weak_deps=False -y \
-                epel-release centos-release-ceph-reef
-        fi
+        get_samba_nightly_repo
+        get_distro_ceph_repo
         package_selection=${package_selection:-nightly}
     ;;
     devbuilds)
+        get_samba_nightly_repo
         # devbuilds - samba nightly dev builds and ceph dev builds
-        get_custom_repo "https://artifacts.ci.centos.org/samba/pkgs/master/${OS_BASE}/samba-nightly-master.repo"
-        generate_repo_from_shaman "https://shaman.ceph.com/api/search/?project=ceph&distros=${OS_BASE}/9/x86_64&flavor=default&ref=main&sha1=latest" ceph-main.repo
-        dnf install --setopt=install_weak_deps=False -y epel-release
+        get_ceph_shaman_repo
+        get_epel_repo_if_needed
         package_selection=${package_selection:-devbuilds}
     ;;
     custom-repo)
         get_custom_repo "${install_custom_repo}"
+        get_distro_ceph_repo
+    ;;
+    custom-devbuilds)
+        get_custom_repo "${install_custom_repo}"
+        get_ceph_shaman_repo
+        get_epel_repo_if_needed
+        package_selection=${package_selection:-devbuilds}
     ;;
 esac