Fix task API stability and Playwright runtime in Docker

backend/src/core/task_manager/persistence.py

@@ -12,6 +12,7 @@ import json
 
 from sqlalchemy.orm import Session
 from ...models.task import TaskRecord, TaskLogRecord
+from ...models.mapping import Environment
 from ..database import TasksSessionLocal
 from .models import Task, TaskStatus, LogEntry, TaskLog, LogFilter, LogStats
 from ..logger import logger, belief_scope
@@ -21,6 +22,40 @@ from ..logger import logger, belief_scope
 # @SEMANTICS: persistence, service, database, sqlalchemy
 # @PURPOSE: Provides methods to save and load tasks from the tasks.db database using SQLAlchemy.
 class TaskPersistenceService:
+    @staticmethod
+    def _json_load_if_needed(value):
+        if value is None:
+            return None
+        if isinstance(value, (dict, list)):
+            return value
+        if isinstance(value, str):
+            stripped = value.strip()
+            if stripped == "" or stripped.lower() == "null":
+                return None
+            try:
+                return json.loads(stripped)
+            except json.JSONDecodeError:
+                return value
+        return value
+
+    @staticmethod
+    def _parse_datetime(value):
+        if value is None or isinstance(value, datetime):
+            return value
+        if isinstance(value, str):
+            try:
+                return datetime.fromisoformat(value)
+            except ValueError:
+                return None
+        return None
+
+    @staticmethod
+    def _resolve_environment_id(session: Session, env_id: Optional[str]) -> Optional[str]:
+        if not env_id:
+            return None
+        exists = session.query(Environment.id).filter(Environment.id == env_id).first()
+        return env_id if exists else None
+
     # [DEF:__init__:Function]
     # @PURPOSE: Initializes the persistence service.
     # @PRE:     None.
@@ -48,7 +83,8 @@ class TaskPersistenceService:
                 
                 record.type = task.plugin_id
                 record.status = task.status.value
-                record.environment_id = task.params.get("environment_id") or task.params.get("source_env_id")
+                raw_env_id = task.params.get("environment_id") or task.params.get("source_env_id")
+                record.environment_id = self._resolve_environment_id(session, raw_env_id)
                 record.started_at = task.started_at
                 record.finished_at = task.finished_at
                 
@@ -123,21 +159,28 @@ class TaskPersistenceService:
                 for record in records:
                     try:
                         logs = []
-                        if record.logs:
-                            for log_data in record.logs:
-                                # Handle timestamp conversion if it's a string
-                                if isinstance(log_data.get('timestamp'), str):
-                                    log_data['timestamp'] = datetime.fromisoformat(log_data['timestamp'])
+                        logs_payload = self._json_load_if_needed(record.logs)
+                        if isinstance(logs_payload, list):
+                            for log_data in logs_payload:
+                                if not isinstance(log_data, dict):
+                                    continue
+                                log_data = dict(log_data)
+                                log_data['timestamp'] = self._parse_datetime(log_data.get('timestamp')) or datetime.utcnow()
                                 logs.append(LogEntry(**log_data))
 
+                        started_at = self._parse_datetime(record.started_at)
+                        finished_at = self._parse_datetime(record.finished_at)
+                        params = self._json_load_if_needed(record.params)
+                        result = self._json_load_if_needed(record.result)
+
                         task = Task(
                             id=record.id,
                             plugin_id=record.type,
                             status=TaskStatus(record.status),
-                            started_at=record.started_at,
-                            finished_at=record.finished_at,
-                            params=record.params or {},
-                            result=record.result,
+                            started_at=started_at,
+                            finished_at=finished_at,
+                            params=params if isinstance(params, dict) else {},
+                            result=result,
                             logs=logs
                         )
                         loaded_tasks.append(task)
@@ -381,4 +424,4 @@ class TaskLogPersistenceService:
     # [/DEF:delete_logs_for_tasks:Function]
 
 # [/DEF:TaskLogPersistenceService:Class]
-# [/DEF:TaskPersistenceModule:Module]
+# [/DEF:TaskPersistenceModule:Module]