New version

core/base.py

@@ -1,5 +1,6 @@
-import time, threading, os, random, socket, hashlib, ipaddress, logging, requests, json, re
-from typing import Union
+import time, threading, os, random, socket, hashlib, ipaddress, logging, requests, json, re, ast
+from dataclasses import fields
+from typing import Union, Literal
 from base64 import b64decode
 from datetime import datetime
 from sqlalchemy import create_engine, Engine, Connection, CursorResult
@@ -16,7 +17,7 @@ class Base:
 
         self.Config = Config                                    # Assigner l'objet de configuration
         self.init_log_system()                                  # Demarrer le systeme de log
-        self.check_for_new_version(True)                            # Verifier si une nouvelle version est disponible
+        self.check_for_new_version(True)                        # Verifier si une nouvelle version est disponible
 
         self.running_timers:list[threading.Timer] = []          # Liste des timers en cours
         self.running_threads:list[threading.Thread] = []        # Liste des threads en cours
@@ -47,7 +48,7 @@ class Base:
     def __get_latest_defender_version(self) -> None:
         try:
             self.logs.debug(f'Looking for a new version available on Github')
-            print(f'===> Looking for a new version available on Github')
+            # print(f'===> Looking for a new version available on Github')
             token = ''
             json_url = f'https://raw.githubusercontent.com/adator85/IRC_DEFENDER_MODULES/main/version.json'
             headers = {
@@ -158,7 +159,7 @@ class Base:
                               encoding='UTF-8',
                               format='%(asctime)s - %(levelname)s - %(filename)s - %(lineno)d - %(funcName)s - %(message)s')
 
-        self.logs.info('#################### STARTING INTERCEPTOR HQ ####################')
+        self.logs.info('#################### STARTING DEFENDER ####################')
 
         return None
 
@@ -190,8 +191,8 @@ class Base:
         Returns:
             bool: True si le module existe déja dans la base de données sinon False
         """
-        query = f"SELECT id FROM {self.Config.table_module} WHERE module = :module"
-        mes_donnes = {'module': module_name}
+        query = f"SELECT id FROM {self.Config.table_module} WHERE module_name = :module_name"
+        mes_donnes = {'module_name': module_name}
         results = self.db_execute_query(query, mes_donnes)
 
         if results.fetchall():
@@ -208,10 +209,9 @@ class Base:
 
         if not self.db_isModuleExist(module_name):
             self.logs.debug(f"Le module {module_name} n'existe pas alors ont le créer")
-            insert_cmd_query = f"INSERT INTO {self.Config.table_module} (datetime, user, module) VALUES (:datetime, :user, :module)"
-            mes_donnees = {'datetime': self.get_datetime(), 'user': user_cmd, 'module': module_name}
+            insert_cmd_query = f"INSERT INTO {self.Config.table_module} (datetime, user, module_name, isdefault) VALUES (:datetime, :user, :module_name, :isdefault)"
+            mes_donnees = {'datetime': self.get_datetime(), 'user': user_cmd, 'module_name': module_name, 'isdefault': 0}
             self.db_execute_query(insert_cmd_query, mes_donnees)
-            # self.db_close_session(self.session)
         else:
             self.logs.debug(f"Le module {module_name} existe déja dans la base de données")
 
@@ -223,12 +223,174 @@ class Base:
         Args:
             cmd (str): le module a enregistrer
         """
-        insert_cmd_query = f"DELETE FROM {self.Config.table_module} WHERE module = :module"
-        mes_donnees = {'module': module_name}
+        insert_cmd_query = f"DELETE FROM {self.Config.table_module} WHERE module_name = :module_name"
+        mes_donnees = {'module_name': module_name}
         self.db_execute_query(insert_cmd_query, mes_donnees)
 
         return False
 
+    def db_sync_core_config(self, module_name: str, dataclassObj: object) -> bool:
+        """Sync module local parameters with the database
+        if new module then local param will be stored in the database
+        if old module then db param will be moved to the local dataclassObj
+        if new local param it will be stored in the database
+        if local param was removed then it will also be removed from the database
+
+        Args:
+            module_name (str): The module name ex. mod_defender
+            dataclassObj (object): The Dataclass object
+
+        Returns:
+            bool: _description_
+        """
+        try:
+            response = True
+            current_date = self.get_datetime()
+            core_table = self.Config.table_config
+
+            # Add local parameters to DB
+            for field in fields(dataclassObj):
+                param_key = field.name
+                param_value = str(getattr(dataclassObj, field.name))
+
+                param_to_search = {'module_name': module_name, 'param_key': param_key}
+
+                search_query = f'''SELECT id FROM {core_table} WHERE module_name = :module_name AND param_key = :param_key'''
+                excecute_search_query = self.db_execute_query(search_query, param_to_search)
+                result_search_query = excecute_search_query.fetchone()
+
+                if result_search_query is None:
+                    # If param and module_name doesn't exist create the record
+                    param_to_insert = {'datetime': current_date,'module_name': module_name,
+                                       'param_key': param_key,'param_value': param_value
+                                    }
+
+                    insert_query = f'''INSERT INTO {core_table} (datetime, module_name, param_key, param_value) 
+                                        VALUES (:datetime, :module_name, :param_key, :param_value)
+                                        '''
+                    execution = self.db_execute_query(insert_query, param_to_insert)
+
+                    if execution.rowcount > 0:
+                        self.logs.debug(f'New parameter added to the database: {param_key} --> {param_value}')
+
+            # Delete from DB unused parameter
+            query_select = f"SELECT module_name, param_key, param_value FROM {core_table} WHERE module_name = :module_name"
+            parameter = {'module_name': module_name}
+            execute_query_select = self.db_execute_query(query_select, parameter)
+            result_query_select = execute_query_select.fetchall()
+
+            for result in result_query_select:
+                db_mod_name, db_param_key, db_param_value = result
+                if not hasattr(dataclassObj, db_param_key):
+                    mes_donnees = {'param_key': db_param_key, 'module_name': db_mod_name}
+                    execute_delete = self.db_execute_query(f'DELETE FROM {core_table} WHERE module_name = :module_name and param_key = :param_key', mes_donnees)
+                    row_affected = execute_delete.rowcount
+                    if row_affected > 0:
+                        self.logs.debug(f'A parameter has been deleted from the database: {db_param_key} --> {db_param_value} | Mod: {db_mod_name}')
+
+            # Sync local variable with Database
+            query = f"SELECT param_key, param_value FROM {core_table} WHERE module_name = :module_name"
+            parameter = {'module_name': module_name}
+            response = self.db_execute_query(query, parameter)
+            result = response.fetchall()
+
+            for param, value in result:
+                if type(getattr(dataclassObj, param)) == list:
+                    value = ast.literal_eval(value)
+
+                setattr(dataclassObj, param, self.int_if_possible(value))
+
+            return response
+
+        except AttributeError as attrerr:
+            self.logs.error(f'Attribute Error: {attrerr}')
+        except Exception as err:
+            self.logs.error(err)
+            return False
+
+    def db_update_core_config(self, module_name:str, dataclassObj: object, param_key:str, param_value: str) -> bool:
+
+        core_table = 'core_config'
+        # Check if the param exist
+        if not hasattr(dataclassObj, param_key):
+            self.logs.error(f"Le parametre {param_key} n'existe pas dans la variable global")
+            return False
+
+        mes_donnees = {'module_name': module_name, 'param_key': param_key, 'param_value': param_value}
+        search_param_query = f"SELECT id FROM {core_table} WHERE module_name = :module_name AND param_key = :param_key"
+        result = self.db_execute_query(search_param_query, mes_donnees)
+        isParamExist = result.fetchone()
+
+        if not isParamExist is None:
+            mes_donnees = {'datetime': self.get_datetime(),
+                           'module_name': module_name,
+                           'param_key': param_key,
+                           'param_value': param_value
+                           }
+            query = f'''UPDATE {core_table} SET datetime = :datetime, param_value = :param_value WHERE module_name = :module_name AND param_key = :param_key'''
+            update = self.db_execute_query(query, mes_donnees)
+            updated_rows = update.rowcount
+            if updated_rows > 0:
+                setattr(dataclassObj, param_key, self.int_if_possible(param_value))
+                self.logs.debug(f'Parameter updated : {param_key} - {param_value} | Module: {module_name}')
+
+        self.logs.debug(dataclassObj)
+
+        return True
+
+    def db_query_channel(self, action: Literal['add','del'], module_name: str, channel_name: str) -> bool:
+        """You can add a channel or delete a channel.
+
+        Args:
+            action (Literal['add','del']): Action on the database
+            module_name (str): The module name (mod_test)
+            channel_name (str): The channel name (With #)
+
+        Returns:
+            bool: True if action done
+        """
+        try:
+            channel_name = channel_name.lower() if self.Is_Channel(channel_name) else None
+            core_table = 'core_channel'
+
+            if not channel_name:
+                self.logs.warn(f'The channel [{channel_name}] is not correct')
+                return False
+
+            match action:
+
+                case 'add':
+                    mes_donnees = {'module_name': module_name, 'channel_name': channel_name}
+                    response = self.db_execute_query(f"SELECT id FROM {core_table} WHERE module_name = :module_name AND channel_name = :channel_name", mes_donnees)
+                    isChannelExist = response.fetchone()
+
+                    if isChannelExist is None:
+                        mes_donnees = {'datetime': self.get_datetime(), 'channel_name': channel_name, 'module_name': module_name}
+                        insert = self.db_execute_query(f"INSERT INTO {core_table} (datetime, channel_name, module_name) VALUES (:datetime, :channel_name, :module_name)", mes_donnees)
+                        if insert.rowcount:
+                            self.logs.debug(f'New channel added: channel={channel_name} / module_name={module_name}')
+                            return True
+                    else:
+                        return False
+                    pass
+
+                case 'del':
+                    mes_donnes = {'channel_name': channel_name, 'module_name': module_name}
+                    response = self.db_execute_query(f"DELETE FROM {core_table} WHERE channel_name = :channel_name AND module_name = :module_name", mes_donnes)
+
+                    if response.rowcount > 0:
+                        self.logs.debug(f'Channel deleted: channel={channel_name} / module: {module_name}')
+                        return True
+                    else:
+                        
+                        return False
+
+                case _:
+                    return False
+
+        except Exception as err:
+            self.logs.error(err)
+
     def db_create_first_admin(self) -> None:
 
         user = self.db_execute_query(f"SELECT id FROM {self.Config.table_admin}")
@@ -267,6 +429,13 @@ class Base:
             self.logs.error(f'Assertion Error -> {ae}')
 
     def create_thread(self, func:object, func_args: tuple = (), run_once:bool = False) -> None:
+        """Create a new thread and store it into running_threads variable
+
+        Args:
+            func (object): The method/function you want to execute via this thread
+            func_args (tuple, optional): Arguments of the function/method. Defaults to ().
+            run_once (bool, optional): If you want to ensure that this method/function run once. Defaults to False.
+        """
         try:
             func_name = func.__name__
 
@@ -275,10 +444,6 @@ class Base:
                     if thread.getName() == func_name:
                         return None
 
-            # if func_name in self.running_threads:
-            #     print(f"HeartBeat is running")
-            #     return None
-
             th = threading.Thread(target=func, args=func_args, name=str(func_name), daemon=True)
             th.start()
 
@@ -376,14 +541,23 @@ class Base:
 
     def __create_db(self) -> None:
 
-        table_logs = f'''CREATE TABLE IF NOT EXISTS {self.Config.table_log} (
+        table_core_log = f'''CREATE TABLE IF NOT EXISTS {self.Config.table_log} (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             datetime TEXT,
             server_msg TEXT
             ) 
         '''
 
-        table_cmds = f'''CREATE TABLE IF NOT EXISTS {self.Config.table_commande} (
+        table_core_config = f'''CREATE TABLE IF NOT EXISTS {self.Config.table_config} (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            datetime TEXT,
+            module_name TEXT,
+            param_key TEXT,
+            param_value TEXT
+            )
+        '''
+
+        table_core_log_command = f'''CREATE TABLE IF NOT EXISTS {self.Config.table_commande} (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             datetime TEXT,
             user TEXT,
@@ -391,15 +565,24 @@ class Base:
             )
         '''
 
-        table_modules = f'''CREATE TABLE IF NOT EXISTS {self.Config.table_module} (
+        table_core_module = f'''CREATE TABLE IF NOT EXISTS {self.Config.table_module} (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             datetime TEXT,
             user TEXT,
-            module TEXT
+            module_name TEXT,
+            isdefault INTEGER
+            )
+        '''
+        
+        table_core_channel = '''CREATE TABLE IF NOT EXISTS core_channel (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            datetime TEXT,
+            module_name TEXT,
+            channel_name TEXT
             )
         '''
 
-        table_admins = f'''CREATE TABLE IF NOT EXISTS {self.Config.table_admin} (
+        table_core_admin = f'''CREATE TABLE IF NOT EXISTS {self.Config.table_admin} (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             createdOn TEXT,
             user TEXT,
@@ -410,10 +593,12 @@ class Base:
             )
         '''
 
-        self.db_execute_query(table_logs)
-        self.db_execute_query(table_cmds)
-        self.db_execute_query(table_modules)
-        self.db_execute_query(table_admins)
+        self.db_execute_query(table_core_log)
+        self.db_execute_query(table_core_log_command)
+        self.db_execute_query(table_core_module)
+        self.db_execute_query(table_core_admin)
+        self.db_execute_query(table_core_channel)
+        self.db_execute_query(table_core_config)
 
         return None
 
@@ -519,7 +704,7 @@ class Base:
         self.periodic_func.clear()
 
     def clean_uid(self, uid:str) -> str:
-        """Clean UID by removing @ / % / + / Owner / and *
+        """Clean UID by removing @ / % / + / ~ / * / :
 
         Args:
             uid (str): The UID to clean
@@ -528,7 +713,7 @@ class Base:
             str: Clean UID without any sign
         """
 
-        pattern = fr'[@|%|\+|~|\*]*'
+        pattern = fr'[:|@|%|\+|~|\*]*'
         parsed_UID = re.sub(pattern, '', uid)
 
         return parsed_UID
@@ -542,11 +727,19 @@ class Base:
         Returns:
             bool: True if the string is a channel / False if this is not a channel
         """
+        try:
+            
+            if channelToCheck is None:
+                return False
 
-        pattern = fr'^#'
-        isChannel = re.findall(pattern, channelToCheck)
+            pattern = fr'^#'
+            isChannel = re.findall(pattern, channelToCheck)
 
-        if not isChannel:
-            return False
-        else:
-            return True
+            if not isChannel:
+                return False
+            else:
+                return True
+        except TypeError as te:
+            self.logs.error(f'TypeError: [{channelToCheck}] - {te}')
+        except Exception as err:
+            self.logs.error(f'TypeError: {err}')

core/connection.py

@@ -0,0 +1,205 @@
+import socket, ssl, time
+from ssl import SSLSocket
+from core.loadConf import Config
+from core.base import Base
+from typing import Union
+
+class Connection:
+
+    def __init__(self, server_port: int, nickname: str, username: str, channels:list[str], ssl:bool = False) -> None:
+        self.Config = Config().ConfigObject
+        self.Base = Base(self.Config)
+        self.IrcSocket: Union[socket.socket, SSLSocket] = None
+        self.signal: bool = True
+        self.nickname = nickname
+        self.username = username
+        self.channels:list[str] = channels
+        self.CHARSET = ['utf-8', 'iso-8859-1']
+
+        self.create_socket(self.Config.SERVEUR_IP, self.Config.SERVEUR_HOSTNAME, server_port, ssl)
+        self.send_connection_information_to_server(self.IrcSocket)
+        self.connect()
+
+    def create_socket(self, server_ip: str,  server_hostname: str, server_port: int, ssl: bool = False) -> bool:
+
+        try:
+            soc = socket.socket(socket.AF_INET, socket.SOCK_STREAM or socket.SOCK_NONBLOCK)
+            connexion_information = (server_ip, server_port)
+
+            if ssl:
+                # Créer un object ssl
+                ssl_context = self.__ssl_context()
+                ssl_connexion = ssl_context.wrap_socket(soc, server_hostname=server_hostname)
+                ssl_connexion.connect(connexion_information)
+                self.IrcSocket:SSLSocket = ssl_connexion
+                self.SSL_VERSION = self.IrcSocket.version()
+                self.Base.logs.debug(f'> Connexion en mode SSL : Version = {self.SSL_VERSION}')
+            else:
+                soc.connect(connexion_information)
+                self.IrcSocket:socket.socket = soc
+                self.Base.logs.debug(f'> Connexion en mode normal')
+
+            return True
+
+        except ssl.SSLEOFError as soe:
+            self.Base.logs.critical(f"SSLEOFError __create_socket: {soe} - {soc.fileno()}")
+            return False
+        except ssl.SSLError as se:
+            self.Base.logs.critical(f"SSLError __create_socket: {se} - {soc.fileno()}")
+            return False
+        except OSError as oe:
+            self.Base.logs.critical(f"OSError __create_socket: {oe} - {soc.fileno()}")
+            return False
+        except AttributeError as ae:
+            self.Base.logs.critical(f"AttributeError __create_socket: {ae} - {soc.fileno()}")
+            return False
+
+    def send2socket(self, send_message:str) -> None:
+        """Envoit les commandes à envoyer au serveur.
+
+        Args:
+            string (Str): contient la commande à envoyer au serveur.
+        """
+        try:
+            with self.Base.lock:
+                # print(f">{str(send_message)}")
+                self.IrcSocket.send(f"{send_message}\r\n".encode(self.CHARSET[0]))
+                self.Base.logs.debug(f'{send_message}')
+
+        except UnicodeDecodeError:
+            self.Base.logs.error(f'Decode Error try iso-8859-1 - message: {send_message}')
+            self.IrcSocket.send(f"{send_message}\r\n".encode(self.CHARSET[1],'replace'))
+        except UnicodeEncodeError:
+            self.Base.logs.error(f'Encode Error try iso-8859-1 - message: {send_message}')
+            self.IrcSocket.send(f"{send_message}\r\n".encode(self.CHARSET[1],'replace'))
+        except AssertionError as ae:
+            self.Base.logs.warning(f'Assertion Error {ae} - message: {send_message}')
+        except ssl.SSLEOFError as soe:
+            self.Base.logs.error(f"SSLEOFError: {soe} - {send_message}")
+        except ssl.SSLError as se:
+            self.Base.logs.error(f"SSLError: {se} - {send_message}")
+        except OSError as oe:
+            self.Base.logs.error(f"OSError: {oe} - {send_message}")
+
+    def send_connection_information_to_server(self, writer:Union[socket.socket, SSLSocket]) -> None:
+        """Créer le link et envoyer les informations nécessaires pour la 
+        connexion au serveur.
+
+        Args:
+            writer (StreamWriter): permet l'envoi des informations au serveur.
+        """
+        try:
+            nickname = self.nickname
+            username = self.username
+
+            # Envoyer un message d'identification
+            writer.send(f"USER {nickname} {username} {username} {nickname} {username} :{username}\r\n".encode('utf-8'))
+            writer.send(f"USER {username} {username} {username} :{username}\r\n".encode('utf-8'))
+            writer.send(f"NICK {nickname}\r\n".encode('utf-8'))
+
+            self.Base.logs.debug('Link information sent to the server')
+
+            return None
+        except AttributeError as ae:
+            self.Base.logs.critical(f'{ae}')
+
+    def connect(self):
+        try:
+
+            while self.signal:
+                try:
+                    # 4072 max what the socket can grab
+                    buffer_size = self.IrcSocket.getsockopt(socket.SOL_SOCKET, socket.SO_RCVBUF)
+                    data_in_bytes = self.IrcSocket.recv(buffer_size)
+                    data = data_in_bytes.splitlines(True)
+                    count_bytes = len(data_in_bytes)
+
+                    while count_bytes > 4070:
+                        # If the received message is > 4070 then loop and add the value to the variable
+                        new_data = self.IrcSocket.recv(buffer_size)
+                        data_in_bytes += new_data
+                        count_bytes = len(new_data)
+
+                    data = data_in_bytes.splitlines(True)
+
+                    if not data:
+                        break
+
+                    self.parser(data)
+
+                except ssl.SSLEOFError as soe:
+                    self.Base.logs.error(f"SSLEOFError __connect_to_irc: {soe} - {data}")
+                    self.signal = False
+                except ssl.SSLError as se:
+                    self.Base.logs.error(f"SSLError __connect_to_irc: {se} - {data}")
+                    self.signal = False
+                except OSError as oe:
+                    self.Base.logs.error(f"OSError __connect_to_irc: {oe} - {data}")
+                    self.signal = False
+
+            self.IrcSocket.shutdown(socket.SHUT_RDWR)
+            self.IrcSocket.close()
+            self.Base.logs.info("--> Clone Disconnected ...")
+
+        except AssertionError as ae:
+            self.Base.logs.error(f'Assertion error : {ae}')
+        except ValueError as ve:
+            self.Base.logs.error(f'Value Error : {ve}')
+        except ssl.SSLEOFError as soe:
+            self.Base.logs.error(f"OS Error __connect_to_irc: {soe}")
+        except AttributeError as atte:
+            self.Base.logs.critical(f"{atte}")
+        except Exception as e:
+            self.Base.logs.error(f"Exception: {e}")
+
+    def parser(self, cmd:list[bytes]):
+        try:
+            for data in cmd:
+                response = data.decode(self.CHARSET[0]).split()
+                # print(response)
+
+                match response[0]:
+                    case 'PING':
+                        pong = str(response[1]).replace(':','')
+                        self.send2socket(f"PONG :{pong}")
+                        return None
+                    case 'ERROR':
+                        error_value = str(response[1]).replace(':','')
+                        if error_value == 'Closing':
+                            self.signal = False
+
+                match response[1]:
+                    case '376':
+                        for channel in self.channels:
+                            self.send2socket(f"JOIN {channel}")
+                        return None
+                    case 'PRIVMSG':
+                        fullname = str(response[0]).replace(':', '')
+                        nickname = fullname.split('!')[0].replace(':','')
+                        if nickname == self.Config.SERVICE_NICKNAME:
+                            command = str(response[3]).replace(':','')
+                            if command == 'KILL':
+                                self.send2socket(f'QUIT :Thanks and goodbye')
+                                self.signal = False
+                            if command == 'JOIN':
+                                channel_to_join = str(response[4])
+                                self.send2socket(f"JOIN {channel_to_join}")
+
+        except UnicodeEncodeError:
+            for data in cmd:
+                response = data.decode(self.CHARSET[1],'replace').split()
+        except UnicodeDecodeError:
+            for data in cmd:
+                response = data.decode(self.CHARSET[1],'replace').split()
+        except AssertionError as ae:
+            self.Base.logs.error(f"Assertion error : {ae}")
+        pass
+
+    def __ssl_context(self) -> ssl.SSLContext:
+        ctx = ssl.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
+        ctx.check_hostname = False
+        ctx.verify_mode = ssl.CERT_NONE
+
+        self.Base.logs.debug(f'SSLContext initiated with verified mode {ctx.verify_mode}')
+
+        return ctx

core/installation.py

@@ -1,51 +1,85 @@
 from importlib.util import find_spec
-from subprocess import check_call, run, CalledProcessError
+from dataclasses import dataclass
+from pathlib import Path
+from subprocess import check_call, run, CalledProcessError, PIPE
 from platform import python_version, python_version_tuple
 from sys import exit
 import os
 
 class Install:
 
+    @dataclass
+    class CoreConfig:
+        unix_systemd_folder: str
+        service_file_name: str
+        service_cmd_executable: list
+        defender_main_executable: str
+        python_min_version: str
+        python_current_version_tuple: tuple[str, str, str]
+        python_current_version: str
+        defender_install_folder: str
+        venv_folder: str
+        venv_cmd_installation: list
+        venv_cmd_requirements: list
+        venv_pip_executable: str
+        venv_python_executable: str
+
     def __init__(self) -> None:
-        self.PYTHON_MIN_VERSION = '3.10'
 
-        self.venv_folder_name = '.pyenv'
-        self.cmd_venv_command = ['python3', '-m', 'venv', self.venv_folder_name]
-        self.module_to_install = ['sqlalchemy','psutil','requests']
+        self.set_configuration()
 
-        if not self.checkPythonVersion():
+        if not self.check_python_version():
             # Tester si c'est la bonne version de python
             exit("Python Version Error")
         else:
             # Sinon tester les dependances python et les installer avec pip
-            self.checkDependencies()
+            if self.do_install():
+
+                self.install_dependencies()
+
+                self.create_service_file()
+
+                self.print_final_message()
 
         return None
 
-    def checkPythonVersion(self) -> bool:
-        """Test si la version de python est autorisée ou non
+    def set_configuration(self):
 
-        Returns:
-            bool: True si la version de python est autorisé sinon False
-        """
-        # Current system version
-        sys_major, sys_minor, sys_patch = python_version_tuple()
+        defender_install_folder = os.getcwd()
+        venv_folder = '.pyenv'
+        unix_user_home_directory = os.path.expanduser("~")
+        unix_systemd_folder = os.path.join(unix_user_home_directory, '.config', 'systemd', 'user')
+        defender_main_executable = os.path.join(defender_install_folder, 'main.py')
 
-        # min python version required
-        python_required_version = self.PYTHON_MIN_VERSION.split('.')
-        min_major, min_minor = tuple((python_required_version[0], python_required_version[1]))
+        self.config = self.CoreConfig(
+                unix_systemd_folder=unix_systemd_folder,
+                service_file_name='defender.service',
+                service_cmd_executable=['systemctl', '--user', 'start', 'defender'],
+                defender_main_executable=defender_main_executable,
+                python_min_version='3.10',
+                python_current_version_tuple=python_version_tuple(),
+                python_current_version=python_version(),
+                defender_install_folder=defender_install_folder,
+                venv_folder=venv_folder,
+                venv_cmd_installation=['python3', '-m', 'venv', venv_folder],
+                venv_cmd_requirements=['sqlalchemy','psutil','requests'],
+                venv_pip_executable=f'{os.path.join(defender_install_folder, venv_folder, "bin")}{os.sep}pip',
+                venv_python_executable=f'{os.path.join(defender_install_folder, venv_folder, "bin")}{os.sep}python'
+            )
 
-        if int(sys_major) < int(min_major):
-            print(f"## Your python version must be greather than or equal to {self.PYTHON_MIN_VERSION} ##")
-            return False
+    def do_install(self) -> bool:
 
-        elif (int(sys_major) <= int(min_major)) and (int(sys_minor) < int(min_minor)):
-            print(f"## Your python version must be greather than or equal to {self.PYTHON_MIN_VERSION} ##")
-            return False
+        full_service_file_path = os.path.join(self.config.unix_systemd_folder, self.config.service_file_name)
 
-        print(f"===> Version of python : {python_version()} ==> OK")
+        if not os.path.exists(full_service_file_path):
+            print(f'/!\\ Service file does not exist /!\\')
+            return True
 
-        return True
+        # Check if virtual env exist
+        if not os.path.exists(f'{os.path.join(self.config.defender_install_folder, self.config.venv_folder)}'):
+            self.run_subprocess(self.config.venv_cmd_installation)
+            print(f'/!\\ Virtual env does not exist run the install /!\\')
+            return True
 
     def run_subprocess(self, command:list) -> None:
 
@@ -58,7 +92,41 @@ class Install:
             print(f"Try to install dependencies ...")
             exit(5)
 
-    def checkDependencies(self) -> None:
+    def check_python_version(self) -> bool:
+        """Test si la version de python est autorisée ou non
+
+        Returns:
+            bool: True si la version de python est autorisé sinon False
+        """
+        # Current system version
+        sys_major, sys_minor, sys_patch = self.config.python_current_version_tuple
+
+        # min python version required
+        python_required_version = self.config.python_min_version.split('.')
+        min_major, min_minor = tuple((python_required_version[0], python_required_version[1]))
+
+        if int(sys_major) < int(min_major):
+            print(f"## Your python version must be greather than or equal to {self.config.python_current_version} ##")
+            return False
+
+        elif (int(sys_major) <= int(min_major)) and (int(sys_minor) < int(min_minor)):
+            print(f"## Your python version must be greather than or equal to {self.config.python_current_version} ##")
+            return False
+
+        print(f"===> Version of python : {self.config.python_current_version} ==> OK")
+
+        return True
+
+    def check_package(self, package_name) -> bool:
+
+        try:
+            # Run a command in the virtual environment's Python to check if the package is installed
+            run([self.config.venv_python_executable, '-c', f'import {package_name}'], check=True, stdout=PIPE, stderr=PIPE)
+            return True
+        except CalledProcessError:
+            return False
+
+    def install_dependencies(self) -> None:
         """### Verifie les dépendances si elles sont installées
         - Test si les modules sont installés
         - Met a jour pip
@@ -67,38 +135,71 @@ class Install:
         do_install = False
 
         # Check if virtual env exist
-        if not os.path.exists(f'{self.venv_folder_name}'):
-            self.run_subprocess(self.cmd_venv_command)
+        if not os.path.exists(f'{os.path.join(self.config.defender_install_folder, self.config.venv_folder)}'):
+            self.run_subprocess(self.config.venv_cmd_installation)
             do_install = True
 
-        for module in self.module_to_install:
-            if find_spec(module) is None:
+        for module in self.config.venv_cmd_requirements:
+            if not self.check_package(module):
                 do_install = True
 
         if not do_install:
             return None
 
         print("===> Vider le cache de pip")
-        check_call(['pip','cache','purge'])
+        self.run_subprocess([self.config.venv_pip_executable, 'cache', 'purge'])
         
         print("===> Verifier si pip est a jour")
-        check_call(['python', '-m', 'pip', 'install', '--upgrade', 'pip'])
+        self.run_subprocess([self.config.venv_python_executable, '-m', 'pip', 'install', '--upgrade', 'pip'])
 
         if find_spec('greenlet') is None:
-            check_call(['pip','install', '--only-binary', ':all:', 'greenlet'])
+            self.run_subprocess([self.config.venv_pip_executable, 'install', '--only-binary', ':all:', 'greenlet'])
             print('====> Module Greenlet installé')
 
-        for module in self.module_to_install:
-            if find_spec(module) is None:
+        for module in self.config.venv_cmd_requirements:
+            if not self.check_package(module):
                 print("### Trying to install missing python packages ###")
-                check_call(['pip','install', module])
+                self.run_subprocess([self.config.venv_pip_executable, 'install', module])
                 print(f"====> Module {module} installé")
             else:
                 print(f"==> {module} already installed")
 
-        print(f"#"*12)
+    def create_service_file(self) -> None:
+
+        full_service_file_path = os.path.join(self.config.unix_systemd_folder, self.config.service_file_name)
+
+        if os.path.exists(full_service_file_path):
+            print(f'/!\\ Service file already exist /!\\')
+            return None
+
+        # Check if user systemd is available (.config/systemd/user/)
+        if not os.path.exists(self.config.unix_systemd_folder):
+            self.run_subprocess(['mkdir', '-p', self.config.unix_systemd_folder])
+
+            contain = f'''[Unit]
+Description=Defender IRC Service
+
+[Service]
+ExecStart={self.config.venv_python_executable} {self.config.defender_main_executable}
+WorkingDirectory={self.config.defender_install_folder}
+SyslogIdentifier=Defender
+Restart=on-failure
+
+[Install]
+WantedBy=multi-user.target
+    '''
+            with open(full_service_file_path, 'w+') as servicefile:
+                servicefile.write(contain)
+                servicefile.close()
+                print(f'Service file generated with current configuration')
+                print(f'Running Defender IRC Service ...')
+                self.run_subprocess(self.config.service_cmd_executable)
+
+    def print_final_message(self) -> None:
+
+        print(f"#"*24)
         print("Installation complete ...")
         print("You must change environment using the command below")
-        print(f"source {self.venv_folder_name}{os.sep}bin{os.sep}activate")
-        print(f"#"*12)
-        exit(1)
+        print(f"source {self.config.defender_install_folder}{os.sep}{self.config.venv_folder}{os.sep}bin{os.sep}activate")
+        print(f"#"*24)
+        exit(1)

core/irc.py

@@ -12,7 +12,6 @@ class Irc:
 
         self.defender_connexion_datetime = datetime.now()   # Date et heure de la premiere connexion de Defender
         self.first_score: int = 100
-        self.db_chan = []                                   # Definir la variable qui contiendra la liste des salons
         self.loaded_classes:dict[str, 'Irc'] = {}           # Definir la variable qui contiendra la liste modules chargés
         self.beat = 30                                      # Lancer toutes les 30 secondes des actions de nettoyages
         self.hb_active = True                               # Heartbeat active
@@ -31,7 +30,7 @@ class Irc:
             0: ['help', 'auth', 'copyright', 'uptime'],
             1: ['load','reload','unload', 'deauth', 'checkversion'],
             2: ['show_modules', 'show_timers', 'show_threads', 'show_channels', 'show_users', 'show_admins'],
-            3: ['quit', 'restart','addaccess','editaccess', 'delaccess']
+            3: ['quit', 'restart','addaccess','editaccess', 'delaccess','umode']
         }
 
         # l'ensemble des commandes.
@@ -44,6 +43,8 @@ class Irc:
         self.User = User(self.Base)
         self.Admin = Admin(self.Base)
         self.Channel = Channel(self.Base)
+
+        self.__create_table()
         self.Base.create_thread(func=self.heartbeat, func_args=(self.beat, ))
 
     ##############################################
@@ -90,7 +91,7 @@ class Irc:
         except OSError as oe:
             self.Base.logs.critical(f"OSError __create_socket: {oe} - {soc.fileno()}")
         except AttributeError as ae:
-            self.Base.logs.critical(f"OSError __create_socket: {oe} - {soc.fileno()}")
+            self.Base.logs.critical(f"AttributeError __create_socket: {ae} - {soc.fileno()}")
 
     def __ssl_context(self) -> ssl.SSLContext:
         ctx = ssl.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
@@ -107,6 +108,7 @@ class Irc:
             self.__link(self.IrcSocket)                         # établir la connexion au serveur IRC
             self.signal = True                                  # Une variable pour initier la boucle infinie
             self.load_existing_modules()                        # Charger les modules existant dans la base de données
+            self.__join_saved_channels()                        # Join existing channels
 
             while self.signal:
                 try:
@@ -158,17 +160,18 @@ class Irc:
             self.IrcSocket.shutdown(socket.SHUT_RDWR)
             self.IrcSocket.close()
             self.Base.logs.info("--> Fermeture de Defender ...")
+            sys.exit(0)
 
         except AssertionError as ae:
-            self.Base.logs.error(f'Assertion error : {ae}')
+            self.Base.logs.error(f'AssertionError: {ae}')
         except ValueError as ve:
-            self.Base.logs.error(f'Value Error : {ve}')
+            self.Base.logs.error(f'ValueError: {ve}')
         except ssl.SSLEOFError as soe:
-            self.Base.logs.error(f"OS Error __connect_to_irc: {soe}")
+            self.Base.logs.error(f"SSLEOFError: {soe}")
         except AttributeError as atte:
-            self.Base.logs.critical(f"{atte}")
-        # except Exception as e:
-        #     self.debug(f"Exception: {e}")
+            self.Base.logs.critical(f"AttributeError: {atte}")
+        except Exception as e:
+            self.Base.logs.critical(f"Exception: {e}")
 
     def __link(self, writer:Union[socket.socket, SSLSocket]) -> None:
         """Créer le link et envoyer les informations nécessaires pour la 
@@ -199,7 +202,8 @@ class Irc:
 
             # Envoyer un message d'identification
             writer.send(f":{sid} PASS :{password}\r\n".encode('utf-8'))
-            writer.send(f":{sid} PROTOCTL NICKv2 VHP UMODE2 NICKIP SJOIN SJOIN2 SJ3 NOQUIT TKLEXT MLOCK SID MTAGS\r\n".encode('utf-8'))
+            writer.send(f":{sid} PROTOCTL SID NOQUIT NICKv2 SJOIN SJ3 NICKIP TKLEXT2 NEXTBANS CLK EXTSWHOIS MLOCK MTAGS\r\n".encode('utf-8'))
+            # writer.send(f":{sid} PROTOCTL NICKv2 VHP UMODE2 NICKIP SJOIN SJOIN2 SJ3 NOQUIT TKLEXT MLOCK SID MTAGS\r\n".encode('utf-8'))
             writer.send(f":{sid} PROTOCTL EAUTH={link},,,{service_name}-v{version}\r\n".encode('utf-8'))
             writer.send(f":{sid} PROTOCTL SID={sid}\r\n".encode('utf-8'))
             writer.send(f":{sid} SERVER {link} 1 :{info}\r\n".encode('utf-8'))
@@ -207,7 +211,7 @@ class Irc:
             writer.send(f":{sid} UID {nickname} 1 {unixtime} {username} {host} {service_id} * {smodes} * * * :{realname}\r\n".encode('utf-8'))
             writer.send(f":{sid} SJOIN {unixtime} {chan} + :{service_id}\r\n".encode('utf-8'))
             writer.send(f":{sid} MODE {chan} +{cmodes}\r\n".encode('utf-8'))
-            writer.send(f":{service_id} SAMODE {chan} +{umodes} {nickname}\r\n".encode('utf-8'))
+            writer.send(f":{sid} SAMODE {chan} +{umodes} {nickname}\r\n".encode('utf-8'))
 
             self.Base.logs.debug('Link information sent to the server')
 
@@ -215,6 +219,19 @@ class Irc:
         except AttributeError as ae:
             self.Base.logs.critical(f'{ae}')
 
+    def __join_saved_channels(self) -> None:
+        
+        core_table = 'core_channel'
+        
+        query = f'''SELECT distinct channel_name FROM {core_table}'''
+        exec_query = self.Base.db_execute_query(query)
+        result_query = exec_query.fetchall()
+
+        if result_query:
+            for chan_name in result_query:
+                chan = chan_name[0]
+                self.send2socket(f":{self.Config.SERVEUR_ID} SJOIN {self.Base.get_unixtime()} {chan} + :{self.Config.SERVICE_ID}")
+
     def send2socket(self, send_message:str) -> None:
         """Envoit les commandes à envoyer au serveur.
 
@@ -268,13 +285,18 @@ class Irc:
     #             FIN CONNEXION IRC              #
     ##############################################
 
+    def __create_table(self):
+        """## Create core tables
+        """
+        pass
+
     def load_existing_modules(self) -> None:
         """Charge les modules qui existe déja dans la base de données
 
         Returns:
             None: Aucun retour requis, elle charge puis c'est tout
         """
-        result = self.Base.db_execute_query(f"SELECT module FROM {self.Config.table_module}")
+        result = self.Base.db_execute_query(f"SELECT module_name FROM {self.Config.table_module}")
         for r in result.fetchall():
             self.load_module('sys', r[0], True)
 
@@ -663,7 +685,6 @@ class Irc:
                             else:
                                 version = f'{current_version}'
 
-                            self.send2socket(f"MODE {self.Config.SERVICE_NICKNAME} +B")
                             self.send2socket(f"JOIN {self.Config.SERVICE_CHANLOG}")
                             print(f"################### DEFENDER ###################")
                             print(f"#               SERVICE CONNECTE                ")
@@ -819,8 +840,10 @@ class Irc:
                     try:
                         # Supprimer la premiere valeur
                         cmd.pop(0)
-
                         get_uid_or_nickname = str(cmd[0].replace(':',''))
+                        user_trigger = self.User.get_nickname(get_uid_or_nickname)
+                        dnickname = self.Config.SERVICE_NICKNAME
+
                         if len(cmd) == 6:
                             if cmd[1] == 'PRIVMSG' and str(cmd[3]).replace('.','') == ':auth':
                                 cmd_copy = cmd.copy()
@@ -830,10 +853,6 @@ class Irc:
                                 self.Base.logs.info(cmd)
                         else:
                             self.Base.logs.info(f'{cmd}')
-                        # user_trigger = get_user.split('!')[0]
-                        # user_trigger = self.get_nickname(get_uid_or_nickname)
-                        user_trigger = self.User.get_nickname(get_uid_or_nickname)
-                        dnickname = self.Config.SERVICE_NICKNAME
 
                         pattern = fr'(:\{self.Config.SERVICE_PREFIX})(.*)$'
                         hcmds = re.search(pattern, ' '.join(cmd)) # va matcher avec tout les caractéres aprés le .
@@ -850,7 +869,8 @@ class Irc:
                             cmd_to_send = convert_to_string.replace(':','')
                             self.Base.log_cmd(user_trigger, cmd_to_send)
 
-                            self._hcmds(user_trigger, arg, cmd)
+                            fromchannel = str(cmd[2]).lower() if self.Base.Is_Channel(cmd[2]) else None
+                            self._hcmds(user_trigger, fromchannel, arg, cmd)
 
                         if cmd[2] == self.Config.SERVICE_ID:
                             pattern = fr'^:.*?:(.*)$'
@@ -888,7 +908,11 @@ class Irc:
                                 cmd_to_send = convert_to_string.replace(':','')
                                 self.Base.log_cmd(self.User.get_nickname(user_trigger), cmd_to_send)
 
-                                self._hcmds(user_trigger, arg, cmd)
+                                fromchannel = None
+                                if len(arg) >= 2:
+                                    fromchannel = str(arg[1]).lower() if self.Base.Is_Channel(arg[1]) else None
+
+                                self._hcmds(user_trigger, fromchannel, arg, cmd)
 
                     except IndexError as io:
                         self.Base.logs.error(f'{io}')
@@ -904,7 +928,7 @@ class Irc:
         except IndexError as ie:
             self.Base.logs.error(f"{ie} / {cmd} / length {str(len(cmd))}")
 
-    def _hcmds(self, user: str, cmd:list, fullcmd: list = []) -> None:
+    def _hcmds(self, user: str, channel: Union[str, None], cmd:list, fullcmd: list = []) -> None:
 
         fromuser = self.User.get_nickname(user)                                   # Nickname qui a lancé la commande
         uid = self.User.get_uid(fromuser)                                         # Récuperer le uid de l'utilisateur
@@ -925,7 +949,7 @@ class Irc:
         # Envoyer la commande aux classes dynamiquement chargées
         if command != 'notallowed':
             for classe_name, classe_object in self.loaded_classes.items():
-                classe_object._hcmds(user, cmd, fullcmd)
+                classe_object._hcmds(user, channel, cmd, fullcmd)
 
         match command:
 
@@ -1229,9 +1253,6 @@ class Irc:
                     reason.append(cmd[i])
                 final_reason = ' '.join(reason)
 
-                # self.db_uid.clear()                     #Vider UID_DB
-                # self.db_chan = []                       #Vider les salons
-
                 self.User.UID_DB.clear()                # Clear User Object
                 self.Channel.UID_CHANNEL_DB.clear()     # Clear Channel Object
 
@@ -1250,7 +1271,7 @@ class Irc:
                 self.Base.logs.debug(self.loaded_classes)
                 all_modules  = self.Base.get_all_modules()
 
-                results = self.Base.db_execute_query(f'SELECT module FROM {self.Config.table_module}')
+                results = self.Base.db_execute_query(f'SELECT module_name FROM {self.Config.table_module}')
                 results = results.fetchall()
 
                 # if len(results) == 0:
@@ -1277,17 +1298,15 @@ class Irc:
             case 'show_timers':
 
                 if self.Base.running_timers:
-                    self.send2socket(f":{dnickname} PRIVMSG {dchanlog} :{self.Base.running_timers}")
+                    for the_timer in self.Base.running_timers:
+                        self.send2socket(f":{dnickname} NOTICE {fromuser} :>> {the_timer.getName()} - {the_timer.is_alive()}")
                 else:
-                    self.send2socket(f":{dnickname} PRIVMSG {dchanlog} :Aucun timers en cours d'execution")
+                    self.send2socket(f":{dnickname} NOTICE {fromuser} :Aucun timers en cours d'execution")
 
             case 'show_threads':
 
-                running_thread_name:list = []
                 for thread in self.Base.running_threads:
-                    running_thread_name.append(f"{thread.getName()} ({thread.is_alive()})")
-
-                self.send2socket(f":{dnickname} PRIVMSG {dchanlog} :{str(running_thread_name)}")
+                    self.send2socket(f":{dnickname} NOTICE {fromuser} :>> {thread.getName()} ({thread.is_alive()})")
 
             case 'show_channels':
 
@@ -1322,5 +1341,15 @@ class Irc:
                     (fromuser, )
                 )
 
+            case 'umode':
+                try:
+                    # .umode nickname +mode
+                    nickname = str(cmd[1])
+                    umode = str(cmd[2])
+
+                    self.send2socket(f':{dnickname} SVSMODE {nickname} {umode}')
+                except KeyError as ke:
+                    self.Base.logs.error(ke)
+
             case _:
                 pass

core/loadConf.py

@@ -52,6 +52,8 @@ class ConfigDataModel:
     table_commande: str
     table_log: str
     table_module: str
+    table_config: str
+    table_channel: str
 
     current_version: str
     latest_version: str
@@ -113,10 +115,12 @@ class Config:
             GLINE_DURATION=import_config["GLINE_DURATION"],
             DEBUG_LEVEL=import_config["DEBUG_LEVEL"],
             CONFIG_COLOR=import_config["CONFIG_COLOR"],
-            table_admin='sys_admins',
-            table_commande='sys_commandes',
-            table_log='sys_logs',
-            table_module='sys_modules',
+            table_admin='core_admin',
+            table_commande='core_command',
+            table_log='core_log',
+            table_module='core_module',
+            table_config='core_config',
+            table_channel='core_channel',
             current_version='',
             latest_version='',
             db_name='defender',