From cdcfd8217b89ffada5d835f353e3b2a8f7ce4e4c Mon Sep 17 00:00:00 2001
From: Adrien Servel <50013774+Kedae@users.noreply.github.com>
Date: Fri, 23 Aug 2024 09:32:06 +0200
Subject: [PATCH] [frontend] Add forgotten feature flag (#8134)

(cherry picked from commit 9eac7e2a433e7e8fe76b05593e19319a15935a27)
---
 .../private/components/data/ImportMenu.tsx    | 13 ++++++
 .../private/components/data/import/Root.jsx   | 43 +++++++++++--------
 2 files changed, 39 insertions(+), 17 deletions(-)

diff --git a/opencti-platform/opencti-front/src/private/components/data/ImportMenu.tsx b/opencti-platform/opencti-front/src/private/components/data/ImportMenu.tsx
index 1922bfccf2a4..ee219b0a87fd 100644
--- a/opencti-platform/opencti-front/src/private/components/data/ImportMenu.tsx
+++ b/opencti-platform/opencti-front/src/private/components/data/ImportMenu.tsx
@@ -1,7 +1,20 @@
 import React from 'react';
 import NavToolbarMenu, { MenuEntry } from '@components/common/menus/NavToolbarMenu';
+import useHelper from '../../../utils/hooks/useHelper';
 
 const ImportMenu = () => {
+  const { isFeatureEnable } = useHelper();
+  const isDataTableEnabled = isFeatureEnable('DATA_TABLES');
+  if (!isDataTableEnabled) {
+    return (
+      <NavToolbarMenu
+        entries={[{
+          path: '/dashboard/data/import',
+          label: 'Import',
+        }]}
+      />
+    );
+  }
   const entries: MenuEntry[] = [
     {
       path: '/dashboard/data/import',
diff --git a/opencti-platform/opencti-front/src/private/components/data/import/Root.jsx b/opencti-platform/opencti-front/src/private/components/data/import/Root.jsx
index bc88581fe3b6..1b116ba5822f 100644
--- a/opencti-platform/opencti-front/src/private/components/data/import/Root.jsx
+++ b/opencti-platform/opencti-front/src/private/components/data/import/Root.jsx
@@ -4,23 +4,32 @@ import Import from './Import';
 import WorkbenchFile from '../../common/files/workbench/WorkbenchFile';
 import ImportFilesContent from './ImportFilesContent';
 import ImportWorkbenchesContent from './ImportWorkbenchesContent';
+import useHelper from '../../../../utils/hooks/useHelper';
 
-const Root = () => (
-  <Routes>
-    <Route path="/" Component={Import} />
-    <Route
-      path="/pending/:fileId"
-      element={<WorkbenchFile />}
-    />
-    <Route
-      path="/file"
-      element={<ImportFilesContent />}
-    />
-    <Route
-      path="/workbench"
-      element={<ImportWorkbenchesContent />}
-    />
-  </Routes>
-);
+const Root = () => {
+  const { isFeatureEnable } = useHelper();
+  const isDataTableEnabled = isFeatureEnable('DATA_TABLES');
+  return (
+    <Routes>
+      <Route path="/" Component={Import} />
+      <Route
+        path="/pending/:fileId"
+        element={<WorkbenchFile />}
+      />
+      {isDataTableEnabled && (
+        <Route
+          path="/file"
+          element={<ImportFilesContent />}
+        />
+      )}
+      {isDataTableEnabled && (
+        <Route
+          path="/workbench"
+          element={<ImportWorkbenchesContent />}
+        />
+      )}
+    </Routes>
+  );
+};
 
 export default Root;