feat(plugins): plugin package basic implementation works

Author: cbrzn

packages/polywrap-core/polywrap_core/types/__init__.py

@@ -11,4 +11,5 @@
 from .uri_resolver_handler import *
 from .uri_wrapper import *
 from .wasm_package import *
+from .wrap_package import *
 from .wrapper import *

packages/polywrap-plugin/polywrap_plugin/package.py

@@ -1,9 +1,13 @@
-from typing import Generic
+from typing import Generic, Optional, cast
 
+from polywrap_core import IWrapPackage, Wrapper, GetManifestOptions
 from polywrap_manifest import AnyWrapManifest
-from polywrap_plugin import PluginModule, TConfig, TResult
+from polywrap_result import Ok, Result
 
-class PluginPackage(Generic[TConfig, TResult]):
+from .module import PluginModule, TConfig, TResult
+from .wrapper import PluginWrapper
+
+class PluginPackage(Generic[TConfig, TResult], IWrapPackage):
     module: PluginModule[TConfig, TResult]
     manifest: AnyWrapManifest
 
@@ -15,3 +19,8 @@ def __init__(
         self.module = module
         self.manifest = manifest
 
+    async def create_wrapper(self) -> Result[Wrapper]:
+        return Ok(PluginWrapper(module=self.module, manifest=self.manifest))
+
+    async def get_manifest(self, options: Optional[GetManifestOptions] = None) -> Result[AnyWrapManifest]:
+        return await super().get_manifest(options)

packages/polywrap-plugin/polywrap_plugin/wrapper.py

@@ -11,11 +11,10 @@
 from polywrap_msgpack import msgpack_decode
 from polywrap_result import Err, Ok, Result
 
-from polywrap_plugin import PluginModule, TConfig, TResult
+from .module import PluginModule, TConfig, TResult
 
 class PluginWrapper(Wrapper, Generic[TConfig, TResult]):
     module: PluginModule[TConfig, TResult]
-    manifest: AnyWrapManifest
 
     def __init__(self, module: PluginModule[TConfig, TResult], manifest: AnyWrapManifest) -> None:
         self.module = module

packages/polywrap-plugin/tests/conftest.py

@@ -0,0 +1,17 @@
+from pytest import fixture
+from typing import Any, Dict
+
+from polywrap_plugin import PluginModule
+from polywrap_core import Invoker
+
+@fixture
+def get_greeting_module():
+    class GreetingModule(PluginModule[Any, str]):
+        def __init__(self, config: Any):
+            super().__init__(config)
+
+        def greeting(self, args: Dict[str, Any], client: Invoker):
+            return f"Greetings from: {args['name']}"
+
+    instance = GreetingModule({})
+    return instance

packages/polywrap-plugin/tests/test_plugin_module.py

@@ -1,24 +1,15 @@
-from typing import Any, Dict
+from typing import Any
 
 import pytest
 from polywrap_client import PolywrapClient
-from polywrap_core import Invoker
 from polywrap_result import Ok
 
 from polywrap_plugin import PluginModule
 
-
-class GreetingModule(PluginModule[Any, str]):
-    def __init__(self, config: Any):
-        super().__init__(config)
-
-    def greeting(self, args: Dict[str, Any], client: Invoker):
-        return f"Greetings from: {args['name']}"
-
 @pytest.mark.asyncio
-async def test_plugin_module():
-    plugin = GreetingModule({})
+async def test_plugin_module(get_greeting_module: PluginModule[Any, Any]):
+    module = get_greeting_module
 
     client = PolywrapClient()
-    result = await plugin._wrap_invoke("greeting", { "name": "Joe" }, client) # type: ignore
+    result = await module._wrap_invoke("greeting", { "name": "Joe" }, client) # type: ignore
     assert result, Ok("Greetings from: Joe")

packages/polywrap-plugin/tests/test_plugin_package.py

@@ -0,0 +1,26 @@
+from typing import Any, cast
+
+import pytest
+from polywrap_core import InvokeOptions, Uri, AnyWrapManifest
+from polywrap_plugin import PluginPackage, PluginModule
+from polywrap_client import PolywrapClient
+from polywrap_result import Ok
+
+@pytest.mark.asyncio
+async def test_plugin_package_invoke(get_greeting_module: PluginModule[Any, Any]):
+    module = get_greeting_module
+    manifest = cast(AnyWrapManifest, {})
+    plugin_package = PluginPackage(module, manifest)
+    wrapper = (await plugin_package.create_wrapper()).unwrap()
+    args = {
+        "name": "Joe"
+    }
+    options = InvokeOptions(
+        uri=Uri("ens/greeting.eth"),
+        method="greeting",
+        args=args
+    )
+
+    client = PolywrapClient()
+    result = await wrapper.invoke(options, client)
+    assert result, Ok("Greetings from: Joe")

packages/polywrap-plugin/tests/test_plugin_wrapper.py

@@ -1,22 +1,19 @@
-from typing import cast
+from typing import cast, Callable, Any
 
 import pytest
 from polywrap_core import InvokeOptions, Uri
 from polywrap_manifest import AnyWrapManifest
-from polywrap_plugin import PluginWrapper
 from polywrap_client import PolywrapClient
 from polywrap_result import Ok
 
-from test_plugin_module import GreetingModule 
+from polywrap_plugin import PluginWrapper, PluginModule
 
 @pytest.mark.asyncio
-async def test_plugin_wrapper_invoke():
-    module = GreetingModule({})
+async def test_plugin_wrapper_invoke(get_greeting_module: PluginModule[Any, Any]):
+    module = get_greeting_module
     manifest = cast(AnyWrapManifest, {})
 
     wrapper = PluginWrapper(module, manifest)
-
-
     args = {
         "name": "Joe"
     }