from typing import TYPE_CHECKING
if TYPE_CHECKING:
    from namara_python.client import Client

from namara_python.utils import gen_request_decorator

from namara_python.rpc.organizations.organizations_pb2_twirp import OrganizationsServiceClient
from namara_python.rpc.catalog.catalog_pb2_twirp import CatalogServiceClient
from namara_python.rpc.category.category_pb2_twirp import CategoryServiceClient
from namara_python.rpc.grants.grants_pb2_twirp import GrantsServiceClient
from namara_python.rpc.sources.sources_pb2_twirp import SourcesServiceClient
from namara_python.rpc.references.references_pb2_twirp import ReferencesServiceClient


class Organization():
    def __init__(self, client: 'Client') -> None:
        request_decorator = gen_request_decorator(client.token)

        self.rpc_client = OrganizationsServiceClient(client.base_api_url, request_decorator)


class Catalog():
    def __init__(self, client: 'Client') -> None:
        request_decorator = gen_request_decorator(client.token)

        self.rpc_client = CatalogServiceClient(client.base_api_url, request_decorator)


class Category():
    def __init__(self, client: 'Client') -> None:
        request_decorator = gen_request_decorator(client.token)

        self.rpc_client = CategoryServiceClient(client.base_api_url, request_decorator)


class Grant():
    def __init__(self, client: 'Client') -> None:
        request_decorator = gen_request_decorator(client.token)

        self.rpc_client = GrantsServiceClient(client.base_api_url, request_decorator)


class Source():
    def __init__(self, client: 'Client') -> None:
        request_decorator = gen_request_decorator(client.token)

        self.rpc_client = SourcesServiceClient(client.base_api_url, request_decorator)


class Reference():
    def __init__(self, client: 'Client') -> None:
        request_decorator = gen_request_decorator(client.token)

        self.rpc_client = ReferencesServiceClient(client.base_api_url, request_decorator)