"""OpenBSD: hw.* sysctls for the hardware, uname for release and kernel."""

from __future__ import annotations

from napalm_device_types import FingerprintRule

from napalm_bsd import parse
from napalm_bsd.base import BsdDriver


class OpenBSDDriver(BsdDriver):
    """NAPALM driver for OpenBSD hosts, over SSH."""

    TYPE_LABEL = "OpenBSD"
    VENDOR = "OpenBSD"
    DRIVER_NAME = "openbsd"
    # OpenBSD's SSH banner names no OS; its sysDescr does ("OpenBSD host 7.9 GENERIC.MP#449").
    SNMP_FINGERPRINT = [FingerprintRule("openbsd", weight=5.0)]

    OS_VERSION_COMMAND = "uname -sr"
    # The kernel's build, e.g. GENERIC.MP#449; the release is in uname -r.
    KERNEL_COMMAND = "uname -v"
    # A sysctl node the machine lacks fails on its own and prints nothing.
    PLATFORM_COMMAND = (
        "for k in vendor product serialno; do "
        'printf "%s=%s\\n" "$k" "$(sysctl -n hw.$k 2>/dev/null)"; done'
    )

    # fstat shows a user only their own processes; netstat shows every socket, unnamed.
    LISTENING_COMMAND = "fstat -n"
    LISTENING_FALLBACK_COMMAND = "netstat -an -f inet; netstat -an -f inet6"

    def _parse_listening(self, output: str, *, attributed: bool) -> list[dict]:
        return parse.fstat_sockets(output) if attributed else parse.netstat_listening(output)

    def _hardware(self) -> tuple[str, str, str]:
        hw = self._platform()
        return hw.get("vendor", ""), hw.get("product", ""), hw.get("serialno", "")