jararaca 0.3.12a13__py3-none-any.whl → 0.3.12a15__py3-none-any.whl

jararaca/tools/typescript/decorators.py

@@ -1,5 +1,7 @@
 from typing import Any, Callable, TypeVar, cast
 
+from pydantic import BaseModel
+
 DECORATED_FUNC = TypeVar("DECORATED_FUNC", bound=Callable[..., Any])
 
 
@@ -60,3 +62,34 @@ class MutationEndpoint:
         Check if the function is marked as a mutation endpoint.
         """
         return getattr(func, MutationEndpoint.METADATA_KEY, False)
+
+
+BASEMODEL_T = TypeVar("BASEMODEL_T", bound=BaseModel)
+
+
+class SplitInputOutput:
+    """
+    Decorator to mark a Pydantic model to generate separate Input and Output TypeScript interfaces.
+
+    Input interface: Used for API inputs (mutations/queries), handles optional fields with defaults
+    Output interface: Used for API outputs, represents the complete object structure
+    """
+
+    METADATA_KEY = "__jararaca_split_input_output__"
+
+    def __init__(self) -> None:
+        pass
+
+    def __call__(self, cls: type[BASEMODEL_T]) -> type[BASEMODEL_T]:
+        """
+        Decorate the Pydantic model class to mark it for split interface generation.
+        """
+        setattr(cls, self.METADATA_KEY, True)
+        return cls
+
+    @staticmethod
+    def is_split_model(cls: type) -> bool:
+        """
+        Check if the Pydantic model is marked for split interface generation.
+        """
+        return getattr(cls, SplitInputOutput.METADATA_KEY, False)