import subprocess
import psutil
from .custom_logging import setup_logging

# Set up logging
log = setup_logging()


class CommandExecutor:
    def __init__(self):
        self.process = None

    def execute_command(self, run_cmd, **kwargs):
        if self.process and self.process.poll() is None:
            log.info(
                'The command is already running. Please wait for it to finish.'
            )
        else:
            self.process = subprocess.Popen(run_cmd, shell=True, **kwargs)

    def kill_command(self):
        if self.process and self.process.poll() is None:
            try:
                parent = psutil.Process(self.process.pid)
                for child in parent.children(recursive=True):
                    child.kill()
                parent.kill()
                log.info('The running process has been terminated.')
            except psutil.NoSuchProcess:
                log.info('The process does not exist.')
            except Exception as e:
                log.info(f'Error when terminating process: {e}')
        else:
            log.info('There is no running process to kill.')