TMP Big updates.

Author: s3rius

Cargo.toml

@@ -11,7 +11,7 @@ publish = false
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 [lib]
 crate-type = ["cdylib"]
-name = "_inner"
+name = "_natsrpy_rs"
 
 [dependencies]
 async-nats = "0.46"

pyproject.toml

@@ -33,7 +33,7 @@ build-backend = "maturin"
 [tool.maturin]
 bindings = "pyo3"
 features = ["pyo3/extension-module"]
-module-name = "natsrpy._inner"
+module-name = "natsrpy._natsrpy_rs"
 python-source = "python"
 
 [tool.mypy]

python/natsrpy/__init__.py

@@ -1,4 +1,4 @@
-from natsrpy._inner import Message, Nats, Subscription
+from natsrpy._natsrpy_rs import Message, Nats, Subscription
 
 __all__ = [
     "Message",

python/natsrpy/_inner/message.pyi

@@ -1,8 +1,8 @@
 from datetime import timedelta
 from typing import Any
 
-from natsrpy._inner.js import JetStream
-from natsrpy._inner.message import Message
+from natsrpy._natsrpy_rs.js import JetStream
+from natsrpy._natsrpy_rs.message import Message
 
 class Subscription:
     def __aiter__(self) -> Subscription: ...

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

@@ -1,5 +1,5 @@
-from natsrpy._inner.js.kv import KeyValue, KVConfig
-from natsrpy._inner.js.stream import Stream, StreamConfig
+from natsrpy._natsrpy_rs.js.kv import KeyValue, KVConfig
+from natsrpy._natsrpy_rs.js.stream import Stream, StreamConfig
 
 class JetStream:
     async def publish(

python/natsrpy/_inner/js/__init__.pyi …thon/natsrpy/_natsrpy_rs/js/__init__.pyipython/natsrpy/_inner/js/__init__.pyi renamed to python/natsrpy/_natsrpy_rs/js/__init__.pyi python/natsrpy/_inner/js/__init__.pyi renamed to python/natsrpy/_natsrpy_rs/js/__init__.pyi

@@ -1,4 +1,4 @@
-from natsrpy._inner.js.stream import Placement, Republish, Source, StorageType
+from natsrpy._natsrpy_rs.js.stream import Placement, Republish, Source, StorageType
 
 class KVConfig:
     """

python/natsrpy/_inner/js/kv.pyi python/natsrpy/_natsrpy_rs/js/kv.pyipython/natsrpy/_inner/js/kv.pyi renamed to python/natsrpy/_natsrpy_rs/js/kv.pyi

@@ -228,8 +228,31 @@ class StreamInfo:
 class Stream:
     async def direct_get(self, sequence: int) -> StreamMessage:
         """
-        Get direct message from a stream.
+        Get direct message from the stream.
 
-        Please note, that this method will throw an error
-        in case of stream being configured without `allow_direct=True`.
+        :param sequence: sequence number of the message to get.
+        :return: Message.
+        """
+
+    async def get_info(self) -> StreamInfo:
+        """
+        Get information about the stream.
+
+        :return: Stream info.
+        """
+
+    async def purge(
+        self,
+        filter: str | None = None,
+        sequence: int | None = None,
+        keep: int | None = None,
+    ) -> int:
+        """
+        Purge current stream.
+
+        :param filter: filter of subjects to purge, defaults to None
+        :param sequence: Message sequence to purge up to (inclusive), defaults to None
+        :param keep: Message count to keep starting from the end of the stream,
+            defaults to None
+        :return: number of messages purged
         """

python/natsrpy/_inner/js/stream.pyi python/natsrpy/_natsrpy_rs/js/stream.pyipython/natsrpy/_inner/js/stream.pyi renamed to python/natsrpy/_natsrpy_rs/js/stream.pyi

@@ -0,0 +1,26 @@
+from typing import Any
+
+class Message:
+    """
+    Simple NATS message.
+
+    Attributes:
+        subject: subject where message was published
+        reply: subject where reply should be sent, if any
+        payload: message payload
+        headers: dictionary of message headers,
+            every value can be a simple value or a list.
+        status: status is used for reply messages to indicate the status of the reply.
+            It is None for regular messages.
+        description: message description is used for reply messages to
+            provide additional information about the status.
+        length: a length of the message payload in bytes.
+    """
+
+    subject: str
+    reply: str | None
+    payload: bytes
+    headers: dict[str, Any]
+    status: int | None
+    description: str | None
+    length: int

python/natsrpy/_natsrpy_rs/message.pyi

@@ -1,4 +1,4 @@
-from natsrpy._inner.js import JetStream
+from natsrpy._natsrpy_rs.js import JetStream
 from natsrpy.js.kv import KeyValue, KVConfig
 from natsrpy.js.stream import (
     Compression,