Corp/backend/communications/utils.py

"""
Утилиты для модуля communications.
Переиспользуемые функции для API, WebSocket и других компонентов.
"""
from __future__ import annotations

import datetime
from datetime import datetime as dt
from datetime import timezone as dt_tz

from django.utils import timezone
from django.utils.dateparse import parse_datetime

from .models import Chat, ChatMembership


def _coerce_ts(val: str | None) -> datetime.datetime:
    """
    Принимает миллисекунды/секунды с эпохи или ISO-дату.
    Возвращает aware-дату (UTC). Фоллбек — timezone.now().
    
    Args:
        val: Строка с timestamp (ms/sec) или ISO-дата
        
    Returns:
        datetime: Aware datetime объект в UTC
        
    Examples:
        >>> _coerce_ts("1640000000000")  # миллисекунды
        datetime.datetime(2021, 12, 20, 13, 33, 20, tzinfo=UTC)
        
        >>> _coerce_ts("1640000000")  # секунды
        datetime.datetime(2021, 12, 20, 13, 33, 20, tzinfo=UTC)
        
        >>> _coerce_ts("2021-12-20T13:33:20Z")  # ISO
        datetime.datetime(2021, 12, 20, 13, 33, 20, tzinfo=UTC)
    """
    if not val:
        return timezone.now()
    
    # Сначала пробуем число (sec/ms)
    try:
        iv = int(val)
        if iv > 10**12:   # пришли миллисекунды
            iv //= 1000
        return dt.fromtimestamp(iv, tz=dt_tz.utc)
    except Exception:
        pass
    
    # Потом пробуем ISO-строку
    d = parse_datetime(val)
    if d is None:
        return timezone.now()
    if timezone.is_naive(d):
        return timezone.make_aware(d, timezone=timezone.get_current_timezone())
    return d


def user_can_access_chat(chat: Chat, user) -> bool:
    """
    Проверяет, имеет ли пользователь доступ к чату.
    
    Правила доступа по типам чатов:
    - global: доступ всем
    - private: только участники (participants)
    - group: участники (participants) или ChatMembership
    - channel/announcement: include_all_users или participants/membership
    - comments: context-based доступ через get_participants()
    
    Args:
        chat: Объект Chat
        user: Объект User
        
    Returns:
        bool: True если пользователь имеет доступ
        
    Note:
        Использует прямые запросы к БД в обход prefetch cache
    """
    if not chat or not user:
        return False

    if chat.type == "global":
        return True

    if chat.type == "private":
        # Личные чаты только через participants
        # Обходим prefetch cache используя прямой запрос
        from .models import Chat
        return Chat.objects.filter(pk=chat.pk, participants=user).exists()

    if chat.type == "group":
        # Групповые чаты с контекстным объектом используют resolver
        if chat.context_object_id:
            try:
                participants = chat.get_participants()
                return participants.filter(pk=user.pk).exists()
            except Exception as e:
                import logging
                logger = logging.getLogger(__name__)
                logger.warning(
                    f"[user_can_access_chat] Failed to resolve for chat "
                    f"{chat.id}: {e}"
                )
        
        # Обычные групповые чаты через participants ИЛИ ChatMembership
        from .models import Chat
        in_participants = Chat.objects.filter(
            pk=chat.pk, participants=user
        ).exists()
        in_membership = ChatMembership.objects.filter(
            chat=chat, user=user
        ).exists()
        return in_participants or in_membership

    if chat.type in ("channel", "announcement"):
        # Для каналов и объявлений может быть include_all_users или membership
        if chat.include_all_users:
            return user.is_active
        # Проверяем и participants и membership для гибкости
        from .models import Chat
        return (
            Chat.objects.filter(pk=chat.pk, participants=user).exists()
            or ChatMembership.objects.filter(chat=chat, user=user).exists()
        )
    
    if chat.type == "comments":
        # Чаты-комментарии ВСЕГДА используют context-based доступ
        if chat.context_object_id:
            try:
                # get_participants() вызовет resolver который учитывает
                # права доступа к контекстному объекту
                participants = chat.get_participants()
                return participants.filter(pk=user.pk).exists()
            except Exception as e:
                import logging
                logger = logging.getLogger(__name__)
                logger.warning(
                    f"[user_can_access_chat] Failed to resolve for "
                    f"comments chat {chat.id}: {e}"
                )
                return False
        
        # Fallback для старых чатов без контекста
        from .models import Chat
        return (
            Chat.objects.filter(pk=chat.pk, participants=user).exists()
            or ChatMembership.objects.filter(chat=chat, user=user).exists()
        )

    return False