mimic/backend/app/serializers.py

"""JSON serializers for API responses."""
from __future__ import annotations

from typing import Any

from backend.app.models import Engagement, User
from backend.app.models.simulation import Simulation


def serialize_user(user: User) -> dict[str, Any]:
    return {
        "id": user.id,
        "username": user.username,
        "role": user.role.value,
        "created_at": user.created_at.isoformat() if user.created_at else None,
    }


def serialize_user_brief(user: User) -> dict[str, Any]:
    return {"id": user.id, "username": user.username}


def serialize_simulation(simulation: Simulation) -> dict[str, Any]:
    return {
        "id": simulation.id,
        "engagement_id": simulation.engagement_id,
        "name": simulation.name,
        "mitre_technique_id": simulation.mitre_technique_id,
        "mitre_technique_name": simulation.mitre_technique_name,
        "description": simulation.description,
        "commands": simulation.commands,
        "prerequisites": simulation.prerequisites,
        "executed_at": simulation.executed_at.isoformat() if simulation.executed_at else None,
        "execution_result": simulation.execution_result,
        "log_source": simulation.log_source,
        "logs": simulation.logs,
        "soc_comment": simulation.soc_comment,
        "incident_number": simulation.incident_number,
        "status": simulation.status.value,
        "created_at": simulation.created_at.isoformat() if simulation.created_at else None,
        "updated_at": simulation.updated_at.isoformat() if simulation.updated_at else None,
        "created_by": serialize_user_brief(simulation.created_by)  # type: ignore[arg-type]
        if simulation.created_by
        else None,
    }


def serialize_engagement(engagement: Engagement) -> dict[str, Any]:
    return {
        "id": engagement.id,
        "name": engagement.name,
        "description": engagement.description,
        "start_date": engagement.start_date.isoformat() if engagement.start_date else None,
        "end_date": engagement.end_date.isoformat() if engagement.end_date else None,
        "status": engagement.status.value,
        "created_at": engagement.created_at.isoformat() if engagement.created_at else None,
        "created_by": serialize_user_brief(engagement.created_by)  # type: ignore[arg-type]
        if engagement.created_by
        else None,
    }