Simple DirectMedia Layer

import os
import subprocess
import sys
from pathlib import Path


def detect_display_server() -> str:
    """
    Detect current display server (Wayland or X11)
    현재 디스플레이 서버 감지
    """
    # Wayland 세션 확인
    if os.environ.get("WAYLAND_DISPLAY"):
        return "wayland"
    if os.environ.get("XDG_SESSION_TYPE", "").lower() == "wayland":
        return "wayland"
    # X11 세션 확인
    if os.environ.get("DISPLAY"):
        return "x11"
    return "unknown"


def find_x11_display() -> str | None:
    """
    Find available X11 DISPLAY value
    사용 가능한 X11 DISPLAY 값 찾기
    """
    # 이미 설정된 경우
    if os.environ.get("DISPLAY"):
        return os.environ["DISPLAY"]

    # /tmp/.X11-unix 소켓 탐색
    x11_dir = Path("/tmp/.X11-unix")
    if x11_dir.exists():
        sockets = sorted(x11_dir.glob("X*"))
        if sockets:
            # X0 -> :0, X1 -> :1 ...
            display_num = sockets[0].name[1:]  # 'X' 제거
            return f":{display_num}"

    return None


def check_opengl_support() -> dict:
    """
    Check OpenGL driver availability via glxinfo
    glxinfo로 OpenGL 드라이버 가용성 확인
    """
    result = {"has_glx": False, "has_egl": False, "renderer": "unknown"}

    try:
        out = subprocess.check_output(
            ["glxinfo", "-B"],
            stderr=subprocess.DEVNULL,
            timeout=3,
        ).decode()
        result["has_glx"] = True
        for line in out.splitlines():
            if "OpenGL renderer" in line:
                result["renderer"] = line.split(":", 1)[-1].strip()
    except (FileNotFoundError, subprocess.CalledProcessError, subprocess.TimeoutExpired):
        pass

    # EGL 확인
    try:
        out = subprocess.check_output(
            ["eglinfo"],
            stderr=subprocess.DEVNULL,
            timeout=3,
        ).decode()
        result["has_egl"] = "EGL_VERSION" in out
    except (FileNotFoundError, subprocess.CalledProcessError, subprocess.TimeoutExpired):
        pass

    return result


def is_ssh_session() -> bool:
    """
    Detect if running inside SSH session
    SSH 세션 여부 감지
    """
    return bool(os.environ.get("SSH_CLIENT") or os.environ.get("SSH_TTY"))


def is_virtual_machine() -> bool:
    """
    Detect if running inside a virtual machine (VirtualBox, VMware, etc.)
    가상머신 여부 감지
    """
    try:
        out = subprocess.check_output(
            ["systemd-detect-virt", "--vm"],
            stderr=subprocess.DEVNULL,
            timeout=2,
        ).decode().strip()
        return out not in ("none", "")
    except Exception:
        pass

    # /sys/class/dmi로 폴백
    try:
        dmi = Path("/sys/class/dmi/id/product_name").read_text().lower()
        return any(v in dmi for v in ("virtualbox", "vmware", "kvm", "qemu"))
    except Exception:
        return False


def resolve_sdl_env() -> dict[str, str]:
    """
    Automatically resolve required SDL environment variables
    필요한 SDL 환경변수 자동 결정

    Returns:
        dict: 설정해야 할 환경변수 딕셔너리
    """
    env_patch: dict[str, str] = {}

    display_server = detect_display_server()
    opengl_info = check_opengl_support()
    ssh = is_ssh_session()
    vm = is_virtual_machine()

    print(f"[sdl_env] Display server : {display_server}")
    print(f"[sdl_env] SSH session    : {ssh}")
    print(f"[sdl_env] Virtual machine: {vm}")
    print(f"[sdl_env] OpenGL renderer: {opengl_info['renderer']}")

    # ── DISPLAY 설정 (X11) ──────────────────────────────────────
    if not os.environ.get("DISPLAY"):
        found = find_x11_display()
        if found:
            env_patch["DISPLAY"] = found
            print(f"[sdl_env] DISPLAY 자동 설정: {found}")

    # ── SDL_VIDEODRIVER ──────────────────────────────────────────
    if not os.environ.get("SDL_VIDEODRIVER"):
        if ssh or vm:
            # SSH/VM 환경: offscreen 또는 x11 강제
            env_patch["SDL_VIDEODRIVER"] = "x11"
        elif display_server == "wayland":
            # Wayland + OpenGL 조합은 x11로 강제하는 게 안정적
            env_patch["SDL_VIDEODRIVER"] = "x11"
        elif display_server == "x11":
            env_patch["SDL_VIDEODRIVER"] = "x11"

    # ── EGL 강제 (Wayland 환경 또는 Mesa 환경) ──────────────────
    if display_server == "wayland" or "mesa" in opengl_info["renderer"].lower():
        if not os.environ.get("SDL_VIDEO_X11_FORCE_EGL"):
            env_patch["SDL_VIDEO_X11_FORCE_EGL"] = "1"

    # ── 소프트웨어 렌더링 폴백 ───────────────────────────────────
    # GPU 드라이버 없고 EGL도 없는 경우
    if not opengl_info["has_glx"] and not opengl_info["has_egl"]:
        print("[sdl_env] ⚠️  HW OpenGL 불가 → 소프트웨어 렌더링 폴백")
        env_patch["LIBGL_ALWAYS_SOFTWARE"] = "1"
        env_patch["SDL_RENDER_DRIVER"] = "software"

    return env_patch


def apply_sdl_env(verbose: bool = True) -> dict[str, str]:
    """
    Apply resolved SDL env vars to the current process environment
    해결된 SDL 환경변수를 현재 프로세스에 적용

    Must be called BEFORE importing pygame
    pygame import 전에 반드시 호출해야 함
    """
    patch = resolve_sdl_env()

    for key, val in patch.items():
        os.environ[key] = val
        if verbose:
            print(f"[sdl_env] SET {key}={val}")

    return patch

# ── main.py ──────────────────────────────────────────────────────
# IMPORTANT: apply_sdl_env() must be called before importing pygame
# 중요: pygame import 전에 반드시 apply_sdl_env() 호출

from sdl_env import apply_sdl_env
apply_sdl_env()  # ← 여기서 먼저 설정

import pygame
from pygame.locals import *
from OpenGL.GL import *

def main():
    pygame.init()
    screen = pygame.display.set_mode(
        (1280, 720),
        DOUBLEBUF | OPENGL,
    )
    pygame.display.set_caption("pygame + OpenGL")

    running = True
    while running:
        for event in pygame.event.get():
            if event.type == QUIT:
                running = False

        glClearColor(0.1, 0.1, 0.15, 1.0)
        glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT)

        pygame.display.flip()

    pygame.quit()

if __name__ == "__main__":
    main()

from sdl_env import apply_sdl_env
apply_sdl_env()  # hello_imgui.run() 전에 호출

from imgui_bundle import hello_imgui, imgui
# ...
hello_imgui.run(runner_params)