AI/ontology_platform/ont_platform/realtime/broadcaster.py

"""이벤트 브로드캐스터 (Phase 8).

기능:
- Neo4j 변경 이벤트 브로드캐스트
- 실시간 그래프 업데이트 알림
"""

import logging
from datetime import datetime, UTC
from typing import Dict, List, Any, Optional

from ont_platform.realtime.websocket import ConnectionManager

logger = logging.getLogger(__name__)


class EventBroadcaster:
    """이벤트 브로드캐스터."""

    def __init__(self, connection_manager: ConnectionManager):
        """초기화.

        Args:
            connection_manager: WebSocket 연결 관리자
        """
        self.manager = connection_manager

    async def broadcast_entity_created(
        self,
        org_id: str,
        entity: Dict[str, Any],
        user_id: str = "system",
    ) -> int:
        """엔티티 생성 이벤트 브로드캐스트.

        Args:
            org_id: 조직 ID
            entity: 엔티티 정보
            user_id: 생성 사용자 ID

        Returns:
            메시지 수신 클라이언트 수
        """
        message = {
            "type": "entity.created",
            "timestamp": datetime.now(UTC).isoformat(),
            "user_id": user_id,
            "entity": entity,
        }

        sent = await self.manager.broadcast(org_id, message)
        logger.info(f"Entity created event broadcast: {entity.get('id')} to {sent} clients")

        return sent

    async def broadcast_entity_updated(
        self,
        org_id: str,
        entity_id: str,
        changes: Dict[str, Any],
        user_id: str = "system",
    ) -> int:
        """엔티티 업데이트 이벤트 브로드캐스트.

        Args:
            org_id: 조직 ID
            entity_id: 엔티티 ID
            changes: 변경사항
            user_id: 업데이트 사용자 ID

        Returns:
            메시지 수신 클라이언트 수
        """
        message = {
            "type": "entity.updated",
            "timestamp": datetime.now(UTC).isoformat(),
            "user_id": user_id,
            "entity_id": entity_id,
            "changes": changes,
        }

        sent = await self.manager.broadcast(org_id, message)
        logger.info(f"Entity updated event broadcast: {entity_id} to {sent} clients")

        return sent

    async def broadcast_entity_deleted(
        self,
        org_id: str,
        entity_id: str,
        user_id: str = "system",
    ) -> int:
        """엔티티 삭제 이벤트 브로드캐스트.

        Args:
            org_id: 조직 ID
            entity_id: 엔티티 ID
            user_id: 삭제 사용자 ID

        Returns:
            메시지 수신 클라이언트 수
        """
        message = {
            "type": "entity.deleted",
            "timestamp": datetime.now(UTC).isoformat(),
            "user_id": user_id,
            "entity_id": entity_id,
        }

        sent = await self.manager.broadcast(org_id, message)
        logger.info(f"Entity deleted event broadcast: {entity_id} to {sent} clients")

        return sent

    async def broadcast_relation_created(
        self,
        org_id: str,
        relation: Dict[str, Any],
        user_id: str = "system",
    ) -> int:
        """관계 생성 이벤트 브로드캐스트.

        Args:
            org_id: 조직 ID
            relation: 관계 정보
            user_id: 생성 사용자 ID

        Returns:
            메시지 수신 클라이언트 수
        """
        message = {
            "type": "relation.created",
            "timestamp": datetime.now(UTC).isoformat(),
            "user_id": user_id,
            "relation": relation,
        }

        sent = await self.manager.broadcast(org_id, message)
        logger.info(f"Relation created event broadcast to {sent} clients")

        return sent

    async def broadcast_relation_deleted(
        self,
        org_id: str,
        relation_id: str,
        user_id: str = "system",
    ) -> int:
        """관계 삭제 이벤트 브로드캐스트.

        Args:
            org_id: 조직 ID
            relation_id: 관계 ID
            user_id: 삭제 사용자 ID

        Returns:
            메시지 수신 클라이언트 수
        """
        message = {
            "type": "relation.deleted",
            "timestamp": datetime.now(UTC).isoformat(),
            "user_id": user_id,
            "relation_id": relation_id,
        }

        sent = await self.manager.broadcast(org_id, message)
        logger.info(f"Relation deleted event broadcast to {sent} clients")

        return sent

    async def broadcast_graph_analyzed(
        self,
        org_id: str,
        analysis_type: str,
        results: Dict[str, Any],
        user_id: str = "system",
    ) -> int:
        """그래프 분석 완료 이벤트 브로드캐스트.

        Args:
            org_id: 조직 ID
            analysis_type: 분석 타입 (centrality, communities, etc.)
            results: 분석 결과
            user_id: 분석 요청 사용자 ID

        Returns:
            메시지 수신 클라이언트 수
        """
        message = {
            "type": "graph.analyzed",
            "timestamp": datetime.now(UTC).isoformat(),
            "user_id": user_id,
            "analysis_type": analysis_type,
            "results": results,
        }

        sent = await self.manager.broadcast(org_id, message)
        logger.info(f"Graph analyzed event broadcast: {analysis_type} to {sent} clients")

        return sent

    async def broadcast_llm_result(
        self,
        org_id: str,
        query: str,
        answer: str,
        user_id: str = "system",
    ) -> int:
        """LLM 쿼리 결과 브로드캐스트.

        Args:
            org_id: 조직 ID
            query: 사용자 질문
            answer: LLM 답변
            user_id: 쿼리 요청 사용자 ID

        Returns:
            메시지 수신 클라이언트 수
        """
        message = {
            "type": "llm.result",
            "timestamp": datetime.now(UTC).isoformat(),
            "user_id": user_id,
            "query": query,
            "answer": answer[:500],  # 처음 500자만
        }

        sent = await self.manager.broadcast(org_id, message)
        logger.info(f"LLM result broadcast to {sent} clients")

        return sent

    async def broadcast_error(
        self,
        org_id: str,
        error_message: str,
        error_type: str = "error",
    ) -> int:
        """에러 이벤트 브로드캐스트.

        Args:
            org_id: 조직 ID
            error_message: 에러 메시지
            error_type: 에러 타입

        Returns:
            메시지 수신 클라이언트 수
        """
        message = {
            "type": f"error.{error_type}",
            "timestamp": datetime.now(UTC).isoformat(),
            "message": error_message,
        }

        sent = await self.manager.broadcast(org_id, message)
        logger.warning(f"Error broadcast: {error_message} to {sent} clients")

        return sent

    async def broadcast_notification(
        self,
        org_id: str,
        title: str,
        message: str,
        severity: str = "info",
    ) -> int:
        """일반 알림 브로드캐스트.

        Args:
            org_id: 조직 ID
            title: 제목
            message: 메시지
            severity: 심각도 (info, warning, error)

        Returns:
            메시지 수신 클라이언트 수
        """
        broadcast_message = {
            "type": "notification",
            "timestamp": datetime.now(UTC).isoformat(),
            "title": title,
            "message": message,
            "severity": severity,
        }

        sent = await self.manager.broadcast(org_id, broadcast_message)
        logger.info(f"Notification broadcast: {title} to {sent} clients")

        return sent