from typing import Dict, Optional
from src.domain.entities.risk_task import RiskTask

class RiskTaskRepository:
    """Simple in-memory repository for tracking tasks in Phase 2."""
    _tasks: Dict[str, RiskTask] = {}

    def save(self, task: RiskTask):
        self._tasks[task.task_id] = task

    def get_by_id(self, task_id: str) -> Optional[RiskTask]:
        return self._tasks.get(task_id)

    def list_all(self):
        return list(self._tasks.values())