"""Vendor SoC recognizers derived from host facts.

Recognition uses OS-reported CPU or PCI identity without environment-variable input.
"""

from __future__ import annotations

from hermes_platform.host import facts

# Match the CPU string because the chassis vendor does not identify the SoC.
_NVIDIA_SOC_VENDOR = "NVIDIA"
_NVIDIA_SOC_MODEL_MARKERS = ("N1X", "SPARK")


def is_nvidia_n1x_pci_id(pci_id: int) -> bool:
    """Recognize N1X; NVML packs device above the 16-bit vendor ID."""
    return pci_id & 0xFFFF == 0x10DE and 0x2E00 <= pci_id >> 16 <= 0x2E3F


def looks_like_nvidia_arm_soc(*, native_arch: str, cpu_model: str, cpu_vendor: str) -> bool:
    """Return whether the supplied facts identify an NVIDIA ARM SoC."""
    if native_arch != "arm64":
        return False
    model = cpu_model.upper()
    vendor_match = _NVIDIA_SOC_VENDOR in model or cpu_vendor.strip().upper() == _NVIDIA_SOC_VENDOR
    return vendor_match and any(marker in model for marker in _NVIDIA_SOC_MODEL_MARKERS)


def is_nvidia_arm_soc() -> bool:
    """Return whether this host has an NVIDIA ARM SoC."""
    return looks_like_nvidia_arm_soc(
        native_arch=facts.native_arch(), cpu_model=facts.cpu_model(), cpu_vendor=facts.cpu_vendor())