defs.py

from config import *  # 导入config.py中定义的全局变量
import shutil  # 导入shutil模块，用于复制、移动、删除文件和目录
import subprocess  # 导入subprocess模块，用于执行系统命令
import fnmatch  # 导入fnmatch模块，用于文件名匹配
import json  # 导入json模块，用于读写JSON格式的数据
from pyaxmlparser import APK  # 导入pyaxmlparser读取apk信息

def move_json(backup, type_name):
    def move_files(type_n):
            if type_n == "ph":
                src_1 = os.path.join("./phone", "app_version.json")
                dst_1 = os.path.join(".", "app_version.json")
                src_2 = os.path.join("./phone", "app_code.json")
                dst_2 = os.path.join(".", "app_code.json")
                # 将文件复制到根目录下
                shutil.copy2(src_1, dst_1)
                shutil.copy2(src_2, dst_2)
                try:
                    with open(JSON_V, 'w') as file:
                        new_content = "Phone"
                        file.write(new_content)
                        print("字典库已变更为 Phone")
                except Exception as e:
                    print(f"异常: {e}")
            elif type_n == "f":
                src_1 = os.path.join("./fold", "app_version.json")
                dst_1 = os.path.join(".", "app_version.json")
                src_2 = os.path.join("./fold", "app_code.json")
                dst_2 = os.path.join(".", "app_code.json")
                # 将文件复制到根目录下
                shutil.copy2(src_1, dst_1)
                shutil.copy2(src_2, dst_2)
                try:
                    with open(JSON_V, 'w') as file:
                        new_content = "Fold"
                        file.write(new_content)
                        print("字典库已变更为 Fold")
                except Exception as e:
                    print(f"异常: {e}")
            elif type_n == "p":
                src_1 = os.path.join("./pad", "app_version.json")
                dst_1 = os.path.join(".", "app_version.json")
                src_2 = os.path.join("./pad", "app_code.json")
                dst_2 = os.path.join(".", "app_code.json")
                # 将文件复制到根目录下
                shutil.copy2(src_1, dst_1)
                shutil.copy2(src_2, dst_2)
                try:
                    with open(JSON_V, 'w') as file:
                        new_content = "Pad"
                        file.write(new_content)
                        print("字典库已变更为 Pad")
                except Exception as e:
                    print(f"异常: {e}")

    # 获取字典库当前列表
    try:
        with open(JSON_V, 'r') as file:
            line = file.readline()
            print("当前字典列表为:", line)
        
        # 同步字典库
        if int(backup) == 1:
            print(f"正在同步到 {line} 字典库目录")
            if line == "Phone":
                src_1 = os.path.join(".", "app_version.json")
                dst_1 = os.path.join("./phone", "app_version.json")
                src_2 = os.path.join(".", "app_code.json")
                dst_2 = os.path.join("./phone", "app_code.json")
                # 将文件移动到 phone 目录下
                shutil.move(src_1, dst_1)
                shutil.move(src_2, dst_2)
                print("字典库已同步到 Phone 目录，正在切换")
                move_files(type_name)
            elif line == "Fold":
                src_1 = os.path.join(".", "app_version.json")
                dst_1 = os.path.join("./fold", "app_version.json")
                src_2 = os.path.join(".", "app_code.json")
                dst_2 = os.path.join("./fold", "app_code.json")
                # 将文件移动到 fold 目录下
                shutil.move(src_1, dst_1)
                shutil.move(src_2, dst_2)
                print("字典库已同步到 Fold 目录，正在切换")
                move_files(type_name)
            elif line == "Pad":
                src_1 = os.path.join(".", "app_version.json")
                dst_1 = os.path.join("./pad", "app_version.json")
                src_2 = os.path.join(".", "app_code.json")
                dst_2 = os.path.join("./pad", "app_code.json")
                # 将文件移动到 pad 目录下
                shutil.move(src_1, dst_1)
                shutil.move(src_2, dst_2)
                print("字典库已同步到 Pad 目录，正在切换")
                move_files(type_name)
        elif int(backup) == 0:
            print("正在覆盖字典库目录")
            move_files(type_name)
    except FileNotFoundError as e:
        print(f"异常，找不到文件: {e}")
    except Exception as e:
        print(f"异常: {e}")

def init_folder():
    """检查并创建所需的文件夹"""
    if not os.path.exists("output_apk"):
        os.mkdir("output_apk")

    if not os.path.exists("update_apk"):
        os.mkdir("update_apk")

    if not os.path.exists("update_name_apk"):
        os.mkdir("update_name_apk")

    if not os.path.exists(output_dir):
        os.makedirs(output_dir)

    if not os.path.exists(update_apk_folder):
        os.makedirs(update_apk_folder)

    if not os.path.exists(update_apk_name_folder):
        os.makedirs(update_apk_name_folder)
    
    if not os.path.exists(APK_CODE) or not os.path.exists(APK_VERSION):
        print("检测到根目录下没有字典库，正在初始化字典库为 Phone")
        move_json(0, "ph")
        print("如有更换字典库需要，请使用 -t 命令进行切换")


def init_json():
    """初始化排除APK列表和APK版本号字典"""
    exclude_apk = []
    apk_version = {}
    apk_code = {}
    apk_code_name = {}
    # 查询 APK 排除列表
    if os.path.exists(EXCLUDE_APK_PATH):
        with open(EXCLUDE_APK_PATH, 'r') as f:
            exclude_apk = [line.strip() for line in f.readlines()]
    # 查询本地字典版本名
    if os.path.exists(APK_VERSION):
        with open(APK_VERSION, 'r') as f:
            apk_version = json.load(f)
    else:
        apk_version = {}
    # 查询本地字典版本号
    if os.path.exists(APK_CODE):
        with open(APK_CODE, 'r') as f:
            apk_code = json.load(f)
    else:
        apk_code = {}
    return exclude_apk, apk_version, apk_code, apk_code_name


def download_rom(url):
    """从给定的URL下载ROM"""
    subprocess.run(["aria2c", "-x16", "-s16",url])


def extract_payload_bin(zip_files):
    """从ZIP文件中提取payload.bin文件"""
    for f in zip_files:
        try:
            subprocess.run(["7z", "x", "{}".format(f), "payload.bin"])
        except Exception as e:
            print(f"异常，报错信息: {e}")


def extract_img():
    # 使用 subprocess 模块运行 shell 命令，执行 payload-dumper-go 的命令，从 payload.bin 文件中提取指定镜像文件
    # -c 参数指定最大并发数为 8，-o 指定提取后的文件输出到当前目录下
    # -p 参数指定提取指定镜像，"payload.bin" 为输入文件
    subprocess.run(["./payload-dumper-go", "-c", "8", "-o","./", "-p", "system,system_ext,product", "payload.bin"])


def extract_files():
    try:
        # 使用 subprocess 模块运行 shell 命令，提取镜像文件中的文件
        output = subprocess.check_output(["file", "system.img"]).decode("utf-8")
        print("当前镜像打包格式:", output)
        if "EROFS filesystem" in output:
            # 如果输出内容包含 EROFS filesystem 则使用 extract.erofs 解压
            # -i 参数指定输入的镜像文件为，-x 参数指定提取文件，-T 参数指定使用线程提取文件
            subprocess.run(["./extract.erofs", "-i", "system.img", "-x", "-T16"])
            subprocess.run(["./extract.erofs", "-i", "system_ext.img", "-x", "-T16"])
            subprocess.run(["./extract.erofs", "-i", "product.img", "-x", "-T16"])
        elif "data" in output:
            # 如果输出内容包含 data 则使用7zip解压
            # x 参数指定输入的镜像文件为，-o 提取指定提取文件到目录下
            subprocess.run(["7z", "x", "system.img", r"-o.\system"])
            subprocess.run(["7z", "x", "system_ext.img", r"-o.\system_ext"])
            subprocess.run(["7z", "x", "product.img", r"-o.\product"])
        else:
            print("未知的文件系统类型")
    except subprocess.CalledProcessError as e:
        print("解包失败:", e)
    except Exception as e:
        print("解包失败:", e)

    # 获取设备代号
    try:
        with open("./system/system/build.prop", "r") as file:
            for line in file:
                if line.startswith("ro.product.system.name"):
                    device_name = line.split("=")[1].strip()
                    print(f"设备名: {device_name}")

                    # 建立一个集合，用来判断是否为 Fold 或者 Pad
                    is_fold = {}
                    is_pad = {"NP01J"}

                    if device_name in is_fold:
                        print("\n检测到包设备为 Fold，请输入-t 0/1(不备份/备份) f 参数切换字库")
                        break
                    elif device_name in is_pad:
                        print("\n检测到包设备为 Pad，请输入-t 0/1(不备份/备份) p 参数切换字库")
                        break
                    else:
                        print("\n检测到包设备为 Phone，请输入-t 0/1(不备份/备份) ph 参数切换字库")
                        break                    
    except FileNotFoundError:
        print("无法获取设备名")

def remove_some_apk(exclude_apk):
    # 遍历当前目录及其子目录
    for root, _, files in os.walk('.'):
        for file in files:
            # 判断文件是否为apk文件，且不在要排除的列表中
            if file.endswith('.apk') and file not in exclude_apk:
                src = os.path.join(root, file)
                dst = os.path.join(output_dir, file)
                # 将文件移动到output_dir目录下
                try:
                    shutil.move(src, dst)
                except PermissionError:
                    print(f"无法移动文件 {src}，请检查你的文件权限或关闭占用该文件的程序。")

    # 遍历output_dir目录及其子目录
    for root, _, files in os.walk(output_dir):
        for filename in fnmatch.filter(files, "*.apk"):
            # 判断文件名中是否包含"Overlay"或"_Sys"，若包含则删除该文件
            if "overlay" in filename.lower() or "_Sys" in filename:
                try:
                    os.remove(os.path.join(root, filename))
                except PermissionError:
                    print(f"无法删除文件 {filename}，请检查你的文件权限或关闭占用该文件的程序。")


def rename_apk(apk_files):
    # 遍历每个 apk 文件
    for apk_file in apk_files:
        apk_path = os.path.join(output_dir, apk_file)

        try:
            # 使用 pyaxmlparser 库读取apk包信息
            apk = APK(apk_path)

            # 获取apk的包名，版本号，version code
            package_name = apk.package
            version_name = apk.get_androidversion_name()
            version_code = apk.get_androidversion_code()

            # 构建新文件名
            new_name = f"{package_name}^{version_name}^{version_code}.apk"

            # 关闭 apk 文件
            apk.zip.close()

            # 重命名 apk 文件
            if not os.path.exists(os.path.join(output_dir, new_name)):
                os.rename(apk_path, os.path.join(output_dir, new_name))
        except Exception as e:
            print(f"异常，报错信息: {e}")


# 定义更新 apk 版本的函数，遍历输出目录下的 apk 文件，并更新本地词典
def update_apk_version(apk_version, apk_code, apk_code_name):
    # 遍历输出目录下的 apk 文件
    for apk_file in os.listdir(output_dir):
        # 如果文件名以 ".apk" 结尾
        if apk_file.endswith('.apk'):
            # 解析文件名，获取包名和版本号
            try:
                x, y, z = os.path.splitext(apk_file)[0].split('^')
                # 如果包名在本地词典中
                if x in apk_code:
                    # 如果本地词典中的版本号比 Apk 记录的版本号低
                    if apk_code[x] < int(z):
                        print(f'更新 {x}：{apk_code[x]} -> {z}')
                        if apk_version[x] == y:
                            # 更新本地词典中的版本号
                            apk_version[x] = y
                            apk_code[x] = int(z) # 以 int 格式写入
                            apk_code_name[x] = int(z) # 以 int 格式写入
                            # 复制新版本的 APK 文件到 update_apk 文件夹
                            src = os.path.join(output_dir, apk_file)
                            dst = os.path.join(update_apk_folder, apk_file)
                            shutil.copy2(src, dst)
                            print(f'已将 {apk_file} 复制到 {update_apk_folder} 文件夹\n')
                        else:
                            # 更新本地词典中的版本号
                            apk_version[x] = y
                            apk_code[x] = int(z) # 以 int 格式写入
                            # 复制新版本的 APK 文件到 update_apk 文件夹
                            src = os.path.join(output_dir, apk_file)
                            dst = os.path.join(update_apk_folder, apk_file)
                            shutil.copy2(src, dst)
                            print(f'已将 {apk_file} 复制到 {update_apk_folder} 文件夹\n')
                    elif apk_code[x] == int(z):
                        if apk_version[x] != y:
                            print(f'疑似更新 {x}：{apk_version[x]} -> {y}')
                            # 复制新版本的 APK 文件到 update_name_apk 文件夹
                            src = os.path.join(output_dir, apk_file)
                            dst = os.path.join(update_apk_name_folder, apk_file)
                            shutil.copy2(src, dst)
                            print(f'已将 {apk_file} 复制到 {update_apk_name_folder} 文件夹\n')
                # 如果包名不在本地词典中
                else:
                    print(f'添加新应用 {x}:{y}({z})\n')
                    # 在本地词典中添加新的包名和版本号
                    apk_version[x] = y
                    apk_code[x] = int(z) # 以 int 格式写入
            except Exception as e:
                print(f"异常，报错信息: {e}")
                return

    # 保存本地词典到json文件
    with open(APK_VERSION, 'w') as f:
        json.dump(apk_version, f)
    with open(APK_CODE, 'w') as f:
        json.dump(apk_code, f)
    with open(APK_CODE_NAME, 'w') as f:
        json.dump(apk_code_name, f)

# 定义更新apk文件名的函数，读取第二个词典并修改apk文件名


def update_apk_name():
    with open(JSON_V, 'r', encoding='utf-8') as file:
        line = file.readline()

    # 判断当前字典库类别
    if line == "Phone" or line == "Fold":
        # 如果第二个词典文件存在，则读取其中的内容
        if os.path.exists(APK_APP_NAME):
            with open(APK_APP_NAME, 'r', encoding='utf-8') as f:
                apk_name = json.load(f)
        # 如果第二个词典文件不存在，则将其设为空字典
        else:
            apk_name = {}
    elif line == "Pad":
        if os.path.exists(APK_APP_NAME_PAD):
            with open(APK_APP_NAME_PAD, 'r', encoding='utf-8') as f:
                apk_name = json.load(f)
        else:
            apk_name = {}

    # 如果临时词典文件存在，则读取其中的内容
    if os.path.exists(APK_CODE_NAME):
        with open(APK_CODE_NAME, 'r', encoding='utf-8') as f:
            apk_code_name = json.load(f)
    # 如果临时词典文件不存在，则将其设为空字典
    else:
        apk_code_name = {}

    def rename_files_in_folder(folder, name_dict, code_dict):
        for apk_file in os.listdir(folder):
            # 如果文件名以".apk"结尾
            if apk_file.endswith('.apk'):
                # 解析文件名，获取包名和版本号
                x, y, z = os.path.splitext(apk_file)[0].split('^')
                # 如果解析的文件名在字典里
                if x in name_dict:
                    if x in code_dict:
                        # 定义修改的文件名
                        new_x = name_dict[x]
                        new_apk_file_1 = f'{new_x}_{y}({z}).apk'
                        # 修改为新定义的文件名
                        os.rename(os.path.join(folder, apk_file),
                                 os.path.join(folder, new_apk_file_1))
                        print(f'修改 {apk_file} -> {new_apk_file_1}')
                    else:
                        # 定义修改的文件名
                        new_x = name_dict[x]
                        new_apk_file_2 = f'{new_x}_{y}({z}).apk'
                        # 修改为新定义的文件名
                        os.rename(os.path.join(folder, apk_file),
                                 os.path.join(folder, new_apk_file_2))
                        print(f'修改 {apk_file} -> {new_apk_file_2}')

    # 重命名 output_dir 中的 APK 文件
    rename_files_in_folder(output_dir, apk_name, apk_code_name)

    # 重命名 update_apk 文件夹中的 APK 文件
    rename_files_in_folder(update_apk_folder, apk_name, apk_code_name)

    # 重命名 update_name_apk 文件夹中的 APK 文件
    rename_files_in_folder(update_apk_name_folder, apk_name, apk_code_name)


def delete_files_and_folders():
    """删除指定的文件和文件夹"""
    files_to_delete = ["payload.bin", "*.img", "app_code_name.json"]
    folders_to_delete = ["output_apk", "update_apk", "update_name_apk", "system_ext", "system", "product", "config"]

    for file in files_to_delete:
        if os.path.exists(file):
            try:
                os.remove(file)
                print(f"{file} 删除成功")
            except OSError as e:
                print(f"无法删除 {file}: {e}")

        else:
            print(f"{file} 不存在")

    for folder in folders_to_delete:
        if os.path.exists(folder):
            if os.path.isdir(folder):
                try:
                    shutil.rmtree(folder)
                    print(f"{folder} 删除成功")
                except OSError as e:
                    print(f"无法删除 {folder}: {e}")
            else:
                print(f"{folder} 不是文件夹")
        else:
            print(f"{folder} 不存在")


def git_push():
    device_name = input("机型：")
    os_version = input("版本号：")
    commit_text = "Database：Update"
    commit = f"{commit_text} {device_name} {os_version}"
    try:
        with open(JSON_V, 'r') as file:
            line = file.readline()
            if line == "Phone":
                move_json(1, "ph")
                subprocess.run(["git", "add", "phone/"]) 
            elif line == "Pad":
                move_json(1, "p")
                subprocess.run(["git", "add", "pad/"]) 
            elif line == "Fold":
                move_json(1, "f")
                subprocess.run(["git", "add", "fold/"]) 
            else:
                print("未检测到字库，无法上传")
    except Exception as e:
        print(f"异常，报错信息: {e}")
    subprocess.run(["git", "commit","-m",commit]) 
    subprocess.run(["git", "push"])