[frontend] entity markings to default file markings for upload file f…

…rom ask ai (#5823)
OpenCTI-Platform · Apr 26, 2024 · 11ba9e5 · 11ba9e5
1 parent 7f82370
commit 11ba9e5
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 7 deletions.
diff --git a/opencti-platform/opencti-front/src/private/components/common/containers/ContainerHeader.jsx b/opencti-platform/opencti-front/src/private/components/common/containers/ContainerHeader.jsx
@@ -1074,6 +1074,7 @@ const ContainerHeader = (props) => {
                 instanceId={container.id}
                 instanceType={container.entity_type}
                 instanceName={getMainRepresentative(container)}
+                instanceMarkings={container.objectMarking.map(({ id }) => id)}
                 type="container"
               />
             )}

diff --git a/...orm/opencti-front/src/private/components/common/stix_core_objects/StixCoreObjectAskAI.tsx b/...orm/opencti-front/src/private/components/common/stix_core_objects/StixCoreObjectAskAI.tsx
@@ -57,6 +57,7 @@ interface StixCoreObjectAskAiProps {
   instanceId: string;
   instanceName: string;
   instanceType: string;
+  instanceMarkings?: string[];
   type: 'container' | 'threat' | 'victim' | 'unsupported';
 }
 
@@ -98,7 +99,7 @@ const actionsExplanation = {
   'convert-files': 'Try to convert the selected files (or all files associated to this entity) in a STIX 2.1 bundle.',
 };
 
-const StixCoreObjectAskAI: FunctionComponent<StixCoreObjectAskAiProps> = ({ instanceId, instanceType, instanceName, type }) => {
+const StixCoreObjectAskAI: FunctionComponent<StixCoreObjectAskAiProps> = ({ instanceId, instanceType, instanceName, type, instanceMarkings }) => {
   const { t_i18n } = useFormatter();
   const navigate = useNavigate();
   const isEnterpriseEdition = useEnterpriseEdition();
@@ -251,10 +252,12 @@ const StixCoreObjectAskAI: FunctionComponent<StixCoreObjectAskAiProps> = ({ inst
     const file = new File([blob], fileName, {
       type,
     });
+    const fileMarkings = instanceMarkings ?? [];
     commitMutationCreateFile({
       variables: {
         id: instanceId,
         file,
+        fileMarkings,
         noTriggerImport: false,
       },
       onCompleted: (response: StixDomainObjectContentFilesUploadStixDomainObjectMutation$data) => {

diff --git a/...pencti-front/src/private/components/common/stix_domain_objects/StixDomainObjectHeader.jsx b/...pencti-front/src/private/components/common/stix_domain_objects/StixDomainObjectHeader.jsx
@@ -523,12 +523,13 @@ const StixDomainObjectHeader = (props) => {
             />
           )}
           {enableAskAi && (
-          <StixCoreObjectAskAI
-            instanceId={stixDomainObject.id}
-            instanceType={stixDomainObject.entity_type}
-            instanceName={getMainRepresentative(stixDomainObject)}
-            type={type}
-          />
+            <StixCoreObjectAskAI
+              instanceId={stixDomainObject.id}
+              instanceType={stixDomainObject.entity_type}
+              instanceName={getMainRepresentative(stixDomainObject)}
+              instanceMarkings={stixDomainObject.objectMarking.map(({ id }) => id) ?? []}
+              type={type}
+            />
           )}
           {(isKnowledgeUpdater || isKnowledgeEnricher) && (
           <div className={classes.popover}>

diff --git a/opencti-platform/opencti-front/src/private/components/threats/intrusion_sets/Root.jsx b/opencti-platform/opencti-front/src/private/components/threats/intrusion_sets/Root.jsx
@@ -45,6 +45,9 @@ const intrusionSetQuery = graphql`
       entity_type
       name
       aliases
+      objectMarking {
+          id
+      } 
       x_opencti_graph_data
       ...IntrusionSet_intrusionSet
       ...IntrusionSetKnowledge_intrusionSet