"""Surface drivers for UI Inspector (windows / browser / mobile stub)."""

from __future__ import annotations

from .base import DumpResult, SurfaceDriver
from .browser_pw import BrowserPlaywrightDriver
from .mobile import MobileDriver
from .windows_uia import WindowsUiaDriver
import target as _target


def get_driver(surface: str) -> SurfaceDriver:
    if surface == _target.SURFACE_WINDOWS:
        return WindowsUiaDriver()
    if surface == _target.SURFACE_BROWSER:
        return BrowserPlaywrightDriver()
    if surface in {_target.SURFACE_ANDROID, _target.SURFACE_IOS}:
        return MobileDriver(surface=surface)
    raise RuntimeError(f"ERROR:SURFACE_UNSUPPORTED unknown surface={surface!r}")


__all__ = [
    "DumpResult",
    "SurfaceDriver",
    "WindowsUiaDriver",
    "BrowserPlaywrightDriver",
    "MobileDriver",
    "get_driver",
]