Update fun.py

fun.py

@@ -1,24 +1,20 @@
-from fastapi import FastAPI, Request, Response, status
+from fastapi import FastAPI, Request, Response, status, WebSocket, WebSocketDisconnect
 import httpx
 import uvicorn
-import time
+import asyncio
+import websockets
+
 app = FastAPI()
-time.sleep(10)
-TARGET_BASE = "http://127.0.0.1:9222"  # 替换为你的目标服务器
-resp = httpx.get(TARGET_BASE)
-print(f'RESP:{resp.text}')
+
+TARGET_BASE = "http://127.0.0.1:9222"  # 目标服务器（Chrome DevTools）
+TARGET_WS_BASE = "ws://127.0.0.1:9222"  # WebSocket 目标
+
 @app.api_route("/{path:path}", methods=["GET", "POST", "PUT", "DELETE", "PATCH", "OPTIONS", "HEAD"])
 async def proxy(request: Request, path: str):
-    resp = httpx.get(TARGET_BASE)
-        # 强制Host头为127.0.0.1
 
-    print(f'RESP:{resp.text}')
-    # if path == "":
-    #     # 拒绝根目录
-    #     return Response(content="Not Found", status_code=status.HTTP_404_NOT_FOUND)
     url = f"{TARGET_BASE}/{path}"
     headers = dict(request.headers)
-    host = headers["host"]
+    # 强制 Host 头为 127.0.0.1，避免 Chrome 拒绝
     headers["host"] = "127.0.0.1"
     body = await request.body()
     async with httpx.AsyncClient(follow_redirects=True) as client:
@@ -29,15 +25,33 @@ async def proxy(request: Request, path: str):
             content=body,
             params=request.query_params
         )
-    headers = dict(resp.headers)
-    headers["host"] = host
     return Response(
         content=resp.content,
         status_code=resp.status_code,
-        headers=headers
+        headers=dict(resp.headers)
     )
 
+@app.websocket("/{path:path}")
+async def websocket_proxy(websocket: WebSocket, path: str):
+    await websocket.accept()
+    target_url = f"{TARGET_WS_BASE}/{path}"
+    try:
+        async with websockets.connect(target_url) as target_ws:
+            async def client_to_server():
+                while True:
+                    data = await websocket.receive_text()
+                    await target_ws.send(data)
+
+            async def server_to_client():
+                while True:
+                    data = await target_ws.recv()
+                    await websocket.send_text(data)
+
+            await asyncio.gather(client_to_server(), server_to_client())
+    except WebSocketDisconnect:
+        pass
+    except Exception:
+        await websocket.close()
 
 if __name__ == "__main__":
-    # 运行FastAPI应用
     uvicorn.run("fun:app", host="0.0.0.0", port=8000)