mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 07:20:44 +08:00
oliver
ba3836f00f
- 在 oclaw/ 下重新初始化 Git 仓库 - 补齐子仓库 .gitignore，避免提交本地运行态数据（_local、node_modules、logs 等） - 提交当前工程代码与配置 Made-with: Cursor
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
|
|
async def close_ws(conn: Any, code: int = 1000, reason: str = "done") -> None:
|
|
try:
|
|
await conn.ws.close(code=code, reason=reason)
|
|
except Exception:
|
|
pass
|
|
|
|
|
|
async def run_connection_loop(conn: Any) -> None:
|
|
await conn.ws.accept()
|
|
await conn.send_event("connect.challenge", {"nonce": conn.connect_nonce, "ts": conn._now_ms()})
|
|
while True:
|
|
frame = await conn._recv_frame(preauth=not conn.connected)
|
|
if frame is None:
|
|
return
|
|
if frame.get("type") != "req":
|
|
await conn.send_res(
|
|
str(frame.get("id") or "invalid"),
|
|
ok=False,
|
|
error=conn._error_shape("INVALID_REQUEST", "expected req frame"),
|
|
)
|
|
if not conn.connected:
|
|
await close_ws(conn, 1008, "invalid handshake")
|
|
return
|
|
continue
|
|
req_id = str(frame.get("id") or "invalid")
|
|
method = str(frame.get("method") or "")
|
|
params = frame.get("params")
|
|
if not conn.connected:
|
|
await conn._handle_connect(req_id, method, params)
|
|
if conn.handshake_failed:
|
|
await close_ws(conn, 1008, "invalid handshake")
|
|
return
|
|
continue
|
|
await conn._dispatch_connected(req_id, method, params)
|
|
|
|
|
|
__all__ = ["run_connection_loop", "close_ws"]
|
|
|