Tmp commit.

Author: s3rius

Cargo.lock

@@ -5,7 +5,7 @@ edition = "2024"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 [lib]
-name = "natsrpy"
+name = "_inner"
 crate-type = ["cdylib"]
 
 [dependencies]
@@ -17,4 +17,5 @@ pyo3 = { version = "0.28", features = ["abi3"] }
 pyo3-async-runtimes = { version = "0.28", features = ["tokio-runtime"] }
 pyo3-log = "0.13.3"
 thiserror = "2.0.18"
+time = "0.3.47"
 tokio = { version = "1.50.0", features = ["full"] }

Cargo.toml

@@ -16,5 +16,5 @@ build-backend = "maturin"
 [tool.maturin]
 bindings = "pyo3"
 python-source = "python"
-module-name = "natsrpy._internal"
+module-name = "natsrpy._inner"
 features = ["pyo3/extension-module"]

pyproject.toml

@@ -1,3 +1,12 @@
-from ._internal import Nats
+from natsrpy._inner import (
+    Nats,
+    # JetStream,
+    # Source,
+    # External,
+    Subscription,
+)
 
-__all__ = ["Nats"]
+__all__ = [
+    "Nats",
+    "Subscription",
+]

python/natsrpy/__init__.py

@@ -1,5 +1,3 @@
-from .jetstream import JetStream
-
 class Message:
     subject: str
     reply: str | None

python/natsrpy/_internal/__init__.pyi python/natsrpy/_inner/__init__.pyipython/natsrpy/_internal/__init__.pyi renamed to python/natsrpy/_inner/__init__.pyi

@@ -0,0 +1,44 @@
+from .kv import KeyValue
+
+class External:
+    def __init__(
+        self,
+        api_prefix: str,
+        delivery_prefix: str | None = None,
+    ) -> None: ...
+
+class Source:
+    def __init__(
+        self,
+        name: str,
+        filter_subject: str | None = None,
+        external: External | None = None,
+    ) -> None: ...
+
+class JetStream:
+    async def publish(
+        self,
+        subject: str,
+        payload: bytes,
+        *,
+        headers: dict[str, str] | None = None,
+        reply: str | None = None,
+        err_on_disconnect: bool = False,
+    ) -> None: ...
+    async def create_kv(
+        self,
+        bucket: str,
+        description=None,
+        max_value_size=None,
+        history=None,
+        max_age=None,
+        max_bytes=None,
+        storage=None,
+        num_replicas=None,
+        republish=None,
+        mirror: Source | None = None,
+        mirror_direct=None,
+        compression=None,
+        limit_markers=None,
+    ) -> KeyValue: ...
+    async def get_kv(self, bucket: str) -> KeyValue: ...

python/natsrpy/_inner/js/__init__.pyi

@@ -0,0 +1,28 @@
+
+class StorageType:
+    FILE: "StorageType"
+    MEMORY: "StorageType"
+
+class KVConfig:
+    def __init__(
+        self,
+        bucket: str,
+        description: str | None = None,
+        max_value_size: int | None = None,
+        history:int|None=None,
+        max_age:float |None=None,
+        max_bytes:int |None=None,
+        storage:StorageType|None=None,
+        num_replicas: int |None=None,
+        republish:bool |None=None,
+        mirror:|None=None,
+        sources:|None=None,
+        mirror_direct:|None=None,
+        compression:|None=None,
+        placement:|None=None,
+        limit_markers:|None=None,
+    ) -> None: ...
+
+class KeyValue:
+    async def get(self, key: str) -> bytes | None: ...
+    async def put(self, key: str, value: bytes) -> int: ...

python/natsrpy/_inner/js/kv.pyi

@@ -0,0 +1,6 @@
+class StorageType:
+    FILE: "StorageType"
+    MEMORY: "StorageType"
+
+class Republish:
+