import os
import pathspec

def collect_repository(root_dir: str, output_file: str, max_file_size_kb: int = 500):
    """
    遍历代码仓库，遵循 .gitignore 规则，将所有源文件结构化打包为适合 1M 上下文的 XML 格式。
    """
    ignore_spec = None
    gitignore_path = os.path.join(root_dir, '.gitignore')
    if os.path.exists(gitignore_path):
        with open(gitignore_path, 'r', encoding='utf-8') as f:
            ignore_spec = pathspec.PathSpec.from_lines('gitwildmatch', f)

    file_entries = []
    tree_lines = []

    for dirpath, dirnames, filenames in os.walk(root_dir):
        # 排除 .git 与 node_modules
        if '.git' in dirnames:
            dirnames.remove('.git')
        if 'node_modules' in dirnames:
            dirnames.remove('node_modules')

        for filename in filenames:
            rel_path = os.path.relpath(os.path.join(dirpath, filename), root_dir).replace('\\', '/')
            if ignore_spec and ignore_spec.match_file(rel_path):
                continue

            full_path = os.path.join(dirpath, filename)
            if os.path.getsize(full_path) > max_file_size_kb * 1024:
                continue

            try:
                with open(full_path, 'r', encoding='utf-8', errors='ignore') as f:
                    content = f.read()
                    file_entries.append((rel_path, content))
                    tree_lines.append(rel_path)
            except Exception:
                continue

    with open(output_file, 'w', encoding='utf-8') as out:
        out.write("<repository_index>\n")
        out.write("\n".join(tree_lines))
        out.write("\n</repository_index>\n\n")

        for path, code in file_entries:
            out.write(f'<file path="{path}">\n')
            out.write(code)
            out.write("\n</file>\n\n")

    print(f"成功打包 {len(file_entries)} 个文件至 {output_file}！")

if __name__ == "__main__":
    collect_repository(".", "codebase_1m_payload.txt")

<!-- 1. 系统角色指令 (Top) -->
<role_directive>
你是一位世界顶尖的系统架构师。你将审查以下全库源码。
所有修改建议必须严格保持类型兼容，并以统一 Git Diff 格式输出。
</role_directive>

<!-- 2. 全局目录索引树 -->
<repository_index>
src/core/auth.ts
src/core/db/pool.ts
src/server/router.ts
... (500 个文件索引)
</repository_index>

<!-- 3. 百万字全量文件内容 -->
<file path="src/core/db/pool.ts">
... (文件源码)
</file>
... (其余文件)

<!-- 4. 具体任务指令 (Bottom: 极度关键，必须放最底部) -->
<task_instruction>
请分析从 `router.ts` 接收到 WebSocket 请求，到 `pool.ts` 释放连接的全链路。
当发生瞬时断网重连时，是否存在连接泄露？
请列出调用栈证据，并输出修复该 Bug 的最小 Git Diff 补丁。
</task_instruction>