From dddf7f409e82f5f6aa43ef7d1a0a3a36795543dd Mon Sep 17 00:00:00 2001
From: Nishant Kumar Thakur <nishant.thakur.ece@gmail.com>
Date: Sun, 26 Apr 2026 14:53:56 +0530
Subject: [PATCH] MWPW-191633: Enable milolibs/unitylibs query params on CC
 stage URLs

---
 unitylibs/scripts/utils.js | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/unitylibs/scripts/utils.js b/unitylibs/scripts/utils.js
index b8fc7c57..275fc119 100644
--- a/unitylibs/scripts/utils.js
+++ b/unitylibs/scripts/utils.js
@@ -5,9 +5,10 @@ export const [setLibs, getLibs] = (() => {
       libs = (() => {
         const { hostname, origin, search } = location || window.location;
         if (hostname.endsWith('acrobat.adobe.com')) return `${origin}/dc-shared/libs`;
-        if (!(hostname.includes('.hlx.') || hostname.includes('.aem.') || hostname.includes('local'))) return prodLibs;
+        if (!['.aem.', '.hlx.', '.stage.', 'localhost', '.da.'].some((i) => hostname.includes(i))) return prodLibs;
         const branch = new URLSearchParams(search).get('milolibs') || 'main';
         if (branch === 'local') return 'http://localhost:6456/libs';
+        if (branch === 'main' && hostname.includes('.stage.')) return prodLibs;
         const env = hostname.includes('.hlx.') ? 'hlx' : 'aem';
         return branch.includes('--') ? `https://${branch}.${env}.live/libs` : `https://${branch}--milo--adobecom.${env}.live/libs`;
       })();
@@ -29,7 +30,7 @@ export const [setUnityLibs, getUnityLibs] = (() => {
 
 export function decorateArea() {}
 
-const miloLibs = setLibs('/libs');
+const miloLibs = setLibs(`${window.location.origin}/libs`);
 
 const {
   createTag, getConfig, loadStyle, loadLink, loadScript, localizeLink, loadArea,