FRAISEMOE-Addons-Installer-NEXT/source/workers/extraction_thread.py

import os
import shutil
import py7zr
import tempfile
import traceback
from PySide6.QtCore import QThread, Signal
from config.config import PLUGIN, GAME_INFO
import time  # 用于时间计算
import threading
import queue
from concurrent.futures import TimeoutError

class ExtractionThread(QThread):
    finished = Signal(bool, str, str)  # success, error_message, game_version
    progress = Signal(int, str)  # 添加进度信号，传递进度百分比和状态信息

    def __init__(self, _7z_path, game_folder, plugin_path, game_version, parent=None, extracted_path=None):
        super().__init__(parent)
        self._7z_path = _7z_path
        self.game_folder = game_folder
        self.plugin_path = plugin_path
        self.game_version = game_version
        self.extracted_path = extracted_path  # 添加已解压文件路径参数

    def run(self):
        try:
            # 确保游戏目录存在
            os.makedirs(self.game_folder, exist_ok=True)

            def update_progress(percent: int, message: str):
                try:
                    self.progress.emit(percent, message)
                except Exception:
                    pass
            
            # 记录调试信息
            from utils.logger import setup_logger
            debug_logger = setup_logger("extraction_thread")
            debug_logger.info(f"====== 开始处理 {self.game_version} 补丁文件 ======")
            debug_logger.info(f"压缩包路径: {self._7z_path}")
            debug_logger.info(f"游戏目录: {self.game_folder}")
            debug_logger.info(f"插件路径: {self.plugin_path}")

            update_progress(0, f"开始处理 {self.game_version} 的补丁文件...")

            # 支持外部请求中断
            if self.isInterruptionRequested():
                self.finished.emit(False, "操作已取消", self.game_version)
                return

            # 如果提供了已解压文件路径，直接使用它
            if self.extracted_path and os.path.exists(self.extracted_path):
                update_progress(20, f"正在复制 {self.game_version} 的补丁文件...\n(在此过程中可能会卡顿或无响应，请不要关闭软件)")

                # 直接复制已解压的文件到游戏目录
                target_file = os.path.join(self.game_folder, os.path.basename(self.plugin_path))
                shutil.copy(self.extracted_path, target_file)

                update_progress(60, f"正在完成 {self.game_version} 的补丁安装...")

                # 对于NEKOPARA After，还需要复制签名文件
                if self.game_version == "NEKOPARA After":
                    try:
                        update_progress(70, f"正在处理 {self.game_version} 的签名文件...")
                        # 从已解压文件的目录中获取签名文件
                        extracted_dir = os.path.dirname(self.extracted_path)
                        sig_filename = os.path.basename(GAME_INFO[self.game_version]["sig_path"])
                        sig_path = os.path.join(extracted_dir, sig_filename)

                        # 尝试多种可能的签名文件路径
                        if not os.path.exists(sig_path):
                            # 尝试在同级目录查找
                            sig_path = os.path.join(os.path.dirname(extracted_dir), sig_filename)
                        
                        # 如果签名文件存在，则复制它
                        if os.path.exists(sig_path):
                            target_sig = os.path.join(self.game_folder, sig_filename)
                            shutil.copy(sig_path, target_sig)
                            update_progress(80, f"签名文件复制完成")
                        else:
                            # 如果签名文件不存在，则使用原始路径
                            sig_path = os.path.join(PLUGIN, GAME_INFO[self.game_version]["sig_path"])
                            if os.path.exists(sig_path):
                                target_sig = os.path.join(self.game_folder, os.path.basename(sig_path))
                                shutil.copy(sig_path, target_sig)
                                update_progress(80, f"使用内置签名文件完成")
                            else:
                                update_progress(80, f"未找到签名文件，继续安装主补丁文件")
                    except Exception as sig_err:
                        # 签名文件处理失败时记录错误但不中断主流程
                        update_progress(80, f"签名文件处理失败: {str(sig_err)}")

                update_progress(100, f"{self.game_version} 补丁文件处理完成")
                self.finished.emit(True, "", self.game_version)
                return

            # 否则解压源压缩包到临时目录，再复制目标文件
            update_progress(10, f"正在打开 {self.game_version} 的补丁压缩包...")

            with py7zr.SevenZipFile(self._7z_path, mode="r") as archive:
                # 获取压缩包内的文件列表
                file_list = archive.getnames()
                
                # 详细记录压缩包中的所有文件
                debug_logger.info(f"压缩包内容分析:")
                debug_logger.info(f"- 文件总数: {len(file_list)}")
                for i, f in enumerate(file_list):
                    debug_logger.info(f"  {i+1}. {f} - 类型: {'文件夹' if f.endswith('/') or f.endswith('\\') else '文件'}")

                update_progress(20, f"正在分析 {self.game_version} 的补丁文件...")

                update_progress(30, f"正在解压 {self.game_version} 的补丁文件...\n(在此过程中可能会卡顿或无响应，请不要关闭软件)")

                with tempfile.TemporaryDirectory() as temp_dir:
                    # 查找主补丁文件和签名文件
                    target_filename = os.path.basename(self.plugin_path)
                    sig_filename = target_filename + ".sig"  # 签名文件名
                    debug_logger.info(f"查找主补丁文件: {target_filename}")
                    debug_logger.info(f"查找签名文件: {sig_filename}")
                    
                    target_file_in_archive = None
                    sig_file_in_archive = None
                    
                    # 对于NEKOPARA After，增加特殊处理
                    if self.game_version == "NEKOPARA After":
                        # 增加专门的检查，同时识别主补丁和签名文件
                        debug_logger.info("执行NEKOPARA After特殊补丁文件识别")
                        
                        # 查找主补丁和签名文件
                        for file_path in file_list:
                            basename = os.path.basename(file_path)
                            
                            # 查找主补丁文件
                            if basename == "afteradult.xp3" and not basename.endswith('.sig'):
                                target_file_in_archive = file_path
                                debug_logger.info(f"找到精确匹配的After主补丁文件: {target_file_in_archive}")
                                
                            # 查找签名文件
                            elif basename == "afteradult.xp3.sig" or basename.endswith('.sig'):
                                sig_file_in_archive = file_path
                                debug_logger.info(f"找到After签名文件: {sig_file_in_archive}")
                        
                        # 如果没找到主补丁文件，寻找可能的替代文件
                        if not target_file_in_archive:
                            for file_path in file_list:
                                if "afteradult.xp3" in file_path and not file_path.endswith('.sig'):
                                    target_file_in_archive = file_path
                                    debug_logger.info(f"找到备选After主补丁文件: {target_file_in_archive}")
                                    break
                    else:
                        # 标准处理逻辑
                        for file_path in file_list:
                            basename = os.path.basename(file_path)
                            
                            # 查找主补丁文件
                            if basename == target_filename and not basename.endswith('.sig'):
                                target_file_in_archive = file_path
                                debug_logger.info(f"在压缩包中找到主补丁文件: {target_file_in_archive}")
                            
                            # 查找签名文件
                            elif basename == sig_filename:
                                sig_file_in_archive = file_path
                                debug_logger.info(f"在压缩包中找到签名文件: {sig_file_in_archive}")
                        
                        # 如果没有找到精确匹配的主补丁文件，使用更宽松的搜索
                        if not target_file_in_archive:
                            debug_logger.warning(f"没有找到精确匹配的主补丁文件，尝试更宽松的搜索")
                            for file_path in file_list:
                                if target_filename in file_path and not file_path.endswith('.sig'):
                                    target_file_in_archive = file_path
                                    debug_logger.info(f"在压缩包中找到可能的主补丁文件: {target_file_in_archive}")
                                    break

                    # 如果找不到主补丁文件，使用回退方案：提取全部内容
                    if not target_file_in_archive:
                        debug_logger.warning(f"未能识别正确的主补丁文件，将提取所有文件并尝试查找")
                        
                        # 提取所有文件到临时目录
                        update_progress(30, f"正在解压所有文件...")
                        archive.extractall(path=temp_dir)
                        debug_logger.info(f"已提取所有文件到临时目录")
                        
                        # 在提取的文件中查找主补丁文件和签名文件
                        found_main = False
                        found_sig = False
                        
                        for root, dirs, files in os.walk(temp_dir):
                            for file in files:
                                # 查找主补丁文件
                                if file == target_filename and not file.endswith('.sig'):
                                    extracted_file_path = os.path.join(root, file)
                                    file_size = os.path.getsize(extracted_file_path)
                                    debug_logger.info(f"在提取的文件中找到主补丁文件: {extracted_file_path}, 大小: {file_size} 字节")
                                    
                                    # 复制到目标位置
                                    target_path = os.path.join(self.game_folder, target_filename)
                                    shutil.copy2(extracted_file_path, target_path)
                                    debug_logger.info(f"已复制主补丁文件到: {target_path}")
                                    found_main = True
                                
                                # 查找签名文件
                                elif file == sig_filename or file.endswith('.sig'):
                                    extracted_sig_path = os.path.join(root, file)
                                    sig_size = os.path.getsize(extracted_sig_path)
                                    debug_logger.info(f"在提取的文件中找到签名文件: {extracted_sig_path}, 大小: {sig_size} 字节")
                                    
                                    # 复制到目标位置
                                    sig_target = os.path.join(self.game_folder, sig_filename)
                                    shutil.copy2(extracted_sig_path, sig_target)
                                    debug_logger.info(f"已复制签名文件到: {sig_target}")
                                    found_sig = True
                                
                                # 如果两个文件都找到，可以停止遍历
                                if found_main and found_sig:
                                    debug_logger.info("已找到所有需要的文件，停止遍历")
                                    break
                            
                            if found_main and found_sig:
                                break
                                    
                        if not found_main:
                            debug_logger.error(f"无法找到主补丁文件，安装失败")
                            raise FileNotFoundError(f"在压缩包中未找到主补丁文件 {target_filename}")
                            
                        # 签名文件没找到不影响主流程，但记录警告
                        if not found_sig:
                            debug_logger.warning(f"未找到签名文件 {sig_filename}，将尝试使用内置签名文件")
                            # 尝试使用内置签名文件
                            self._try_use_builtin_signature(sig_filename, debug_logger, update_progress)
                    else:
                        # 准备要解压的文件列表
                        files_to_extract = [target_file_in_archive]
                        if sig_file_in_archive:
                            files_to_extract.append(sig_file_in_archive)
                            debug_logger.info(f"将同时解压主补丁文件和签名文件: {files_to_extract}")
                        else:
                            debug_logger.info(f"将仅解压主补丁文件: {files_to_extract}")
                        
                        # 解压选定的文件到临时目录
                        debug_logger.info(f"开始解压选定文件到临时目录: {temp_dir}")
                        
                        # 设置解压超时时间（秒）
                        extract_timeout = 180  # 3分钟超时
                        debug_logger.info(f"设置解压超时: {extract_timeout}秒")
                        
                        # 创建子线程执行解压
                        import threading
                        import queue
                        
                        extract_result = queue.Queue()
                        
                        def extract_files():
                            try:
                                archive.extract(path=temp_dir, targets=files_to_extract)
                                extract_result.put(("success", None))
                            except Exception as e:
                                extract_result.put(("error", e))
                        
                        extract_thread = threading.Thread(target=extract_files)
                        extract_thread.daemon = True
                        extract_thread.start()
                        
                        # 每5秒更新一次进度，最多等待设定的超时时间
                        total_waited = 0
                        while extract_thread.is_alive() and total_waited < extract_timeout:
                            update_progress(30 + int(30 * total_waited / extract_timeout), 
                                f"正在解压文件...已等待{total_waited}秒")
                            extract_thread.join(5)  # 等待5秒
                            total_waited += 5
                        
                        # 检查是否超时
                        if extract_thread.is_alive():
                            debug_logger.error(f"解压超时（超过{extract_timeout}秒）")
                            raise TimeoutError(f"解压超时（超过{extract_timeout}秒），请检查补丁文件是否完整")
                        
                        # 检查解压结果
                        if not extract_result.empty():
                            status, error = extract_result.get()
                            if status == "error":
                                debug_logger.error(f"解压错误: {error}")
                                raise error
                        
                        debug_logger.info(f"文件解压完成")

                        update_progress(60, f"正在复制 {self.game_version} 的补丁文件...")

                        # 复制主补丁文件到游戏目录
                        extracted_file_path = os.path.join(temp_dir, target_file_in_archive)
                        
                        # 检查解压后的文件是否存在及其大小
                        if os.path.exists(extracted_file_path):
                            file_size = os.path.getsize(extracted_file_path)
                            debug_logger.info(f"解压后的主补丁文件存在: {extracted_file_path}, 大小: {file_size} 字节")
                        else:
                            debug_logger.error(f"解压后的主补丁文件不存在: {extracted_file_path}")
                            raise FileNotFoundError(f"解压后的文件不存在: {extracted_file_path}")

                        # 构建目标路径并复制
                        target_path = os.path.join(self.game_folder, target_filename)
                        debug_logger.info(f"复制主补丁文件: {extracted_file_path} 到 {target_path}")
                        shutil.copy2(extracted_file_path, target_path)
                        
                        # 验证主补丁文件是否成功复制
                        if os.path.exists(target_path):
                            target_size = os.path.getsize(target_path)
                            debug_logger.info(f"主补丁文件成功复制: {target_path}, 大小: {target_size} 字节")
                        else:
                            debug_logger.error(f"主补丁文件复制失败: {target_path}")
                            raise FileNotFoundError(f"目标文件复制失败: {target_path}")
                            
                        # 如果有找到签名文件，也复制它
                        if sig_file_in_archive:
                            update_progress(80, f"正在复制签名文件...")
                            extracted_sig_path = os.path.join(temp_dir, sig_file_in_archive)
                            
                            if os.path.exists(extracted_sig_path):
                                sig_size = os.path.getsize(extracted_sig_path)
                                debug_logger.info(f"解压后的签名文件存在: {extracted_sig_path}, 大小: {sig_size} 字节")
                                
                                # 复制签名文件到游戏目录
                                sig_target = os.path.join(self.game_folder, sig_filename)
                                shutil.copy2(extracted_sig_path, sig_target)
                                debug_logger.info(f"签名文件成功复制: {sig_target}")
                            else:
                                debug_logger.warning(f"解压后的签名文件不存在: {extracted_sig_path}")
                                # 尝试使用内置签名文件
                                self._try_use_builtin_signature(sig_filename, debug_logger, update_progress)
                        else:
                            debug_logger.warning(f"没有找到签名文件，尝试使用内置签名文件")
                            # 尝试使用内置签名文件
                            self._try_use_builtin_signature(sig_filename, debug_logger, update_progress)

                update_progress(100, f"{self.game_version} 补丁文件解压完成")
                self.finished.emit(True, "", self.game_version)
        except (py7zr.Bad7zFile, FileNotFoundError, Exception) as e:
            try:
                self.progress.emit(100, f"处理 {self.game_version} 的补丁文件失败")
            except Exception:
                pass
            self.finished.emit(False, f"\n文件操作失败，请重试\n\n【错误信息】：{e}\n", self.game_version) 

# 添加一个新的辅助方法用于使用内置签名文件
def _try_use_builtin_signature(self, sig_filename, debug_logger, update_progress):
    """尝试使用内置的签名文件"""
    try:
        # 如果签名文件不在压缩包中，则尝试使用原始路径
        update_progress(85, f"尝试使用内置签名文件...")
        sig_path = os.path.join(PLUGIN, GAME_INFO[self.game_version]["sig_path"])
        debug_logger.info(f"内置签名文件路径: {sig_path}")
        
        if os.path.exists(sig_path):
            sig_size = os.path.getsize(sig_path)
            debug_logger.info(f"内置签名文件存在，大小: {sig_size} 字节")
            
            # 确保使用正确的签名文件名
            sig_target = os.path.join(self.game_folder, sig_filename)
            debug_logger.info(f"目标签名文件路径: {sig_target}")
            
            # 增加异常处理
            try:
                shutil.copy2(sig_path, sig_target)
                
                if os.path.exists(sig_target):
                    target_sig_size = os.path.getsize(sig_target)
                    debug_logger.info(f"内置签名文件成功复制到目标位置，大小: {target_sig_size} 字节")
                else:
                    debug_logger.error(f"内置签名文件复制失败，目标文件不存在: {sig_target}")
            except Exception as copy_err:
                debug_logger.error(f"复制内置签名文件时出错: {str(copy_err)}")
                debug_logger.error(f"错误详情: {traceback.format_exc()}")
            
            update_progress(90, f"使用内置签名文件完成")
        else:
            debug_logger.warning(f"内置签名文件不存在: {sig_path}")
            update_progress(85, f"未找到内置签名文件，继续安装主补丁文件")
    except Exception as sig_err:
        # 签名文件处理失败时记录错误但不中断主流程
        debug_logger.error(f"内置签名文件处理异常: {str(sig_err)}")
        debug_logger.error(f"异常详情: {traceback.format_exc()}")
        update_progress(85, f"内置签名文件处理失败: {str(sig_err)}")