更改项目名为NebulaShell

store/@{NebulaShell}/ws-api/middleware.py

@@ -0,0 +1,44 @@
+"""WebSocket 中间件链"""
+from typing import Callable, Optional, Any
+
+
+class WsMiddleware:
+    """WebSocket 中间件基类"""
+    async def process(self, client: Any, message: str, next_fn: Callable) -> Optional[str]:
+        """处理消息"""
+        return await next_fn()
+
+
+class AuthMiddleware(WsMiddleware):
+    """认证中间件"""
+    async def process(self, client, message, next_fn):
+        # 可以在这里验证 token
+        return await next_fn()
+
+
+class WsMiddlewareChain:
+    """WebSocket 中间件链"""
+
+    def __init__(self):
+        self.middlewares: list[WsMiddleware] = []
+
+    def add(self, middleware: WsMiddleware):
+        """添加中间件"""
+        self.middlewares.append(middleware)
+
+    async def run(self, client, message) -> Optional[str]:
+        """执行中间件链"""
+        idx = 0
+        current_message = message
+
+        async def next_fn(msg=None):
+            nonlocal idx, current_message
+            if msg is not None:
+                current_message = msg
+            if idx < len(self.middlewares):
+                mw = self.middlewares[idx]
+                idx += 1
+                return await mw.process(client, current_message, next_fn)
+            return current_message
+
+        return await next_fn()