Update fun.py

fun.py

@@ -38,33 +38,25 @@ async def proxy(request: Request, path: str):
 async def websocket_proxy(websocket: WebSocket, path: str):
     await websocket.accept()
     target_url = f"{TARGET_WS_BASE}/{path}"
-    logger.info(f"Forwarding WebSocket to: {target_url}")
+    logger.info(f"Forwarding to: {target_url}")
 
     try:
         async with websockets.connect(target_url) as target_ws:
-            async def forward_client_to_server():
+            async def forward(from_ws, to_ws, direction):
                 while True:
-                    message = await websocket.receive()
-                    if message["type"] == "websocket.receive.text":
-                        await target_ws.send(message["text"])
-                        logger.info(f"Sent to Chrome: {message['text'][:100]}")
-
-            async def forward_server_to_client():
-                while True:
-                    message = await target_ws.recv()
-                    if isinstance(message, str):
-                        await websocket.send_text(message)
-                        logger.info(f"Received from Chrome: {message[:100]}")
-                    elif isinstance(message, bytes):
-                        await websocket.send_bytes(message)
+                    data = await from_ws.receive()
+                    logger.info(f"{direction} RAW DATA: {data}")  # 打印原始数据
+                    if isinstance(data, str):
+                        await to_ws.send(data)
+                    else:
+                        await to_ws.send(data.bytes if hasattr(data, 'bytes') else data)
 
             await asyncio.gather(
-                forward_client_to_server(),
-                forward_server_to_client()
+                forward(websocket, target_ws, "CLIENT->CHROME"),
+                forward(target_ws, websocket, "CHROME->CLIENT")
             )
     except Exception as e:
         logger.error(f"Error: {e}")
-        await websocket.close()
  
 if __name__ == "__main__":
     uvicorn.run("fun:app", host="0.0.0.0", port=8000, reload=True)