# -*- coding: utf-8 -*-
"""
@Time ： 2023/12/11 12:49
@Auth ： Juexiao Zhou
@File ：executor.py
@IDE ：PyCharm
@Page: www.joshuachou.ink
"""

import subprocess
import time

class CodeExecutor:
    def __init__(self):
        self.bash_code_path = None
        self.code_prefix = [
            'mamba activate abc_runtime',
        ]
        self.code_postfix = [
        ]

    def execute(self, bash_code_path):

        self.bash_code_path = bash_code_path
        with open(self.bash_code_path, 'r') as input_file:
            bash_content = input_file.read()

        self.bash_code_path_execute = self.bash_code_path + '.execute.sh'

        # 打开新生成的 Bash 文件以供写入
        with open(self.bash_code_path_execute, 'w') as output_file:
            for code in self.code_prefix:
                output_file.write(code + '\n')
            # 写入原始内容
            output_file.write(bash_content)
            output_file.write('\n')  # 确保在新行开始
            for code in self.code_postfix:
                output_file.write(code + '\n')

        # 使用 subprocess 执行 Bash 文件，将输出捕获到一个字符串中
        process = subprocess.Popen(['bash', '-i', '-e', self.bash_code_path_execute],
                                            stdout=subprocess.PIPE,
                                            stderr=subprocess.PIPE,
                                            text=True)

        # 实时读取输出并打印
        stdout = []
        while True:
            output = process.stdout.readline()
            if output == '' and process.poll() is not None:
                break
            print(f'[stdout] {output.strip()}')
            stdout.append(f'[stdout] {output.strip()}')

        stderr = []
        for _ in process.stderr.readlines():
            if 'EnvironmentNameNotFound' in _ or '\n' == _:
                pass
            else:
                print(f"[stderr] {_}", end='')
                stderr.append(_)

        if len(stdout) > 10:
            stdout = stdout[-10:]
        if len(stderr) > 10:
            stderr = stderr[-10:]

        stdout = '\n'.join(stdout)
        stderr = '\n'.join(stderr)

        process.communicate()

        executor_info = stdout + '\n' + stderr
        return executor_info