From 0cc35e39ef7a42ad9c9353bd4c543a7654c0c61f Mon Sep 17 00:00:00 2001
From: "Eric D. Helms" <ericdhelms@gmail.com>
Date: Tue, 29 Aug 2023 16:01:09 -0400
Subject: [PATCH] Allow yumrepo to deploy from stagingyum

---
 puppet/modules/web/files/deploy-yumrepo.sh | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/puppet/modules/web/files/deploy-yumrepo.sh b/puppet/modules/web/files/deploy-yumrepo.sh
index 9cdd21762..cf860f754 100644
--- a/puppet/modules/web/files/deploy-yumrepo.sh
+++ b/puppet/modules/web/files/deploy-yumrepo.sh
@@ -23,7 +23,10 @@ do_rsync() {
 	fi
 
 	rsync "${opts[@]}" --log-file-format 'CHANGED %f' "${REPO_SOURCE_RPM}/*" "${REPO_INSTANCE_PATH}/"
-	rsync "${opts[@]}" --log-file-format 'CHANGED source/%f' "${REPO_SOURCE_SRPM}/" "${REPO_INSTANCE_PATH}/source/"
+
+	if [[ -n "$REPO_SOURCE_RPM" ]]; then
+		rsync "${opts[@]}" --log-file-format 'CHANGED source/%f' "${REPO_SOURCE_SRPM}/" "${REPO_INSTANCE_PATH}/source/"
+	fi
 
 	set +f
 	for d in "${REPO_INSTANCE_PATH}"/*; do
@@ -93,15 +96,21 @@ REPO_SOURCE=$1
 REPO_DEST=$2
 OVERWRITE=${3:-false}
 MERGE=${4:-false}
+STAGING=${5:-false}
 
 if [[ -z $REPO_SOURCE ]] || [[ -z $REPO_DEST ]] ; then
 	echo "Usage: $0 REPO_SOURCE REPO_DEST OVERWRITE MERGE"
 	exit 1
 fi
 
-REPO_SOURCE_BASE="rsync://koji.katello.org/releases"
-REPO_SOURCE_RPM="${REPO_SOURCE_BASE}/yum/${REPO_SOURCE}"
-REPO_SOURCE_SRPM="${REPO_SOURCE_BASE}/source/${REPO_SOURCE}"
+if [[ $STAGING == true ]] ; then
+	REPO_SOURCE_BASE="rsync://stagingyum.theforeman.org/"
+	REPO_SOURCE_RPM="${REPO_SOURCE_BASE}/${REPO_SOURCE}"
+else
+	REPO_SOURCE_BASE="rsync://koji.katello.org/releases"
+	REPO_SOURCE_RPM="${REPO_SOURCE_BASE}/yum/${REPO_SOURCE}"
+	REPO_SOURCE_SRPM="${REPO_SOURCE_BASE}/source/${REPO_SOURCE}"
+fi
 
 DEPLOY_TO="/var/www/vhosts/yum/htdocs"
 REPO_PATH="${DEPLOY_TO}/${REPO_DEST}"