From 8190b1f1cbd055f0022580f9a284bd8a9e198880 Mon Sep 17 00:00:00 2001
From: Shenoy Pratik <sgguruda@amazon.com>
Date: Fri, 11 Oct 2024 09:59:17 -0700
Subject: [PATCH] fix workbench routes to support modal mounting (#401)

Signed-off-by: Shenoy Pratik <sgguruda@amazon.com>
---
 public/components/app.tsx | 73 +++++++++++++--------------------------
 1 file changed, 24 insertions(+), 49 deletions(-)

diff --git a/public/components/app.tsx b/public/components/app.tsx
index 2f90332..7a2882e 100644
--- a/public/components/app.tsx
+++ b/public/components/app.tsx
@@ -44,6 +44,24 @@ export const WorkbenchApp = ({
   const isNavGroupEnabled = coreRefs?.chrome?.navGroup.getNavGroupEnabled();
   const basePath = isNavGroupEnabled ? '/opensearch-query-workbench' : '';
 
+  const renderMain = (props: any, isAccelerationFlyoutOpen: boolean, urlDataSource: string) => {
+    return (
+      <Main
+        httpClient={http}
+        {...props}
+        setBreadcrumbs={chrome.setBreadcrumbs}
+        isAccelerationFlyoutOpen={isAccelerationFlyoutOpen}
+        urlDataSource={urlDataSource}
+        notifications={notifications}
+        savedObjects={savedObjects}
+        dataSourceEnabled={dataSourceEnabled}
+        dataSourceManagement={dataSourceManagement}
+        dataSourceMDSId={dataSourceId}
+        setActionMenu={setActionMenu}
+      />
+    );
+  };
+
   return (
     <HashRouter>
       <I18nProvider>
@@ -51,62 +69,19 @@ export const WorkbenchApp = ({
           <EuiPage>
             <EuiPageBody>
               <Switch>
-                <Route
-                  exact
-                  path={isNavGroupEnabled ? [`${basePath}`, `/`] : `/${basePath}`}
-                  render={(props) => (
-                    <Main
-                      httpClient={http}
-                      {...props}
-                      setBreadcrumbs={chrome.setBreadcrumbs}
-                      isAccelerationFlyoutOpen={false}
-                      urlDataSource=""
-                      notifications={notifications}
-                      savedObjects={savedObjects}
-                      dataSourceEnabled={dataSourceEnabled}
-                      dataSourceManagement={dataSourceManagement}
-                      dataSourceMDSId={dataSourceId}
-                      setActionMenu={setActionMenu}
-                    />
-                  )}
-                />
                 <Route
                   exact
                   path={`${basePath}/:dataSource`}
-                  render={(props) => (
-                    <Main
-                      httpClient={http}
-                      {...props}
-                      setBreadcrumbs={chrome.setBreadcrumbs}
-                      isAccelerationFlyoutOpen={false}
-                      urlDataSource={props.match.params.dataSource}
-                      notifications={notifications}
-                      savedObjects={savedObjects}
-                      dataSourceEnabled={dataSourceEnabled}
-                      dataSourceManagement={dataSourceManagement}
-                      dataSourceMDSId={dataSourceId}
-                      setActionMenu={setActionMenu}
-                    />
-                  )}
+                  render={(props) => renderMain(props, false, props.match.params.dataSource)}
                 />
                 <Route
                   exact
                   path={`${basePath}/accelerate/:dataSource`}
-                  render={(props) => (
-                    <Main
-                      httpClient={http}
-                      {...props}
-                      setBreadcrumbs={chrome.setBreadcrumbs}
-                      isAccelerationFlyoutOpen={true}
-                      urlDataSource={props.match.params.dataSource}
-                      notifications={notifications}
-                      savedObjects={savedObjects}
-                      dataSourceEnabled={dataSourceEnabled}
-                      dataSourceManagement={dataSourceManagement}
-                      dataSourceMDSId={dataSourceId}
-                      setActionMenu={setActionMenu}
-                    />
-                  )}
+                  render={(props) => renderMain(props, true, props.match.params.dataSource)}
+                />
+                <Route
+                  path={isNavGroupEnabled ? [`${basePath}`, `/`] : `/${basePath}`}
+                  render={(props) => renderMain(props, false, '')}
                 />
               </Switch>
             </EuiPageBody>