From fa66ca1e16c2fe86bbe70e4e90e5e555caacabe9 Mon Sep 17 00:00:00 2001
From: vgoodric <vgoodric@adobe.com>
Date: Thu, 2 Jan 2025 13:51:09 -0700
Subject: [PATCH] account for no stageDomainsMap

---
 libs/features/personalization/preview.js | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/libs/features/personalization/preview.js b/libs/features/personalization/preview.js
index f55deedbb8..f1ff53fd7d 100644
--- a/libs/features/personalization/preview.js
+++ b/libs/features/personalization/preview.js
@@ -74,16 +74,18 @@ function addPillEventListeners(div) {
 export function parsePageAndUrl(config, windowLocation, prefix) {
   const { stageDomainsMap, env } = config;
   const { pathname, origin } = windowLocation;
-  if (env?.name === 'prod' || !stageDomainsMap) {
-    return { page: pathname.replace(`/${prefix}/`, '/'), url: `${origin}${pathname}` };
-  }
-  let path = pathname;
-  let domain = origin;
   const allowedHosts = [
     'business.stage.adobe.com',
     'www.stage.adobe.com',
     'milo.stage.adobe.com',
   ];
+  if (env?.name === 'prod' || !stageDomainsMap) {
+    const domain = allowedHosts.includes(origin.replace('https://', ''))
+      ? origin.replace('stage.adobe.com', 'adobe.com') : origin;
+    return { page: pathname.replace(`/${prefix}/`, '/'), url: `${domain}${pathname}` };
+  }
+  let path = pathname;
+  let domain = origin;
   const domainCheck = Object.keys(stageDomainsMap)
     .find((key) => {
       try {