Global :
- Reorganize folder - Reorganize function - Add "todo" in mains
This commit is contained in:
61
site/interface/server/main.py
Normal file
61
site/interface/server/main.py
Normal file
@ -0,0 +1,61 @@
|
||||
# **************************************************************************** #
|
||||
# #
|
||||
# ::: :::::::: #
|
||||
# main.py :+: :+: :+: #
|
||||
# +:+ +:+ +:+ #
|
||||
# By: edbernar <edbernar@student.42angouleme. +#+ +:+ +#+ #
|
||||
# +#+#+#+#+#+ +#+ #
|
||||
# Created: 2024/08/03 08:10:40 by edbernar #+# #+# #
|
||||
# Updated: 2024/08/04 14:31:26 by edbernar ### ########.fr #
|
||||
# #
|
||||
# **************************************************************************** #
|
||||
|
||||
from typeRequets.getPrivateListMessage import getPrivateListMessage
|
||||
from typeRequets.getPrivateListUser import getPrivateListUser
|
||||
from typeRequets.sendPrivateMessage import sendPrivateMessage
|
||||
from typeRequets.login import login
|
||||
from Class.User import User
|
||||
import websockets
|
||||
import asyncio
|
||||
import json
|
||||
|
||||
# Todo :
|
||||
# - verifier que l'utilisateur n'est pas déjà connecté pour éviter les doublons
|
||||
|
||||
typeRequest = ["login", "get_private_list_user", "get_private_list_message", "send_private_message"]
|
||||
functionRequest = [login, getPrivateListUser, getPrivateListMessage, sendPrivateMessage]
|
||||
|
||||
async def handler(websocket, path):
|
||||
if (path != "/"):
|
||||
await websocket.sendError("Invalid path", 9003)
|
||||
await websocket.close()
|
||||
return
|
||||
userClass = User(websocket)
|
||||
try:
|
||||
async for resquet in userClass.websocket:
|
||||
try:
|
||||
jsonRequest = json.loads(resquet)
|
||||
except json.JSONDecodeError:
|
||||
await userClass.sendError("Invalid JSON", 9002)
|
||||
continue
|
||||
try:
|
||||
userClass.printDebug(jsonRequest, 0)
|
||||
if (jsonRequest["type"] in typeRequest):
|
||||
if jsonRequest["type"] == "login":
|
||||
await functionRequest[typeRequest.index(jsonRequest["type"])](userClass, jsonRequest["content"])
|
||||
else:
|
||||
if (await userClass.verifyToken(jsonRequest["token"]) == False):
|
||||
continue
|
||||
await functionRequest[typeRequest.index(jsonRequest["type"])](userClass, jsonRequest["content"])
|
||||
else:
|
||||
await userClass.sendError("Invalid type", 9004)
|
||||
except Exception as e:
|
||||
await userClass.sendError("Invalid request", 9005, e)
|
||||
except websockets.ConnectionClosed:
|
||||
pass
|
||||
await userClass.close()
|
||||
start_server = websockets.serve(handler, "localhost", 8000, subprotocols=['123456'])
|
||||
|
||||
asyncio.get_event_loop().run_until_complete(start_server)
|
||||
print("Server started")
|
||||
asyncio.get_event_loop().run_forever()
|
Reference in New Issue
Block a user