FRAISEMOE-Addons-Installer-NEXT/source/core/handlers/extraction_handler.py

import os
import shutil
from PySide6 import QtWidgets
from PySide6.QtWidgets import QMessageBox
from PySide6.QtCore import QTimer, QCoreApplication

from utils.logger import setup_logger
from workers.extraction_thread import ExtractionThread

# 初始化logger
logger = setup_logger("extraction_handler")

class ExtractionHandler:
    """解压处理器，负责管理解压任务和结果处理"""
    
    def __init__(self, main_window):
        """初始化解压处理器
        
        Args:
            main_window: 主窗口实例，用于访问UI和状态
        """
        self.main_window = main_window
        self.APP_NAME = main_window.APP_NAME if hasattr(main_window, 'APP_NAME') else ""
        self.extraction_progress_window = None
        
    def start_extraction(self, _7z_path, game_folder, plugin_path, game_version, extracted_path=None):
        """开始解压任务
        
        Args:
            _7z_path: 7z文件路径
            game_folder: 游戏文件夹路径
            plugin_path: 插件路径
            game_version: 游戏版本名称
            extracted_path: 已解压的补丁文件路径，如果提供则直接使用它而不进行解压
        """
        # 检查是否处于离线模式
        is_offline = False
        if hasattr(self.main_window, 'offline_mode_manager'):
            is_offline = self.main_window.offline_mode_manager.is_in_offline_mode()
        
        # 创建并显示解压进度窗口，替代原来的消息框
        self.extraction_progress_window = self.main_window.create_extraction_progress_window()
        self.extraction_progress_window.show()
        
        # 确保UI更新
        QCoreApplication.processEvents()
        
        # 创建并启动解压线程
        self.main_window.extraction_thread = ExtractionThread(
            _7z_path, game_folder, plugin_path, game_version, self.main_window, extracted_path
        )
        
        # 连接进度信号
        self.main_window.extraction_thread.progress.connect(self.update_extraction_progress)
        
        # 连接完成信号
        self.main_window.extraction_thread.finished.connect(self.on_extraction_finished_with_hash_check)
        
        # 启动线程
        self.main_window.extraction_thread.start()
        
    def update_extraction_progress(self, progress, status_text):
        """更新解压进度
        
        Args:
            progress: 进度百分比
            status_text: 状态文本
        """
        if self.extraction_progress_window and hasattr(self.extraction_progress_window, 'progress_bar'):
            self.extraction_progress_window.progress_bar.setValue(progress)
            self.extraction_progress_window.status_label.setText(status_text)
            
            # 确保UI更新
            QCoreApplication.processEvents()
            
    def on_extraction_finished_with_hash_check(self, success, error_message, game_version):
        """解压完成后进行哈希校验
        
        Args:
            success: 是否解压成功
            error_message: 错误信息
            game_version: 游戏版本
        """
        # 关闭解压进度窗口
        if self.extraction_progress_window:
            self.extraction_progress_window.close()
            self.extraction_progress_window = None

        # 如果解压失败，显示错误并询问是否继续
        if not success:
            # 临时启用窗口以显示错误消息
            self.main_window.setEnabled(True)
            
            QtWidgets.QMessageBox.critical(self.main_window, f"错误 - {self.APP_NAME}", error_message)
            self.main_window.installed_status[game_version] = False
            
            # 询问用户是否继续其他游戏的安装
            reply = QtWidgets.QMessageBox.question(
                self.main_window,
                f"继续安装? - {self.APP_NAME}",
                f"\n{game_version} 的补丁安装失败。\n\n是否继续安装其他游戏的补丁？\n",
                QtWidgets.QMessageBox.StandardButton.Yes | QtWidgets.QMessageBox.StandardButton.No,
                QtWidgets.QMessageBox.StandardButton.Yes
            )
            
            if reply == QtWidgets.QMessageBox.StandardButton.Yes:
                # 继续下一个，重新禁用窗口
                self.main_window.setEnabled(False)
                # 通知DownloadManager继续下一个下载任务
                self.main_window.download_manager.on_extraction_finished(True)
            else:
                # 用户选择停止，保持窗口启用状态
                if hasattr(self.main_window, 'window_manager'):
                    self.main_window.window_manager.change_window_state(self.main_window.window_manager.STATE_READY)
                # 通知DownloadManager停止下载队列
                self.main_window.download_manager.on_extraction_finished(False)
            return
            
        # 解压成功，进行哈希校验
        self._perform_hash_check(game_version)
        
    def _perform_hash_check(self, game_version):
        """解压成功后进行哈希校验
        
        Args:
            game_version: 游戏版本
        """
        # 导入所需模块
        from config.config import GAME_INFO, PLUGIN_HASH
        from workers.hash_thread import HashThread
        
        # 获取安装路径
        install_paths = {}
        if hasattr(self.main_window, 'game_detector') and hasattr(self.main_window, 'download_manager'):
            game_dirs = self.main_window.game_detector.identify_game_directories_improved(
                self.main_window.download_manager.selected_folder
            )
            
            for game, info in GAME_INFO.items():
                if game in game_dirs and game == game_version:
                    game_dir = game_dirs[game]
                    install_path = os.path.join(game_dir, os.path.basename(info["install_path"]))
                    install_paths[game] = install_path
                    break
        
        if not install_paths:
            # 如果找不到安装路径，直接认为安装成功
            logger.warning(f"未找到 {game_version} 的安装路径，跳过哈希校验")
            self.main_window.installed_status[game_version] = True
            self.main_window.download_manager.on_extraction_finished(True)
            return
        
        # 关闭可能存在的哈希校验窗口
        self.main_window.close_hash_msg_box()
            
        # 显示哈希校验窗口
        self.main_window.hash_msg_box = self.main_window.hash_manager.hash_pop_window(
            check_type="post", 
            auto_close=True,  # 添加自动关闭参数
            close_delay=1000  # 1秒后自动关闭
        )
        
        # 直接创建并启动哈希线程进行校验
        hash_thread = HashThread(
            "after", 
            install_paths, 
            PLUGIN_HASH, 
            self.main_window.installed_status,
            self.main_window
        )
        hash_thread.after_finished.connect(self.on_hash_check_finished)
        
        # 保存引用以便后续使用
        self.hash_thread = hash_thread
        hash_thread.start()
        
    def on_hash_check_finished(self, result):
        """哈希校验完成后的处理
        
        Args:
            result: 校验结果，包含通过状态、游戏版本和消息
        """
        # 导入所需模块
        from config.config import GAME_INFO
        
        # 关闭哈希检查窗口
        self.main_window.close_hash_msg_box()
            
        if not result["passed"]:
            # 校验失败，删除已解压的文件并提示重新下载
            game_version = result["game"]
            error_message = result["message"]
            
            # 临时启用窗口以显示错误消息
            self.main_window.setEnabled(True)
            
            # 获取安装路径
            install_path = None
            if hasattr(self.main_window, 'game_detector') and hasattr(self.main_window, 'download_manager'):
                game_dirs = self.main_window.game_detector.identify_game_directories_improved(
                    self.main_window.download_manager.selected_folder
                )
                
                if game_version in game_dirs and game_version in GAME_INFO:
                    game_dir = game_dirs[game_version]
                    install_path = os.path.join(game_dir, os.path.basename(GAME_INFO[game_version]["install_path"]))
            
            # 如果找到安装路径，尝试删除已解压的文件
            if install_path and os.path.exists(install_path):
                try:
                    os.remove(install_path)
                    logger.debug(f"已删除校验失败的文件: {install_path}")
                except Exception as e:
                    logger.error(f"删除文件失败: {e}")
            
            # 显示错误消息并询问是否重试
            reply = QtWidgets.QMessageBox.question(
                self.main_window,
                f"校验失败 - {self.APP_NAME}",
                f"{error_message}\n\n是否重新下载并安装？",
                QtWidgets.QMessageBox.StandardButton.Yes | QtWidgets.QMessageBox.StandardButton.No,
                QtWidgets.QMessageBox.StandardButton.Yes
            )
            
            if reply == QtWidgets.QMessageBox.StandardButton.Yes:
                # 重新下载，将游戏重新添加到下载队列
                self.main_window.setEnabled(False)
                self.main_window.installed_status[game_version] = False
                
                # 获取游戏目录和下载URL
                if hasattr(self.main_window, 'download_manager') and hasattr(self.main_window, 'game_detector'):
                    game_dirs = self.main_window.game_detector.identify_game_directories_improved(
                        self.main_window.download_manager.selected_folder
                    )
                    
                    if game_version in game_dirs:
                        # 重新将游戏添加到下载队列
                        self.main_window.download_manager.download_queue.appendleft([game_version])
                        # 继续下一个下载任务
                        self.main_window.download_manager.next_download_task()
                    else:
                        # 如果找不到游戏目录，继续下一个
                        self.main_window.download_manager.on_extraction_finished(True)
                else:
                    # 如果无法重新下载，继续下一个
                    self.main_window.download_manager.on_extraction_finished(True)
            else:
                # 用户选择不重试，继续下一个
                self.main_window.installed_status[game_version] = False
                self.main_window.download_manager.on_extraction_finished(True)
        else:
            # 校验通过，更新安装状态
            game_version = result["game"]
            self.main_window.installed_status[game_version] = True
            # 通知DownloadManager继续下一个下载任务
            self.main_window.download_manager.on_extraction_finished(True)
            
    def on_extraction_finished(self, success, error_message, game_version):
        """兼容旧版本的回调函数
        
        Args:
            success: 是否解压成功
            error_message: 错误信息
            game_version: 游戏版本
        """
        # 调用新的带哈希校验的回调函数
        self.on_extraction_finished_with_hash_check(success, error_message, game_version)