From c5528272b1f45d0bb199e803c7e2c61e50d642f9 Mon Sep 17 00:00:00 2001
From: mathuria <mathuria@adobe.com>
Date: Tue, 28 Jan 2025 11:32:33 +0530
Subject: [PATCH] Unity V2 for CC

---
 creativecloud/blocks/unity/unity.js | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/creativecloud/blocks/unity/unity.js b/creativecloud/blocks/unity/unity.js
index 8e709866d..bc0c8556b 100644
--- a/creativecloud/blocks/unity/unity.js
+++ b/creativecloud/blocks/unity/unity.js
@@ -25,6 +25,8 @@ function getUnityLibs(prodLibs = '/unitylibs') {
   const { hostname } = window.location;
   if (!hostname.includes('hlx.page')
     && !hostname.includes('hlx.live')
+    && !hostname.includes('aem.page')
+    && !hostname.includes('aem.live')
     && !hostname.includes('localhost')) {
     return prodLibs;
   }
@@ -39,12 +41,9 @@ export default async function init(el) {
     `${unitylibs}/core/styles/styles.css`,
     `${unitylibs}/core/workflow/workflow.js`,
     `${unitylibs}/scripts/utils.js`,
-    `${unitylibs}/core/workflow/workflow-photoshop/workflow-photoshop.js`,
-    `${unitylibs}/core/workflow/workflow-photoshop/workflow-photoshop.css`,
-    `${unitylibs}/core/steps/upload-btn.js`,
-    `${unitylibs}/core/steps/app-connector.js`,
+    `${unitylibs}/core/workflow/workflow.js`,
   ];
   await priorityLoad(promiseArr);
   const { default: wfinit } = await import(`${unitylibs}/core/workflow/workflow.js`);
-  await wfinit(el, 'cc', unitylibs);
+  await wfinit(el, 'cc', unitylibs, 'v2');
 }