mirror of
https://github.com/hyb-oyqq/FRAISEMOE-Addons-Installer-NEXT.git
synced 2025-12-16 03:40:27 +00:00
535 lines
21 KiB
Python
535 lines
21 KiB
Python
import os
|
||
import sys
|
||
import subprocess
|
||
import shutil
|
||
import json
|
||
import webbrowser
|
||
|
||
from PySide6 import QtWidgets
|
||
from PySide6.QtCore import QTimer, Qt, QPoint, QRect, QSize
|
||
from PySide6.QtWidgets import QMainWindow, QMessageBox, QGraphicsOpacityEffect, QGraphicsColorizeEffect
|
||
from PySide6.QtGui import QPalette, QColor, QPainterPath, QRegion, QFont
|
||
from PySide6.QtGui import QAction # Added for menu actions
|
||
|
||
from ui.Ui_install import Ui_MainWindows
|
||
from data.config import (
|
||
APP_NAME, PLUGIN, GAME_INFO, BLOCK_SIZE,
|
||
PLUGIN_HASH, UA, CONFIG_URL, LOG_FILE,
|
||
DOWNLOAD_THREADS, DEFAULT_DOWNLOAD_THREAD_LEVEL # 添加下载线程常量
|
||
)
|
||
from utils import (
|
||
load_config, save_config, HashManager, AdminPrivileges, msgbox_frame, load_image_from_file
|
||
)
|
||
from workers import (
|
||
DownloadThread, ProgressWindow, IpOptimizerThread,
|
||
HashThread, ExtractionThread, ConfigFetchThread
|
||
)
|
||
from core import (
|
||
MultiStageAnimations, UIManager, DownloadManager, DebugManager,
|
||
WindowManager, GameDetector, PatchManager, ConfigManager
|
||
)
|
||
from core.ipv6_manager import IPv6Manager
|
||
from handlers import PatchToggleHandler, UninstallHandler
|
||
|
||
class MainWindow(QMainWindow):
|
||
def __init__(self):
|
||
super().__init__()
|
||
|
||
# 设置窗口为无边框
|
||
self.setWindowFlags(Qt.WindowType.FramelessWindowHint)
|
||
# 设置窗口背景透明
|
||
self.setAttribute(Qt.WidgetAttribute.WA_TranslucentBackground)
|
||
|
||
# 调整窗口大小以适应背景图片比例 (1280x720)
|
||
self.resize(1280, 720)
|
||
# 设置固定尺寸范围
|
||
self.setMinimumSize(QSize(1024, 576))
|
||
self.setMaximumSize(QSize(1280, 720))
|
||
|
||
# 初始化UI (从Ui_install.py导入)
|
||
self.ui = Ui_MainWindows()
|
||
self.ui.setupUi(self)
|
||
|
||
# 初始化配置
|
||
self.config = load_config()
|
||
|
||
# 初始化工具类
|
||
self.hash_manager = HashManager(BLOCK_SIZE)
|
||
self.admin_privileges = AdminPrivileges()
|
||
|
||
# 初始化各种管理器
|
||
# 1. 首先创建必要的基础管理器
|
||
self.animator = MultiStageAnimations(self.ui, self)
|
||
self.window_manager = WindowManager(self)
|
||
self.debug_manager = DebugManager(self)
|
||
|
||
# 2. 初始化IPv6Manager(应在UIManager之前)
|
||
self.ipv6_manager = IPv6Manager(self)
|
||
|
||
# 3. 创建UIManager(依赖IPv6Manager)
|
||
self.ui_manager = UIManager(self)
|
||
|
||
# 4. 为debug_manager设置ui_manager引用
|
||
self.debug_manager.set_ui_manager(self.ui_manager)
|
||
|
||
# 设置UI - 确保debug_action已初始化
|
||
self.ui_manager.setup_ui()
|
||
|
||
# 5. 初始化其他管理器
|
||
self.config_manager = ConfigManager(APP_NAME, CONFIG_URL, UA, self.debug_manager)
|
||
self.game_detector = GameDetector(GAME_INFO, self.debug_manager)
|
||
self.patch_manager = PatchManager(APP_NAME, GAME_INFO, self.debug_manager)
|
||
|
||
# 6. 初始化下载管理器 - 放在最后，因为它可能依赖于其他管理器
|
||
self.download_manager = DownloadManager(self)
|
||
|
||
# 7. 初始化功能处理程序
|
||
self.uninstall_handler = UninstallHandler(self)
|
||
self.patch_toggle_handler = PatchToggleHandler(self)
|
||
|
||
# 加载用户下载线程设置
|
||
if "download_thread_level" in self.config and self.config["download_thread_level"] in DOWNLOAD_THREADS:
|
||
self.download_manager.download_thread_level = self.config["download_thread_level"]
|
||
|
||
# 初始化状态变量
|
||
self.cloud_config = None
|
||
self.config_valid = False # 添加配置有效标志
|
||
self.patch_manager.initialize_status()
|
||
self.installed_status = self.patch_manager.get_status() # 获取初始化后的状态
|
||
self.hash_msg_box = None
|
||
self.progress_window = None
|
||
|
||
# 设置关闭按钮事件连接
|
||
if hasattr(self.ui, 'close_btn'):
|
||
self.ui.close_btn.clicked.connect(self.close)
|
||
|
||
if hasattr(self.ui, 'minimize_btn'):
|
||
self.ui.minimize_btn.clicked.connect(self.showMinimized)
|
||
|
||
# 检查管理员权限和进程
|
||
self.admin_privileges.request_admin_privileges()
|
||
self.admin_privileges.check_and_terminate_processes()
|
||
|
||
# 备份hosts文件
|
||
self.download_manager.hosts_manager.backup()
|
||
|
||
# 创建缓存目录
|
||
if not os.path.exists(PLUGIN):
|
||
try:
|
||
os.makedirs(PLUGIN)
|
||
except OSError as e:
|
||
QtWidgets.QMessageBox.critical(
|
||
self,
|
||
f"错误 - {APP_NAME}",
|
||
f"\n无法创建缓存位置\n\n使用管理员身份运行或检查文件读写权限\n\n【错误信息】：{e}\n",
|
||
)
|
||
sys.exit(1)
|
||
|
||
# 连接信号 - 绑定到新按钮
|
||
self.ui.start_install_btn.clicked.connect(self.handle_install_button_click)
|
||
self.ui.uninstall_btn.clicked.connect(self.uninstall_handler.handle_uninstall_button_click) # 使用卸载处理程序
|
||
self.ui.toggle_patch_btn.clicked.connect(self.patch_toggle_handler.handle_toggle_patch_button_click) # 使用补丁切换处理程序
|
||
self.ui.exit_btn.clicked.connect(self.shutdown_app)
|
||
|
||
# 初始化按钮状态标记
|
||
self.install_button_enabled = False
|
||
self.last_error_message = ""
|
||
|
||
# 根据初始配置决定是否开启Debug模式
|
||
if hasattr(self.ui_manager, 'debug_action') and self.ui_manager.debug_action:
|
||
if self.ui_manager.debug_action.isChecked():
|
||
self.debug_manager.start_logging()
|
||
|
||
# 在窗口显示前设置初始状态
|
||
self.animator.initialize()
|
||
|
||
# 窗口显示后延迟100ms启动动画
|
||
QTimer.singleShot(100, self.start_animations)
|
||
|
||
# 窗口事件处理 - 委托给WindowManager
|
||
def mousePressEvent(self, event):
|
||
self.window_manager.handle_mouse_press(event)
|
||
|
||
def mouseMoveEvent(self, event):
|
||
self.window_manager.handle_mouse_move(event)
|
||
|
||
def mouseReleaseEvent(self, event):
|
||
self.window_manager.handle_mouse_release(event)
|
||
|
||
def resizeEvent(self, event):
|
||
self.window_manager.handle_resize(event)
|
||
super().resizeEvent(event)
|
||
|
||
def start_animations(self):
|
||
"""开始启动动画"""
|
||
# 不再禁用退出按钮的交互性，只通过样式表控制外观
|
||
# 但仍然需要跟踪动画状态，防止用户在动画播放过程中退出
|
||
self.animation_in_progress = True
|
||
|
||
# 按钮容器初始是隐藏的，无需在这里禁用
|
||
# 但确保开始安装按钮仍然处于禁用状态
|
||
self.set_start_button_enabled(False)
|
||
|
||
self.animator.animation_finished.connect(self.on_animations_finished)
|
||
self.animator.start_animations()
|
||
# 在动画开始时获取云端配置
|
||
self.fetch_cloud_config()
|
||
|
||
def on_animations_finished(self):
|
||
"""动画完成后启用按钮"""
|
||
self.animation_in_progress = False
|
||
|
||
# 启用所有菜单按钮
|
||
self.ui.start_install_btn.setEnabled(True)
|
||
self.ui.uninstall_btn.setEnabled(True)
|
||
self.ui.toggle_patch_btn.setEnabled(True) # 启用禁/启用补丁按钮
|
||
self.ui.exit_btn.setEnabled(True)
|
||
|
||
# 只有在配置有效时才启用开始安装按钮
|
||
if self.config_valid:
|
||
self.set_start_button_enabled(True)
|
||
else:
|
||
self.set_start_button_enabled(False)
|
||
|
||
def set_start_button_enabled(self, enabled, installing=False):
|
||
"""[已弃用] 设置按钮启用状态的旧方法，保留以兼容旧代码
|
||
|
||
现在推荐使用主窗口的setEnabled方法和直接设置按钮文本
|
||
|
||
Args:
|
||
enabled: 是否启用按钮
|
||
installing: 是否正在安装中
|
||
"""
|
||
# 直接设置按钮文本，不改变窗口启用状态
|
||
if installing:
|
||
self.ui.start_install_text.setText("正在安装")
|
||
self.install_button_enabled = False
|
||
else:
|
||
if enabled:
|
||
self.ui.start_install_text.setText("开始安装")
|
||
else:
|
||
self.ui.start_install_text.setText("!无法安装!")
|
||
|
||
self.install_button_enabled = enabled
|
||
|
||
def fetch_cloud_config(self):
|
||
"""获取云端配置（异步方式）"""
|
||
self.config_manager.fetch_cloud_config(
|
||
lambda url, headers, debug_mode, parent=None: ConfigFetchThread(url, headers, debug_mode, self),
|
||
self.on_config_fetched
|
||
)
|
||
|
||
def on_config_fetched(self, data, error_message):
|
||
"""云端配置获取完成的回调处理
|
||
|
||
Args:
|
||
data: 获取到的配置数据
|
||
error_message: 错误信息，如果有
|
||
"""
|
||
# 处理返回结果
|
||
result = self.config_manager.on_config_fetched(data, error_message)
|
||
|
||
# 根据返回的操作执行相应动作
|
||
if result and "action" in result:
|
||
if result["action"] == "exit":
|
||
# 强制关闭程序
|
||
self.shutdown_app(force_exit=True)
|
||
elif result["action"] == "disable_button":
|
||
# 禁用开始安装按钮
|
||
self.set_start_button_enabled(False)
|
||
|
||
# 检查是否有后续操作
|
||
if "then" in result and result["then"] == "exit":
|
||
# 强制关闭程序
|
||
self.shutdown_app(force_exit=True)
|
||
elif result["action"] == "enable_button":
|
||
# 启用开始安装按钮
|
||
self.set_start_button_enabled(True)
|
||
|
||
# 同步状态
|
||
self.cloud_config = self.config_manager.get_cloud_config()
|
||
self.config_valid = self.config_manager.is_config_valid()
|
||
self.last_error_message = self.config_manager.get_last_error()
|
||
|
||
# 重新启用窗口，恢复用户交互
|
||
self.setEnabled(True)
|
||
|
||
def toggle_debug_mode(self, checked):
|
||
"""切换调试模式
|
||
|
||
Args:
|
||
checked: 是否启用调试模式
|
||
"""
|
||
self.debug_manager.toggle_debug_mode(checked)
|
||
|
||
def save_config(self, config):
|
||
"""保存配置的便捷方法"""
|
||
self.config_manager.save_config(config)
|
||
|
||
def create_download_thread(self, url, _7z_path, game_version):
|
||
"""创建下载线程
|
||
|
||
Args:
|
||
url: 下载URL
|
||
_7z_path: 7z文件保存路径
|
||
game_version: 游戏版本名称
|
||
|
||
Returns:
|
||
DownloadThread: 下载线程实例
|
||
"""
|
||
from workers import DownloadThread
|
||
return DownloadThread(url, _7z_path, game_version, parent=self)
|
||
|
||
def create_progress_window(self):
|
||
"""创建下载进度窗口
|
||
|
||
Returns:
|
||
ProgressWindow: 进度窗口实例
|
||
"""
|
||
return ProgressWindow(self)
|
||
|
||
def create_hash_thread(self, mode, install_paths):
|
||
"""创建哈希检查线程
|
||
|
||
Args:
|
||
mode: 检查模式，"pre"或"after"
|
||
install_paths: 安装路径字典
|
||
|
||
Returns:
|
||
HashThread: 哈希检查线程实例
|
||
"""
|
||
return HashThread(mode, install_paths, PLUGIN_HASH, self.installed_status, self)
|
||
|
||
def create_extraction_thread(self, _7z_path, game_folder, plugin_path, game_version):
|
||
"""创建解压线程
|
||
|
||
Args:
|
||
_7z_path: 7z文件路径
|
||
game_folder: 游戏文件夹路径
|
||
plugin_path: 插件路径
|
||
game_version: 游戏版本
|
||
|
||
Returns:
|
||
ExtractionThread: 解压线程实例
|
||
"""
|
||
return ExtractionThread(_7z_path, game_folder, plugin_path, game_version, self)
|
||
|
||
def after_hash_compare(self):
|
||
"""进行安装后哈希比较"""
|
||
# 禁用窗口已在安装流程开始时完成
|
||
|
||
self.hash_msg_box = self.hash_manager.hash_pop_window(check_type="after")
|
||
|
||
install_paths = self.download_manager.get_install_paths()
|
||
|
||
self.hash_thread = self.create_hash_thread("after", install_paths)
|
||
self.hash_thread.after_finished.connect(self.on_after_hash_finished)
|
||
self.hash_thread.start()
|
||
|
||
def on_after_hash_finished(self, result):
|
||
"""哈希比较完成后的处理
|
||
|
||
Args:
|
||
result: 哈希比较结果
|
||
"""
|
||
# 确保哈希检查窗口关闭，无论是否还在显示
|
||
if self.hash_msg_box:
|
||
try:
|
||
if self.hash_msg_box.isVisible():
|
||
self.hash_msg_box.close()
|
||
else:
|
||
# 如果窗口已经不可见但没有关闭，也要尝试关闭
|
||
self.hash_msg_box.close()
|
||
except:
|
||
pass # 忽略任何关闭窗口时的错误
|
||
self.hash_msg_box = None
|
||
|
||
if not result["passed"]:
|
||
# 启用窗口以显示错误消息
|
||
self.setEnabled(True)
|
||
|
||
game = result.get("game", "未知游戏")
|
||
message = result.get("message", "发生未知错误。")
|
||
msg_box = msgbox_frame(
|
||
f"文件校验失败 - {APP_NAME}",
|
||
message,
|
||
QMessageBox.StandardButton.Ok,
|
||
)
|
||
msg_box.exec()
|
||
|
||
# 恢复窗口状态
|
||
self.setEnabled(True)
|
||
self.ui.start_install_text.setText("开始安装")
|
||
|
||
# 添加短暂延迟确保UI更新
|
||
QTimer.singleShot(100, self.show_result)
|
||
|
||
def show_result(self):
|
||
"""显示安装结果，区分不同情况"""
|
||
# 获取当前安装状态
|
||
installed_versions = [] # 成功安装的版本
|
||
skipped_versions = [] # 已有补丁跳过的版本
|
||
failed_versions = [] # 安装失败的版本
|
||
not_found_versions = [] # 未找到的版本
|
||
|
||
# 获取所有游戏版本路径
|
||
install_paths = self.download_manager.get_install_paths() if hasattr(self.download_manager, "get_install_paths") else {}
|
||
|
||
for game_version, is_installed in self.installed_status.items():
|
||
# 只处理install_paths中存在的游戏版本
|
||
if game_version in install_paths:
|
||
path = install_paths[game_version]
|
||
|
||
# 检查游戏是否存在但未通过本次安装补丁
|
||
if is_installed:
|
||
# 游戏已安装补丁
|
||
if hasattr(self, 'download_queue_history') and game_version not in self.download_queue_history:
|
||
# 已有补丁，被跳过下载
|
||
skipped_versions.append(game_version)
|
||
else:
|
||
# 本次成功安装
|
||
installed_versions.append(game_version)
|
||
else:
|
||
# 游戏未安装补丁
|
||
if os.path.exists(path):
|
||
# 游戏文件夹存在，但安装失败
|
||
failed_versions.append(game_version)
|
||
else:
|
||
# 游戏文件夹不存在
|
||
not_found_versions.append(game_version)
|
||
|
||
# 构建结果信息
|
||
result_text = f"\n安装结果：\n"
|
||
|
||
# 总数统计 - 不再显示已跳过的数量
|
||
total_installed = len(installed_versions)
|
||
total_failed = len(failed_versions)
|
||
|
||
result_text += f"安装成功：{total_installed} 个 安装失败：{total_failed} 个\n\n"
|
||
|
||
# 详细列表
|
||
if installed_versions:
|
||
result_text += f"【成功安装】:\n{chr(10).join(installed_versions)}\n\n"
|
||
|
||
if failed_versions:
|
||
result_text += f"【安装失败】:\n{chr(10).join(failed_versions)}\n\n"
|
||
|
||
if not_found_versions:
|
||
# 只有在真正检测到了游戏但未安装补丁时才显示
|
||
result_text += f"【尚未安装补丁的游戏】:\n{chr(10).join(not_found_versions)}\n"
|
||
|
||
QMessageBox.information(
|
||
self,
|
||
f"安装完成 - {APP_NAME}",
|
||
result_text
|
||
)
|
||
|
||
def closeEvent(self, event):
|
||
"""窗口关闭事件处理
|
||
|
||
Args:
|
||
event: 关闭事件
|
||
"""
|
||
self.shutdown_app(event)
|
||
|
||
def shutdown_app(self, event=None, force_exit=False):
|
||
"""关闭应用程序
|
||
|
||
Args:
|
||
event: 关闭事件，如果是从closeEvent调用的
|
||
force_exit: 是否强制退出
|
||
"""
|
||
# 检查是否有动画或任务正在进行
|
||
if hasattr(self, 'animation_in_progress') and self.animation_in_progress and not force_exit:
|
||
# 如果动画正在进行，阻止退出
|
||
if event:
|
||
event.ignore()
|
||
return
|
||
|
||
# 检查是否有下载任务正在进行
|
||
if hasattr(self.download_manager, 'current_download_thread') and \
|
||
self.download_manager.current_download_thread and \
|
||
self.download_manager.current_download_thread.isRunning() and not force_exit:
|
||
# 询问用户是否确认退出
|
||
reply = QMessageBox.question(
|
||
self,
|
||
f"确认退出 - {APP_NAME}",
|
||
"\n下载任务正在进行中，确定要退出吗？\n",
|
||
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
|
||
QMessageBox.StandardButton.No
|
||
)
|
||
if reply == QMessageBox.StandardButton.No:
|
||
if event:
|
||
event.ignore()
|
||
return
|
||
|
||
# 恢复hosts文件（如果未禁用自动还原）
|
||
self.download_manager.hosts_manager.restore()
|
||
|
||
# 额外检查并清理hosts文件中的残留记录（如果未禁用自动还原）
|
||
self.download_manager.hosts_manager.check_and_clean_all_entries()
|
||
|
||
# 停止日志记录
|
||
self.debug_manager.stop_logging()
|
||
|
||
if not force_exit:
|
||
reply = QMessageBox.question(
|
||
self,
|
||
f"确认退出 - {APP_NAME}",
|
||
"\n确定要退出吗？\n",
|
||
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
|
||
QMessageBox.StandardButton.No
|
||
)
|
||
if reply == QMessageBox.StandardButton.No:
|
||
if event:
|
||
event.ignore()
|
||
return
|
||
|
||
# 退出应用程序
|
||
if event:
|
||
event.accept()
|
||
else:
|
||
sys.exit(0)
|
||
|
||
def handle_install_button_click(self):
|
||
"""处理安装按钮点击事件
|
||
根据按钮当前状态决定是显示错误还是执行安装
|
||
"""
|
||
if not self.install_button_enabled:
|
||
# 按钮处于"无法安装"状态
|
||
if self.last_error_message == "update_required":
|
||
msg_box = msgbox_frame(
|
||
f"更新提示 - {APP_NAME}",
|
||
"\n当前版本过低，请及时更新。\n",
|
||
QMessageBox.StandardButton.Ok,
|
||
)
|
||
msg_box.exec()
|
||
elif self.last_error_message == "directory_not_found":
|
||
# 目录识别失败的特定错误信息
|
||
reply = msgbox_frame(
|
||
f"目录错误 - {APP_NAME}",
|
||
"\n未能识别游戏目录，请确认选择的是游戏的上级目录，并且目录中包含Nekopara游戏文件夹。\n\n是否重新选择目录？\n",
|
||
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
|
||
)
|
||
if reply.exec() == QMessageBox.StandardButton.Yes:
|
||
# 重新启用按钮并允许用户选择目录
|
||
self.set_start_button_enabled(True)
|
||
# 直接调用文件对话框
|
||
self.download_manager.file_dialog()
|
||
else:
|
||
# 网络错误或其他错误
|
||
reply = msgbox_frame(
|
||
f"错误 - {APP_NAME}",
|
||
"\n访问云端配置失败，请检查网络状况或稍后再试。\n\n是否重新尝试连接？\n",
|
||
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
|
||
)
|
||
if reply.exec() == QMessageBox.StandardButton.Yes:
|
||
# 重试获取配置
|
||
self.fetch_cloud_config()
|
||
else:
|
||
# 按钮处于"开始安装"状态，正常执行安装流程
|
||
self.download_manager.file_dialog()
|
||
|
||
|
||
|
||
|