dockered

django/notice/admin.py

@@ -0,0 +1,3 @@
+from django.contrib import admin
+
+# Register your models here.

django/notice/apps.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class NoticeConfig(AppConfig):
+    default_auto_field = 'django.db.models.BigAutoField'
+    name = 'notice'

django/notice/consumers.py

@@ -0,0 +1,95 @@
+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()
+
+        for friend in consumer.user.profilemodel.get_friends():
+            self.notify_user(friend.user, {'type': 'online',
+                                           'user': ProfileSerializer(consumer.user.profilemodel).data})
+
+    def remove(self, consumer: NoticeConsumer):
+        for friend in consumer.user.profilemodel.get_friends():
+            self.notify_user(friend.user, {'type': 'offline',
+                                           'user': ProfileSerializer(consumer.user.profilemodel).data})
+        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):
+        serialized_data = ProfileSerializer(friend).data
+        if self.get_consumer_by_user(user) is not None:
+            status = 'online'
+        else:
+            status = 'offline'
+        print(status)
+
+        self.notify_user(user, {'type': 'new_friend', 'friend': serialized_data})
+        self.notify_user(user, {'type': 'online', 'user': serialized_data})
+        self.notify_user(friend.user, {'type': status, 'user': ProfileSerializer(user.profilemodel).data})
+
+    def notify_friend_removed(self, user: User, friend: ProfileModel):
+        self.notify_user(user, {'type': 'friend_removed', 'friend': ProfileSerializer(friend).data})
+
+    def ask_game(self, asker:User, asked: User):
+        self.notify_user(asker, {'type': 'game_asked', 'asker': ProfileSerializer(asked).data})
+
+    def ask_game_canceled(self, asker:User, asked: User):
+        self.notify_user(asker, {'type': 'game_canceled', 'asker': ProfileSerializer(asked).data})
+
+    def refuse_game(self, asked: User, asker: User):
+        self.notify_user(asked, {'type': 'game_refused', 'asker': ProfileSerializer(asker).data})
+
+    def accept_game(self, asked: User, asker: User):
+        self.notify_user(asked, {'type': 'game_accepted', 'asker': ProfileSerializer(asker).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)

django/notice/models.py

@@ -0,0 +1,7 @@
+from django.db.models import Model, ForeignKey, CharField, CASCADE
+from django.contrib.auth.models import User
+
+
+class NoticeModel(Model):
+    user = ForeignKey(User, on_delete=CASCADE)
+    data = CharField(max_length=100)

django/notice/routing.py

@@ -0,0 +1,7 @@
+from django.urls import re_path
+
+from .consumers import NoticeConsumer
+
+websocket_urlpatterns = [
+    re_path(r'ws/notice$', NoticeConsumer.as_asgi()),
+]

django/notice/tests.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

django/notice/views.py

@@ -0,0 +1,3 @@
+from django.shortcuts import render
+
+# Create your views here.