Feat/viewer app controller

requirements.txt

@@ -63,4 +63,3 @@ wslink==1.12.4
 yarl>=1
     # via aiohttp
 
-opengeodeweb-microservice==1.*,>=1.2.1

src/opengeodeweb_viewer/rpc/mesh/points/points_protocols.py

@@ -1,8 +1,9 @@
 # Standard library imports
+from typing import Any
 import os
 
 # Third party imports
-from wslink import register as exportRpc  # type: ignore
+from opengeodeweb_viewer.utils_functions import exportRpc
 from opengeodeweb_microservice.schemas import get_schemas_dict
 
 # Local application imports
@@ -24,14 +25,15 @@ def __init__(self) -> None:
         super().__init__()
 
     @exportRpc(mesh_points_prefix + mesh_points_schemas_dict["visibility"]["rpc"])
-    def setMeshPointsVisibility(self, rpc_params: RpcParams) -> None:
+    def setMeshPointsVisibility(self, rpc_params: RpcParams) -> dict[str, Any]:
         validate_schema(
             rpc_params,
             self.mesh_points_schemas_dict["visibility"],
             self.mesh_points_prefix,
         )
         params = schemas.Visibility.from_dict(rpc_params)
         self.SetPointsVisibility(params.id, params.visibility)
+        return params.to_dict()
 
     @exportRpc(mesh_points_prefix + mesh_points_schemas_dict["color"]["rpc"])
     def setMeshPointsColor(self, rpc_params: RpcParams) -> None:

src/opengeodeweb_viewer/rpc/viewer/viewer_protocols.py

@@ -4,7 +4,6 @@
 from typing import cast, Any
 
 # Third party imports
-from wslink import register as exportRpc  # type: ignore
 from vtkmodules.vtkIOImage import vtkPNGWriter, vtkJPEGWriter
 from vtkmodules.vtkRenderingAnnotation import vtkCubeAxesActor, vtkAxesActor
 from vtkmodules.vtkRenderingCore import (
@@ -27,6 +26,7 @@
 
 # Local application imports
 from opengeodeweb_viewer.utils_functions import (
+    exportRpc,
     validate_schema,
     RpcParams,
 )

src/opengeodeweb_viewer/utils_functions.py

@@ -2,13 +2,34 @@
 
 # Third party imports
 import fastjsonschema  # type: ignore
+import functools
 import math
+from typing import Any, Callable, TypeVar
+from wslink import register  # type: ignore
 from vtkmodules.vtkRenderingCore import vtkColorTransferFunction
 
 from opengeodeweb_microservice.schemas import SchemaDict
 
 type RpcParams = dict[str, str]
 
+R = TypeVar("R")
+
+
+def exportRpc(rpc_id: str) -> Callable[[Callable[..., R]], Callable[..., R]]:
+    def decorator(function: Callable[..., R]) -> Callable[..., R]:
+        def wrapper(self: Any, *args: Any, **kwargs: Any) -> R:
+            do_stream = bool(kwargs.pop("stream", False))
+            print("do_stream", do_stream, flush=True)
+            result = function(self, *args, **kwargs)
+            if do_stream:
+                rpc_params = args[0] if args else None
+                self.publish(rpc_id, rpc_params)
+            return result
+
+        return register(rpc_id)(wrapper)  # type: ignore[no-any-return]
+
+    return decorator
+
 
 def validate_schema(
     rpc_params: RpcParams, schema: SchemaDict, prefix: str = ""