#!/usr/bin/env python
"""
Скрипт для тестирования уведомлений модуля Requests.
Создает тестовые заявления и проверяет отправку уведомлений.

Использование:
    python scripts/utils/test_request_notifications.py
"""

import os
import sys

import django

# Добавляем backend в путь
sys.path.insert(0, os.path.join(os.path.dirname(__file__), "../.."))

os.environ.setdefault("DJANGO_SETTINGS_MODULE", "eusrr_backend.settings")
django.setup()

from django.contrib.auth import get_user_model

from employees.models import Department
from notifications.models import Notification
from requests_app.models import Request

Employee = get_user_model()

print("\n" + "=" * 70)
print(" ТЕСТИРОВАНИЕ УВЕДОМЛЕНИЙ МОДУЛЯ REQUESTS")
print("=" * 70 + "\n")

# Получаем тестовых пользователей
users = Employee.objects.filter(is_active=True)[:3]

if len(users) < 2:
    print("[ERROR] Нужно минимум 2 активных пользователя!")
    sys.exit(1)

author = users[0]
recipient = users[1]
cc_user = users[2] if len(users) > 2 else None

print(f"👤 Автор заявления: {author.get_full_name()} ({author.email})")
print(f"👤 Получатель: {recipient.get_full_name()} ({recipient.email})")
if cc_user:
    print(f"👤 В копии: {cc_user.get_full_name()} ({cc_user.email})")

# Получаем первый отдел (если есть)
department = Department.objects.first()
if not department:
    print("\n[WARNING] Нет отделов в системе. Создаем тестовый...")
    department = Department.objects.create(name="Тестовый отдел", head=recipient)

print(f"\n🏢 Отдел: {department.name}")
print("\n" + "=" * 70)

# Меню
print("\nВыберите тест:\n")
print("1. Новое заявление (уведомление получателям)")
print("2. Изменение статуса заявления (одобрение)")
print("3. Изменение статуса заявления (отклонение)")
print("4. Комментарий к заявлению")
print("5. Проверить все уведомления пользователя")
print("0. Выход")
print()

choice = input("Ваш выбор: ").strip()

if choice == "0":
    print("Выход.")
    sys.exit(0)

if choice == "1":
    print("\n📝 Создаем новое заявление...")

    # Создаем заявление
    request = Request.objects.create(
        employee=author,
        type="vacation",
        comment="Тестовое заявление на отпуск",
        status="pending",
        department=department,
    )

    # Устанавливаем получателей (это триггерит m2m_changed сигнал)
    request.recipients.set([recipient])
    if cc_user:
        request.cc_users.set([cc_user])

    print(f"✅ Создано заявление #{request.id}")

    # Проверяем уведомления
    notifications = Notification.objects.filter(
        recipient__in=[recipient, cc_user] if cc_user else [recipient]
    ).order_by("-timestamp")[:5]

    print(f"\n📬 Отправлено уведомлений: {notifications.count()}")
    for notif in notifications:
        print(f"  - [{notif.recipient.username}] {notif.data.get('title', 'N/A')}")
        print(f"    Verb: {notif.verb} | Read: {notif.unread}")

elif choice == "2":
    # Находим или создаем заявление
    request = Request.objects.filter(status="pending").first()

    if not request:
        print("⚠️  Нет заявлений со статусом 'pending', создаем новое...")
        request = Request.objects.create(
            employee=author,
            type="vacation",
            comment="Тестовое заявление для одобрения",
            status="pending",
            department=department,
            approver=recipient,
        )
        request.recipients.set([recipient])

    print(f"\n✅ Одобряем заявление #{request.id}...")

    request.approver = recipient
    request.status = "approved"
    request.save()

    print("✅ Заявление одобрено!")

    # Проверяем уведомления
    notifications = Notification.objects.filter(
        verb="request_approved", recipient=author
    ).order_by("-timestamp")[:3]

    print(f"\n📬 Уведомлений об одобрении: {notifications.count()}")
    for notif in notifications:
        print(f"  - [{notif.recipient.username}] {notif.data.get('title', 'N/A')}")

elif choice == "3":
    # Находим или создаем заявление
    request = Request.objects.filter(status="pending").first()

    if not request:
        print("⚠️  Нет заявлений со статусом 'pending', создаем новое...")
        request = Request.objects.create(
            employee=author,
            type="vacation",
            comment="Тестовое заявление для отклонения",
            status="pending",
            department=department,
            approver=recipient,
        )
        request.recipients.set([recipient])

    print(f"\n❌ Отклоняем заявление #{request.id}...")

    request.approver = recipient
    request.status = "rejected"
    request.save()

    print("❌ Заявление отклонено!")

    # Проверяем уведомления
    notifications = Notification.objects.filter(
        verb="request_rejected", recipient=author
    ).order_by("-timestamp")[:3]

    print(f"\n📬 Уведомлений об отклонении: {notifications.count()}")
    for notif in notifications:
        print(f"  - [{notif.recipient.username}] {notif.data.get('title', 'N/A')}")

elif choice == "4":
    # Находим или создаем заявление
    request = Request.objects.first()

    if not request:
        print("⚠️  Нет заявлений, создаем новое...")
        request = Request.objects.create(
            employee=author,
            type="vacation",
            comment="Тестовое заявление",
            status="pending",
            department=department,
        )
        request.recipients.set([recipient])

    print(f"\n💬 Добавляем комментарий к заявлению #{request.id}...")

    from requests_app.models import RequestComment

    comment = RequestComment.objects.create(
        request=request, author=recipient, text="Это тестовый комментарий к заявлению"
    )

    print("✅ Комментарий добавлен!")

    # Проверяем уведомления
    notifications = Notification.objects.filter(
        verb="request_comment", recipient=author
    ).order_by("-timestamp")[:3]

    print(f"\n📬 Уведомлений о комментарии: {notifications.count()}")
    for notif in notifications:
        print(f"  - [{notif.recipient.username}] {notif.data.get('title', 'N/A')}")
        print(f"    {notif.description[:80]}...")

elif choice == "5":
    print("\n📊 Все уведомления по заявлениям:\n")

    request_verbs = [
        "request_new",
        "request_approved",
        "request_rejected",
        "request_comment",
        "request_status_changed",
    ]

    for user in users:
        count = Notification.objects.filter(
            recipient=user, verb__in=request_verbs
        ).count()

        unread = Notification.objects.filter(
            recipient=user, verb__in=request_verbs, unread=True
        ).count()

        print(f"👤 {user.username}: {count} всего, {unread} непрочитанных")

        recent = Notification.objects.filter(
            recipient=user, verb__in=request_verbs
        ).order_by("-timestamp")[:3]

        for notif in recent:
            read_icon = "✉️" if notif.unread else "✅"
            print(f"  {read_icon} {notif.verb}: {notif.data.get('title', 'N/A')[:50]}")

else:
    print("[ERROR] Неверный выбор")
    sys.exit(1)

print("\n" + "=" * 70)
print(" ГОТОВО!")
print("=" * 70)
print("\nДля проверки уведомлений:")
print("- Откройте http://localhost:9000/ и войдите под тестовым пользователем")
print("- Проверьте колокольчик в navbar")
print("- Проверьте вкладку в Chrome DevTools 'Network' → WS (WebSocket)")
print()