From 69bed52d613faf1905ab62c6d810cf8756302d55 Mon Sep 17 00:00:00 2001
From: tygao <tygao@amazon.com>
Date: Thu, 11 Apr 2024 16:18:25 +0800
Subject: [PATCH] hide datasource and settings menu

Signed-off-by: tygao <tygao@amazon.com>
---
 .../management_app/management_app.tsx         | 20 ++++++++++++++++---
 src/plugins/management/public/plugin.ts       |  2 ++
 2 files changed, 19 insertions(+), 3 deletions(-)

diff --git a/src/plugins/management/public/components/management_app/management_app.tsx b/src/plugins/management/public/components/management_app/management_app.tsx
index b2109ceb08ca..c78eb77c5d13 100644
--- a/src/plugins/management/public/components/management_app/management_app.tsx
+++ b/src/plugins/management/public/components/management_app/management_app.tsx
@@ -37,7 +37,8 @@ import { ManagementSection, MANAGEMENT_BREADCRUMB } from '../../utils';
 import { ManagementRouter } from './management_router';
 import { ManagementSidebarNav } from '../management_sidebar_nav';
 import { reactRouterNavigate } from '../../../../opensearch_dashboards_react/public';
-import { SectionsServiceStart } from '../../types';
+import { SectionsServiceStart, ManagementSectionId } from '../../types';
+import { ApplicationStart, WorkspacesStart } from '../../../../../core/public';
 
 import './management_app.scss';
 
@@ -51,10 +52,12 @@ export interface ManagementAppDependencies {
   sections: SectionsServiceStart;
   opensearchDashboardsVersion: string;
   setBreadcrumbs: (newBreadcrumbs: ChromeBreadcrumb[]) => void;
+  capabilities: ApplicationStart['capabilities'];
+  workspaces: WorkspacesStart;
 }
 
 export const ManagementApp = ({ dependencies, history }: ManagementAppProps) => {
-  const { setBreadcrumbs } = dependencies;
+  const { setBreadcrumbs, capabilities, workspaces } = dependencies;
   const [selectedId, setSelectedId] = useState<string>('');
   const [sections, setSections] = useState<ManagementSection[]>();
 
@@ -79,8 +82,19 @@ export const ManagementApp = ({ dependencies, history }: ManagementAppProps) =>
   );
 
   useEffect(() => {
+    // If workspace is enabled and user has entered workspace, hide advance settings and dataSource menu
+    if (capabilities.workspaces.enabled && workspaces.currentWorkspace$.getValue()) {
+      const opensearchDashboardsSection = dependencies.sections
+        .getSectionsEnabled()
+        .find((section) => section.id === ManagementSectionId.OpenSearchDashboards);
+      const disabledApps = opensearchDashboardsSection?.apps.filter(
+        (app) => app.id === 'settings' || app.id === 'dataSources'
+      );
+      disabledApps?.forEach((app) => app.disable());
+    }
+
     setSections(dependencies.sections.getSectionsEnabled());
-  }, [dependencies.sections]);
+  }, [dependencies.sections, capabilities, workspaces]);
 
   if (!sections) {
     return null;
diff --git a/src/plugins/management/public/plugin.ts b/src/plugins/management/public/plugin.ts
index 81a970a0fc48..2c1bca7907de 100644
--- a/src/plugins/management/public/plugin.ts
+++ b/src/plugins/management/public/plugin.ts
@@ -87,6 +87,8 @@ export class ManagementPlugin implements Plugin<ManagementSetup, ManagementStart
           sections: getSectionsServiceStartPrivate(),
           opensearchDashboardsVersion,
           setBreadcrumbs: coreStart.chrome.setBreadcrumbs,
+          capabilities: coreStart.application.capabilities,
+          workspaces: coreStart.workspaces,
         });
       },
     });