From 31683d6e13eb4fd7c61c36730f5a8ed2c9cbbb5b Mon Sep 17 00:00:00 2001
From: Jeremy Fee <jmfee@usgs.gov>
Date: Fri, 6 Nov 2020 10:29:18 -0700
Subject: [PATCH] Update how DATA_HOST is configured

---
 scripts/custom.config.sh | 4 ++++
 scripts/custom.funcs.sh  | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/scripts/custom.config.sh b/scripts/custom.config.sh
index fc96e1983..a00c5b9a5 100644
--- a/scripts/custom.config.sh
+++ b/scripts/custom.config.sh
@@ -8,3 +8,7 @@ export SERVICE_MAP=(
 export DATA_HOST=${DATA_HOST:-cwbpub.cr.usgs.gov};
 export DATA_PORT=${DATA_PORT:-2060};
 export DATA_TYPE=${DATA_TYPE:-edge};
+
+if [[ $TARGET_HOSTNAME == *"mage"* ]]; then
+  export DATA_HOST=${TARGET_HOSTNAME}
+fi
diff --git a/scripts/custom.funcs.sh b/scripts/custom.funcs.sh
index 060c80b55..ef5b23b38 100644
--- a/scripts/custom.funcs.sh
+++ b/scripts/custom.funcs.sh
@@ -99,7 +99,7 @@ services:
       - 8000
     environment:
       - BASE_HREF=${BASE_HREF}
-      - DATA_HOST=${TARGET_HOSTNAME}
+      - DATA_HOST=${DATA_HOST}
       - DATA_PORT=${DATA_PORT}
       - DATA_TYPE=${DATA_TYPE}
       - SITE_URL=${SITE_URL}
-- 
GitLab