# 传入参数 0:SDK_PATH 1:ChannelPath 2:Obfuz(可选, true/false)
import sys
import os
import yaml

def main():
    if len(sys.argv) < 3:
        print("Usage: python PreHandle.py <SDK_PATH> <ChannelPath> [Obfuz]")
        sys.exit(1)  # 参数不足时返回错误码1

    sdk_path = sys.argv[1]
    channel_path = sys.argv[2]
    obfuz = sys.argv[3].lower() == 'true' if len(sys.argv) >= 4 else None

    if not os.path.exists(sdk_path):
        print(f"Error: SDK path '{sdk_path}' does not exist.")
        sys.exit(2)  # SDK路径不存在时返回错误码2

    if not os.path.exists(channel_path):
        print(f"Error: Channel path '{channel_path}' does not exist.")
        sys.exit(3)  # Channel路径不存在时返回错误码3

    # 这里可以添加更多的处理逻辑
    print(f"SDK Path: {sdk_path}")
    print(f"Channel Path: {channel_path}")

    # 读取Channel Path/ChannelSetting.yaml
    channelSettingPath = os.path.join(channel_path, "ChannelSetting.yaml")
    if not os.path.exists(channelSettingPath):
        print(f"Error: Channel setting file '{channelSettingPath}' does not exist.")
        sys.exit(4)  # Channel设置文件不存在时返回错误码4
    with open(channelSettingPath, 'r', encoding='utf-8') as file:
        channel_setting = file.read()
        print(f"Channel Setting: {channel_setting}")
        # 使用yaml解析
        try:
            channel_data = yaml.safe_load(channel_setting)
            print(f"Parsed Channel Data: {channel_data}")
        except yaml.YAMLError as e:
            print(f"Error parsing YAML: {e}")
            sys.exit(5)

    # 检查Packages是否存在
    # 这里假设Packages是一个列表 内容是 packageName@version
    # 例如: ["package1@1.0.0", "package2@latest"]
    packageList = channel_data.get("Packages", [])
    if not packageList:
        print("Info: PackageList is empty or not found in ChannelSetting.yaml.")
        # 如果PackageList为空,则不处理
        return
    for package in packageList:
        HandlePackage(package, sdk_path, channel_path)

    # 同步 USE_OBFUZ 宏（如果传入了 Obfuz 参数）
    if obfuz is not None:
        SyncUseObfuzDefine(channel_path, obfuz)

    # 处理完成，正常退出
    print("All packages processed successfully.")
    sys.exit(0)  # 改为成功退出

def SyncUseObfuzDefine(channel_path, enable):
    """根据 enable 参数在 ChannelSetting.yaml 的 Define_list 中添加或移除 USE_OBFUZ"""
    channel_setting_path = os.path.join(channel_path, "ChannelSetting.yaml")
    if not os.path.exists(channel_setting_path):
        print(f"[PreHandle] WARNING: ChannelSetting.yaml not found at {channel_setting_path}, skip USE_OBFUZ sync")
        return

    try:
        with open(channel_setting_path, 'r', encoding='utf-8') as f:
            data = yaml.safe_load(f) or {}
    except (IOError, yaml.YAMLError) as e:
        print(f"[PreHandle] ERROR: Failed to read ChannelSetting.yaml: {e}")
        return

    defines = data.get('Define_list') or []
    changed = False
    if enable:
        if 'USE_OBFUZ' not in defines:
            defines.append('USE_OBFUZ')
            data['Define_list'] = defines
            changed = True
            print('[PreHandle] Added USE_OBFUZ to Define_list')
        else:
            print('[PreHandle] USE_OBFUZ already in Define_list, skip')
    else:
        if 'USE_OBFUZ' in defines:
            defines.remove('USE_OBFUZ')
            data['Define_list'] = defines
            changed = True
            print('[PreHandle] Removed USE_OBFUZ from Define_list')
        else:
            print('[PreHandle] USE_OBFUZ not in Define_list, skip')

    if changed:
        try:
            with open(channel_setting_path, 'w', encoding='utf-8') as f:
                yaml.dump(data, f, allow_unicode=True)
        except IOError as e:
            print(f"[PreHandle] ERROR: Failed to write ChannelSetting.yaml: {e}")


def HandlePackage(packageDefineName, sdk_path, channel_path):
    # 处理包定义名称
    if '@' not in packageDefineName:
        print(f"Error: Invalid package format '{packageDefineName}'. Expected format 'packageName@version'.")
        sys.exit(8)  # 返回错误码8
    packageName, version = packageDefineName.split('@', 1)
    print(f"Start Handle Package Name: {packageName}, Version: {version}")
    # 检查sdk_path/packageName 是否存在
    package_path = os.path.join(sdk_path, packageName)
    if not os.path.exists(package_path):
        print(f"Error: Package path '{package_path}' does not exist.")
        sys.exit(7)  # 包路径不存在时返回错误码7

    # 如果version是latest, 则读取package_path/latest.txt
    if version == "latest":
        latest_file_path = os.path.join(package_path, "latest.txt")
        if not os.path.exists(latest_file_path):
            print(f"Error: Latest file '{latest_file_path}' does not exist.")
            sys.exit(9)
        with open(latest_file_path, 'r', encoding='utf-8') as file:
            latest_version = file.read().strip()
            print(f"Latest Version: {latest_version}")
            version = latest_version  # 添加这行：更新version变量
    
    #判断package_path/version文件夹是否存在
    version_path = os.path.join(package_path, version)
    if not os.path.exists(version_path):
        print(f"Error: Version path '{version_path}' does not exist.")
        sys.exit(10)
    # 把version_path/Assets下所有的文件和文件夹复制到channel_path/Assets
    srcAssetPath = os.path.join(version_path, "Assets")
    targetAssetPath = os.path.join(channel_path, "Assets")
    CopyAssets(srcAssetPath, targetAssetPath)
    ModifyDefineList(version_path, channel_path)

def ModifyDefineList(versionPath, channel_path):
    # 读取versionPath/OtherConfig.yaml
    other_config_path = os.path.join(versionPath, "OtherConfig.yaml")
    if not os.path.exists(other_config_path):
        print(f"Info: OtherConfig file '{other_config_path}' does not exist.")
        return
    
    try:
        with open(other_config_path, 'r', encoding='utf-8') as file:
            other_config = yaml.safe_load(file)
            print(f"Other Config: {other_config}")
    except (IOError, yaml.YAMLError) as e:
        print(f"Error reading OtherConfig file '{other_config_path}': {e}")
        return
        
    # 检查是否有DefineList
    define_list = other_config.get("Define_list", [])
    if not define_list:
        print("Info: Define_list is empty or not found in OtherConfig.yaml.")
        return
    
    # 修改channel_path/ChannelSetting.yaml中的DefineList
    channel_setting_path = os.path.join(channel_path, "ChannelSetting.yaml")
    if not os.path.exists(channel_setting_path):
        print(f"Error: Channel setting file '{channel_setting_path}' does not exist.")
        sys.exit(11)
    
    try:
        with open(channel_setting_path, 'r', encoding='utf-8') as file:
            channel_setting = yaml.safe_load(file)
            # print(f"Channel Setting Before Modification: {channel_setting}")
    except (IOError, yaml.YAMLError) as e:
        print(f"Error reading channel setting file '{channel_setting_path}': {e}")
        sys.exit(11)

    print(f"Old Define_List: {channel_setting.get('Define_list', [])}")
    
    # 修改DefineList（添加去重功能）
    # 注意：Define_list 可能不存在，或存在但值为 None（YAML 中写了 key 但没有值）
    if not channel_setting.get("Define_list"):
        channel_setting["Define_list"] = []
    
    # 合并并去重
    current_defines = set(channel_setting["Define_list"])
    new_defines = set(define_list)
    
    # 输出将要添加的新定义
    added_defines = new_defines - current_defines
    if added_defines:
        print(f"Adding new defines: {list(added_defines)}")
    
    # 重复项警告
    duplicate_defines = new_defines & current_defines
    if duplicate_defines:
        print(f"Warning: Duplicate defines found and will be ignored: {list(duplicate_defines)}")
    
    # 更新定义列表（保持原有顺序，添加新项）
    channel_setting["Define_list"] = list(current_defines | new_defines)
    
    try:
        with open(channel_setting_path, 'w', encoding='utf-8') as file:
            yaml.dump(channel_setting, file, allow_unicode=True)
            # print(f"Channel Setting After Modification: {channel_setting}")
    except IOError as e:
        print(f"Error writing channel setting file '{channel_setting_path}': {e}")
        sys.exit(15)

    print(f"New Define_List: {channel_setting['Define_list']}")

def CopyAssets(srcAssetPath, targetAssetPath):
    # 如果srcAssetPath不存在,则返回
    if not os.path.exists(srcAssetPath):
        print(f"Info: Source asset path '{srcAssetPath}' does not exist.")
        return
    # 如果targetAssetPath不存在,则创建
    if not os.path.exists(targetAssetPath):
        try:
            os.makedirs(targetAssetPath)
            # print(f"Created target asset path: {targetAssetPath}")
        except OSError as e:
            print(f"Error: Failed to create directory '{targetAssetPath}': {e}")
            sys.exit(12)
    
    # print(f"Start copying assets from '{srcAssetPath}' to '{targetAssetPath}'")

    # 遍历srcAssetPath下的所有文件和文件夹
    try:
        for item in os.listdir(srcAssetPath):
            src_item_path = os.path.join(srcAssetPath, item)
            target_item_path = os.path.join(targetAssetPath, item)
            # 如果是文件,则复制
            if os.path.isfile(src_item_path):
                # 检查目标文件是否已存在，如果存在则输出警告
                if os.path.exists(target_item_path):
                    print(f"Warning: Target file '{target_item_path}' already exists and will be overwritten.")
                
                try:
                    with open(src_item_path, 'rb') as src_file:
                        with open(target_item_path, 'wb') as target_file:
                            target_file.write(src_file.read())
                    # print(f"Copied file: {src_item_path} to {target_item_path}")
                except IOError as e:
                    print(f"Error: Failed to copy file '{src_item_path}': {e}")
                    sys.exit(13)
            # 如果是文件夹,则递归调用CopyAssets
            elif os.path.isdir(src_item_path):
                # 检查目标目录是否已存在，如果存在则输出警告
                if os.path.exists(target_item_path):
                    print(f"Warning: Target directory '{target_item_path}' already exists, merging contents.")
                CopyAssets(src_item_path, target_item_path)
    except OSError as e:
        print(f"Error: Failed to list directory '{srcAssetPath}': {e}")
        sys.exit(14)

    # print(f"Assets copied successfully from '{srcAssetPath}' to '{targetAssetPath}'")

if __name__ == "__main__":
    main()