[frontend] trials

Author: Archidoit

pycti/api/opencti_api_client.py

@@ -18,6 +18,7 @@
 from pycti.api.opencti_api_trash import OpenCTIApiTrash
 from pycti.api.opencti_api_work import OpenCTIApiWork
 from pycti.api.opencti_api_workspace import OpenCTIApiWorkspace
+from pycti.api.opencti_api_internal_file import OpenCTIApiInternalFile
 from pycti.entities.opencti_attack_pattern import AttackPattern
 from pycti.entities.opencti_campaign import Campaign
 from pycti.entities.opencti_capability import Capability
@@ -182,6 +183,7 @@ def __init__(
         self.connector = OpenCTIApiConnector(self)
         self.stix2 = OpenCTIStix2(self)
         self.pir = OpenCTIApiPir(self)
+        self.internal_file = OpenCTIApiInternalFile(self)
 
         # Define the entities
         self.vocabulary = Vocabulary(self)

pycti/api/opencti_api_internal_file.py

@@ -0,0 +1,29 @@
+class OpenCTIApiInternalFile:
+    """OpenCTIApiInternalFile"""
+
+    def __init__(self, api):
+        self.api = api
+
+    def delete(self, **kwargs):
+        fileName = kwargs.get("fileName", None)
+        self.opencti.app_logger.info(
+            "-------- fileName ----------",
+            {"filename": fileName},
+        )
+        if fileName is not None:
+            query = """
+                mutation InternalFileDelete($fileName: String) {
+                    deleteImport(fileName: $fileName)
+                }
+               """
+            self.api.query(
+                query,
+                {
+                    "fileName": fileName,
+                },
+            )
+        else:
+            self.opencti.app_logger.error(
+                "[stix_internal_file] Cant delete internal file, missing parameters: fileName"
+            )
+            return None

pycti/utils/opencti_stix2.py

@@ -941,6 +941,7 @@ def get_internal_helper(self):
             "workspace": self.opencti.workspace,
             "publicdashboard": self.opencti.public_dashboard,
             "notification": self.opencti.notification,
+            "internalfile": self.opencti.internal_file
         }
 
     def generate_standard_id_from_stix(self, data):
@@ -2598,7 +2599,22 @@ def element_operation_delete(self, item, operation):
         else:
             # Element is not knowledge we need to use the right api
             stix_helper = self.get_internal_helper().get(item["type"])
-            if stix_helper and hasattr(stix_helper, "delete"):
+
+            self.opencti.app_logger.info(stix_helper)
+            self.opencti.app_logger.info(
+                "-------- TYPE ----------",
+                {"type": item["type"]},
+            )
+            if item["type"] == "internalfile":
+                self.opencti.app_logger.info(
+                    "-------- internal file ----------",
+                    {"type": item["type"]},
+                )
+                self.opencti.app_logger.info(item)
+                fileName = self.opencti.get_attribute_in_extension("id", item)
+                self.opencti.app_logger.info(fileName)
+                stix_helper.delete(fileName=item["fileName"])
+            elif stix_helper and hasattr(stix_helper, "delete"):
                 stix_helper.delete(id=item["id"])
             else:
                 raise ValueError(