debugging.py

import atexit
import cProfile
import logging
import os
import pstats
from datetime import datetime

__LOG_DIR = os.path.expanduser("~/.local/share/vectorcode/logs/")

logger = logging.getLogger(name=__name__)

__profiler: cProfile.Profile | None = None


def _ensure_log_dir():
    """Ensure the log directory exists"""
    os.makedirs(__LOG_DIR, exist_ok=True)


def finish():
    """Clean up profiling and save results"""
    if __profiler is not None:
        try:
            __profiler.disable()
            stats_file = os.path.join(
                __LOG_DIR,
                f"cprofile-{datetime.now().strftime('%Y-%m-%d_%H-%M-%S')}.stats",
            )
            __profiler.dump_stats(stats_file)
            print(f"cProfile stats saved to: {stats_file}")

            # Print summary stats
            stats = pstats.Stats(__profiler)
            stats.sort_stats("cumulative")
            stats.print_stats(20)
        except Exception as e:
            logger.warning(f"Failed to save cProfile output: {e}")


def enable():
    """Enable cProfile-based profiling and crash debugging"""
    global __profiler

    try:
        _ensure_log_dir()

        # Initialize cProfile for comprehensive profiling
        __profiler = cProfile.Profile()
        __profiler.enable()
        atexit.register(finish)
        logger.info("cProfile profiling enabled successfully")

        try:
            import coredumpy  # noqa: F401

            logger.info("coredumpy crash debugging enabled successfully")
            coredumpy.patch_except(directory=__LOG_DIR)
        except Exception as e:
            logger.warning(
                f"Crash debugging will not be available. Failed to import coredumpy: {e}"
            )

    except Exception as e:
        logger.error(f"Failed to initialize cProfile: {e}")
        logger.warning("Profiling will not be available for this session")