fix: update types

Author: phi-friday

src/socketio-stubs/_types.pyi

@@ -2,7 +2,7 @@ from collections.abc import Awaitable, Callable, Mapping, Sequence
 from contextlib import AbstractAsyncContextManager, AbstractContextManager
 from threading import Event as ThreadingEvent
 from types import ModuleType
-from typing import Any, Concatenate, Literal, TypeAlias, overload
+from typing import Any, ClassVar, Concatenate, Literal, TypeAlias, overload
 
 import engineio
 from _typeshed import Incomplete
@@ -13,6 +13,7 @@ from engineio.async_drivers.threading import DaemonThread
 from engineio.socket import Socket
 from gevent.event import Event as GeventEvent
 from socketio.admin import InstrumentedServer
+from socketio.msgpack_packet import MsgPackPacket
 from socketio.server import Server
 from typing_extensions import NotRequired, Required, TypedDict
 
@@ -208,6 +209,10 @@ class JsonModule(ModuleType):
     @staticmethod
     def loads(s: str | bytes | bytearray, **kwargs: Any) -> Any: ...
 
+class CustomMsgPackPacket(MsgPackPacket):
+    dumps_default: ClassVar[Callable[[Any], Any] | None]
+    ext_hook: ClassVar[Callable[[int, bytes], Any]]
+
 ## handlers
 
 ServerConnectHandler: TypeAlias = Callable[[str, dict[str, Any]], Any]

src/socketio-stubs/async_redis_manager.pyi

@@ -1,8 +1,12 @@
 import logging
 from typing import Any
 
+from redis.asyncio import Redis
+from redis.asyncio.client import PubSub as RedisPubSub
 from socketio._types import RedisArgs
 from socketio.async_pubsub_manager import AsyncPubSubManager
+from valkey.asyncio import Valkey
+from valkey.asyncio.client import PubSub as ValkeyPubSub
 
 def parse_redis_sentinel_url(
     url: str,
@@ -12,6 +16,9 @@ class AsyncRedisManager(AsyncPubSubManager):
     name: str
     redis_url: str
     redis_options: dict[str, Any]
+    connected: bool
+    redis: Redis | Valkey | None
+    pubsub: RedisPubSub | ValkeyPubSub | None
     def __init__(
         self,
         url: str = ...,

src/socketio-stubs/msgpack_packet.pyi

@@ -1,7 +1,17 @@
+from collections.abc import Callable
+from typing import Any
+
+from socketio._types import CustomMsgPackPacket
 from socketio.packet import Packet
 from typing_extensions import Buffer
 
 class MsgPackPacket(Packet):
     uses_binary_events: bool
     def encode(self) -> bytes: ...
     def decode(self, encoded_packet: Buffer) -> None: ...  # pyright: ignore[reportIncompatibleMethodOverride]
+    @classmethod
+    def configure(
+        cls,
+        dump_default: Callable[[Any], Any] | None = ...,
+        ext_hook: Callable[[int, bytes], Any] = ...,
+    ) -> type[CustomMsgPackPacket]: ...

src/socketio-stubs/packet.pyi

@@ -1,4 +1,4 @@
-from typing import Literal
+from typing import Any, Literal
 
 from socketio._types import DataType, JsonModule
 from typing_extensions import Buffer
@@ -43,4 +43,9 @@ class Packet:
     def encode(self) -> DataType: ...
     def decode(self, encoded_packet: Buffer) -> int: ...
     def add_attachment(self, attachment: bytes) -> bool: ...
-    def reconstruct_binary(self, attachments: list[bytes]) -> None: ...
+    @classmethod
+    def reconstruct_binary(cls, data: Any, attachments: list[bytes]) -> None: ...
+    @classmethod
+    def deconstruct_binary(cls, data: Any) -> tuple[Any, list[bytes]]: ...
+    @classmethod
+    def data_is_binary(cls, data: Any) -> bool: ...

src/socketio-stubs/redis_manager.pyi

@@ -1,8 +1,12 @@
 import logging
 from typing import Any
 
+from redis import Redis
+from redis.client import PubSub as RedisPubSub
 from socketio._types import RedisArgs
 from socketio.pubsub_manager import PubSubManager
+from valkey import Valkey
+from valkey.client import PubSub as ValkeyPubSub
 
 logger: logging.Logger
 
@@ -14,6 +18,9 @@ class RedisManager(PubSubManager):
     name: str
     redis_url: str
     redis_options: dict[str, Any]
+    connected: bool
+    redis: Redis | Valkey | None
+    pubsub: RedisPubSub | ValkeyPubSub | None
     def __init__(
         self,
         url: str = ...,