from __future__ import annotations

from openstatus._gen.openstatus.health.v1.health_pb2 import (
    CheckRequest,
    CheckResponse,
)
from openstatus._transport import _AsyncJsonTransport, _JsonTransport

__all__ = ["AsyncHealthServiceClient", "HealthServiceClient"]

_FQN = "openstatus.health.v1.HealthService"


class HealthServiceClient:
    def __init__(self, transport: _JsonTransport) -> None:
        self._t = transport

    def check(
        self,
        request: CheckRequest,
        *,
        headers: dict[str, str] | None = None,
    ) -> CheckResponse:
        return self._t.call(_FQN, "Check", request, CheckResponse, headers)


class AsyncHealthServiceClient:
    def __init__(self, transport: _AsyncJsonTransport) -> None:
        self._t = transport

    async def check(
        self,
        request: CheckRequest,
        *,
        headers: dict[str, str] | None = None,
    ) -> CheckResponse:
        return await self._t.call(_FQN, "Check", request, CheckResponse, headers)