Added stream consumers. (#7)

Author: s3rius

.github/workflows/test.yml

@@ -48,6 +48,7 @@ jobs:
       - uses: auguwu/clippy-action@1.4.0
         with:
           token: ${{secrets.GITHUB_TOKEN}}
+          deny: warnings
   pytest:
     runs-on: ubuntu-latest
     steps:

Cargo.lock

@@ -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"

Cargo.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]

pyproject.toml

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

python/natsrpy/__init__.py

@@ -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: ...
@@ -28,7 +28,7 @@ class Nats:
     async def publish(
         self,
         subject: str,
-        payload: bytes,
+        payload: bytes | str | bytearray | memoryview,
         *,
         headers: dict[str, Any] | None = None,
         reply: str | None = None,

python/natsrpy/_inner/js/__init__.pyi

@@ -0,0 +1,29 @@
+from typing import Any
+
+from .managers import KVManager, StreamsManager
+
+class JetStream:
+    async def publish(
+        self,
+        subject: str,
+        payload: str | bytes | bytearray | memoryview,
+        *,
+        headers: dict[str, str] | None = None,
+        reply: str | None = None,
+        err_on_disconnect: bool = False,
+    ) -> None: ...
+    @property
+    def kv(self) -> KVManager: ...
+    @property
+    def streams(self) -> StreamsManager: ...
+
+class JetStreamMessage:
+    @property
+    def subject(self) -> str: ...
+    @property
+    def reply(self) -> str | None: ...
+    @property
+    def payload(self) -> bytes: ...
+    @property
+    def headers(self) -> dict[str, Any]: ...
+    async def ack(self) -> None: ...