[frontend] Implement missing enrich buttons (#6265)

OpenCTI-Platform · Mar 4, 2024 · 1cf4cac · 1cf4cac
1 parent a3a61cc
commit 1cf4cac
Show file tree

Hide file tree

Showing 4 changed files with 72 additions and 4 deletions.
diff --git a/opencti-platform/opencti-front/src/private/components/arsenal/channels/ChannelPopover.jsx b/opencti-platform/opencti-front/src/private/components/arsenal/channels/ChannelPopover.jsx
@@ -12,12 +12,13 @@ import DialogContentText from '@mui/material/DialogContentText';
 import MoreVert from '@mui/icons-material/MoreVert';
 import { graphql } from 'react-relay';
 import ToggleButton from '@mui/material/ToggleButton';
+import StixCoreObjectEnrichment from '../../common/stix_core_objects/StixCoreObjectEnrichment';
 import inject18n from '../../../../components/i18n';
 import { commitMutation, QueryRenderer } from '../../../../relay/environment';
 import { channelEditionQuery } from './ChannelEdition';
 import ChannelEditionContainer from './ChannelEditionContainer';
 import Security from '../../../../utils/Security';
-import { KNOWLEDGE_KNUPDATE_KNDELETE } from '../../../../utils/hooks/useGranted';
+import { KNOWLEDGE_KNENRICHMENT, KNOWLEDGE_KNUPDATE_KNDELETE } from '../../../../utils/hooks/useGranted';
 import Transition from '../../../../components/Transition';
 
 const ChannelPopoverDeletionMutation = graphql`
@@ -33,6 +34,7 @@ class ChannelPopover extends Component {
       anchorEl: null,
       displayDelete: false,
       displayEdit: false,
+      displayEnrichment: false,
       deleting: false,
     };
   }
@@ -78,6 +80,15 @@ class ChannelPopover extends Component {
     this.setState({ displayEdit: false });
   }
 
+  handleOpenEnrichment() {
+    this.setState({ displayEnrichment: true });
+    this.handleClose();
+  }
+
+  handleCloseEnrichment() {
+    this.setState({ displayEnrichment: false });
+  }
+
   render() {
     const { t, id } = this.props;
     return (
@@ -98,12 +109,18 @@ class ChannelPopover extends Component {
           <MenuItem onClick={this.handleOpenEdit.bind(this)}>
             {t('Update')}
           </MenuItem>
+          <Security needs={[KNOWLEDGE_KNENRICHMENT]}>
+            <MenuItem onClick={this.handleOpenEnrichment.bind(this)}>
+              {t('Enrich')}
+            </MenuItem>
+          </Security>
           <Security needs={[KNOWLEDGE_KNUPDATE_KNDELETE]}>
             <MenuItem onClick={this.handleOpenDelete.bind(this)}>
               {t('Delete')}
             </MenuItem>
           </Security>
         </Menu>
+        <StixCoreObjectEnrichment stixCoreObjectId={id} open={this.state.displayEnrichment} handleClose={this.handleCloseEnrichment.bind(this)} />
         <Dialog
           PaperProps={{ elevation: 1 }}
           open={this.state.displayDelete}

diff --git a/opencti-platform/opencti-front/src/private/components/arsenal/tools/ToolPopover.jsx b/opencti-platform/opencti-front/src/private/components/arsenal/tools/ToolPopover.jsx
@@ -12,12 +12,13 @@ import DialogContentText from '@mui/material/DialogContentText';
 import MoreVert from '@mui/icons-material/MoreVert';
 import { graphql } from 'react-relay';
 import ToggleButton from '@mui/material/ToggleButton';
+import StixCoreObjectEnrichment from '../../common/stix_core_objects/StixCoreObjectEnrichment';
 import inject18n from '../../../../components/i18n';
 import { commitMutation, QueryRenderer } from '../../../../relay/environment';
 import { toolEditionQuery } from './ToolEdition';
 import ToolEditionContainer from './ToolEditionContainer';
 import Security from '../../../../utils/Security';
-import { KNOWLEDGE_KNUPDATE_KNDELETE } from '../../../../utils/hooks/useGranted';
+import { KNOWLEDGE_KNENRICHMENT, KNOWLEDGE_KNUPDATE_KNDELETE } from '../../../../utils/hooks/useGranted';
 import Transition from '../../../../components/Transition';
 
 const ToolPopoverDeletionMutation = graphql`
@@ -35,6 +36,7 @@ class ToolPopover extends Component {
       anchorEl: null,
       displayDelete: false,
       displayEdit: false,
+      displayEnrichment: false,
       deleting: false,
     };
   }
@@ -80,6 +82,15 @@ class ToolPopover extends Component {
     this.setState({ displayEdit: false });
   }
 
+  handleOpenEnrichment() {
+    this.setState({ displayEnrichment: true });
+    this.handleClose();
+  }
+
+  handleCloseEnrichment() {
+    this.setState({ displayEnrichment: false });
+  }
+
   render() {
     const { t, id } = this.props;
     return (
@@ -100,12 +111,18 @@ class ToolPopover extends Component {
           <MenuItem onClick={this.handleOpenEdit.bind(this)}>
             {t('Update')}
           </MenuItem>
+          <Security needs={[KNOWLEDGE_KNENRICHMENT]}>
+            <MenuItem onClick={this.handleOpenEnrichment.bind(this)}>
+              {t('Enrich')}
+            </MenuItem>
+          </Security>
           <Security needs={[KNOWLEDGE_KNUPDATE_KNDELETE]}>
             <MenuItem onClick={this.handleOpenDelete.bind(this)}>
               {t('Delete')}
             </MenuItem>
           </Security>
         </Menu>
+        <StixCoreObjectEnrichment stixCoreObjectId={id} open={this.state.displayEnrichment} handleClose={this.handleCloseEnrichment.bind(this)} />
         <Dialog
           PaperProps={{ elevation: 1 }}
           open={this.state.displayDelete}

diff --git a/...orm/opencti-front/src/private/components/arsenal/vulnerabilities/VulnerabilityPopover.jsx b/...orm/opencti-front/src/private/components/arsenal/vulnerabilities/VulnerabilityPopover.jsx
@@ -12,12 +12,13 @@ import DialogContentText from '@mui/material/DialogContentText';
 import MoreVert from '@mui/icons-material/MoreVert';
 import { graphql } from 'react-relay';
 import ToggleButton from '@mui/material/ToggleButton';
+import StixCoreObjectEnrichment from '../../common/stix_core_objects/StixCoreObjectEnrichment';
 import inject18n from '../../../../components/i18n';
 import { commitMutation, QueryRenderer } from '../../../../relay/environment';
 import { vulnerabilityEditionQuery } from './VulnerabilityEdition';
 import VulnerabilityEditionContainer from './VulnerabilityEditionContainer';
 import Security from '../../../../utils/Security';
-import { KNOWLEDGE_KNUPDATE_KNDELETE } from '../../../../utils/hooks/useGranted';
+import { KNOWLEDGE_KNENRICHMENT, KNOWLEDGE_KNUPDATE_KNDELETE } from '../../../../utils/hooks/useGranted';
 import Transition from '../../../../components/Transition';
 
 const VulnerabilityPopoverDeletionMutation = graphql`
@@ -35,6 +36,7 @@ class VulnerabilityPopover extends Component {
       anchorEl: null,
       displayDelete: false,
       displayEdit: false,
+      displayEnrichment: false,
       deleting: false,
     };
   }
@@ -80,6 +82,15 @@ class VulnerabilityPopover extends Component {
     this.setState({ displayEdit: false });
   }
 
+  handleOpenEnrichment() {
+    this.setState({ displayEnrichment: true });
+    this.handleClose();
+  }
+
+  handleCloseEnrichment() {
+    this.setState({ displayEnrichment: false });
+  }
+
   render() {
     const { t, id } = this.props;
     return (
@@ -100,12 +111,18 @@ class VulnerabilityPopover extends Component {
           <MenuItem onClick={this.handleOpenEdit.bind(this)}>
             {t('Update')}
           </MenuItem>
+          <Security needs={[KNOWLEDGE_KNENRICHMENT]}>
+            <MenuItem onClick={this.handleOpenEnrichment.bind(this)}>
+              {t('Enrich')}
+            </MenuItem>
+          </Security>
           <Security needs={[KNOWLEDGE_KNUPDATE_KNDELETE]}>
             <MenuItem onClick={this.handleOpenDelete.bind(this)}>
               {t('Delete')}
             </MenuItem>
           </Security>
         </Menu>
+        <StixCoreObjectEnrichment stixCoreObjectId={id} open={this.state.displayEnrichment} handleClose={this.handleCloseEnrichment.bind(this)} />
         <Dialog
           PaperProps={{ elevation: 1 }}
           open={this.state.displayDelete}

diff --git a/...latform/opencti-front/src/private/components/observations/indicators/IndicatorPopover.jsx b/...latform/opencti-front/src/private/components/observations/indicators/IndicatorPopover.jsx
@@ -12,12 +12,13 @@ import DialogContent from '@mui/material/DialogContent';
 import DialogContentText from '@mui/material/DialogContentText';
 import MoreVert from '@mui/icons-material/MoreVert';
 import { graphql } from 'react-relay';
+import StixCoreObjectEnrichment from '../../common/stix_core_objects/StixCoreObjectEnrichment';
 import inject18n from '../../../../components/i18n';
 import { commitMutation, QueryRenderer } from '../../../../relay/environment';
 import { indicatorEditionQuery } from './IndicatorEdition';
 import IndicatorEditionContainer from './IndicatorEditionContainer';
 import Security from '../../../../utils/Security';
-import { KNOWLEDGE_KNUPDATE_KNDELETE } from '../../../../utils/hooks/useGranted';
+import { KNOWLEDGE_KNENRICHMENT, KNOWLEDGE_KNUPDATE_KNDELETE } from '../../../../utils/hooks/useGranted';
 import Transition from '../../../../components/Transition';
 
 const IndicatorPopoverDeletionMutation = graphql`
@@ -33,6 +34,7 @@ class IndicatorPopover extends Component {
       anchorEl: null,
       displayDelete: false,
       displayEdit: false,
+      displayEnrichment: false,
       deleting: false,
     };
   }
@@ -78,6 +80,15 @@ class IndicatorPopover extends Component {
     this.setState({ displayEdit: false });
   }
 
+  handleOpenEnrichment() {
+    this.setState({ displayEnrichment: true });
+    this.handleClose();
+  }
+
+  handleCloseEnrichment() {
+    this.setState({ displayEnrichment: false });
+  }
+
   render() {
     const { t, id } = this.props;
     return (
@@ -98,12 +109,18 @@ class IndicatorPopover extends Component {
           <MenuItem onClick={this.handleOpenEdit.bind(this)}>
             {t('Update')}
           </MenuItem>
+          <Security needs={[KNOWLEDGE_KNENRICHMENT]}>
+            <MenuItem onClick={this.handleOpenEnrichment.bind(this)}>
+              {t('Enrich')}
+            </MenuItem>
+          </Security>
           <Security needs={[KNOWLEDGE_KNUPDATE_KNDELETE]}>
             <MenuItem onClick={this.handleOpenDelete.bind(this)}>
               {t('Delete')}
             </MenuItem>
           </Security>
         </Menu>
+        <StixCoreObjectEnrichment stixCoreObjectId={id} open={this.state.displayEnrichment} handleClose={this.handleCloseEnrichment.bind(this)} />
         <Dialog
           open={this.state.displayDelete}
           PaperProps={{ elevation: 1 }}