refactor: client-config-builder package

Author: Niraj-Kamdar

packages/polywrap-client-config-builder/polywrap_client_config_builder/__init__.py

@@ -1,3 +1,3 @@
 """This package contains modules related to client config builder."""
 
-from .client_config_builder import *
+from .polywrap_client_config_builder import *

packages/polywrap-client-config-builder/polywrap_client_config_builder/client_config_builder.py

@@ -0,0 +1,7 @@
+from .base_configure import *
+from .env_configure import *
+from .interface_configure import *
+from .package_configure import *
+from .redirect_configure import *
+from .resolver_configure import *
+from .wrapper_configure import *

packages/polywrap-client-config-builder/polywrap_client_config_builder/configures/__init__.py

@@ -0,0 +1,27 @@
+from ..types import BuilderConfig, ClientConfigBuilder
+
+
+class BaseConfigure(ClientConfigBuilder):
+    """BaseConfigure is the base class for builder configures.
+
+    Attributes:
+        config (BuilderConfig): The internal configuration.
+    """
+
+    config: BuilderConfig
+
+    def add(self, config: BuilderConfig) -> ClientConfigBuilder:
+        """Add the values from the given config to the builder's config."""
+        if config.envs:
+            self.config.envs.update(config.envs)
+        if config.interfaces:
+            self.config.interfaces.update(config.interfaces)
+        if config.redirects:
+            self.config.redirects.update(config.redirects)
+        if config.resolvers:
+            self.config.resolvers.extend(config.resolvers)
+        if config.wrappers:
+            self.config.wrappers.update(config.wrappers)
+        if config.packages:
+            self.config.packages.update(config.packages)
+        return self

packages/polywrap-client-config-builder/polywrap_client_config_builder/configures/base_configure.py

@@ -0,0 +1,56 @@
+from typing import Dict, List, Union
+
+from polywrap_core import Env, Uri
+
+from ..types import ClientConfigBuilder
+
+
+class EnvConfigure(ClientConfigBuilder):
+    """Allows configuring the environment variables."""
+
+    def get_env(self, uri: Uri) -> Union[Env, None]:
+        """Return the env for the given uri."""
+        return self.config.envs.get(uri)
+
+    def get_envs(self) -> Dict[Uri, Env]:
+        """Return the envs from the builder's config."""
+        return self.config.envs
+
+    def set_env(self, uri: Uri, env: Env) -> ClientConfigBuilder:
+        """Set the env by uri in the builder's config, overiding any existing values."""
+        self.config.envs[uri] = env
+        return self
+
+    def set_envs(self, uri_envs: Dict[Uri, Env]) -> ClientConfigBuilder:
+        """Set the envs in the builder's config, overiding any existing values."""
+        self.config.envs.update(uri_envs)
+        return self
+
+    def add_env(self, uri: Uri, env: Env) -> ClientConfigBuilder:
+        """Add an env for the given uri.
+
+        If an Env is already associated with the uri, it is modified.
+        """
+        if self.config.envs.get(uri):
+            for key in self.config.envs[uri]:
+                self.config.envs[uri][key] = env[key]
+        else:
+            self.config.envs[uri] = env
+        return self
+
+    def add_envs(self, uri_envs: Dict[Uri, Env]) -> ClientConfigBuilder:
+        """Add a list of envs to the builder's config."""
+        for uri, env in uri_envs.items():
+            self.add_env(uri, env)
+        return self
+
+    def remove_env(self, uri: Uri) -> ClientConfigBuilder:
+        """Remove the env for the given uri."""
+        self.config.envs.pop(uri, None)
+        return self
+
+    def remove_envs(self, uris: List[Uri]) -> ClientConfigBuilder:
+        """Remove the envs for the given uris."""
+        for uri in uris:
+            self.remove_env(uri)
+        return self

packages/polywrap-client-config-builder/polywrap_client_config_builder/configures/env_configure.py

@@ -0,0 +1,44 @@
+from typing import Dict, List, Union
+
+from polywrap_core import Uri
+
+from ..types import ClientConfigBuilder
+
+
+class InterfaceConfigure(ClientConfigBuilder):
+    """Allows configuring the interface-implementations."""
+
+    def get_interfaces(self) -> Dict[Uri, List[Uri]]:
+        """Return all registered interface and its implementations\
+            from the builder's config."""
+        return self.config.interfaces
+
+    def get_interface_implementations(self, uri: Uri) -> Union[List[Uri], None]:
+        """Return the interface for the given uri."""
+        return self.config.interfaces.get(uri)
+
+    def add_interface_implementations(
+        self, interface_uri: Uri, implementations_uris: List[Uri]
+    ) -> ClientConfigBuilder:
+        """Add a list of implementation URIs for the given interface URI to the builder's config."""
+        if interface_uri in self.config.interfaces.keys():
+            self.config.interfaces[interface_uri].extend(implementations_uris)
+        else:
+            self.config.interfaces[interface_uri] = implementations_uris
+        return self
+
+    def remove_interface_implementations(
+        self, interface_uri: Uri, implementations_uris: List[Uri]
+    ) -> ClientConfigBuilder:
+        """Remove the implementations for the given interface uri."""
+        self.config.interfaces[interface_uri] = [
+            uri
+            for uri in self.config.interfaces[interface_uri]
+            if uri not in implementations_uris
+        ]
+        return self
+
+    def remove_interface(self, interface_uri: Uri) -> ClientConfigBuilder:
+        """Remove the interface for the given uri."""
+        self.config.interfaces.pop(interface_uri, None)
+        return self

packages/polywrap-client-config-builder/polywrap_client_config_builder/configures/interface_configure.py

@@ -0,0 +1,42 @@
+from typing import Dict, List, Union
+
+from polywrap_core import Uri, UriPackageOrWrapper, WrapPackage
+
+from ..types import ClientConfigBuilder
+
+
+class PackageConfigure(ClientConfigBuilder):
+    """Allows configuring the WRAP packages."""
+
+    def get_package(self, uri: Uri) -> Union[WrapPackage[UriPackageOrWrapper], None]:
+        """Return the package for the given uri."""
+        return self.config.packages.get(uri)
+
+    def get_packages(self) -> Dict[Uri, WrapPackage[UriPackageOrWrapper]]:
+        """Return the packages from the builder's config."""
+        return self.config.packages
+
+    def set_package(
+        self, uri: Uri, package: WrapPackage[UriPackageOrWrapper]
+    ) -> ClientConfigBuilder:
+        """Set the package by uri in the builder's config, overiding any existing values."""
+        self.config.packages[uri] = package
+        return self
+
+    def set_packages(
+        self, uri_packages: Dict[Uri, WrapPackage[UriPackageOrWrapper]]
+    ) -> ClientConfigBuilder:
+        """Set the packages in the builder's config, overiding any existing values."""
+        self.config.packages.update(uri_packages)
+        return self
+
+    def remove_package(self, uri: Uri) -> ClientConfigBuilder:
+        """Remove the package for the given uri."""
+        self.config.packages.pop(uri, None)
+        return self
+
+    def remove_packages(self, uris: List[Uri]) -> ClientConfigBuilder:
+        """Remove the packages for the given uris."""
+        for uri in uris:
+            self.remove_package(uri)
+        return self