mirror of
https://github.com/hyb-oyqq/FRAISEMOE-Addons-Installer-NEXT.git
synced 2025-12-16 03:40:27 +00:00
393 lines
16 KiB
Python
393 lines
16 KiB
Python
import os
|
||
from PySide6.QtWidgets import (
|
||
QDialog, QVBoxLayout, QLabel, QListWidget, QPushButton, QHBoxLayout,
|
||
QAbstractItemView, QRadioButton, QButtonGroup, QFileDialog, QMessageBox
|
||
)
|
||
from PySide6.QtCore import QObject
|
||
from PySide6.QtGui import QFont
|
||
from utils import msgbox_frame
|
||
|
||
class PatchToggleHandler(QObject):
|
||
"""
|
||
处理补丁启用/禁用功能的类
|
||
"""
|
||
def __init__(self, main_window):
|
||
"""
|
||
初始化补丁切换处理程序
|
||
|
||
Args:
|
||
main_window: 主窗口实例，用于访问其他组件
|
||
"""
|
||
super().__init__()
|
||
self.main_window = main_window
|
||
self.debug_manager = main_window.debug_manager
|
||
self.game_detector = main_window.game_detector
|
||
self.patch_manager = main_window.patch_manager
|
||
self.app_name = main_window.patch_manager.app_name
|
||
|
||
def handle_toggle_patch_button_click(self):
|
||
"""
|
||
处理禁/启用补丁按钮点击事件
|
||
打开文件选择对话框选择游戏目录，然后禁用或启用对应游戏的补丁
|
||
"""
|
||
# 获取游戏目录
|
||
debug_mode = self.debug_manager._is_debug_mode()
|
||
|
||
# 提示用户选择目录
|
||
file_dialog_info = "选择游戏上级目录" if debug_mode else "选择游戏目录"
|
||
selected_folder = QFileDialog.getExistingDirectory(self.main_window, file_dialog_info, "")
|
||
|
||
if not selected_folder or selected_folder == "":
|
||
return # 用户取消了选择
|
||
|
||
if debug_mode:
|
||
print(f"DEBUG: 禁/启用功能 - 用户选择了目录: {selected_folder}")
|
||
|
||
# 首先尝试将选择的目录视为上级目录，使用增强的目录识别功能
|
||
game_dirs = self.game_detector.identify_game_directories_improved(selected_folder)
|
||
|
||
if game_dirs and len(game_dirs) > 0:
|
||
self._handle_multiple_games(game_dirs, debug_mode)
|
||
else:
|
||
self._handle_single_game(selected_folder, debug_mode)
|
||
|
||
def _handle_multiple_games(self, game_dirs, debug_mode):
|
||
"""
|
||
处理多个游戏的补丁切换
|
||
|
||
Args:
|
||
game_dirs: 游戏目录字典
|
||
debug_mode: 是否为调试模式
|
||
"""
|
||
if debug_mode:
|
||
print(f"DEBUG: 禁/启用功能 - 在上级目录中找到以下游戏: {list(game_dirs.keys())}")
|
||
|
||
# 查找已安装补丁的游戏，只处理那些已安装补丁的游戏
|
||
games_with_patch = {}
|
||
for game_version, game_dir in game_dirs.items():
|
||
if self.patch_manager.check_patch_installed(game_dir, game_version):
|
||
# 检查补丁当前状态（是否禁用）
|
||
is_disabled, disabled_path = self.patch_manager.check_patch_disabled(game_dir, game_version)
|
||
status = "已禁用" if is_disabled else "已启用"
|
||
games_with_patch[game_version] = {
|
||
"dir": game_dir,
|
||
"disabled": is_disabled,
|
||
"status": status
|
||
}
|
||
if debug_mode:
|
||
print(f"DEBUG: 禁/启用功能 - {game_version} 已安装补丁，当前状态: {status}")
|
||
|
||
# 检查是否有已安装补丁的游戏
|
||
if not games_with_patch:
|
||
QMessageBox.information(
|
||
self.main_window,
|
||
f"提示 - {self.app_name}",
|
||
"\n未在选择的目录中找到已安装补丁的游戏。\n请确认您选择了正确的游戏目录，并且该目录中的游戏已安装过补丁。\n",
|
||
QMessageBox.StandardButton.Ok
|
||
)
|
||
return
|
||
|
||
# 显示选择对话框
|
||
selected_games, operation = self._show_multi_game_dialog(games_with_patch)
|
||
|
||
if not selected_games:
|
||
return # 用户取消了操作
|
||
|
||
# 过滤games_with_patch，只保留选中的游戏
|
||
selected_game_dirs = {}
|
||
for game in selected_games:
|
||
if game in games_with_patch:
|
||
selected_game_dirs[game] = games_with_patch[game]["dir"]
|
||
|
||
# 确认操作
|
||
operation_text = "禁用" if operation == "disable" else "启用" if operation == "enable" else "切换"
|
||
game_list = '\n'.join([f"{game} ({games_with_patch[game]['status']})" for game in selected_games])
|
||
reply = QMessageBox.question(
|
||
self.main_window,
|
||
f"确认{operation_text}操作 - {self.app_name}",
|
||
f"\n确定要{operation_text}以下游戏补丁吗？\n\n{game_list}\n",
|
||
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
|
||
QMessageBox.StandardButton.No
|
||
)
|
||
|
||
if reply == QMessageBox.StandardButton.No:
|
||
return
|
||
|
||
# 执行批量操作
|
||
self._execute_batch_toggle(selected_game_dirs, operation, debug_mode)
|
||
|
||
def _handle_single_game(self, selected_folder, debug_mode):
|
||
"""
|
||
处理单个游戏的补丁切换
|
||
|
||
Args:
|
||
selected_folder: 选择的游戏目录
|
||
debug_mode: 是否为调试模式
|
||
"""
|
||
# 未找到游戏目录，尝试将选择的目录作为游戏目录
|
||
if debug_mode:
|
||
print(f"DEBUG: 禁/启用功能 - 未在上级目录找到游戏，尝试将选择的目录视为游戏目录")
|
||
|
||
game_version = self.game_detector.identify_game_version(selected_folder)
|
||
|
||
if game_version:
|
||
if debug_mode:
|
||
print(f"DEBUG: 禁/启用功能 - 识别为游戏: {game_version}")
|
||
|
||
# 检查是否已安装补丁
|
||
if self.patch_manager.check_patch_installed(selected_folder, game_version):
|
||
# 检查补丁当前状态
|
||
is_disabled, disabled_path = self.patch_manager.check_patch_disabled(selected_folder, game_version)
|
||
current_status = "已禁用" if is_disabled else "已启用"
|
||
|
||
# 显示单游戏操作对话框
|
||
operation = self._show_single_game_dialog(game_version, current_status, is_disabled)
|
||
|
||
if not operation:
|
||
return # 用户取消了操作
|
||
|
||
# 执行操作
|
||
result = self.patch_manager.toggle_patch(selected_folder, game_version, operation=operation)
|
||
if not result["success"]:
|
||
# 操作失败的消息已在toggle_patch中显示
|
||
pass
|
||
else:
|
||
# 没有安装补丁
|
||
QMessageBox.information(
|
||
self.main_window,
|
||
f"提示 - {self.app_name}",
|
||
f"\n未在 {game_version} 中找到已安装的补丁。\n请确认该游戏已经安装过补丁。\n",
|
||
QMessageBox.StandardButton.Ok
|
||
)
|
||
else:
|
||
# 两种方式都未识别到游戏
|
||
if debug_mode:
|
||
print(f"DEBUG: 禁/启用功能 - 无法识别游戏")
|
||
|
||
msg_box = msgbox_frame(
|
||
f"错误 - {self.app_name}",
|
||
"\n所选目录不是有效的NEKOPARA游戏目录。\n请选择包含游戏可执行文件的目录或其上级目录。\n",
|
||
QMessageBox.StandardButton.Ok,
|
||
)
|
||
msg_box.exec()
|
||
|
||
def _show_multi_game_dialog(self, games_with_patch):
|
||
"""
|
||
显示多游戏选择对话框
|
||
|
||
Args:
|
||
games_with_patch: 已安装补丁的游戏信息
|
||
|
||
Returns:
|
||
tuple: (选择的游戏列表, 操作类型)
|
||
"""
|
||
dialog = QDialog(self.main_window)
|
||
dialog.setWindowTitle("选择要操作的游戏补丁")
|
||
dialog.resize(400, 400) # 增加高度以适应新增的单选按钮
|
||
|
||
layout = QVBoxLayout(dialog)
|
||
|
||
# 添加"已安装补丁的游戏"标签
|
||
already_installed_label = QLabel("已安装补丁的游戏:", dialog)
|
||
already_installed_label.setFont(QFont(already_installed_label.font().family(), already_installed_label.font().pointSize(), QFont.Bold))
|
||
layout.addWidget(already_installed_label)
|
||
|
||
# 添加游戏列表和状态
|
||
games_status_text = ""
|
||
for game, info in games_with_patch.items():
|
||
games_status_text += f"{game} ({info['status']})\n"
|
||
games_status_label = QLabel(games_status_text.strip(), dialog)
|
||
layout.addWidget(games_status_label)
|
||
|
||
# 添加一些间距
|
||
layout.addSpacing(10)
|
||
|
||
# 添加"请选择要操作的游戏"标签
|
||
info_label = QLabel("请选择要操作的游戏:", dialog)
|
||
info_label.setFont(QFont(info_label.font().family(), info_label.font().pointSize(), QFont.Bold))
|
||
layout.addWidget(info_label)
|
||
|
||
# 添加列表控件，只显示已安装补丁的游戏
|
||
list_widget = QListWidget(dialog)
|
||
list_widget.setSelectionMode(QAbstractItemView.SelectionMode.ExtendedSelection) # 允许多选
|
||
for game, info in games_with_patch.items():
|
||
list_widget.addItem(f"{game} ({info['status']})")
|
||
layout.addWidget(list_widget)
|
||
|
||
# 添加全选按钮
|
||
select_all_btn = QPushButton("全选", dialog)
|
||
select_all_btn.clicked.connect(lambda: list_widget.selectAll())
|
||
layout.addWidget(select_all_btn)
|
||
|
||
# 添加操作选择单选按钮
|
||
operation_label = QLabel("请选择要执行的操作:", dialog)
|
||
operation_label.setFont(QFont(operation_label.font().family(), operation_label.font().pointSize(), QFont.Bold))
|
||
layout.addWidget(operation_label)
|
||
|
||
# 创建单选按钮组
|
||
radio_button_group = QButtonGroup(dialog)
|
||
|
||
# 添加"自动切换状态"单选按钮（默认选中）
|
||
auto_toggle_radio = QRadioButton("自动切换状态（禁用<->启用）", dialog)
|
||
auto_toggle_radio.setChecked(True)
|
||
radio_button_group.addButton(auto_toggle_radio, 0)
|
||
layout.addWidget(auto_toggle_radio)
|
||
|
||
# 添加"全部禁用"单选按钮
|
||
disable_all_radio = QRadioButton("禁用选中的补丁", dialog)
|
||
radio_button_group.addButton(disable_all_radio, 1)
|
||
layout.addWidget(disable_all_radio)
|
||
|
||
# 添加"全部启用"单选按钮
|
||
enable_all_radio = QRadioButton("启用选中的补丁", dialog)
|
||
radio_button_group.addButton(enable_all_radio, 2)
|
||
layout.addWidget(enable_all_radio)
|
||
|
||
# 添加确定和取消按钮
|
||
buttons_layout = QHBoxLayout()
|
||
ok_button = QPushButton("确定", dialog)
|
||
cancel_button = QPushButton("取消", dialog)
|
||
buttons_layout.addWidget(ok_button)
|
||
buttons_layout.addWidget(cancel_button)
|
||
layout.addLayout(buttons_layout)
|
||
|
||
# 连接按钮事件
|
||
ok_button.clicked.connect(dialog.accept)
|
||
cancel_button.clicked.connect(dialog.reject)
|
||
|
||
# 显示对话框并等待用户选择
|
||
result = dialog.exec()
|
||
|
||
if result != QDialog.DialogCode.Accepted or list_widget.selectedItems() == []:
|
||
# 用户取消或未选择任何游戏
|
||
return [], None
|
||
|
||
# 获取用户选择的游戏
|
||
selected_items = [item.text() for item in list_widget.selectedItems()]
|
||
selected_games = []
|
||
|
||
# 从选中项文本中提取游戏名称
|
||
for item in selected_items:
|
||
# 去除状态后缀 " (已启用)" 或 " (已禁用)"
|
||
game_name = item.split(" (")[0]
|
||
selected_games.append(game_name)
|
||
|
||
# 获取选中的操作类型
|
||
operation = None
|
||
if radio_button_group.checkedId() == 1: # 禁用选中的补丁
|
||
operation = "disable"
|
||
elif radio_button_group.checkedId() == 2: # 启用选中的补丁
|
||
operation = "enable"
|
||
# 否则为None，表示自动切换状态
|
||
|
||
return selected_games, operation
|
||
|
||
def _show_single_game_dialog(self, game_version, current_status, is_disabled):
|
||
"""
|
||
显示单游戏操作对话框
|
||
|
||
Args:
|
||
game_version: 游戏版本
|
||
current_status: 当前补丁状态
|
||
is_disabled: 是否已禁用
|
||
|
||
Returns:
|
||
str: 操作类型，"enable"或"disable"，或None表示取消
|
||
"""
|
||
dialog = QDialog(self.main_window)
|
||
dialog.setWindowTitle(f"{game_version} 补丁操作")
|
||
dialog.resize(300, 200)
|
||
|
||
layout = QVBoxLayout(dialog)
|
||
|
||
# 添加当前状态标签
|
||
status_label = QLabel(f"当前补丁状态: {current_status}", dialog)
|
||
status_label.setFont(QFont(status_label.font().family(), status_label.font().pointSize(), QFont.Bold))
|
||
layout.addWidget(status_label)
|
||
|
||
# 添加操作选择单选按钮
|
||
operation_label = QLabel("请选择要执行的操作:", dialog)
|
||
layout.addWidget(operation_label)
|
||
|
||
# 创建单选按钮组
|
||
radio_button_group = QButtonGroup(dialog)
|
||
|
||
# 添加可选操作
|
||
if is_disabled:
|
||
# 当前是禁用状态，显示启用选项
|
||
enable_radio = QRadioButton("启用补丁", dialog)
|
||
enable_radio.setChecked(True)
|
||
radio_button_group.addButton(enable_radio, 0)
|
||
layout.addWidget(enable_radio)
|
||
else:
|
||
# 当前是启用状态，显示禁用选项
|
||
disable_radio = QRadioButton("禁用补丁", dialog)
|
||
disable_radio.setChecked(True)
|
||
radio_button_group.addButton(disable_radio, 0)
|
||
layout.addWidget(disable_radio)
|
||
|
||
# 添加确定和取消按钮
|
||
buttons_layout = QHBoxLayout()
|
||
ok_button = QPushButton("确定", dialog)
|
||
cancel_button = QPushButton("取消", dialog)
|
||
buttons_layout.addWidget(ok_button)
|
||
buttons_layout.addWidget(cancel_button)
|
||
layout.addLayout(buttons_layout)
|
||
|
||
# 连接按钮事件
|
||
ok_button.clicked.connect(dialog.accept)
|
||
cancel_button.clicked.connect(dialog.reject)
|
||
|
||
# 显示对话框并等待用户选择
|
||
result = dialog.exec()
|
||
|
||
if result != QDialog.DialogCode.Accepted:
|
||
# 用户取消
|
||
return None
|
||
|
||
# 根据当前状态确定操作
|
||
return "enable" if is_disabled else "disable"
|
||
|
||
def _execute_batch_toggle(self, selected_game_dirs, operation, debug_mode):
|
||
"""
|
||
执行批量补丁切换操作
|
||
|
||
Args:
|
||
selected_game_dirs: 选择的游戏目录
|
||
operation: 操作类型
|
||
debug_mode: 是否为调试模式
|
||
"""
|
||
success_count = 0
|
||
fail_count = 0
|
||
results = []
|
||
|
||
for game_version, game_dir in selected_game_dirs.items():
|
||
try:
|
||
# 使用静默模式进行操作
|
||
result = self.patch_manager.toggle_patch(game_dir, game_version, operation=operation, silent=True)
|
||
|
||
if result["success"]:
|
||
success_count += 1
|
||
else:
|
||
fail_count += 1
|
||
|
||
results.append({
|
||
"version": game_version,
|
||
"success": result["success"],
|
||
"message": result["message"],
|
||
"action": result["action"]
|
||
})
|
||
|
||
except Exception as e:
|
||
if debug_mode:
|
||
print(f"DEBUG: 切换 {game_version} 补丁状态时出错: {str(e)}")
|
||
fail_count += 1
|
||
results.append({
|
||
"version": game_version,
|
||
"success": False,
|
||
"message": f"操作出错: {str(e)}",
|
||
"action": "none"
|
||
})
|
||
|
||
# 显示操作结果
|
||
self.patch_manager.show_toggle_result(success_count, fail_count, results) |