"""Connectors."""

from functools import cache

from content_utils.connectors.graphql import LambdaGraphQLConnector
from content_utils.connectors.opensearch import LambdaOpensearchConnector

from src import config


@cache
def get_os_connector():
    """Get OpenSearch connector."""
    os_opts = {}
    if config.OPENSEARCH_PORT:
        os_opts.update(port=int(config.OPENSEARCH_PORT))

    if config.OPENSEARCH_INDEX_ALIAS:
        os_opts.update(alias_name=config.OPENSEARCH_INDEX_ALIAS)

    return LambdaOpensearchConnector(
        config.OPENSEARCH_ENDPOINT,
        config.app_logger,
        **os_opts,
    )


@cache
def get_graphql_connector():
    """Get GraphQL connector."""
    return LambdaGraphQLConnector(
        config.GRAPHQL_GATEWAY_URL,
        config.APPLICATION_NAME,
        config.app_logger,
        raise_on_error=True
    )