From e188de29a0108101c070aa9bde62ca526b380ad9 Mon Sep 17 00:00:00 2001
From: Alexander-D-Karpov <alexandr.d.karpov@gmail.com>
Date: Thu, 23 Jun 2022 14:45:34 +0300
Subject: [PATCH] added algorithm to find opponent

---
 room/consumers.py | 72 +++++++++++++++++++++++++++++++++++++++++++----
 room/models.py    |  1 +
 2 files changed, 67 insertions(+), 6 deletions(-)

diff --git a/room/consumers.py b/room/consumers.py
index fa8e20f..f139aba 100644
--- a/room/consumers.py
+++ b/room/consumers.py
@@ -1,13 +1,18 @@
 import json
 
-from asgiref.sync import sync_to_async
+from asgiref.sync import sync_to_async, async_to_sync
 from channels.generic.websocket import AsyncWebsocketConsumer
+from channels.layers import get_channel_layer
 
 from game.models import Deck
 from room.models import PlayerInQueue
 
 
 class QueueConsumer(AsyncWebsocketConsumer):
+    def __init__(self, *args, **kwargs):
+        super().__init__(args, kwargs)
+        self.room_group_name = None
+
     async def connect(self):
         self.room_group_name = "queue"
 
@@ -17,6 +22,7 @@ class QueueConsumer(AsyncWebsocketConsumer):
         await self.channel_layer.group_add(self.room_group_name, self.channel_name)
 
     async def disconnect(self, close_code):
+        await self.delete_user_in_queue()
         await self.channel_layer.group_discard(self.room_group_name, self.channel_name)
 
     # Receive message from WebSocket
@@ -61,6 +67,7 @@ class QueueConsumer(AsyncWebsocketConsumer):
                                 )
                             )
                         if deck:
+                            # add to que, start finding players
                             await self.queue_connector(deck)
                             await self.send(
                                 text_data=json.dumps(
@@ -70,6 +77,37 @@ class QueueConsumer(AsyncWebsocketConsumer):
                                     }
                                 )
                             )
+                            opponent = await self.find_user_by_score()
+
+                            if not opponent:
+                                await self.send(
+                                    text_data=json.dumps(
+                                        {
+                                            "type": "INFO",
+                                            "message": "no user found, awaiting in queue",
+                                        }
+                                    )
+                                )
+                            else:
+                                # add to group and send message that opponent found to players
+                                channel_layer = get_channel_layer()
+
+                                await channel_layer.send(
+                                    opponent[0],
+                                    {
+                                        "type": "info",
+                                        "message": f"user found, with score {self.scope['score']}",
+                                    },
+                                )
+
+                                await self.send(
+                                    text_data=json.dumps(
+                                        {
+                                            "type": "INFO",
+                                            "message": f"user found, with score {opponent[1]}",
+                                        }
+                                    )
+                                )
                         else:
                             await self.send(
                                 text_data=json.dumps(
@@ -80,6 +118,23 @@ class QueueConsumer(AsyncWebsocketConsumer):
                                 )
                             )
 
+    @sync_to_async
+    def delete_user_in_queue(self):
+        try:
+            PlayerInQueue.objects.get(player_id=self.scope["player"]).delete()
+        except PlayerInQueue.DoesNotExist:
+            return False
+
+    @sync_to_async
+    def find_user_by_score(self):
+        s_min = self.scope["score"] * 0.95
+        s_max = self.scope["score"] * 1.05
+        for el in PlayerInQueue.objects.all():
+            if el.player_id != self.scope["player"]:
+                if s_min <= el.score <= s_max:
+                    return el.channel_name, el.score
+        return False
+
     @sync_to_async
     def check_user_deck(self, deck_id: int):
         try:
@@ -95,16 +150,22 @@ class QueueConsumer(AsyncWebsocketConsumer):
         try:
             queue = PlayerInQueue.objects.get(player_id=self.scope["player"])
             queue.score = deck.score()
+            queue.channel_name = self.channel_name
+            queue.save()
+
         except PlayerInQueue.DoesNotExist:
             queue = PlayerInQueue.objects.create(
-                player_id=self.scope["player"], score=deck.score()
+                player_id=self.scope["player"],
+                score=deck.score(),
+                channel_name=self.channel_name,
             )
 
         self.scope["queue"] = queue.id
         self.scope["score"] = queue.score
 
-    async def chat_message(self, event):
-        pass
+    async def info(self, event):
+        message = event["message"]
+        await self.send(text_data=json.dumps({"type": "INFO", "message": message}))
 
     async def check_origin(self):
         if not self.scope["player"]:
@@ -124,7 +185,7 @@ class RoomConsumer(AsyncWebsocketConsumer):
 
     async def connect(self):
         self.room_name = self.scope["url_route"]["kwargs"]["room_name"]
-        self.room_group_name = "room_%s" % self.room_name
+        self.room_group_name = f"room_{self.room_name}"
 
         # Join room group
         await self.channel_layer.group_add(self.room_group_name, self.channel_name)
@@ -137,7 +198,6 @@ class RoomConsumer(AsyncWebsocketConsumer):
 
     # Receive message from WebSocket
     async def receive(self, text_data):
-        print(text_data)
         # Send message to room group
         await self.channel_layer.group_send(
             self.room_group_name,
diff --git a/room/models.py b/room/models.py
index 89a6bb8..f6a110b 100644
--- a/room/models.py
+++ b/room/models.py
@@ -8,6 +8,7 @@ class PlayerInQueue(models.Model):
     # TODO use redis for storing
     player = models.OneToOneField(Player, unique=True, on_delete=models.CASCADE)
     score = models.IntegerField()
+    channel_name = models.CharField(max_length=50, blank=False)
 
     def __str__(self):
         return f"{self.player.name} in que with score {self.score}"