Refacto

src/discord/discord_oauth.py

@@ -3,7 +3,7 @@ import webbrowser
 import os
 from urllib.parse import urlencode
 from http.server import HTTPServer, BaseHTTPRequestHandler
-from tools.get_server_token import GetServerTokenForDiscord
+from fivemserver.get_server_token import GetServerTokenForDiscord
 from config.constants import CLIENT_ID, REDIRECT_URI, SCOPES, AUTENTICATION_SUCCESS_MESSAGE
 
 # Disable stderr output

src/discord/discord_tools.py

@@ -1,9 +1,9 @@
 import psutil
 
 from pypresence.presence import Presence
-from tools.get_server_token import GetServerTokenForDiscord
+from fivemserver.get_server_token import GetServerTokenForDiscord
 from config.constants import Urls
-from tools.discord_oauth import CLIENT_ID
+from discord.discord_oauth import CLIENT_ID
 
 
 class DiscordToken:

src/fivemserver/get_server_token.py

@@ -65,12 +65,12 @@ class GetServerTokenForDiscord:
         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"]
+    @staticmethod
+    def register_discord_user(user_id: str, server = Urls.API_URL.value) -> str:
+        if server is None:
+            server = Urls.API_URL.value
+        registeredId = requests.post(server + "/api_v2/connection/register", verify=False, json={
+            "x-session-id": user_id
+        }).json()
+
+        return registeredId["discord_id"]

src/fivemserver/whitelistmanager.py

@@ -0,0 +1,23 @@
+import requests
+from urllib3 import disable_warnings
+from urllib3.exceptions import InsecureRequestWarning
+
+# Supress only InsecureRequestWarning
+disable_warnings(InsecureRequestWarning)
+
+WHITELIST_URL_ENDPOINT = f'iswhitelist/'
+
+class WhiteList:
+    @staticmethod
+    def checkwhitelist(url, discord_user_id: str) -> bool:
+        print('🗒️ Vérification de la whitelist...')
+
+        response = requests.get(url + WHITELIST_URL_ENDPOINT + discord_user_id, verify=False)
+        api_data =  response.json()
+
+        if api_data['whitelisted']:
+            print("👍 Vous êtes en whitelist")
+            return True
+        else:
+            print('🙅‍♂️ Désole mais vous n\'êtes pas whitelisté sur le serveur.')
+            return False

src/main.py

@@ -9,8 +9,8 @@ from PySide6.QtWidgets import QApplication
 from config.config_manager import ConfigManager
 
 # Imports pour la vérification Discord
-from tools.discord_tools import CheckDiscord
-from tools.custom_message_box import CustomMessageBox
+from discord.discord_tools import CheckDiscord
+from ui.custom_message_box import CustomMessageBox
 
 # Ne pas supprimer ! Enregistre les ressources Qt
 import resources  # noqa: F401 - required to register Qt resources

src/ui/main_window.py

@@ -11,7 +11,7 @@ from config.config_manager import ConfigManager
 from controllers.audio_controller import AudioController
 from controllers.glow_animator import GlowAnimator
 from controllers.window_dragger import WindowDragger
-from tools import discord_oauth
+from discord import discord_oauth
 from config.constants import NO_STAFF, Urls
 
 from fake_patch_notes import patch_note