From e9df89765d890a59ca64c543a8ae058a2f40c992 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mose=20M=C3=BCller?= <mosemueller@gmail.com>
Date: Mon, 27 Nov 2023 15:38:35 +0100
Subject: [PATCH] adds SocketIOHandler emitting error messages via
 socketio.AsyncServer

---
 src/pydase/utils/logging.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/src/pydase/utils/logging.py b/src/pydase/utils/logging.py
index e7d1abb..60de7ff 100644
--- a/src/pydase/utils/logging.py
+++ b/src/pydase/utils/logging.py
@@ -1,8 +1,10 @@
+import asyncio
 import logging
 import sys
 from copy import copy
 from typing import Optional
 
+import socketio
 import uvicorn.logging
 from uvicorn.config import LOGGING_CONFIG
 
@@ -34,6 +36,29 @@ class DefaultFormatter(uvicorn.logging.ColourizedFormatter):
         return sys.stderr.isatty()  # pragma: no cover
 
 
+class SocketIOHandler(logging.Handler):
+    def __init__(self, sio: socketio.AsyncServer, level: int = 40) -> None:
+        super().__init__(level)
+        self.sio = sio
+
+    def format(self, record: logging.LogRecord) -> str:
+        return f"{record.name}:{record.funcName}:{record.lineno} - {record.message}"
+
+    def emit(self, record: logging.LogRecord) -> None:
+        log_entry = self.format(record)
+
+        loop = asyncio.get_event_loop()
+        loop.create_task(
+            self.sio.emit(  # type: ignore[reportUnknownMemberType]
+                "log",
+                {
+                    "levelname": record.levelname,
+                    "message": log_entry,
+                },
+            )
+        )
+
+
 def setup_logging(level: Optional[str | int] = None) -> None:
     """
     Configures the logging settings for the application.