feat: add support for interface

Author: Niraj-Kamdar

packages/polywrap-client/polywrap_client/client.py

@@ -20,6 +20,7 @@
     UriPackage,
     UriPackageOrWrapper,
     UriResolutionContext,
+    InterfaceImplementations,
     Wrapper,
 )
 from polywrap_msgpack import msgpack_decode, msgpack_encode
@@ -53,6 +54,17 @@ def get_uri_resolver(
     def get_envs(self, options: Optional[GetEnvsOptions] = None) -> List[Env]:
         return self._config.envs
 
+    def get_interfaces(self) -> List[InterfaceImplementations]:
+        return self._config.interfaces
+
+    def get_implementations(self, uri: Uri) -> Result[List[Uri], Exception]:
+        if interface_implementations := next(
+            filter(lambda x: x.interface == uri, self._config.interfaces), None
+        ):
+            return Ok(interface_implementations.implementations)
+        else:
+            return Err(ValueError(f"Unable to find implementations for uri: {uri}"))
+
     def get_env_by_uri(
         self, uri: Uri, options: Optional[GetEnvsOptions] = None
     ) -> Union[Env, None]:
@@ -62,7 +74,9 @@ async def get_file(self, uri: Uri, options: GetFileOptions) -> Union[bytes, str]
         loaded_wrapper = (await self.load_wrapper(uri)).unwrap()
         return await loaded_wrapper.get_file(options)
 
-    async def get_manifest(self, uri: Uri, options: Optional[GetManifestOptions] = None) -> AnyWrapManifest:
+    async def get_manifest(
+        self, uri: Uri, options: Optional[GetManifestOptions] = None
+    ) -> AnyWrapManifest:
         loaded_wrapper = (await self.load_wrapper(uri)).unwrap()
         return loaded_wrapper.get_manifest()
 
@@ -124,4 +138,5 @@ async def invoke(self, options: InvokerOptions) -> InvokeResult:
                 return result
 
         except Exception as e:
-            return InvokeResult(result=None, error=e)
+            raise e
+            # return InvokeResult(result=None, error=e)

packages/polywrap-client/tests/cases/simple-interface/implementation/wrap.info

@@ -1,7 +1,7 @@
 from pathlib import Path
 
 from polywrap_client import PolywrapClient
-from polywrap_core import Uri, InvokerOptions
+from polywrap_core import Uri, InvokerOptions, InterfaceImplementations
 from polywrap_uri_resolvers import BaseUriResolver, SimpleFileReader
 
 from polywrap_client.client import PolywrapClientConfig
@@ -31,17 +31,43 @@ async def test_subinvoke():
         },
     )
 
-    client = PolywrapClient(config=PolywrapClientConfig(
-        envs=[], resolver=uri_resolver
-    ))
+    client = PolywrapClient(config=PolywrapClientConfig(envs=[], resolver=uri_resolver))
     uri = Uri(
         f'fs/{Path(__file__).parent.joinpath("cases", "simple-subinvoke", "invoke").absolute()}'
     )
-    args = b'\x82\xa1a\x01\xa1b\x02'
-    options = InvokerOptions(
-        uri=uri, method="add", args=args, encode_result=False
-    )
+    args = {"a": 1, "b": 2}
+    options = InvokerOptions(uri=uri, method="add", args=args, encode_result=False)
     result = await client.invoke(options)
 
     assert result.result == "1 + 2 = 3"
 
+
+async def test_interface_implementation():
+    uri_resolver = BaseUriResolver(
+        file_reader=SimpleFileReader(),
+        redirects={},
+    )
+
+    impl_uri = Uri(
+        f'fs/{Path(__file__).parent.joinpath("cases", "simple-interface", "implementation").absolute()}'
+    )
+
+    client = PolywrapClient(
+        config=PolywrapClientConfig(
+            envs=[],
+            resolver=uri_resolver,
+            interfaces=[
+                InterfaceImplementations(
+                    interface=Uri("ens/interface.eth"), implementations=[impl_uri]
+                )
+            ],
+        )
+    )
+    uri = Uri(
+        f'fs/{Path(__file__).parent.joinpath("cases", "simple-interface", "wrapper").absolute()}'
+    )
+    args = {"arg": {"str": "hello", "uint8": 2}}
+    options = InvokerOptions(uri=uri, method="moduleMethod", args=args, encode_result=False)
+    result = await client.invoke(options)
+
+    assert result.result == {"str": "hello", "uint8": 2}

packages/polywrap-client/tests/cases/simple-interface/implementation/wrap.wasm

@@ -6,6 +6,7 @@
 
 from polywrap_manifest import DeserializeManifestOptions, AnyWrapManifest
 
+from .interface_implementation import InterfaceImplementations
 from .env import Env
 from .invoke import Invoker
 from .uri import Uri
@@ -16,6 +17,7 @@
 @dataclass(slots=True, kw_only=True)
 class ClientConfig:
     envs: List[Env] = field(default_factory=list)
+    interfaces: List[InterfaceImplementations] = field(default_factory=list)
     resolver: IUriResolver
 
 
@@ -41,6 +43,10 @@ class GetManifestOptions(DeserializeManifestOptions):
 
 
 class Client(Invoker, UriResolverHandler):
+    @abstractmethod
+    def get_interfaces(self) -> List[InterfaceImplementations]:
+        pass
+
     @abstractmethod
     def get_envs(self, options: Optional[GetEnvsOptions] = None) -> List[Env]:
         pass

packages/polywrap-client/tests/cases/simple-interface/interface/wrap.info

@@ -0,0 +1,9 @@
+from dataclasses import dataclass
+from typing import List
+
+from .uri import Uri
+
+@dataclass(slots=True, kw_only=True)
+class InterfaceImplementations:
+    interface: Uri
+    implementations: List[Uri]

packages/polywrap-client/tests/cases/simple-interface/wrapper/wrap.info

@@ -2,7 +2,9 @@
 
 from abc import ABC, abstractmethod
 from dataclasses import dataclass, field
-from typing import Any, Dict, Optional, Union
+from typing import Any, Dict, List, Optional, Union
+
+from result import Result
 
 from .uri import Uri
 from .uri_resolution_context import IUriResolutionContext
@@ -57,6 +59,10 @@ class Invoker(ABC):
     async def invoke(self, options: InvokerOptions) -> InvokeResult:
         pass
 
+    @abstractmethod
+    def get_implementations(self, uri: Uri) -> Result[List[Uri], Exception]:
+        pass
+
 
 class Invocable(ABC):
     @abstractmethod