add: webhook

events/WebhookEventHandlers/MarkdownWebhookEventHandlers/MarkdownCreatedWebhookEventHandler.py

@@ -0,0 +1,20 @@
+from events import MARKDOWN_CREATED_EVENT, markdown_created
+from events.WebhookEventHandlers import auto_instantiate
+from events.WebhookEventHandlers.MarkdownWebhookEventHandlers import MarkdownWebhookEventHandler
+from misc import Singleton
+
+
+@auto_instantiate
+class MarkdownCreatedWebhookEventHandler(MarkdownWebhookEventHandler, Singleton):
+    def __init__(self):
+        if getattr(self, "_initialized", False):
+            return
+        super().__init__(MARKDOWN_CREATED_EVENT)
+        markdown_created.connect(self)
+        self._initialized = True
+
+
+
+
+
+

events/WebhookEventHandlers/MarkdownWebhookEventHandlers/MarkdownDeletedWebhookEventHandler.py

@@ -0,0 +1,15 @@
+from events import MARKDOWN_DELETED_EVENT, markdown_deleted
+from events.WebhookEventHandlers import auto_instantiate
+from events.WebhookEventHandlers.MarkdownWebhookEventHandlers import MarkdownWebhookEventHandler
+from misc import Singleton
+
+
+@auto_instantiate
+class MarkdownDeletedWebhookEventHandler(MarkdownWebhookEventHandler, Singleton):
+    def __init__(self):
+        if getattr(self, "_initialized", False):
+            return
+        super().__init__(MARKDOWN_DELETED_EVENT)
+        markdown_deleted.connect(self)
+        self._initialized = True
+

events/WebhookEventHandlers/MarkdownWebhookEventHandlers/MarkdownUpdatedWebhookEventHandler.py

@@ -0,0 +1,20 @@
+from events import MARKDOWN_UPDATED_EVENT, markdown_updated
+from events.WebhookEventHandlers import auto_instantiate
+from events.WebhookEventHandlers.MarkdownWebhookEventHandlers import MarkdownWebhookEventHandler
+from misc import Singleton
+
+
+@auto_instantiate
+class MarkdownUpdatedWebhookEventHandler(MarkdownWebhookEventHandler, Singleton):
+    def __init__(self):
+        if getattr(self, "_initialized", False):
+            return
+        super().__init__(MARKDOWN_UPDATED_EVENT)
+        markdown_updated.connect(self)
+        self._initialized = True
+
+
+
+
+
+

events/WebhookEventHandlers/MarkdownWebhookEventHandlers/__init__.py

@@ -0,0 +1,9 @@
+from events.WebhookEventHandlers import WebhookEventHandler
+
+
+class MarkdownWebhookEventHandler(WebhookEventHandler):
+    def __init__(self, event_type):
+        super().__init__(event_type)
+
+    def get_path_id(self, payload):
+        return payload["path_id"]

events/WebhookEventHandlers/PathWebhookEventHandlers/PathCreatedWebhookEventHandler.py

@@ -0,0 +1,26 @@
+from events import PATH_CREATED_EVENT, path_created
+from events.WebhookEventHandlers import auto_instantiate
+from events.WebhookEventHandlers.PathWebhookEventHandlers import PathWebhookEventHandler
+from misc import Singleton
+
+
+@auto_instantiate
+class PathCreatedWebhookEventHandler(PathWebhookEventHandler,Singleton):
+
+    _instance = None
+    def __new__(cls, *args, **kwargs):
+        if cls._instance is None:
+            cls._instance = super(PathCreatedWebhookEventHandler, cls).__new__(cls)
+        return cls._instance
+    def __init__(self):
+        if getattr(self, "_initialized", False):
+            return
+        super().__init__(PATH_CREATED_EVENT)
+        path_created.connect(self)
+        self._initialized = True
+
+
+
+
+
+

events/WebhookEventHandlers/PathWebhookEventHandlers/PathDeletedWebhookEventHandler.py

@@ -0,0 +1,20 @@
+from events import PATH_DELETED_EVENT, path_deleted
+from events.WebhookEventHandlers import auto_instantiate
+from events.WebhookEventHandlers.PathWebhookEventHandlers import PathWebhookEventHandler
+from misc import Singleton
+
+
+@auto_instantiate
+class PathDeletedWebhookEventHandler(PathWebhookEventHandler, Singleton):
+    _instance = None
+    def __new__(cls, *args, **kwargs):
+        if cls._instance is None:
+            cls._instance = super(PathDeletedWebhookEventHandler, cls).__new__(cls)
+        return cls._instance
+    def __init__(self):
+        if getattr(self, "_initialized", False):
+            return
+        super().__init__(PATH_DELETED_EVENT)
+        path_deleted.connect(self)
+        self._initialized = True
+

events/WebhookEventHandlers/PathWebhookEventHandlers/PathUpdatedWebhookEventHandler.py

@@ -0,0 +1,26 @@
+from events import PATH_UPDATED_EVENT, path_updated
+from events.WebhookEventHandlers import auto_instantiate
+from events.WebhookEventHandlers.PathWebhookEventHandlers import PathWebhookEventHandler
+from misc import Singleton
+
+
+@auto_instantiate
+class PathUpdatedWebhookEventHandler(PathWebhookEventHandler, Singleton):
+
+    _instance = None
+    def __new__(cls, *args, **kwargs):
+        if cls._instance is None:
+            cls._instance = super(PathUpdatedWebhookEventHandler, cls).__new__(cls)
+        return cls._instance
+    def __init__(self):
+        if getattr(self, "_initialized", False):
+            return
+        super().__init__(PATH_UPDATED_EVENT)
+        path_updated.connect(self)
+        self._initialized = True
+
+
+
+
+
+

events/WebhookEventHandlers/PathWebhookEventHandlers/__init__.py

@@ -0,0 +1,9 @@
+from events.WebhookEventHandlers import WebhookEventHandler
+
+
+class PathWebhookEventHandler(WebhookEventHandler):
+    def __init__(self, event_type):
+        super().__init__(event_type)
+
+    def get_path_id(self, payload):
+        return payload["id"]

events/WebhookEventHandlers/__init__.py

@@ -0,0 +1,69 @@
+from sqlalchemy.orm import Session
+from db.models.Path import Path
+from db.models.Webhook import Webhook
+from db.models.WebhookSetting import WebhookSetting
+import abc
+import importlib
+import json
+import pkgutil
+import requests
+import db
+
+class WebhookEventHandler(abc.ABC):
+    def __init__(self, event_type=0):
+        self.event_type = event_type
+
+    @abc.abstractmethod
+    def get_path_id(self, payload):
+        pass
+
+    def __call__(self, payload):
+        path_id = self.get_path_id(payload)
+        with db.get_db() as session:
+            setting = self.get_setting(session, path_id)
+            if setting is None:
+                return
+            headers = {'Content-Type': 'application/json'}
+            if setting["additional_headers"] is not None:
+                headers.update(json.loads(setting["additional_headers"]))
+            try:
+                response = requests.post(setting["webhook_url"], json=payload, headers=headers, timeout=5)
+                response.raise_for_status()
+            except requests.RequestException as e:
+                print(e)
+
+    def get_setting(self, session: Session, path_id):
+
+        path = session.query(Path).filter(Path.id == path_id).first()
+        if path is None:
+            return None
+        p = path.to_dict()
+        webhook_setting = session.query(WebhookSetting).filter(WebhookSetting.path_id == path_id).first()
+
+        if webhook_setting is None and p["parent_id"] != 1:
+            return self.get_setting(session, p["parent_id"])
+
+        setting = webhook_setting.to_dict()
+        if not setting["enabled"] or setting["on_events"] & self.event_type == 0:
+            return None
+        webhook = session.query(Webhook).filter(Webhook.id == webhook_setting.webhook_id).first()
+        if webhook is None:
+            return None
+        setting["webhook_url"] = webhook.to_dict()["hook_url"]
+        return setting
+
+_auto_instantiate_classes = set()
+
+def auto_instantiate(cls):
+    _auto_instantiate_classes.add(cls)
+    return cls
+
+def register_all_webhook_event_handlers():
+    package = __name__
+    package_path = __path__
+
+    for finder, name, ispkg in pkgutil.walk_packages(package_path, package+"."):
+        importlib.import_module(name)
+
+    for cls in _auto_instantiate_classes:
+        cls()