Update all Protocol calls

core/classes/protocols/unreal6.py

@@ -9,6 +9,8 @@ if TYPE_CHECKING:
 class Unrealircd6:
 
     def  __init__(self, ircInstance: 'Irc'):
+        self.name = 'UnrealIRCD-6'
+
         self.__Irc = ircInstance
         self.__Config = ircInstance.Config
         self.__Base = ircInstance.Base
@@ -171,6 +173,12 @@ class Unrealircd6:
 
         return None
 
+    def unkline(self, nickname:str, hostname: str) -> None:
+
+        self.send2socket(f":{self.__Config.SERVEUR_ID} TKL - K {nickname} {hostname} {self.__Config.SERVICE_NICKNAME}")
+
+        return None
+
     def sjoin(self, channel: str) -> None:
 
         if not self.__Irc.Channel.Is_Channel(channel):
@@ -183,8 +191,36 @@ class Unrealircd6:
         self.__Irc.Channel.insert(self.__Irc.Loader.Definition.MChannel(name=channel, uids=[self.__Config.SERVICE_ID]))
         return None
 
+    def sendSapart(self, nick_to_sapart: str, channel_name: str) -> None:
+        """_summary_
+
+        Args:
+            from_nick (str): _description_
+            nick_to (str): _description_
+            channel_name (str): _description_
+        """
+        try:
+
+            userObj = self.__Irc.User.get_User(uidornickname=nick_to_sapart)
+            chanObj = self.__Irc.Channel.get_Channel(channel_name)
+            service_uid = self.__Config.SERVICE_ID
+
+            if userObj is None or chanObj is None:
+                return None
+
+            self.send2socket(f":{service_uid} SAPART {userObj.nickname} {chanObj.name}")
+            self.__Irc.Channel.delete_user_from_channel(chanObj.name, userObj.uid)
+
+            return None
+
+        except Exception as err:
+            self.__Base.logs.error(f"{__name__} - General Error: {err}")
+
     def sendQuit(self, uid: str, reason: str, print_log: True) -> None:
         """Send quit message
+        - Delete uid from User object
+        - Delete uid from Clone object
+        - Delete uid from Reputation object
 
         Args:
             uidornickname (str): The UID or the Nickname
@@ -211,7 +247,7 @@ class Unrealircd6:
 
     def sendUID(self, nickname:str, username: str, hostname: str, uid:str, umodes: str, vhost: str, remote_ip: str, realname: str, print_log: bool = True) -> None:
         """Send UID to the server
-
+        - Insert User to User Object
         Args:
             nickname (str): Nickname of the client
             username (str): Username of the client
@@ -298,12 +334,6 @@ class Unrealircd6:
         self.__Irc.Channel.delete_user_from_channel(channel, userObj.uid)
         return None
 
-    def unkline(self, nickname:str, hostname: str) -> None:
-
-        self.send2socket(f":{self.__Config.SERVEUR_ID} TKL - K {nickname} {hostname} {self.__Config.SERVICE_NICKNAME}")
-
-        return None
-
     def on_umode2(self, serverMsg: list[str]) -> None:
         """Handle umode2 coming from a server
 
@@ -357,6 +387,41 @@ class Unrealircd6:
         except Exception as err:
             self.__Base.logs.error(f"{__name__} - General Error: {err}")
 
+    def on_squit(self, serverMsg: list[str]) -> None:
+        """Handle squit coming from a server
+
+        Args:
+            serverMsg (list[str]): Original server message
+        """
+        # ['@msgid=QOEolbRxdhpVW5c8qLkbAU;time=2024-09-21T17:33:16.547Z', 'SQUIT', 'defender.deb.biz.st', ':Connection', 'closed']
+
+        server_hostname = serverMsg[2]
+        uid_to_delete = None
+        for s_user in self.__Irc.User.UID_DB:
+            if s_user.hostname == server_hostname and 'S' in s_user.umodes:
+                uid_to_delete = s_user.uid
+
+        if uid_to_delete is None:
+            return None
+
+        self.__Irc.User.delete(uid_to_delete)
+        self.__Irc.Channel.delete_user_from_all_channel(uid_to_delete)
+
+        return None
+
+    def on_protoctl(self, serverMsg: list[str]) -> None:
+        """Handle protoctl coming from a server
+
+        Args:
+            serverMsg (list[str]): Original server message
+        """
+        if len(serverMsg) > 5:
+            if '=' in serverMsg[5]:
+                serveur_hosting_id = str(serverMsg[5]).split('=')
+                self.__Config.HSID = serveur_hosting_id[1]
+
+        return None
+
     def on_nick(self, serverMsg: list[str]) -> None:
         """Handle nick coming from a server
         new nickname
@@ -366,7 +431,7 @@ class Unrealircd6:
         """
         try:
             # ['@unrealircd.org/geoip=FR;unrealircd.org/', ':001OOU2H3', 'NICK', 'WebIrc', '1703795844']
-                    # Changement de nickname
+            # Changement de nickname
 
             uid = str(serverMsg[1]).lstrip(':')
             newnickname = serverMsg[3]
@@ -610,11 +675,26 @@ class Unrealircd6:
             self.__Base.logs.error(f"{__name__} - General Error: {err}")
 
     def on_version_msg(self, serverMsg: list[str]) -> None:
+        """Handle version coming from the server
 
-        # ['@label=0073', ':0014E7P06', 'VERSION', 'PyDefender']
-        getUser  = self.__Irc.User.get_User(self.__Irc.User.clean_uid(serverMsg[1]))
+        Args:
+            serverMsg (list[str]): Original message from the server
+        """
+        try:
+            # ['@label=0073', ':0014E7P06', 'VERSION', 'PyDefender']
+            getUser  = self.__Irc.User.get_User(self.__Irc.User.clean_uid(serverMsg[1]))
+
+            if getUser is None:
+                return None
+
+            response_351 = f"{self.__Config.SERVICE_NAME.capitalize()}-{self.__Config.CURRENT_VERSION} {self.__Config.SERVICE_HOST} {self.name}"
+            self.send2socket(f':{self.__Config.SERVICE_HOST} 351 {getUser.nickname} {response_351}')
+
+            modules = self.__Base.get_all_modules()
+            response_005 = ' | '.join(modules)
+            self.send2socket(f':{self.__Config.SERVICE_HOST} 005 {getUser.nickname} {response_005} are supported by this server')
 
-        if getUser is None:
             return None
 
-        self.send2socket(f'{self.__Config.SERVEUR_ID} 351 {getUser.nickname} {self.__Config.CURRENT_VERSION} {self.__Config.SERVICE_NAME} *:')
+        except Exception as err:
+            self.__Base.logs.error(f"{__name__} - General Error: {err}")