From ab52db2b561f960b363de7312bf282e756a933e5 Mon Sep 17 00:00:00 2001
From: terrtia <or1994@hotmail.fr>
Date: Wed, 10 Jul 2024 15:10:07 +0200
Subject: [PATCH] fix: [trackers] fix webhook export

---
 bin/exporter/WebHookExporter.py | 44 +++++++++++++++++++--------------
 bin/trackers/Tracker_Regex.py   |  4 ++-
 bin/trackers/Tracker_Yara.py    |  4 ++-
 3 files changed, 32 insertions(+), 20 deletions(-)

diff --git a/bin/exporter/WebHookExporter.py b/bin/exporter/WebHookExporter.py
index b4391922..49fd00b3 100755
--- a/bin/exporter/WebHookExporter.py
+++ b/bin/exporter/WebHookExporter.py
@@ -7,6 +7,8 @@
 Import Content
 
 """
+import json
+import logging
 import os
 import requests
 import sys
@@ -18,10 +20,9 @@
 # Import Project packages
 #################################
 from exporter.abstract_exporter import AbstractExporter
+from lib.ail_core import get_ail_uuid
 
-# from ConfigLoader import ConfigLoader
-# from lib.objects.abstract_object import AbstractObject
-# from lib.Tracker import Tracker
+logger = logging.getLogger()
 
 class WebHookExporter(AbstractExporter, ABC):
     def __init__(self, url=''):
@@ -35,11 +36,9 @@ def _export(self, data):
         try:
             response = requests.post(self.url, json=data)
             if response.status_code >= 400:
-                print(f"Webhook request failed for {self.url}\nReason: {response.reason}")
-                # self.redis_logger.error(f"Webhook request failed for {webhook_to_post}\nReason: {response.reason}")
+                logger.error(f"Webhook request failed for {self.url}\nReason: {response.reason}")
         except Exception as e:
-            print(f"Webhook request failed for {self.url}\nReason: Something went wrong {e}")
-            # self.redis_logger.error(f"Webhook request failed for {webhook_to_post}\nReason: Something went wrong")
+            logger.error(f"Webhook request failed for {self.url}\nReason: Something went wrong {e}")
 
 
 class WebHookExporterTracker(WebHookExporter):
@@ -48,17 +47,26 @@ def __init__(self, url=''):
         super().__init__(url=url)
 
     # TODO Change exported keys
-    def export(self, tracker, obj):
+    def export(self, tracker, obj, matches=[]):
         self.set_url(tracker.get_webhook())
-        data = {'trackerId': tracker.get_uuid(),
-                'trackerType': tracker.get_type(),
-                'tags': tracker.get_tags(),
-                'tracker': tracker.get_tracked(),
-                # object
-                'itemId': obj.get_id(),
-                'itemURL': obj.get_link()}
-        # Item
-        # data['itemDate'] = obj.get_date()
-        # data["itemSource"] = obj.get_source()
+        data = {'version': 0,
+                'type': 'tracker:match',
+                'ail_uuid': get_ail_uuid(),
+                'tracker': {
+                    'uuid': tracker.get_uuid(),
+                    'type': tracker.get_type(),
+                    'tags': list(tracker.get_tags()),
+                    'tracker': tracker.get_tracked(),
+                },
+                'obj': {'type': obj.get_type(),
+                        'subtype': obj.get_subtype(r_str=True),
+                        'id': obj.get_id(),
+                        'tags': list(obj.get_tags()),
+                        'url': obj.get_link()
+                        },
+                }
+        if matches:
+            data['matches'] = matches
 
+        # data = json.dumps(data)
         self._export(data)
diff --git a/bin/trackers/Tracker_Regex.py b/bin/trackers/Tracker_Regex.py
index c23d1589..9d5f8418 100755
--- a/bin/trackers/Tracker_Regex.py
+++ b/bin/trackers/Tracker_Regex.py
@@ -133,7 +133,9 @@ def new_tracker_found(self, tracker_name, tracker_type, obj, re_matches):
                 self.exporters['mail'].export(tracker, obj, matches)
 
             if tracker.webhook_export():
-                self.exporters['webhook'].export(tracker, obj)
+                if not matches:
+                    matches = self.extract_matches(re_matches)
+                self.exporters['webhook'].export(tracker, obj, matches)
 
 
 if __name__ == "__main__":
diff --git a/bin/trackers/Tracker_Yara.py b/bin/trackers/Tracker_Yara.py
index 29542553..0bf6db4e 100755
--- a/bin/trackers/Tracker_Yara.py
+++ b/bin/trackers/Tracker_Yara.py
@@ -151,7 +151,9 @@ def yara_rules_match(self, data):
 
             # Webhook
             if tracker.webhook_export():
-                self.exporters['webhook'].export(tracker, self.obj)
+                if not matches:
+                    matches = self.extract_matches(data)
+                self.exporters['webhook'].export(tracker, self.obj, matches)
 
         return yara.CALLBACK_CONTINUE