vention-communication 0.2.2__py3-none-any.whl → 0.3.0__py3-none-any.whl

communication/app.py

@@ -7,7 +7,7 @@ from .connect_router import ConnectRouter
 from .decorators import collect_bundle, set_global_app
 from .codegen import generate_proto, sanitize_service_name
 from .entries import RpcBundle
-from .registry import RpcRegistry
+from .rpc_registry import RpcRegistry
 
 
 class VentionApp(FastAPI):

communication/rpc_registry.py

@@ -0,0 +1,90 @@
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from typing import List, Optional, Set, Type
+
+from pydantic import BaseModel
+
+from .entries import RpcBundle
+from .typing_utils import extract_nested_models, is_pydantic_model, apply_aliases
+
+
+@dataclass
+class RpcRegistry:
+    """
+    Central registry that collects RPC bundles, applies model normalization
+    (Pydantic field aliasing), and exposes a unified bundle.
+
+    - Plugins and decorators add RpcBundle instances.
+    - Registry merges them.
+    - Registry applies camelCase aliases to all Pydantic models exactly once.
+    """
+
+    service_name: str = "VentionApp"
+    _bundles: List[RpcBundle] = field(default_factory=list)
+    _models_normalized: bool = False
+
+    # ------------- Bundle registration -------------
+
+    def add_bundle(self, bundle: RpcBundle) -> None:
+        """Register a bundle for inclusion in the unified RPC view."""
+        self._bundles.append(bundle)
+
+    @property
+    def bundle(self) -> RpcBundle:
+        """Return a merged RpcBundle (does not mutate stored bundles)."""
+        merged = RpcBundle()
+        for bundle in self._bundles:
+            merged.extend(bundle)
+        return merged
+
+    # ------------- Model normalization / aliasing -------------
+    def _normalize_model(
+        self, model: Optional[Type[BaseModel]], seen: Set[Type[BaseModel]]
+    ) -> None:
+        """Recursively normalize a model and all its nested models."""
+        if model is None:
+            return
+        if not is_pydantic_model(model):
+            return
+        if model in seen:
+            return
+
+        seen.add(model)
+        apply_aliases(model)
+
+        if hasattr(model, "model_fields"):
+            for _, field_info in model.model_fields.items():
+                nested_models = extract_nested_models(field_info.annotation)
+                for nested_model in nested_models:
+                    self._normalize_model(nested_model, seen)
+
+    def normalize_models_and_apply_aliases(self) -> None:
+        """
+        Walk all RPCs in all bundles and apply camelCase JSON aliases
+        to every Pydantic model exactly once, including nested models.
+        """
+        if self._models_normalized:
+            return
+
+        seen: Set[Type[BaseModel]] = set()
+
+        for bundle in self._bundles:
+            for action in bundle.actions:
+                self._normalize_model(action.input_type, seen)
+                self._normalize_model(action.output_type, seen)
+            for stream in bundle.streams:
+                self._normalize_model(stream.payload_type, seen)
+
+        self._models_normalized = True
+
+    # ------------- Unified, normalized view -------------
+
+    def get_unified_bundle(self) -> RpcBundle:
+        """
+        Get the fully merged, normalized RPC bundle.
+
+        This will apply aliasing exactly once and then return a merged RpcBundle.
+        """
+        self.normalize_models_and_apply_aliases()
+        return self.bundle

communication/typing_utils.py

@@ -1,6 +1,17 @@
 from __future__ import annotations
 import inspect
-from typing import Any, Callable, Optional, Type, get_type_hints, cast
+from typing import (
+    Any,
+    Callable,
+    List,
+    Optional,
+    Type,
+    Union,
+    get_args,
+    get_origin,
+    get_type_hints,
+    cast,
+)
 
 from pydantic import BaseModel, ConfigDict
 
@@ -122,3 +133,40 @@ def apply_aliases(model_cls: Type[BaseModel]) -> None:
 
     # Force rebuild
     model_cls.model_rebuild(force=True)
+
+
+def unwrap_optional(field_type: Any) -> Any:
+    """Unwrap Optional[Type] or Union[Type, None] to get the non-None type."""
+    origin = get_origin(field_type)
+    if origin is not Union:
+        return field_type
+
+    args = get_args(field_type)
+    non_none_args = [arg for arg in args if arg is not type(None)]
+    if len(non_none_args) == 1:
+        return non_none_args[0]
+    return field_type
+
+
+def unwrap_list(field_type: Any) -> Any:
+    """Unwrap List[Type] to get the inner type, handling Optional inside List."""
+    origin = get_origin(field_type)
+    if origin not in (list, List):
+        return field_type
+
+    args = get_args(field_type)
+    if not args:
+        return field_type
+
+    inner_type = args[0]
+    return unwrap_optional(inner_type)
+
+
+def extract_nested_models(field_type: Any) -> List[Type[BaseModel]]:
+    """Extract Pydantic models from a field type, handling Optional, List, etc."""
+    field_type = unwrap_optional(field_type)
+    field_type = unwrap_list(field_type)
+
+    if is_pydantic_model(field_type):
+        return [field_type]
+    return []

{vention_communication-0.2.2.dist-info → vention_communication-0.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vention-communication
-Version: 0.2.2
+Version: 0.3.0
 Summary: A framework for communication between machine apps and other services.
 License: Proprietary
 Author: VentionCo

{vention_communication-0.2.2.dist-info → vention_communication-0.3.0.dist-info}/RECORD

@@ -1,12 +1,13 @@
 communication/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-communication/app.py,sha256=FPfnjIGMDClTysq6DvHoMudY_s2XnqmNlZCV93zYFgA,3419
+communication/app.py,sha256=dxbm7oG8-uQSu_dY4S8exHsBXND1N1nC2GdsMthqZE0,3423
 communication/codegen.py,sha256=_1LYma-MXedwRHn_P3CqPs1xIMEZrZgiUAsGSj9yff4,7089
 communication/connect_router.py,sha256=tjNl9dRukjLecZH1y-GYIQXH-0xzz9JESbikSH5XhFo,10527
 communication/decorators.py,sha256=3pVlXUSX4KXSKlweskF0RfD8pST2zisTuFGJQHHIvcI,3419
 communication/entries.py,sha256=vdZc8GAQztRWEiav6R2wM4l35GE-EiEdRH0ZJR4GShM,1065
 communication/errors.py,sha256=hdJBB9jPJNWx8hbxIxwLBNKt2JVpmhZ1YF8q9VKk-dI,1773
 communication/registry.py,sha256=acbhwU0z1iRHqzOahXG47GfJS5VQHjf69UiruoXrr7g,4517
-communication/typing_utils.py,sha256=VHyRyclAcADKXETCklgHUM-k0mehnW8p5v_FmIMtRvk,3393
-vention_communication-0.2.2.dist-info/METADATA,sha256=t0TQRQOIApZjxQ7GTz_k1tPeC_ctuC6KRH0QTfzXnLs,11255
-vention_communication-0.2.2.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-vention_communication-0.2.2.dist-info/RECORD,,
+communication/rpc_registry.py,sha256=90r4YYHKveVGBO-pOD91qAs9GSB3651WPclC6oRnrZo,2986
+communication/typing_utils.py,sha256=6S6LvtjFBZKo-gWPc8fbh4F1rlPWFgrH_mX8esZbzWM,4559
+vention_communication-0.3.0.dist-info/METADATA,sha256=e6h67l5DGs9LSAL_eBan_spH60FDQfBVcmeyoNWrlvg,11255
+vention_communication-0.3.0.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+vention_communication-0.3.0.dist-info/RECORD,,