def encrypt_file(key, in_path, out_path):
    """使用AES-256 CBC模式加密文件"""
    iv = get_random_bytes(16)  # 生成隨機初始化向量
    cipher = AES.new(key, AES.MODE_CBC, iv)  # 創(chuàng)建加密器
    
    with open(in_path, 'rb') as f_in:
        data = f_in.read()  # 讀取原始文件內容
    
    # 加密并填充數據
    ct_bytes = cipher.encrypt(pad(data, AES.block_size))
    encrypted = iv + ct_bytes  # 組合IV和密文
    
    with open(out_path, 'wb') as f_out:
        f_out.write(encrypted)  # 寫入加密文件

def generate_stub_code(password_hash, encrypted_data_base64):
    """生成包含密碼驗證界面的Python代碼"""
    return f'''#!/usr/bin/env python
# Tkinter運行時修復 - 解決打包環(huán)境下的關鍵問題
def fix_tkinter_runtime():
    if getattr(sys, 'frozen', False):
        base_path = sys._MEIPASS
        tk_data_dir = os.path.join(base_path, '_tk_data')
        if not os.path.exists(tk_data_dir):
            tk_data_dir = os.path.join(base_path, 'tk', 'data')
        os.environ['TKDATA'] = tk_data_dir  # 設置環(huán)境變量

# 在創(chuàng)建Tkinter界面之前調用修復函數
fix_tkinter_runtime()

# 密碼驗證邏輯
def check_password():
    password = password_entry.get()
    key = hashlib.sha256(password.encode()).digest()  # 從密碼生成密鑰
    decrypted_data = decrypt_data(key)  # 嘗試解密
    if decrypted_data:
        root.destroy()
        execute_decrypted(decrypted_data)  # 執(zhí)行解密后的程序

# 創(chuàng)建美觀的解鎖界面
root = tk.Tk()
root.title("程序解鎖")
root.geometry("260x230")
root.resizable(False, False)
root.configure(bg="#f0f0f0")

# 界面組件：圖標、輸入框、按鈕等
icon_label = ttk.Label(main_frame, text="??", font=("Arial", 24))
password_entry = ttk.Entry(main_frame, show="*", width=25)
submit_btn = ttk.Button(main_frame, text="解鎖程序", command=check_password, width=15)
'''

def main():
    """主GUI界面和處理流程"""
    # 文件選擇函數
    def select_exe():
        # 自動設置輸出路徑
        output_path = os.path.join(dir_name, f"{name_without_ext}_protected.exe")
    
    # 加密處理核心邏輯
    def encrypt():
        # 驗證輸入
        if not all([exe_path, output_path, password, confirm]):
            messagebox.showerror("錯誤", "所有字段都必須填寫")
            return
        
        # 生成密鑰和哈希
        password_hash = hashlib.sha256(password.encode()).hexdigest()
        key = hashlib.sha256(password.encode()).digest()
        
        # 加密文件
        encrypt_file(key, exe_path, encrypted_temp_path)
        
        # 生成stub代碼
        stub_code = generate_stub_code(password_hash, encrypted_data_base64)
        
        # PyInstaller打包處理
        pyinstaller_cmd = [
            *find_pyinstaller(),  # 自動檢測PyInstaller
            '--onefile', '--noconsole',
            '--name', output_basename,
            *icon_cmd,  # 圖標參數
            stub_path
        ]
        
        # 執(zhí)行打包
        subprocess.run(pyinstaller_cmd, shell=sys.platform.startswith('win'))
        
        # 處理結果
        if os.path.exists(generated_exe):
            shutil.copy(generated_exe, output_path)
            status_label.config(text=f"加密成功! 文件已保存到:\n{output_path}", fg="green")

if __name__ == "__main__":
    # 自動安裝缺失依賴
    try:
        from Crypto.Cipher import AES
    except ImportError:
        print("正在安裝依賴庫pycryptodome...")
        subprocess.check_call([sys.executable, "-m", "pip", "install", "pycryptodome"])
        os.execl(sys.executable, sys.executable, *sys.argv)  # 重啟腳本
    
    # 檢查PyInstaller
    try:
        import PyInstaller
    except ImportError:
        subprocess.check_call([sys.executable, "-m", "pip", "install", "pyinstaller"])
    
    main()

def fix_tkinter_runtime():
    """解決PyInstaller打包環(huán)境下Tkinter的資源路徑問題"""
    if getattr(sys, 'frozen', False):
        base_path = sys._MEIPASS
        tk_data_dir = os.path.join(base_path, '_tk_data')
        if not os.path.exists(tk_data_dir):
            tk_data_dir = os.path.join(base_path, 'tk', 'data')
        os.environ['TKDATA'] = tk_data_dir

def find_pyinstaller():
    """自動檢測系統(tǒng)中可用的PyInstaller安裝方式"""
    if shutil.which('pyinstaller'):
        return ['pyinstaller']
    
    # 嘗試通過Python解釋器調用
    for py_exe in ['python', 'python3']:
        if shutil.which(py_exe):
            try:
                subprocess.run([py_exe, '-c', 'import PyInstaller'], check=True)
                return [py_exe, '-m', 'PyInstaller']
            except:
                continue
    return None

try:
    # 加密和打包過程
except Exception as e:
    error_details = traceback.format_exc()
    messagebox.showerror(
        "錯誤", 
        f"加密過程中出錯:\n\n{str(e)}\n\n"
        f"詳細錯誤信息:\n{error_details[:1000]}"
    )
finally:
    # 確保清理臨時文件
    shutil.rmtree(temp_dir, ignore_errors=True)