New version

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)