hoodini.utils.runtime_env

  1"""
  2Runtime environment utilities for managing LD_LIBRARY_PATH across conda/pixi/mamba.
  3
  4This module handles automatic detection and setup of library paths needed by
  5Playwright Firefox in conda/mamba/pixi environments without requiring sudo.
  6"""
  7
  8import os
  9from pathlib import Path
 10
 11
 12def find_candidate_lib_dirs() -> list[str]:
 13    """
 14    Find candidate library directories from conda/mamba/pixi environments.
 15
 16    Searches for:
 17    - $CONDA_PREFIX/lib (conda/mamba active environment)
 18    - ~/.pixi/envs/*/lib (pixi environments)
 19    - ~/.mamba/envs/*/lib (mamba environments)
 20
 21    Returns:
 22        List of absolute paths to lib directories, deduplicated and ordered.
 23    """
 24    candidates: list[str] = []
 25
 26    # Check CONDA_PREFIX (active conda/mamba environment)
 27    if conda_prefix := os.environ.get("CONDA_PREFIX"):
 28        lib_dir = os.path.join(conda_prefix, "lib")
 29        if os.path.isdir(lib_dir):
 30            candidates.append(lib_dir)
 31
 32    # Check pixi environments
 33    pixi_envs_dir = Path.home() / ".pixi" / "envs"
 34    if pixi_envs_dir.exists():
 35        for env_dir in pixi_envs_dir.iterdir():
 36            if env_dir.is_dir():
 37                lib_dir = env_dir / "lib"
 38                if lib_dir.exists():
 39                    candidates.append(str(lib_dir))
 40
 41    # Check mamba environments
 42    mamba_envs_dir = Path.home() / ".mamba" / "envs"
 43    if mamba_envs_dir.exists():
 44        for env_dir in mamba_envs_dir.iterdir():
 45            if env_dir.is_dir():
 46                lib_dir = env_dir / "lib"
 47                if lib_dir.exists():
 48                    candidates.append(str(lib_dir))
 49
 50    # Deduplicate while preserving order
 51    seen = set()
 52    unique_candidates = []
 53    for path in candidates:
 54        if path not in seen:
 55            seen.add(path)
 56            unique_candidates.append(path)
 57
 58    return unique_candidates
 59
 60
 61def verify_gtk_availability() -> str | None:
 62    """
 63    Verify that GTK3 libraries are available in the current environment.
 64
 65    Returns:
 66        Path to the lib directory containing libgtk-3.so.0, or None if not found.
 67    """
 68    for lib_dir in find_candidate_lib_dirs():
 69        gtk_lib = Path(lib_dir) / "libgtk-3.so.0"
 70        if gtk_lib.exists():
 71            return lib_dir
 72    return None
 73
 74
 75def apply_ld_library_path() -> None:
 76    """
 77    Apply LD_LIBRARY_PATH modifications for Playwright Firefox in conda/pixi/mamba.
 78
 79    Prepends candidate library directories to LD_LIBRARY_PATH so that Playwright
 80    Firefox can find GTK3 and other required libraries.
 81
 82    This should be called before launching Playwright browser instances.
 83    """
 84    candidates = find_candidate_lib_dirs()
 85    if not candidates:
 86        # No conda/pixi/mamba environments found, nothing to do
 87        return
 88
 89    current_ld_path = os.environ.get("LD_LIBRARY_PATH", "")
 90
 91    # Build new LD_LIBRARY_PATH: candidates first, then existing path
 92    ld_path_parts = candidates.copy()
 93    if current_ld_path:
 94        ld_path_parts.append(current_ld_path)
 95
 96    new_ld_path = ":".join(ld_path_parts)
 97    os.environ["LD_LIBRARY_PATH"] = new_ld_path
 98
 99
100__all__ = [
101    "find_candidate_lib_dirs",
102    "verify_gtk_availability",
103    "apply_ld_library_path",
104]