Update app.py

app.py

@@ -0,0 +1,460 @@
+from fastapi import FastAPI, WebSocket, WebSocketDisconnect
+import uvicorn
+import json
+import os
+import uuid
+
+app = FastAPI()
+
+PLAYERS_FILE = "players.json"
+ENTREPRISES_FILE = "entreprises.json"
+
+clients = {}
+logged_users = {}
+player_positions = {}
+
+# ======================================================
+# PLAYERS
+# ======================================================
+
+def load_players():
+
+    if os.path.exists(PLAYERS_FILE):
+
+        with open(
+            PLAYERS_FILE,
+            "r",
+            encoding="utf-8"
+        ) as f:
+
+            return json.load(f)
+
+    return {}
+
+def save_players():
+
+    with open(
+        PLAYERS_FILE,
+        "w",
+        encoding="utf-8"
+    ) as f:
+
+        json.dump(
+            players_data,
+            f,
+            indent=4,
+            ensure_ascii=False
+        )
+
+# ======================================================
+# ENTREPRISES
+# ======================================================
+
+def load_entreprises():
+
+    if os.path.exists(
+        ENTREPRISES_FILE
+    ):
+
+        with open(
+            ENTREPRISES_FILE,
+            "r",
+            encoding="utf-8"
+        ) as f:
+
+            return json.load(f)
+
+    return {}
+
+def save_entreprises():
+
+    with open(
+        ENTREPRISES_FILE,
+        "w",
+        encoding="utf-8"
+    ) as f:
+
+        json.dump(
+            entreprises_data,
+            f,
+            indent=4,
+            ensure_ascii=False
+        )
+
+players_data = load_players()
+entreprises_data = load_entreprises()
+
+# ======================================================
+# WEBSOCKET
+# ======================================================
+
+@app.websocket("/ws")
+async def websocket_endpoint(
+    ws: WebSocket
+):
+
+    await ws.accept()
+
+    session_id = str(
+        uuid.uuid4()
+    )
+
+    clients[session_id] = ws
+
+    username = None
+
+    await ws.send_text(json.dumps({
+        "type":"welcome",
+        "id":session_id
+    }))
+
+    try:
+
+        while True:
+
+            text = await ws.receive_text()
+
+            data = json.loads(text)
+
+            msg_type = data.get(
+                "type",
+                ""
+            )
+
+            # =====================================
+            # LOGIN
+            # =====================================
+
+            if msg_type == "login":
+
+                user = str(
+                    data.get(
+                        "username",
+                        ""
+                    )
+                ).strip()
+
+                password = str(
+                    data.get(
+                        "password",
+                        ""
+                    )
+                ).strip()
+
+                if (
+                    user == ""
+                    or
+                    password == ""
+                ):
+
+                    await ws.send_text(
+                        json.dumps({
+                            "type":"login_failed"
+                        })
+                    )
+
+                    continue
+
+                if user not in players_data:
+
+                    players_data[user] = {
+                        "password":password,
+                        "money":0,
+                        "progress":1
+                    }
+
+                    save_players()
+
+                elif (
+                    players_data[user]["password"]
+                    != password
+                ):
+
+                    await ws.send_text(
+                        json.dumps({
+                            "type":"login_failed"
+                        })
+                    )
+
+                    continue
+
+                username = user
+
+                logged_users[
+                    session_id
+                ] = username
+
+                await ws.send_text(
+                    json.dumps({
+                        "type":"login_success"
+                    })
+                )
+
+                await ws.send_text(
+                    json.dumps({
+                        "type":"player_data",
+                        "money":
+                        players_data[user]["money"],
+                        "progress":
+                        players_data[user]["progress"]
+                    })
+                )
+
+                for sid in clients.keys():
+
+                    if sid == session_id:
+                        continue
+
+                    await ws.send_text(
+                        json.dumps({
+                            "type":"player_join",
+                            "id":sid
+                        })
+                    )
+
+                for sid, client in clients.items():
+
+                    if sid == session_id:
+                        continue
+
+                    try:
+
+                        await client.send_text(
+                            json.dumps({
+                                "type":"player_join",
+                                "id":session_id
+                            })
+                        )
+
+                    except:
+                        pass
+
+                for path, owner in (
+                    entreprises_data.items()
+                ):
+
+                    await ws.send_text(
+                        json.dumps({
+                            "type":"entreprise_owner",
+                            "path":path,
+                            "owner":owner
+                        })
+                    )
+
+            # =====================================
+            # POSITION
+            # =====================================
+
+            elif msg_type == "player_update":
+
+                player_positions[
+                    session_id
+                ] = {
+
+                    "x":
+                    data.get("x",0),
+
+                    "y":
+                    data.get("y",0),
+
+                    "z":
+                    data.get("z",0)
+                }
+
+                data["id"] = session_id
+
+                for sid, client in clients.items():
+
+                    if sid == session_id:
+                        continue
+
+                    try:
+
+                        await client.send_text(
+                            json.dumps(
+                                data
+                            )
+                        )
+
+                    except:
+                        pass
+
+            # =====================================
+            # MONEY
+            # =====================================
+
+            elif msg_type == "save_money":
+
+                if username:
+
+                    players_data[
+                        username
+                    ]["money"] = int(
+                        data.get(
+                            "money",
+                            0
+                        )
+                    )
+
+                    save_players()
+
+            # =====================================
+            # PROGRESS
+            # =====================================
+
+            elif msg_type == "save_progress":
+
+                if username:
+
+                    players_data[
+                        username
+                    ]["progress"] = int(
+                        data.get(
+                            "progress",
+                            1
+                        )
+                    )
+
+                    save_players()
+
+            # =====================================
+            # ENTREPRISE
+            # =====================================
+
+            elif msg_type == "save_entreprise":
+
+                path = data.get(
+                    "path",
+                    ""
+                )
+
+                owner = data.get(
+                    "owner",
+                    ""
+                )
+
+                entreprises_data[
+                    path
+                ] = owner
+
+                save_entreprises()
+
+            # =====================================
+            # CHANGE OWNER
+            # =====================================
+
+            elif msg_type == "set_entreprise_owner":
+
+                path = data.get(
+                    "path",
+                    ""
+                )
+
+                owner = data.get(
+                    "owner",
+                    ""
+                )
+
+                entreprises_data[
+                    path
+                ] = owner
+
+                save_entreprises()
+
+                broadcast = {
+                    "type":"entreprise_owner",
+                    "path":path,
+                    "owner":owner
+                }
+
+                for client in clients.values():
+
+                    try:
+
+                        await client.send_text(
+                            json.dumps(
+                                broadcast
+                            )
+                        )
+
+                    except:
+                        pass
+
+    except WebSocketDisconnect:
+
+        pass
+
+    finally:
+
+        # Libérer les entreprises du joueur
+
+        to_clear = []
+
+        for path, owner in entreprises_data.items():
+
+            if owner == session_id:
+
+                to_clear.append(path)
+
+        for path in to_clear:
+
+            entreprises_data[path] = ""
+
+            for client in clients.values():
+
+                try:
+
+                    await client.send_text(
+                        json.dumps({
+                            "type":"entreprise_owner",
+                            "path":path,
+                            "owner":""
+                        })
+                    )
+
+                except:
+                    pass
+
+        save_entreprises()
+
+        if session_id in clients:
+            del clients[session_id]
+
+        if session_id in logged_users:
+            del logged_users[session_id]
+
+        if session_id in player_positions:
+            del player_positions[session_id]
+
+        for client in clients.values():
+
+            try:
+
+                await client.send_text(
+                    json.dumps({
+                        "type":"player_leave",
+                        "id":session_id
+                    })
+                )
+
+            except:
+                pass
+
+# ======================================================
+# HUGGING FACE
+# ======================================================
+
+if __name__ == "__main__":
+
+    port = int(
+        os.environ.get(
+            "PORT",
+            7860
+        )
+    )
+
+    uvicorn.run(
+        app,
+        host="0.0.0.0",
+        port=port
+    )