import zipfile
import tarfile
import gzip
import bz2
import os
import shutil

def extract_archive(archive_path, extract_to='.', encoding='utf-8'):
    """支持中文路徑的解壓函數(shù)"""
    comp_type = detect_compression(archive_path)

    # 根據(jù)類(lèi)型調(diào)用對(duì)應(yīng)解壓方法
    if comp_type == 'zip':
        # 關(guān)鍵修復(fù)：強(qiáng)制使用指定編碼
        with zipfile.ZipFile(archive_path, 'r') as zip_ref:
            # 處理亂碼：將亂碼文件名轉(zhuǎn)換為正確編碼
            for file in zip_ref.namelist():
                try:
                    # 嘗試用指定編碼解析文件名
                    fixed_name = file.encode('cp437').decode(encoding)
                except:
                    fixed_name = file
                # 重命名文件
                zip_ref.NameToInfo[file].filename = fixed_name
            zip_ref.extractall(extract_to)
            
    elif comp_type == 'tar':
        # 關(guān)鍵修復(fù)：指定編碼打開(kāi)tar
        with tarfile.open(archive_path, 'r', encoding=encoding) as tar_ref:
            tar_ref.extractall(extract_to)
            
    elif comp_type == 'gz':
        # 注意：gz通常只壓縮單個(gè)文件，文件名需手動(dòng)處理
        raw_name = os.path.basename(archive_path)[:-3]
        output_path = os.path.join(extract_to, raw_name)
        with gzip.open(archive_path, 'rb') as gz_ref:
            with open(output_path, 'wb') as out_file:
                shutil.copyfileobj(gz_ref, out_file)
                
    elif comp_type == 'bz2':
        raw_name = os.path.basename(archive_path)[:-4]
        output_path = os.path.join(extract_to, raw_name)
        with bz2.open(archive_path, 'rb') as bz2_ref:
            with open(output_path, 'wb') as out_file:
                shutil.copyfileobj(bz2_ref, out_file)
    else:
        print(f"不支持的壓縮格式: {archive_path}")