G4F/ReactDev/__init__.py

# Инициализация пакета
import os

from .core.ai_interface import AIInterface
from .core.project_map import ProjectMap
from .core.file_analyzer import FileAnalyzer
from .core.commands import CommandGenerator
from .utils.logger import log
from .utils.file_utils import read_file, write_file
from .config import REACT_PROJECT_PATH

__all__ = ["AIInterface", "ProjectMap", "FileAnalyzer", "CommandGenerator", "run_workflow"]


def ensure_default_files():
    """
    Проверяет наличие необходимых файлов и создает их с базовым содержимым, если они отсутствуют.
    """
    default_files = {
        "workingon.md": "/path/to/react/project/src/Auth.tsx",
        "prompt.md": "Добавить функцию авторизации в файл Auth.tsx",
        "instructions.md": "# Инструкции для ИИ\n\n1. Функции должны быть написаны в TypeScript.\n2. Используйте современные практики React (например, хуки).\n3. При создании новых компонентов добавьте комментарии.",
    }

    for file_name, content in default_files.items():
        if not os.path.exists(file_name):
            log(f"File {file_name} not found. Creating it with default content.")
            write_file(file_name, content)


def parse_workingon_content(content: str) -> dict:
    """
    Парсит содержимое файла workingon.md.
    Возвращает словарь с путем к файлу и дополнительным контекстом.
    """
    lines = content.strip().split("\n")
    result = {"file_path": None, "context": ""}

    if lines:
        # Первая строка — путь к файлу
        result["file_path"] = lines[0].strip()
        # Остальные строки — дополнительный контекст
        result["context"] = "\n".join(lines[1:]).strip()

    return result



def run_workflow():
    """
    Основной алгоритм работы библиотеки ReactDev.
    1. Создает карту структуры проекта.
    2. Анализирует файлы на наличие функций.
    3. Отправляет запрос к ИИ с контекстом.
    4. Генерирует план модификации.
    """
    log("Starting ReactDev workflow...")

    # Шаг 0: Убедиться, что все необходимые файлы существуют
    ensure_default_files()

    # Шаг 1: Создание карты структуры проекта
    project_map = ProjectMap(REACT_PROJECT_PATH)
    project_structure = project_map.generate_map()
    project_map.save_map()

    log(f"Generated project structure:\n{project_structure}")

    # Шаг 2: Чтение и парсинг workingon.md
    workingon_content = read_file("workingon.md")
    workingon_data = parse_workingon_content(workingon_content)

    target_file = workingon_data["file_path"]
    additional_context = workingon_data["context"]

    if target_file and not os.path.exists(target_file):
        log(f"Target file {target_file} does not exist.", level="error")
        return

    # Шаг 3: Анализ целевого файла (если указан)
    functions_info = {}
    if target_file:
        file_analyzer = FileAnalyzer()
        functions_info = file_analyzer.analyze_file(target_file)
        log(f"Analyzed functions in {target_file}: {functions_info}")

    # Шаг 4: Отправка запроса к ИИ
    ai_interface = AIInterface()
    user_prompt = read_file("prompt.md").strip()

    # Добавляем дополнительный контекст из workingon.md
    full_context = {
        "project_structure": project_structure,
        "target_file": target_file,
        "functions_info": functions_info,
        "additional_context": additional_context,
    }

    ai_response = ai_interface.analyze_project(user_prompt, full_context)
    log(f"AI response: {ai_response}")

    ai_response = ai_interface.get_modifications(user_prompt, full_context)
    log(f"AI response: {ai_response}")


    # # Шаг 5: Генерация плана модификации
    # command_generator = CommandGenerator()
    # modifications = ai_response.get("modifications", [])
    # modification_plan = command_generator.generate_commands(modifications)

    with open("modification_plan.md", "w", encoding="utf-8") as f:
        f.write(ai_response["response"])

    # log(f"Generated modification plan:\n{modification_plan}")
    log("ReactDev workflow completed.")