[client] add draft api

Author: richard-julien

pycti/api/opencti_api_client.py

@@ -10,6 +10,7 @@
 
 from pycti import __version__
 from pycti.api.opencti_api_connector import OpenCTIApiConnector
+from pycti.api.opencti_api_draft import OpenCTIApiDraft
 from pycti.api.opencti_api_playbook import OpenCTIApiPlaybook
 from pycti.api.opencti_api_trash import OpenCTIApiTrash
 from pycti.api.opencti_api_work import OpenCTIApiWork
@@ -169,6 +170,7 @@ def __init__(
         # Define the dependencies
         self.work = OpenCTIApiWork(self)
         self.trash = OpenCTIApiTrash(self)
+        self.draft = OpenCTIApiDraft(self)
         self.playbook = OpenCTIApiPlaybook(self)
         self.connector = OpenCTIApiConnector(self)
         self.stix2 = OpenCTIStix2(self)

pycti/api/opencti_api_draft.py

@@ -0,0 +1,18 @@
+class OpenCTIApiDraft:
+    """OpenCTIApiDraft"""
+
+    def __init__(self, api):
+        self.api = api
+
+    def delete(self, draft_id: str):
+        query = """
+            mutation DraftWorkspaceDelete($id: ID!) {
+                draftWorkspaceDelete(id: $id)
+            }
+           """
+        self.api.query(
+            query,
+            {
+                "id": draft_id,
+            },
+        )

pycti/entities/opencti_stix_core_object.py

@@ -1888,3 +1888,26 @@ def delete(self, **kwargs):
         else:
             self.opencti.app_logger.error("[stix_core_object] Missing parameters: id")
             return None
+
+    """
+        Remove a Stix-Core-Object object from draft (revert)
+
+        :param id: the Stix-Core-Object id
+        :return void
+    """
+
+    def remove_from_draft(self, **kwargs):
+        id = kwargs.get("id", None)
+        if id is not None:
+            self.opencti.app_logger.info("Draft remove stix_core_object", {"id": id})
+            query = """
+                mutation StixCoreObjectEditDraftRemove($id: ID!) {
+                    stixCoreObjectEdit(id: $id) {
+                        removeFromDraft
+                    }
+                }
+            """
+            self.opencti.query(query, {"id": id})
+        else:
+            self.opencti.app_logger.error("[stix_core_object] Missing parameters: id")
+            return None

pycti/utils/opencti_stix2.py

@@ -2497,7 +2497,11 @@ def apply_opencti_operation(self, item, operation):
             delete_id = item["id"]
             force_delete = operation == "delete-force"
             self.opencti.stix.delete(id=delete_id)
-        elif operation == "restore":
+        elif operation == "delete-draft":
+                    self.opencti.draft.delete(item["id"])
+                elif item["opencti_operation"] == "revert-draft":
+                    self.opencti.stix_core_object.remove_from_draft(id=item["id"])
+                elif item["opencti_operation"] == "restore":
                     self.opencti.trash.delete_operation_restore(item["id"])
                 elif item["opencti_operation"] == "merge":
             target_id = item["merge_target_id"]