bb1d662ee1
Rebrand the Rust client crate (coven/ → hh/, package+binary "hack-house"), README, CLI strings, and branch (coven → hack-house). Gitea repo renamed cmd-chat → hack-house to match. Crypto/server logic unchanged; selftest + golden-vector test still green, binary is now `hack-house`. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
83 lines
2.2 KiB
Python
83 lines
2.2 KiB
Python
from collections.abc import Awaitable, MutableMapping
|
|
from typing import Any, Callable
|
|
|
|
from sanic.exceptions import InvalidUsage
|
|
|
|
|
|
ASGIMessage = MutableMapping[str, Any]
|
|
|
|
|
|
class WebSocketConnection:
|
|
"""
|
|
This is for ASGI Connections.
|
|
It provides an interface similar to WebsocketProtocol, but
|
|
sends/receives over an ASGI connection.
|
|
"""
|
|
|
|
# TODO
|
|
# - Implement ping/pong
|
|
|
|
def __init__(
|
|
self,
|
|
send: Callable[[ASGIMessage], Awaitable[None]],
|
|
receive: Callable[[], Awaitable[ASGIMessage]],
|
|
subprotocols: list[str] | None = None,
|
|
) -> None:
|
|
self._send = send
|
|
self._receive = receive
|
|
self._subprotocols = subprotocols or []
|
|
|
|
async def send(self, data: str | bytes, *args, **kwargs) -> None:
|
|
message: dict[str, str | bytes] = {"type": "websocket.send"}
|
|
|
|
if isinstance(data, bytes):
|
|
message.update({"bytes": data})
|
|
else:
|
|
message.update({"text": str(data)})
|
|
|
|
await self._send(message)
|
|
|
|
async def recv(self, *args, **kwargs) -> str | bytes | None:
|
|
message = await self._receive()
|
|
|
|
if message["type"] == "websocket.receive":
|
|
try:
|
|
return message["text"]
|
|
except KeyError:
|
|
try:
|
|
return message["bytes"]
|
|
except KeyError:
|
|
raise InvalidUsage("Bad ASGI message received")
|
|
elif message["type"] == "websocket.disconnect":
|
|
pass
|
|
|
|
return None
|
|
|
|
receive = recv
|
|
|
|
async def accept(self, subprotocols: list[str] | None = None) -> None:
|
|
subprotocol = None
|
|
if subprotocols:
|
|
for subp in subprotocols:
|
|
if subp in self.subprotocols:
|
|
subprotocol = subp
|
|
break
|
|
|
|
await self._send(
|
|
{
|
|
"type": "websocket.accept",
|
|
"subprotocol": subprotocol,
|
|
}
|
|
)
|
|
|
|
async def close(self, code: int = 1000, reason: str = "") -> None:
|
|
pass
|
|
|
|
@property
|
|
def subprotocols(self):
|
|
return self._subprotocols
|
|
|
|
@subprotocols.setter
|
|
def subprotocols(self, subprotocols: list[str] | None = None):
|
|
self._subprotocols = subprotocols or []
|