from __future__ import annotations
import json

from channels.generic.websocket import WebsocketConsumer
from django.contrib.auth.models import User

from profiles.serializers import ProfileSerializer
from profiles.models import ProfileModel
from .models import NoticeModel


class NoticeManager:
    def __init__(self):
        self._list: list[NoticeConsumer] = []

    def add(self, consumer: NoticeConsumer):
        self._list.append(consumer)

        unsend_notices = NoticeModel.objects.filter(user=consumer.user)
        for notice in unsend_notices:
            consumer.send(notice.data)
            notice.delete()

    def remove(self, consumer: NoticeConsumer):
        self._list.remove(consumer)

    def get_consumer_by_user(self, user: User):
        for consumer in self._list:
            if consumer.user == user:
                return consumer

    def notify_user(self, user: User, data: dict):
        consumer = self.get_consumer_by_user(user)
        data_str: str = json.dumps(data)
        if consumer:
            consumer.send(data_str)
        else:
            NoticeModel(user=user, data=data_str).save()

    def notify_friend_request(self, user: User, friend: ProfileModel):
        self.notify_user(user, {'type': 'friend_request', 'author': ProfileSerializer(friend).data})

    def notify_friend_request_canceled(self, user: User, friend: ProfileModel):
        self.notify_user(user, {'type': 'friend_request_canceled', 'author': ProfileSerializer(friend).data})

    def notify_new_friend(self, user: User, friend: ProfileModel):
        self.notify_user(user, {'type': 'new_friend', 'friend': ProfileSerializer(friend).data})

    def notify_friend_removed(self, user: User, friend: ProfileModel):
        self.notify_user(user, {'type': 'friend_removed', 'friend': ProfileSerializer(friend).data})


notice_manager = NoticeManager()


class NoticeConsumer(WebsocketConsumer):
    def connect(self):
        self.user: User = self.scope['user']
        if not self.user.is_authenticated:
            self.close()
            return

        self.accept()
        notice_manager.add(self)

    def disconnect(self, code):
        notice_manager.remove(self)
        super().disconnect(code)