Update app.py

app.py

@@ -12,6 +12,7 @@ ENTREPRISES_FILE = "entreprises.json"
 clients = {}
 logged_users = {}
 player_positions = {}
+coop_requests = {}
 
 # ======================================================
 # PLAYERS
@@ -229,8 +230,6 @@ async def websocket_endpoint(
                     except:
                         pass
 
-                # Envoi des propriétaires des entreprises
-
                 for path, owner in (
                     entreprises_data.items()
                 ):
@@ -426,12 +425,185 @@ async def websocket_endpoint(
                     except:
                         pass
 
+            # =====================================
+            # COOP INVITE
+            # =====================================
+
+            elif (
+                msg_type
+                ==
+                "coop_invite"
+            ):
+
+                if username:
+
+                    target = str(
+                        data.get(
+                            "target",
+                            ""
+                        )
+                    ).strip()
+
+                    target_sid = None
+
+                    for sid, uname in logged_users.items():
+
+                        if uname == target:
+
+                            target_sid = sid
+
+                            break
+
+                    if (
+                        target_sid
+                        and
+                        target_sid in clients
+                    ):
+
+                        coop_requests[
+                            target_sid
+                        ] = session_id
+
+                        try:
+
+                            await clients[
+                                target_sid
+                            ].send_text(
+                                json.dumps({
+                                    "type":"coop_invite",
+                                    "sender":username
+                                })
+                            )
+
+                        except:
+                            pass
+
+            # =====================================
+            # COOP ACCEPT
+            # =====================================
+
+            elif (
+                msg_type
+                ==
+                "coop_accept"
+            ):
+
+                if username:
+
+                    sender_name = str(
+                        data.get(
+                            "sender",
+                            ""
+                        )
+                    ).strip()
+
+                    sender_sid = None
+
+                    for sid, uname in logged_users.items():
+
+                        if uname == sender_name:
+
+                            sender_sid = sid
+
+                            break
+
+                    if (
+                        sender_sid
+                        and
+                        sender_sid in clients
+                    ):
+
+                        if session_id in coop_requests:
+
+                            del coop_requests[session_id]
+
+                        try:
+
+                            await clients[
+                                sender_sid
+                            ].send_text(
+                                json.dumps({
+                                    "type":"coop_accept",
+                                    "from":username
+                                })
+                            )
+
+                        except:
+                            pass
+
+            # =====================================
+            # COOP REFUSE
+            # =====================================
+
+            elif (
+                msg_type
+                ==
+                "coop_refuse"
+            ):
+
+                if username:
+
+                    sender_name = str(
+                        data.get(
+                            "sender",
+                            ""
+                        )
+                    ).strip()
+
+                    sender_sid = None
+
+                    for sid, uname in logged_users.items():
+
+                        if uname == sender_name:
+
+                            sender_sid = sid
+
+                            break
+
+                    if (
+                        sender_sid
+                        and
+                        sender_sid in clients
+                    ):
+
+                        if session_id in coop_requests:
+
+                            del coop_requests[session_id]
+
+                        try:
+
+                            await clients[
+                                sender_sid
+                            ].send_text(
+                                json.dumps({
+                                    "type":"coop_refuse",
+                                    "from":username
+                                })
+                            )
+
+                        except:
+                            pass
+
     except WebSocketDisconnect:
 
         pass
 
     finally:
 
+        # Libération coop
+
+        if session_id in coop_requests:
+
+            del coop_requests[session_id]
+
+        for k, v in list(
+            coop_requests.items()
+        ):
+
+            if v == session_id:
+
+                del coop_requests[k]
+
         # Libération entreprises
 
         to_clear = []
@@ -485,7 +657,6 @@ async def websocket_endpoint(
 
             except:
                 pass
-
 if __name__ == "__main__":
 
     port = int(