game: add: online

games/consumers.py

@@ -33,11 +33,23 @@ class TicTacToeWebSocket(WebsocketConsumer):
 
         self.game_id = int(self.scope['url_route']['kwargs']['game_id'])
         
-        self.game: PongGame = game_manager.get(self.game_id, "pong")
-    
+        self.game: TicTacToeGame = game_manager.get(self.game_id, "tictactoe")
+
+        self.member = self.game.join(self.user.pk, self)
+		
+        self.member.send("x" if self.game._everbody_is_here() else "o")
+
+        if (self.game._everbody_is_here()):
+            self.game.broadcast("game_start")
+
     def receive(self, text_data=None, bytes_data=None):
+        print(text_data)
+        self.game.broadcast("", json.loads(text_data), [self.member])
         pass
 
+    def disconnect(self, event):
+        self.member.socket = None
+
 class PongWebSocket(WebsocketConsumer):
 
     def __init__(self, *args, **kwargs):

games/objects/tictactoe/TicTacToeGame.py

@@ -1,5 +1,47 @@
 from ..AGame import AGame
 
+from channels.generic.websocket import WebsocketConsumer
+
+from .TicTacToePlayer import TicTacToePlayer
+
+from .TicTacToeSpectator import TicTacToeSpectator
+
 
 class TicTacToeGame(AGame):
-    pass
+    
+    def __init__(self, game_id: int, game_manager):
+        super().__init__("tictactoe", game_id, game_manager)
+
+    def _everbody_is_here(self):
+        return len(self.players) == len(self.get_players_connected())
+
+    def _player_join(self, user_id: int, socket: WebsocketConsumer):
+        
+        if (self.model.started):
+            return None
+        
+        player = self.get_player_by_user_id(user_id)
+        if (player is None):
+            return None
+
+        # check if player is already connected
+        if (player.is_connected()):
+            player.disconnect(1001)
+
+        player.socket = socket
+
+        return player
+
+    def _spectator_join(self, user_id: int, socket: WebsocketConsumer):
+        
+        spectator:TicTacToeSpectator = TicTacToeSpectator(user_id, socket, self)
+        
+        self.spectators.append(spectator)
+        
+        return spectator    
+    
+    def join(self, user_id: int, socket: WebsocketConsumer) -> TicTacToeSpectator | TicTacToePlayer:
+        member: TicTacToePlayer = self._player_join(user_id, socket)
+        if (member is None):
+            member: TicTacToeSpectator = self._spectator_join(user_id, socket)
+        return member