From 7ba1e3676df8669b987933273094949113f58393 Mon Sep 17 00:00:00 2001
From: Kyrylo Shmidt <kshmidt@digma.ai>
Date: Wed, 27 Nov 2024 14:07:58 +0100
Subject: [PATCH] Add baseUrl variable

---
 packages/jaeger-ui/index.html                 |  1 +
 .../TracePageHeader/AltViewOptions.tsx        | 62 ++++++++++---------
 .../TracePageHeader/TracePageHeader.tsx       |  3 +-
 .../src/components/common/ErrorMessage.tsx    |  7 ++-
 packages/jaeger-ui/typings/custom.d.ts        |  1 +
 5 files changed, 41 insertions(+), 33 deletions(-)

diff --git a/packages/jaeger-ui/index.html b/packages/jaeger-ui/index.html
index b1e4634d98..b11fe17a68 100644
--- a/packages/jaeger-ui/index.html
+++ b/packages/jaeger-ui/index.html
@@ -39,6 +39,7 @@
       window.global = {};
     </script>
     <script>
+      window.baseUrl;
       window.apiBaseUrl;
       window.initialRoutePath;
       window.embeddedMode;
diff --git a/packages/jaeger-ui/src/components/TracePage/TracePageHeader/AltViewOptions.tsx b/packages/jaeger-ui/src/components/TracePage/TracePageHeader/AltViewOptions.tsx
index c6c41c5ffc..a486032729 100644
--- a/packages/jaeger-ui/src/components/TracePage/TracePageHeader/AltViewOptions.tsx
+++ b/packages/jaeger-ui/src/components/TracePage/TracePageHeader/AltViewOptions.tsx
@@ -71,37 +71,41 @@ export default function AltViewOptions(props: Props) {
     onTraceViewChange(item);
   };
 
-  const menu = (
-    <Menu>
-      {MENU_ITEMS.filter(item => item.viewType !== viewType).map(item => (
-        <Menu.Item key={item.viewType}>
-          <a onClick={() => handleSelectView(item.viewType)} role="button">
-            {item.label}
+  const menu = () => {
+    const baseUrl = window.baseUrl ?? window.apiBaseUrl;
+
+    return (
+      <Menu>
+        {MENU_ITEMS.filter(item => item.viewType !== viewType).map(item => (
+          <Menu.Item key={item.viewType}>
+            <a onClick={() => handleSelectView(item.viewType)} role="button">
+              {item.label}
+            </a>
+          </Menu.Item>
+        ))}
+        <Menu.Item>
+          <a
+            href={`${baseUrl}/api/traces/${traceID}?prettyPrint=true`}
+            rel="noopener noreferrer"
+            target="_blank"
+            onClick={trackJsonView}
+          >
+            Trace JSON
           </a>
         </Menu.Item>
-      ))}
-      <Menu.Item>
-        <a
-          href={`${window.apiBaseUrl}/api/traces/${traceID}?prettyPrint=true`}
-          rel="noopener noreferrer"
-          target="_blank"
-          onClick={trackJsonView}
-        >
-          Trace JSON
-        </a>
-      </Menu.Item>
-      <Menu.Item>
-        <a
-          href={`${window.apiBaseUrl}/api/traces/${traceID}?raw=true&prettyPrint=true`}
-          rel="noopener noreferrer"
-          target="_blank"
-          onClick={trackRawJsonView}
-        >
-          Trace JSON (unadjusted)
-        </a>
-      </Menu.Item>
-    </Menu>
-  );
+        <Menu.Item>
+          <a
+            href={`${baseUrl}/api/traces/${traceID}?raw=true&prettyPrint=true`}
+            rel="noopener noreferrer"
+            target="_blank"
+            onClick={trackRawJsonView}
+          >
+            Trace JSON (unadjusted)
+          </a>
+        </Menu.Item>
+      </Menu>
+    );
+  };
 
   const currentItem = MENU_ITEMS.find(item => item.viewType === viewType);
   const dropdownText = currentItem ? currentItem.label : 'Alternate Views';
diff --git a/packages/jaeger-ui/src/components/TracePage/TracePageHeader/TracePageHeader.tsx b/packages/jaeger-ui/src/components/TracePage/TracePageHeader/TracePageHeader.tsx
index 6c3337739e..0fd861ef9f 100644
--- a/packages/jaeger-ui/src/components/TracePage/TracePageHeader/TracePageHeader.tsx
+++ b/packages/jaeger-ui/src/components/TracePage/TracePageHeader/TracePageHeader.tsx
@@ -159,11 +159,12 @@ export function TracePageHeaderFn(props: TracePageHeaderEmbedProps & { forwarded
   );
 
   const handleStandaloneLinkClick = (e: React.MouseEvent<HTMLAnchorElement>) => {
+    const baseUrl = window.baseUrl ?? window.apiBaseUrl;
     e.preventDefault();
     window.sendMessageToDigma<OpenURLInDefaultBrowserPayload>({
       action: globalActions.OPEN_URL_IN_DEFAULT_BROWSER,
       payload: {
-        url: `${window.apiBaseUrl}${window.location.pathname}${window.location.search}`,
+        url: `${baseUrl}${window.location.pathname}${window.location.search}`,
       },
     });
   };
diff --git a/packages/jaeger-ui/src/components/common/ErrorMessage.tsx b/packages/jaeger-ui/src/components/common/ErrorMessage.tsx
index 5749f8e6a5..7a4356637d 100644
--- a/packages/jaeger-ui/src/components/common/ErrorMessage.tsx
+++ b/packages/jaeger-ui/src/components/common/ErrorMessage.tsx
@@ -164,6 +164,7 @@ export default function ErrorMessage({
   }
 
   if (error.message.includes('Failed to fetch') || error.httpStatus === 502) {
+    const baseUrl = window.baseUrl ?? window.apiBaseUrl;
     const isUserDefinedJaegerQueryURL = window.isUserDefinedJaegerQueryURL === true;
     return (
       <DigmaErrorMessage
@@ -173,16 +174,16 @@ export default function ErrorMessage({
           isUserDefinedJaegerQueryURL ? (
             <>
               The Jaeger link
-              {isString(window.apiBaseUrl) && (
+              {isString(baseUrl) && (
                 <>
                   {' '}
                   <a
                     className="CustomErrorMessage--link"
-                    href={window.apiBaseUrl}
+                    href={baseUrl}
                     target="_blank"
                     rel="noopener noreferrer"
                   >
-                    {window.apiBaseUrl}
+                    {baseUrl}
                   </a>
                 </>
               )}{' '}
diff --git a/packages/jaeger-ui/typings/custom.d.ts b/packages/jaeger-ui/typings/custom.d.ts
index 3df25a59db..56a51213f3 100644
--- a/packages/jaeger-ui/typings/custom.d.ts
+++ b/packages/jaeger-ui/typings/custom.d.ts
@@ -36,6 +36,7 @@ declare interface Window {
   ) => string | undefined;
   cancelMessageToDigma: (request_id: string) => void;
   platform?: unknown;
+  baseUrl?: unknown
   apiBaseUrl?: unknown;
   initialRoutePath?: unknown;
   embeddedMode?: unknown;