Source code for polywrap_client_config_builder.configures.resolver_configure

"""This module contains the resolver configure class for the client config builder."""
from typing import List, cast

from polywrap_core import UriResolver

from ..types import BuilderConfig, ClientConfigBuilder


class ResolverConfigure:
    """Allows configuring the URI resolvers."""

    config: BuilderConfig

    def get_resolvers(self) -> List[UriResolver]:
        """Return the resolvers from the builder's config."""
        return self.config.resolvers

    def add_resolver(self, resolver: UriResolver) -> ClientConfigBuilder:
        """Add a resolver to the builder's config."""
        self.config.resolvers.append(resolver)
        return cast(ClientConfigBuilder, self)

    def add_resolvers(self, resolvers_list: List[UriResolver]) -> ClientConfigBuilder:
        """Add a list of resolvers to the builder's config."""
        for resolver in resolvers_list:
            self.add_resolver(resolver)
        return cast(ClientConfigBuilder, self)