"""
Strategy Infrastructure — Causal Link Repository (Django ORM).
"""

from typing import List, Optional
from uuid import UUID

from ...domain.entities.causal_link import CausalLink
from ...domain.repositories.interfaces import CausalLinkRepository
from ..persistence.models import CausalLinkModel
from .mappers import CausalLinkMapper


class DjangoCausalLinkRepository(CausalLinkRepository):

    def find_by_id(self, link_id: UUID) -> Optional[CausalLink]:
        try:
            model = CausalLinkModel.objects.get(id=link_id)
            return CausalLinkMapper.to_entity(model)
        except CausalLinkModel.DoesNotExist:
            return None

    def find_by_map(self, map_id: UUID) -> List[CausalLink]:
        qs = CausalLinkModel.objects.filter(strategy_map_id=map_id)
        return [CausalLinkMapper.to_entity(m) for m in qs]

    def save(self, entity: CausalLink) -> CausalLink:
        try:
            model = CausalLinkModel.objects.get(id=entity.id)
            model = CausalLinkMapper.to_model(entity, model)
        except CausalLinkModel.DoesNotExist:
            model = CausalLinkMapper.to_model(entity)
        model.save()
        return CausalLinkMapper.to_entity(model)

    def delete(self, link_id: UUID) -> bool:
        deleted, _ = CausalLinkModel.objects.filter(id=link_id).delete()
        return deleted > 0