rmser/pack_go_files.py

#!/usr/bin/env python3
"""
Скрипт для упаковки всех файлов проекта (.go, .json, .mod, .md)
в один Python-файл для удобной передачи ИИ.
Формирует дерево проекта и экранирует содержимое всех файлов.
"""

import os
import sys
import json


# ---------------------------------------------------------
#   Список имён файлов/папок, которые нужно игнорировать.
#   Работает по вхождению: "vendor" исключит любую vendor/*
# ---------------------------------------------------------
IGNORE_LIST = [
    ".git",
    ".kilocode",
    "tools",
    "project_dump.py",
    ".idea",
    ".vscode",
    "node_modules",
    "ftp_cache",
    "ocr-service",
    "rmser-view"
]


def should_ignore(path: str) -> bool:
    """
    Проверяет, должен ли путь быть проигнорирован.
    Смотрит и на файлы, и на каталоги.
    """
    for ignore in IGNORE_LIST:
        if ignore in path.replace("\\", "/"):
            return True
    return False


def escape_content(content: str) -> str:
    """
    Экранирует содержимое файла для корректного помещения в Python-строку.
    Используем json.dumps для максимальной безопасности и читаемости.
    """
    return json.dumps(content, ensure_ascii=False)


def collect_files(root_dir: str, extensions):
    """
    Рекурсивно собирает пути ко всем файлам с указанными расширениями.
    Учитывает IGNORE_LIST.
    """
    collected = []
    for dirpath, dirnames, filenames in os.walk(root_dir):

        # фильтрация каталогов
        dirnames[:] = [d for d in dirnames if not should_ignore(os.path.join(dirpath, d))]

        for file in filenames:
            full_path = os.path.join(dirpath, file)
            if should_ignore(full_path):
                continue
            if any(file.endswith(ext) for ext in extensions):
                collected.append(os.path.normpath(full_path))

    return sorted(collected)


def build_tree(root_dir: str) -> str:
    """
    Создаёт строковое представление дерева проекта.
    Учитывает IGNORE_LIST.
    """
    tree_lines = []

    def walk(dir_path: str, prefix: str = ""):
        try:
            entries = sorted(os.listdir(dir_path))
        except PermissionError:
            return

        # фильтрация по IGNORE_LIST
        entries = [e for e in entries if not should_ignore(os.path.join(dir_path, e))]

        for idx, entry in enumerate(entries):
            path = os.path.join(dir_path, entry)
            connector = "└── " if idx == len(entries) - 1 else "├── "
            tree_lines.append(f"{prefix}{connector}{entry}")
            if os.path.isdir(path):
                new_prefix = prefix + ("    " if idx == len(entries) - 1 else "│   ")
                walk(path, new_prefix)

    tree_lines.append(".")
    walk(root_dir)
    return "\n".join(tree_lines)


def write_to_py(files, tree_str, output_file):
    """
    Записывает дерево проекта и содержимое файлов в один .py файл.
    """
    with open(output_file, "w", encoding="utf-8") as f:
        f.write("# -*- coding: utf-8 -*-\n")
        f.write("# Этот файл сгенерирован автоматически.\n")
        f.write("# Содержит дерево проекта и файлы (.go, .json, .mod, .md) в экранированном виде.\n\n")

        f.write("project_tree = '''\n")
        f.write(tree_str)
        f.write("\n'''\n\n")

        f.write("project_files = {\n")
        for path in files:
            rel_path = os.path.relpath(path)
            try:
                with open(path, "r", encoding="utf-8", errors="ignore") as src:
                    content = src.read()
            except Exception as e:
                content = f"<<Ошибка чтения файла: {e}>>"

            escaped_content = escape_content(content)
            f.write(f'    "{rel_path}": {escaped_content},\n')
        f.write("}\n\n")

        f.write("if __name__ == '__main__':\n")
        f.write("    print('=== Дерево проекта ===')\n")
        f.write("    print(project_tree)\n")
        f.write("    print('\\n=== Список файлов ===')\n")
        f.write("    for name in project_files:\n")
        f.write("        print(f'- {name}')\n")


def main():
    root_dir = "."
    output_file = "project_dump.py"

    if len(sys.argv) > 1:
        output_file = sys.argv[1]

    exts = [".go", ".yaml", ".json", ".mod", ".md"]

    files = collect_files(root_dir, exts)
    tree_str = build_tree(root_dir)
    write_to_py(files, tree_str, output_file)

    print(f"Собрано {len(files)} файлов. Результат в {output_file}")


if __name__ == "__main__":
    main()