worked backup

backend/src/core/task_manager.py

@@ -61,6 +61,7 @@ class TaskManager:
     def __init__(self, plugin_loader):
         self.plugin_loader = plugin_loader
         self.tasks: Dict[str, Task] = {}
+        self.subscribers: Dict[str, List[asyncio.Queue]] = {}
         self.executor = ThreadPoolExecutor(max_workers=5) # For CPU-bound plugin execution
         self.loop = asyncio.get_event_loop()
 # [/DEF]
@@ -92,7 +93,7 @@ class TaskManager:
 
         task.status = TaskStatus.RUNNING
         task.started_at = datetime.utcnow()
-        task.logs.append(LogEntry(level="INFO", message=f"Task started for plugin '{plugin.name}'"))
+        self._add_log(task_id, "INFO", f"Task started for plugin '{plugin.name}'")
 
         try:
             # Execute plugin in a separate thread to avoid blocking the event loop
@@ -103,10 +104,10 @@ class TaskManager:
                 lambda: asyncio.run(plugin.execute(task.params)) if asyncio.iscoroutinefunction(plugin.execute) else plugin.execute(task.params)
             )
             task.status = TaskStatus.SUCCESS
-            task.logs.append(LogEntry(level="INFO", message=f"Task completed successfully for plugin '{plugin.name}'"))
+            self._add_log(task_id, "INFO", f"Task completed successfully for plugin '{plugin.name}'")
         except Exception as e:
             task.status = TaskStatus.FAILED
-            task.logs.append(LogEntry(level="ERROR", message=f"Task failed: {e}", context={"error_type": type(e).__name__}))
+            self._add_log(task_id, "ERROR", f"Task failed: {e}", {"error_type": type(e).__name__})
         finally:
             task.finished_at = datetime.utcnow()
             # In a real system, you might notify clients via WebSocket here
@@ -129,3 +130,38 @@ class TaskManager:
         """
         task = self.tasks.get(task_id)
         return task.logs if task else []
+
+    def _add_log(self, task_id: str, level: str, message: str, context: Optional[Dict[str, Any]] = None):
+        """
+        Adds a log entry to a task and notifies subscribers.
+        """
+        task = self.tasks.get(task_id)
+        if not task:
+            return
+
+        log_entry = LogEntry(level=level, message=message, context=context)
+        task.logs.append(log_entry)
+
+        # Notify subscribers
+        if task_id in self.subscribers:
+            for queue in self.subscribers[task_id]:
+                self.loop.call_soon_threadsafe(queue.put_nowait, log_entry)
+
+    async def subscribe_logs(self, task_id: str) -> asyncio.Queue:
+        """
+        Subscribes to real-time logs for a task.
+        """
+        queue = asyncio.Queue()
+        if task_id not in self.subscribers:
+            self.subscribers[task_id] = []
+        self.subscribers[task_id].append(queue)
+        return queue
+
+    def unsubscribe_logs(self, task_id: str, queue: asyncio.Queue):
+        """
+        Unsubscribes from real-time logs for a task.
+        """
+        if task_id in self.subscribers:
+            self.subscribers[task_id].remove(queue)
+            if not self.subscribers[task_id]:
+                del self.subscribers[task_id]