from __future__ import annotations

from openstatus._gen.openstatus.maintenance.v1.service_pb2 import (
    CreateMaintenanceRequest,
    CreateMaintenanceResponse,
    DeleteMaintenanceRequest,
    DeleteMaintenanceResponse,
    GetMaintenanceRequest,
    GetMaintenanceResponse,
    ListMaintenancesRequest,
    ListMaintenancesResponse,
    UpdateMaintenanceRequest,
    UpdateMaintenanceResponse,
)
from openstatus._transport import _AsyncJsonTransport, _JsonTransport

__all__ = ["AsyncMaintenanceServiceClient", "MaintenanceServiceClient"]

_FQN = "openstatus.maintenance.v1.MaintenanceService"


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

    def create_maintenance(
        self,
        request: CreateMaintenanceRequest,
        *,
        headers: dict[str, str] | None = None,
    ) -> CreateMaintenanceResponse:
        return self._t.call(_FQN, "CreateMaintenance", request, CreateMaintenanceResponse, headers)

    def get_maintenance(
        self,
        request: GetMaintenanceRequest,
        *,
        headers: dict[str, str] | None = None,
    ) -> GetMaintenanceResponse:
        return self._t.call(_FQN, "GetMaintenance", request, GetMaintenanceResponse, headers)

    def list_maintenances(
        self,
        request: ListMaintenancesRequest,
        *,
        headers: dict[str, str] | None = None,
    ) -> ListMaintenancesResponse:
        return self._t.call(_FQN, "ListMaintenances", request, ListMaintenancesResponse, headers)

    def update_maintenance(
        self,
        request: UpdateMaintenanceRequest,
        *,
        headers: dict[str, str] | None = None,
    ) -> UpdateMaintenanceResponse:
        return self._t.call(_FQN, "UpdateMaintenance", request, UpdateMaintenanceResponse, headers)

    def delete_maintenance(
        self,
        request: DeleteMaintenanceRequest,
        *,
        headers: dict[str, str] | None = None,
    ) -> DeleteMaintenanceResponse:
        return self._t.call(_FQN, "DeleteMaintenance", request, DeleteMaintenanceResponse, headers)


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

    async def create_maintenance(
        self,
        request: CreateMaintenanceRequest,
        *,
        headers: dict[str, str] | None = None,
    ) -> CreateMaintenanceResponse:
        return await self._t.call(
            _FQN, "CreateMaintenance", request, CreateMaintenanceResponse, headers
        )

    async def get_maintenance(
        self,
        request: GetMaintenanceRequest,
        *,
        headers: dict[str, str] | None = None,
    ) -> GetMaintenanceResponse:
        return await self._t.call(_FQN, "GetMaintenance", request, GetMaintenanceResponse, headers)

    async def list_maintenances(
        self,
        request: ListMaintenancesRequest,
        *,
        headers: dict[str, str] | None = None,
    ) -> ListMaintenancesResponse:
        return await self._t.call(
            _FQN, "ListMaintenances", request, ListMaintenancesResponse, headers
        )

    async def update_maintenance(
        self,
        request: UpdateMaintenanceRequest,
        *,
        headers: dict[str, str] | None = None,
    ) -> UpdateMaintenanceResponse:
        return await self._t.call(
            _FQN, "UpdateMaintenance", request, UpdateMaintenanceResponse, headers
        )

    async def delete_maintenance(
        self,
        request: DeleteMaintenanceRequest,
        *,
        headers: dict[str, str] | None = None,
    ) -> DeleteMaintenanceResponse:
        return await self._t.call(
            _FQN, "DeleteMaintenance", request, DeleteMaintenanceResponse, headers
        )