modules/cmd.py

import os
import time
import threading
import subprocess
import gradio as gr
from utils import EN_US

ZH2EN = {
    "状态栏": "Status",
    "执行结果": "Command output",
    "命令执行测试工具": "Command executor",
    "输入一个命令, 点击查看其执行结果": "Enter a command, and click to see its result",
    "执行超时时间": "Execution timeout",
}


def _L(zh_txt: str):
    return ZH2EN[zh_txt] if EN_US else zh_txt


class Inject:
    def __init__(self):
        self.output = ""
        self.status = "Success"
        self.thread: threading.Thread = None
        self.timeout = 10
        self.delay = 0.1

    def run(self, cmd: str):
        try:
            self.output = subprocess.check_output(
                cmd,
                shell=True,
                stderr=subprocess.STDOUT,
                text=True,
            )

        except Exception as e:
            self.status = f" {e} "
            self.output = ""

    def infer(self, cmd: str):
        self.status = "Success"
        self.output = ""
        try:
            if self.thread and self.thread.is_alive():
                self.thread.join(timeout=0)

            self.thread = threading.Thread(
                target=self.run,
                args=(cmd,),
                daemon=True,
            )
            self.thread.start()
            delay = 0
            while self.thread and self.thread.is_alive():
                delay += self.delay
                time.sleep(self.delay)
                if delay > self.timeout:
                    self.thread.join(timeout=0)
                    self.status = "Killed"
                    self.output = _L("执行超时时间") + f": {self.timeout}s"
                    break

        except Exception as e:
            self.status = f" {e} "
            self.output = ""

        return self.status, self.output


def cmd_inject():
    inject = Inject()
    return gr.Interface(
        fn=inject.infer,
        inputs=gr.Textbox(
            label=_L("命令执行测试工具"),
            value="dir" if os.name == "nt" else "ls /",
            placeholder=_L("输入一个命令, 点击查看其执行结果"),
        ),
        outputs=[
            gr.Textbox(label=_L("状态栏"), show_copy_button=True),
            gr.TextArea(label=_L("执行结果"), show_copy_button=True),
        ],
        flagging_mode="never",
    )