add validation to url mode

Signed-off-by: Adam Tackett <tackadam@amazon.com>
opensearch-project · Jan 8, 2025 · 1b5ab44 · 1b5ab44
1 parent 4ee5670
commit 1b5ab44
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/public/components/trace_analytics/home.tsx b/public/components/trace_analytics/home.tsx
@@ -210,6 +210,10 @@ export const Home = (props: HomeProps) => {
     ]);
   };
 
+  const isValidTraceAnalyticsMode = (urlMode: string | null): urlMode is TraceAnalyticsMode => {
+    return ['jaeger', 'data_prepper', 'custom_data_prepper'].includes(urlMode || '');
+  };
+
   useEffect(() => {
     handleDataPrepperIndicesExistRequest(
       props.http,
@@ -406,7 +410,8 @@ export const Home = (props: HomeProps) => {
           render={(_routerProps) => {
             const queryParams = new URLSearchParams(window.location.href.split('?')[1]);
             const traceId = queryParams.get('traceId');
-            const traceMode = queryParams.get('mode');
+            const traceModeFromURL = queryParams.get('mode');
+            const traceMode = isValidTraceAnalyticsMode(traceModeFromURL) ? traceModeFromURL : mode;
 
             const SideBarComponent = !isNavGroupEnabled ? TraceSideBar : React.Fragment;
             if (!traceId) {
@@ -451,7 +456,10 @@ export const Home = (props: HomeProps) => {
           render={(_routerProps) => {
             const queryParams = new URLSearchParams(window.location.href.split('?')[1]);
             const serviceId = queryParams.get('serviceId');
-            const serviceMode = queryParams.get('mode');
+            const serviceModeFromURL = queryParams.get('mode');
+            const serviceMode = isValidTraceAnalyticsMode(serviceModeFromURL)
+              ? serviceModeFromURL
+              : mode;
 
             const SideBarComponent = !isNavGroupEnabled ? TraceSideBar : React.Fragment;
             if (!serviceId) {