Vous pouvez maintenant fermer cette fenêtre et revenir au launcher de La Tanière.
+ + + """.encode('utf-8') + +class OAuthCallbackHandler(BaseHTTPRequestHandler): + code: str | None = None + + def do_GET(self): + if "/callback" in self.path: + query = self.path.split("?")[1] + params = dict(p.split("=") for p in query.split("&")) + OAuthCallbackHandler.code = params.get("code") + + self.send_response(200) + self.send_header("Content-type", "text/html") + self.end_headers() + self.wfile.write(AUTENTICATION_SUCCESS_MESSAGE) + +# return discord application id (client id) +def get_discord_client_id() -> str: + return CLIENT_ID + +# return discord user id +def get_discord_user_id() -> str: + # récupération des infos serveur lataupe + session_id = GetServerTokenForDiscord.authenticate() + client_secret = GetServerTokenForDiscord.get_token(session_id) + + auth_url = "https://discord.com/api/oauth2/authorize" + params = { + "client_id": CLIENT_ID, + "redirect_uri": REDIRECT_URI, + "response_type": "code", + "scope": " ".join(SCOPES), + } + + webbrowser.open(f"{auth_url}?{urlencode(params)}") + + server = HTTPServer(("localhost", 5000), OAuthCallbackHandler) + # celle ligne cache le stderr output pour ne pas afficher l'url de callback dans la console + # valable en debug mode + os.dup2(os.open(os.devnull, os.O_WRONLY), 2) + server.handle_request() + + if not OAuthCallbackHandler.code: + raise RuntimeError("OAuth échoué") + + token = requests.post( + "https://discord.com/api/oauth2/token", + data={ + "client_id": CLIENT_ID, + "client_secret": client_secret, + "grant_type": "authorization_code", + "code": OAuthCallbackHandler.code, + "redirect_uri": REDIRECT_URI, + }, + headers={"Content-Type": "application/x-www-form-urlencoded"}, + ).json() + + user = requests.get( + "https://discord.com/api/users/@me", + headers={"Authorization": f"Bearer {token['access_token']}"}, + ).json() + + return user["id"] diff --git a/src/tools/discord_tools.py b/src/tools/discord_tools.py new file mode 100644 index 0000000..e079941 --- /dev/null +++ b/src/tools/discord_tools.py @@ -0,0 +1,33 @@ +import psutil + +from pypresence import Presence +from get_server_token import GetServerTokenForDiscord +from constants import Urls + +class DiscordToken: + @staticmethod + def decode_discord_token(): + discord_token = GetServerTokenForDiscord.get_token(GetServerTokenForDiscord.authenticate(Urls.API_URL.value)) + return discord_token + +class CheckDiscord: + + @staticmethod + def isdiscordrunning() -> bool: + for process in psutil.process_iter(["name"]): + if process.info["name"].lower() == "discord.exe": + return True + return False + + @staticmethod + def isuserconnected(clientid: str) -> bool: + rpc = Presence(clientid) + try: + return True + except Exception: + return False + finally: + try: + rpc.close() + except Exception: + pass diff --git a/src/tools/get_server_token.py b/src/tools/get_server_token.py new file mode 100644 index 0000000..a5327e8 --- /dev/null +++ b/src/tools/get_server_token.py @@ -0,0 +1,76 @@ +import base64 +import requests +from cryptography.hazmat.primitives import serialization, hashes +from cryptography.hazmat.primitives.asymmetric import ec +from cryptography.hazmat.primitives.kdf.hkdf import HKDF +from cryptography.hazmat.primitives.ciphers.aead import AESGCM + +API_URL = 'https://prod.la-taniere.fun:30121/' + +class GetServerTokenForDiscord: + derived_key: bytes | None = None + + @staticmethod + def authenticate(server = API_URL): + + if server is None: + server = API_URL + # ========================== + # Génération clé ECDH client + # ========================== + client_private = ec.generate_private_key(ec.SECP256R1()) + client_public = client_private.public_key() + + client_pub_pem = client_public.public_bytes( + encoding=serialization.Encoding.PEM, + format=serialization.PublicFormat.SubjectPublicKeyInfo + ) + + # ========================== + # AUTH + # ========================== + auth = requests.post(server + "/api_v2/auth", verify=False, json={ + "client_pub": base64.b64encode(client_pub_pem).decode() + }).json() + + server_pub = serialization.load_pem_public_key( + base64.b64decode(auth["server_pub"]) + ) + + shared_key = client_private.exchange(ec.ECDH(), server_pub) + + GetServerTokenForDiscord.derived_key = HKDF( + algorithm=hashes.SHA256(), + length=32, + salt=None, + info=b"fivem-private-server" + ).derive(shared_key) + + return auth["session_id"] + + @staticmethod + def get_token(session_id: bytes, server = API_URL): + # ========================== + # DISCORD TOKEN + # ========================== + if server is None: + server = API_URL + download = requests.post(server + "/api_v2/tkn_auth", verify=False, headers={ + "x-session-id": session_id + }).json() + + nonce = base64.b64decode(download["nonce"]) + encrypted_data = base64.b64decode(download["data"]) + + aesgcm = AESGCM(GetServerTokenForDiscord.derived_key) # type: ignore[arg-type] + return aesgcm.decrypt(nonce, encrypted_data, None) + + # @staticmethod + # def register_discord_user(user_id: str, server = API_URL) -> str: + # if server is None: + # server = API_URL + # registeredId = requests.post(server + "/api_v2/connection/register", verify=False, json={ + # "x-session-id": user_id + # }).json() + # + # return registeredId["discord_id"] diff --git a/src/tools/utils.py b/src/tools/utils.py new file mode 100644 index 0000000..ceb80b9 --- /dev/null +++ b/src/tools/utils.py @@ -0,0 +1,19 @@ +import sys +from pathlib import Path + +PROJECT_ROOT = Path(__file__).resolve().parents[2] + +def get_internal_dir() -> Path: + # Retourne le chemin vers les ressources figées à l'intérieur de l'EXE (_MEIPASS). + # En mode script, retourne le dossier du fichier .py. + if getattr(sys, 'frozen', False) and hasattr(sys, '_MEIPASS'): + return Path(sys._MEIPASS).resolve() + return Path(__file__).resolve().parents[2] + +def get_executable_dir() -> Path: + # Retourne le chemin du dossier contenant réellement le fichier .exe. + # C'est ici que se trouve votre 'config.json'. + if getattr(sys, 'frozen', False): + # sys.executable est le chemin complet vers l'application .exe + return Path(sys.executable).parent.resolve() + return Path(__file__).resolve().parents[2] diff --git a/src/ui/main_window.py b/src/ui/main_window.py index 0c7b621..1321c80 100644 --- a/src/ui/main_window.py +++ b/src/ui/main_window.py @@ -8,7 +8,7 @@ from PySide6.QtUiTools import QUiLoader from PySide6.QtWidgets import QMainWindow, QSizePolicy from config.config_manager import ConfigManager -from constants import NO_DISCORD, NO_STAFF, Urls +from tools.constants import NO_DISCORD, NO_STAFF, Urls from controllers.audio_controller import AudioController from controllers.glow_animator import GlowAnimator from controllers.window_dragger import WindowDragger diff --git a/src/utils.py b/src/utils.py deleted file mode 100644 index 5a7d6ef..0000000 --- a/src/utils.py +++ /dev/null @@ -1,7 +0,0 @@ -import sys -from pathlib import Path - -def get_bundle_dir() -> Path: - if getattr(sys, 'frozen', False) and hasattr(sys, '_MEIPASS'): - return Path(sys._MEIPASS) - return Path(__file__).resolve().parent.parent