#!/usr/bin/env python3
"""
更新所有 C# 文件中的命名空间声明和 using 语句
"""

import os
import re
from pathlib import Path

ROOT_DIR = "/mnt/f/gewuyou/System/Documents/WorkSpace/GameDev/GFramework"

# 命名空间替换规则（按优先级排序，长的先匹配）
NAMESPACE_RULES = [
    # CQRS 子命名空间
    (r'\.cqrs\.notification\b', '.CQRS.Notification'),
    (r'\.cqrs\.command\b', '.CQRS.Command'),
    (r'\.cqrs\.request\b', '.CQRS.Request'),
    (r'\.cqrs\.query\b', '.CQRS.Query'),
    (r'\.cqrs\.behaviors\b', '.CQRS.Behaviors'),
    (r'\.cqrs\b', '.CQRS'),

    # 嵌套命名空间
    (r'\.coroutine\.instructions\b', '.Coroutine.Instructions'),
    (r'\.coroutine\.extensions\b', '.Coroutine.Extensions'),
    (r'\.coroutine\b', '.Coroutine'),

    (r'\.events\.filters\b', '.Events.Filters'),
    (r'\.events\b', '.Events'),

    (r'\.logging\.appenders\b', '.Logging.Appenders'),
    (r'\.logging\.filters\b', '.Logging.Filters'),
    (r'\.logging\.formatters\b', '.Logging.Formatters'),
    (r'\.logging\b', '.Logging'),

    (r'\.functional\.async\b', '.Functional.Async'),
    (r'\.functional\.control\b', '.Functional.Control'),
    (r'\.functional\.functions\b', '.Functional.Functions'),
    (r'\.functional\.pipe\b', '.Functional.Pipe'),
    (r'\.functional\.result\b', '.Functional.Result'),
    (r'\.functional\b', '.Functional'),

    (r'\.services\.modules\b', '.Services.Modules'),
    (r'\.services\b', '.Services'),

    (r'\.extensions\.signal\b', '.Extensions.Signal'),
    (r'\.extensions\b', '.Extensions'),

    (r'\.setting\.data\b', '.Setting.Data'),
    (r'\.setting\.events\b', '.Setting.Events'),
    (r'\.setting\b', '.Setting'),

    (r'\.scene\.handler\b', '.Scene.Handler'),
    (r'\.scene\b', '.Scene'),

    (r'\.ui\.handler\b', '.UI.Handler'),
    (r'\.ui\b', '.UI'),

    (r'\.data\.events\b', '.Data.Events'),
    (r'\.data\b', '.Data'),

    # 单层命名空间
    (r'\.architecture\b', '.Architecture'),
    (r'\.bases\b', '.Bases'),
    (r'\.command\b', '.Command'),
    (r'\.configuration\b', '.Configuration'),
    (r'\.constants\b', '.Constants'),
    (r'\.enums\b', '.Enums'),
    (r'\.environment\b', '.Environment'),
    (r'\.internals\b', '.Internals'),
    (r'\.ioc\b', '.IoC'),
    (r'\.lifecycle\b', '.Lifecycle'),
    (r'\.model\b', '.Model'),
    (r'\.pause\b', '.Pause'),
    (r'\.pool\b', '.Pool'),
    (r'\.properties\b', '.Properties'),
    (r'\.property\b', '.Property'),
    (r'\.query\b', '.Query'),
    (r'\.registries\b', '.Registries'),
    (r'\.resource\b', '.Resource'),
    (r'\.rule\b', '.Rule'),
    (r'\.serializer\b', '.Serializer'),
    (r'\.state\b', '.State'),
    (r'\.storage\b', '.Storage'),
    (r'\.system\b', '.System'),
    (r'\.time\b', '.Time'),
    (r'\.utility\b', '.Utility'),
    (r'\.versioning\b', '.Versioning'),
    (r'\.asset\b', '.Asset'),
    (r'\.components\b', '.Components'),
    (r'\.systems\b', '.Systems'),
    (r'\.ecs\b', '.ECS'),
    (r'\.integration\b', '.Integration'),
    (r'\.mediator\b', '.Mediator'),
    (r'\.tests\b', '.Tests'),
    (r'\.analyzers\b', '.Analyzers'),
    (r'\.diagnostics\b', '.Diagnostics'),
    (r'\.generator\b', '.Generator'),
    (r'\.info\b', '.Info'),
]

def update_file(file_path):
    """更新单个文件中的命名空间"""
    try:
        with open(file_path, 'r', encoding='utf-8') as f:
            content = f.read()

        original_content = content
        replacements = 0

        for pattern, replacement in NAMESPACE_RULES:
            matches = re.findall(pattern, content, re.IGNORECASE)
            if matches:
                content = re.sub(pattern, replacement, content, flags=re.IGNORECASE)
                replacements += len(matches)

        if content != original_content:
            with open(file_path, 'w', encoding='utf-8') as f:
                f.write(content)
            return replacements

        return 0
    except Exception as e:
        print(f"错误处理文件 {file_path}: {e}")
        return 0

def main():
    print("开始更新命名空间...")

    # 查找所有 C# 文件
    cs_files = []
    for root, dirs, files in os.walk(ROOT_DIR):
        # 跳过 bin, obj, Generated 目录
        dirs[:] = [d for d in dirs if d not in ['bin', 'obj', 'Generated', '.git', 'node_modules']]

        for file in files:
            if file.endswith('.cs'):
                cs_files.append(os.path.join(root, file))

    print(f"找到 {len(cs_files)} 个 C# 文件")

    updated_files = 0
    total_replacements = 0

    for file_path in cs_files:
        replacements = update_file(file_path)
        if replacements > 0:
            updated_files += 1
            total_replacements += replacements
            print(f"更新: {os.path.basename(file_path)} ({replacements} 处替换)")

    print(f"\n完成！更新了 {updated_files} 个文件，共 {total_replacements} 处替换")

if __name__ == '__main__':
    main()