import wx
import os
import subprocess

class BatManagerFrame(wx.Frame):
    def __init__(self):
        super().__init__(parent=None, title='BAT文件管理器')
        panel = wx.Panel(self)

        # 創(chuàng)建控件
        self.bat_combo = wx.ComboBox(panel, style=wx.CB_DROPDOWN | wx.TE_PROCESS_ENTER)
        self.memo = wx.TextCtrl(panel, style=wx.TE_MULTILINE)
        save_button = wx.Button(panel, label='保存')
        run_button = wx.Button(panel, label='運(yùn)行')

        # 綁定事件
        self.bat_combo.Bind(wx.EVT_COMBOBOX, self.on_select)
        self.bat_combo.Bind(wx.EVT_TEXT_ENTER, self.on_select)
        save_button.Bind(wx.EVT_BUTTON, self.on_save)
        run_button.Bind(wx.EVT_BUTTON, self.on_run)

        # 設(shè)置布局
        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(self.bat_combo, 0, wx.ALL | wx.EXPAND, 5)
        sizer.Add(self.memo, 1, wx.ALL | wx.EXPAND, 5)
        button_sizer = wx.BoxSizer(wx.HORIZONTAL)
        button_sizer.Add(save_button, 0, wx.ALL, 5)
        button_sizer.Add(run_button, 0, wx.ALL, 5)
        sizer.Add(button_sizer, 0, wx.CENTER)

        panel.SetSizer(sizer)

        # 加載現(xiàn)有的.bat文件
        self.load_bat_files()

    def load_bat_files(self):
        bat_dir = './bat'
        if not os.path.exists(bat_dir):
            os.makedirs(bat_dir)
        for file in os.listdir(bat_dir):
            if file.endswith('.bat'):
                self.bat_combo.Append(file)

    def on_select(self, event):
        selected_file = self.bat_combo.GetValue()
        if selected_file:
            file_path = os.path.join('./bat', selected_file)
            if os.path.exists(file_path):
                with open(file_path, 'r') as f:
                    self.memo.SetValue(f.read())
            else:
                self.memo.Clear()

    def on_save(self, event):
        file_name = self.bat_combo.GetValue()
        if not file_name.endswith('.bat'):
            file_name += '.bat'
        file_path = os.path.join('./bat', file_name)
        with open(file_path, 'w') as f:
            f.write(self.memo.GetValue())
        if file_name not in self.bat_combo.GetItems():
            self.bat_combo.Append(file_name)
        wx.MessageBox(f'已保存文件: {file_name}', '保存成功')

    # def on_run(self, event):
    #     file_name = self.bat_combo.GetValue()
    #     if file_name:
    #         file_path = os.path.join('./bat', file_name)
    #         if os.path.exists(file_path):
    #             try:
    #                 subprocess.run([file_path], check=True, shell=True)
    #                 wx.MessageBox(f'成功執(zhí)行: {file_name}', '執(zhí)行成功')
    #             except subprocess.CalledProcessError as e:
    #                 wx.MessageBox(f'執(zhí)行失敗: {e}', '錯(cuò)誤', wx.ICON_ERROR)
    #         else:
    #             wx.MessageBox(f'文件不存在: {file_name}', '錯(cuò)誤', wx.ICON_ERROR)
    def on_run(self, event):
        file_name = self.bat_combo.GetValue()
        if file_name:
            file_path = os.path.join('bat', file_name)
            if os.path.exists(file_path):
                try:
                    result = subprocess.run([file_path], capture_output=True, text=True, shell=True)
                    if result.returncode == 0:
                        wx.MessageBox(f'成功執(zhí)行: {file_name}\n\n輸出:\n{result.stdout}', '執(zhí)行成功')
                    else:
                        error_msg = f'執(zhí)行失敗: {file_name}\n\n錯(cuò)誤代碼: {result.returncode}\n'
                        error_msg += f'標(biāo)準(zhǔn)輸出:\n{result.stdout}\n'
                        error_msg += f'錯(cuò)誤輸出:\n{result.stderr}'
                        wx.MessageBox(error_msg, '執(zhí)行失敗', wx.ICON_ERROR)
                except Exception as e:
                    wx.MessageBox(f'執(zhí)行時(shí)發(fā)生異常: {str(e)}', '錯(cuò)誤', wx.ICON_ERROR)
            else:
                wx.MessageBox(f'文件不存在: {file_path}', '錯(cuò)誤', wx.ICON_ERROR)
        else:
            wx.MessageBox('請(qǐng)選擇或輸入一個(gè).bat文件名', '錯(cuò)誤', wx.ICON_ERROR)

    
if __name__ == '__main__':
    app = wx.App()
    frame = BatManagerFrame()
    frame.Show()
    app.MainLoop()