V6.1.0 update the help command

core/classes/client.py

@@ -0,0 +1,243 @@
+from re import sub
+from typing import Union, TYPE_CHECKING
+from dataclasses import asdict
+
+if TYPE_CHECKING:
+    from core.base import Base
+    from core.definition import MClient
+
+class Client:
+
+    CLIENT_DB: list['MClient'] = []
+
+    def __init__(self, baseObj: 'Base') -> None:
+
+        self.Logs = baseObj.logs
+        self.Base = baseObj
+
+        return None
+
+    def insert(self, newUser: 'MClient') -> bool:
+        """Insert a new User object
+
+        Args:
+            newUser (UserModel): New userModel object
+
+        Returns:
+            bool: True if inserted
+        """
+
+        userObj = self.get_Client(newUser.uid)
+
+        if not userObj is None:
+            # User already created return False
+            return False
+
+        self.CLIENT_DB.append(newUser)
+
+        return True
+
+    def update(self, uid: str, newNickname: str) -> bool:
+        """Update the nickname starting from the UID
+
+        Args:
+            uid (str): UID of the user
+            newNickname (str): New nickname
+
+        Returns:
+            bool: True if updated
+        """
+        userObj = self.get_Client(uidornickname=uid)
+
+        if userObj is None:
+            return False
+
+        userObj.nickname = newNickname
+
+        return True
+
+    def update_mode(self, uidornickname: str, modes: str) -> bool:
+        """Updating user mode
+
+        Args:
+            uidornickname (str): The UID or Nickname of the user
+            modes (str): new modes to update
+
+        Returns:
+            bool: True if user mode has been updaed
+        """
+        response = True
+        userObj = self.get_Client(uidornickname=uidornickname)
+
+        if userObj is None:
+            return False
+
+        action = modes[0]
+        new_modes = modes[1:]
+
+        existing_umodes = userObj.umodes
+        umodes = userObj.umodes
+
+        if action == '+':
+
+            for nm in new_modes:
+                if nm not in existing_umodes:
+                    umodes += nm
+
+        elif action == '-':
+            for nm in new_modes:
+                if nm in existing_umodes:
+                    umodes = umodes.replace(nm, '')
+        else:
+            return False
+
+        liste_umodes = list(umodes)
+        final_umodes_liste = [x for x in self.Base.Settings.PROTOCTL_USER_MODES if x in liste_umodes]
+        final_umodes = ''.join(final_umodes_liste)
+
+        userObj.umodes = f"+{final_umodes}"
+
+        return response
+
+    def delete(self, uid: str) -> bool:
+        """Delete the User starting from the UID
+
+        Args:
+            uid (str): UID of the user
+
+        Returns:
+            bool: True if deleted
+        """
+
+        userObj = self.get_Client(uidornickname=uid)
+
+        if userObj is None:
+            return False
+
+        self.CLIENT_DB.remove(userObj)
+
+        return True
+
+    def get_Client(self, uidornickname: str) -> Union['MClient', None]:
+        """Get The Client Object model
+
+        Args:
+            uidornickname (str): UID or Nickname
+
+        Returns:
+            UserModel|None: The UserModel Object | None
+        """
+        User = None
+        for record in self.CLIENT_DB:
+            if record.uid == uidornickname:
+                User = record
+            elif record.nickname == uidornickname:
+                User = record
+
+        return User
+
+    def get_uid(self, uidornickname:str) -> Union[str, None]:
+        """Get the UID of the user starting from the UID or the Nickname
+
+        Args:
+            uidornickname (str): UID or Nickname
+
+        Returns:
+            str|None: Return the UID
+        """
+
+        userObj = self.get_Client(uidornickname=uidornickname)
+
+        if userObj is None:
+            return None
+
+        return userObj.uid
+
+    def get_nickname(self, uidornickname:str) -> Union[str, None]:
+        """Get the Nickname starting from UID or the nickname
+
+        Args:
+            uidornickname (str): UID or Nickname of the user
+
+        Returns:
+            str|None: the nickname
+        """
+        userObj = self.get_Client(uidornickname=uidornickname)
+
+        if userObj is None:
+            return None
+
+        return userObj.nickname
+
+    def get_Client_AsDict(self, uidornickname: str) -> Union[dict[str, any], None]:
+        """Transform User Object to a dictionary
+
+        Args:
+            uidornickname (str): The UID or The nickname
+
+        Returns:
+            Union[dict[str, any], None]: User Object as a dictionary or None
+        """
+        userObj = self.get_Client(uidornickname=uidornickname)
+
+        if userObj is None:
+            return None
+
+        return asdict(userObj)
+
+    def is_exist(self, uidornikname: str) -> bool:
+        """Check if the UID or the nickname exist in the USER DB
+
+        Args:
+            uidornickname (str): The UID or the NICKNAME
+
+        Returns:
+            bool: True if exist
+        """
+        userObj = self.get_Client(uidornickname=uidornikname)
+
+        if userObj is None:
+            return False
+
+        return True
+
+    def db_is_account_exist(self, account: str) -> bool:
+        """Check if the account exist in the database
+
+        Args:
+            account (str): The account to check
+
+        Returns:
+            bool: True if exist
+        """
+
+        table_client = self.Base.Config.TABLE_CLIENT
+        account_to_check = {'account': account.lower()}
+        account_to_check_query = self.Base.db_execute_query(f"""
+                    SELECT id FROM {table_client} WHERE LOWER(account) = :account
+                    """, account_to_check)
+
+        account_to_check_result = account_to_check_query.fetchone()
+        if account_to_check_result:
+            self.Logs.error(f"Account ({account}) already exist")
+            return True
+
+        return False
+
+    def clean_uid(self, uid: str) -> Union[str, None]:
+        """Clean UID by removing @ / % / + / ~ / * / :
+
+        Args:
+            uid (str): The UID to clean
+
+        Returns:
+            str: Clean UID without any sign
+        """
+
+        pattern = fr'[:|@|%|\+|~|\*]*'
+        parsed_UID = sub(pattern, '', uid)
+
+        if not parsed_UID:
+            return None
+
+        return parsed_UID

core/classes/protocols/unreal6.py

@@ -420,6 +420,18 @@ class Unrealircd6:
 
         # Add defender to the channel uids list
         self.__Irc.Channel.insert(self.__Irc.Loader.Definition.MChannel(name=channel, uids=[userObj.uid]))
+
+        # Set the automode to the user
+        if 'r' not in userObj.umodes and 'o' not in userObj.umodes:
+            return None
+
+        db_data: dict[str, str] = {"nickname": userObj.nickname, "channel": channel}
+        db_query = self.__Base.db_execute_query("SELECT id, mode FROM command_automode WHERE nickname = :nickname AND channel = :channel", db_data)
+        db_result = db_query.fetchone()
+        if db_result is not None:
+            id, mode = db_result
+            self.send2socket(f":{self.__Config.SERVICE_ID} MODE {channel} {mode} {userObj.nickname}")
+
         return None
 
     def send_part_chan(self, uidornickname:str, channel: str, print_log: bool = True) -> None:
@@ -447,6 +459,16 @@ class Unrealircd6:
         self.__Irc.Channel.delete_user_from_channel(channel, userObj.uid)
         return None
 
+    def send_raw(self, raw_command: str) -> None:
+
+        self.send2socket(f":{self.__Config.SERVICE_NICKNAME} {raw_command}")
+
+        return None
+
+    #####################
+    #   HANDLE EVENTS   #
+    #####################
+
     def on_svs2mode(self, serverMsg: list[str]) -> None:
         """Handle svs2mode coming from a server
 
@@ -527,6 +549,7 @@ class Unrealircd6:
 
             self.__Irc.Channel.delete_user_from_all_channel(uid_who_quit)
             self.__Irc.User.delete(uid_who_quit)
+            self.__Irc.Client.delete(uid_who_quit)
             self.__Irc.Reputation.delete(uid_who_quit)
             self.__Irc.Clone.delete(uid_who_quit)
 
@@ -609,6 +632,7 @@ class Unrealircd6:
             uid = str(serverMsg[1]).lstrip(':')
             newnickname = serverMsg[3]
             self.__Irc.User.update(uid, newnickname)
+            self.__Irc.Client.update(uid, newnickname)
 
             return None
 
@@ -777,7 +801,7 @@ class Unrealircd6:
                 self.__Irc.first_score = int(str(server_msg_copy[3]).replace('*',''))
                 for user in self.__Irc.User.UID_DB:
                     if user.remote_ip == self.__Irc.first_connexion_ip:
-                        user.score_connexion = self.first_score
+                        user.score_connexion = self.__Irc.first_score
             else:
                 self.__Irc.first_score = int(server_msg_copy[3])
 
@@ -890,7 +914,7 @@ class Unrealircd6:
                 convert_to_string = ' '.join(liste_des_commandes)
                 arg = convert_to_string.split()
                 arg.remove(f':{self.__Config.SERVICE_PREFIX}')
-                if not arg[0].lower() in self.__Irc.commands:
+                if not arg[0].lower() in self.__Irc.module_commands_list:
                     self.__Base.logs.debug(f"This command {arg[0]} is not available")
                     self.send_notice(
                         nick_from=self.__Config.SERVICE_NICKNAME,
@@ -929,7 +953,7 @@ class Unrealircd6:
                         self.on_ping(srv_msg)
                         return False
 
-                    if not arg[0].lower() in self.__Irc.commands:
+                    if not arg[0].lower() in self.__Irc.module_commands_list:
                         self.__Base.logs.debug(f"This command {arg[0]} sent by {user_trigger} is not available")
                         return False
 
@@ -945,8 +969,10 @@ class Unrealircd6:
 
         except KeyError as ke:
             self.__Base.logs.error(f"Key Error: {ke}")
+        except AttributeError as ae:
+            self.__Base.logs.error(f"Attribute Error: {ae}")
         except Exception as err:
-            self.__Base.logs.error(f"General Error: {err}")
+            self.__Base.logs.error(f"General Error: {err} - {srv_msg}")
 
     def on_server_ping(self, serverMsg: list[str]) -> None:
         """Send a PONG message to the server

core/classes/settings.py

@@ -3,13 +3,13 @@ from socket import socket
 
 class Settings:
 
-    RUNNING_TIMERS: list[Timer]     = []
-    RUNNING_THREADS: list[Thread]   = []
-    RUNNING_SOCKETS: list[socket]   = []
-    PERIODIC_FUNC: dict[object]     = {}
-    LOCK: RLock                     = RLock()
+    RUNNING_TIMERS: list[Timer]         = []
+    RUNNING_THREADS: list[Thread]       = []
+    RUNNING_SOCKETS: list[socket]       = []
+    PERIODIC_FUNC: dict[object]         = {}
+    LOCK: RLock                         = RLock()
 
-    CONSOLE: bool                   = False
+    CONSOLE: bool                       = False
 
-    PROTOCTL_USER_MODES: list[str]   = []
-    PROTOCTL_PREFIX: list[str]       = []
+    PROTOCTL_USER_MODES: list[str]      = []
+    PROTOCTL_PREFIX: list[str]          = []