# Copyright 2026 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import typing

_METHOD_REGISTRY: dict[int, tuple[typing.Any | None, typing.Any | None]] = {}


def register_method(
    ordinal: int,
    request_cls: typing.Any | None,
    response_cls: typing.Any | None,
) -> None:
    _METHOD_REGISTRY[ordinal] = (request_cls, response_cls)


def get_registered_method(
    ordinal: int,
) -> tuple[typing.Any | None, typing.Any | None] | None:
    return _METHOD_REGISTRY.get(ordinal)


def get_method_ordinal(class_name: str) -> int | None:
    for ord_val, (req_cls, resp_cls) in _METHOD_REGISTRY.items():
        if (req_cls and req_cls.__name__ == class_name) or (
            resp_cls and resp_cls.__name__ == class_name
        ):
            return ord_val
    return None