Update app.py

app.py

@@ -1,39 +1,108 @@
-from fastapi import FastAPI, Request
-from fastapi.responses import Response, HTMLResponse
-import requests
+from fastapi import FastAPI, Request, Response
+from fastapi.responses import StreamingResponse
+from fastapi.middleware.cors import CORSMiddleware
+import httpx
+import asyncio
+from typing import Optional
 
 app = FastAPI()
 
-@app.get("/", response_class=HTMLResponse)
-@app.get("/{path:path}", response_class=HTMLResponse)
-async def proxy(request: Request, path: str = ""):
+# 添加 CORS 支持
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# 目标网站
+TARGET_URL = "http://beibeioo.top"
+
+# 创建一个全局的 httpx 客户端
+async def get_http_client():
+    return httpx.AsyncClient(
+        timeout=30.0,  # 30秒超时
+        follow_redirects=True,
+        http2=True
+    )
+
+@app.api_route("/{path:path}", methods=["GET", "POST", "PUT", "DELETE", "OPTIONS", "HEAD", "PATCH"])
+async def proxy(path: str, request: Request):
     try:
-        # 构建目标 URL
-        if path.startswith("api/"):
-            actual_path = path[4:]  # 跳过 "api/"
-            target_url = f"http://beibeioo.top/{actual_path}"
-        else:
-            target_url = "http://beibeioo.top"
+        # 构建目标URL
+        url = f"{TARGET_URL}/{path}"
         
-        # 转发请求
-        response = requests.get(
-            target_url,
-            headers={
-                'User-Agent': 'Mozilla/5.0',
-                'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8',
-                'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8',
-            }
-        )
+        # 获取并处理请求头
+        headers = dict(request.headers)
+        # 删除可能导致问题的头部
+        headers.pop('host', None)
+        headers.pop('connection', None)
+        headers.pop('content-length', None)
+        headers.pop('transfer-encoding', None)
+        
+        # 如果有 referer，修改它
+        if 'referer' in headers:
+            headers['referer'] = headers['referer'].replace(str(request.base_url), TARGET_URL)
+        
+        # 获取查询参数
+        params = dict(request.query_params)
+        
+        # 获取请求体
+        body = await request.body()
         
-        # 返回响应
+        async with await get_http_client() as client:
+            response = await client.request(
+                method=request.method,
+                url=url,
+                params=params,
+                headers=headers,
+                content=body if body else None,
+            )
+            
+            # 处理响应头
+            response_headers = dict(response.headers)
+            # 删除可能导致问题的响应头
+            response_headers.pop('transfer-encoding', None)
+            response_headers.pop('content-encoding', None)
+            
+            # 修改所有涉及原始域名的响应头
+            for key, value in response_headers.items():
+                if isinstance(value, str) and TARGET_URL in value:
+                    response_headers[key] = value.replace(TARGET_URL, str(request.base_url).rstrip('/'))
+            
+            # 返回响应
+            return StreamingResponse(
+                response.iter_bytes(),
+                status_code=response.status_code,
+                headers=response_headers,
+                media_type=response.headers.get('content-type')
+            )
+            
+    except httpx.TimeoutException:
         return Response(
-            content=response.content,
-            media_type=response.headers.get('content-type', 'text/html'),
-            headers={
-                'Access-Control-Allow-Origin': '*',
-                'Access-Control-Allow-Methods': 'GET, POST, OPTIONS',
-                'Access-Control-Allow-Headers': '*'
-            }
+            content="请求超时",
+            status_code=504,
+            media_type="text/plain"
+        )
+    except httpx.RequestError:
+        return Response(
+            content="无法连接到目标服务器",
+            status_code=502,
+            media_type="text/plain"
         )
     except Exception as e:
-        return HTMLResponse(content=f"Error: {str(e)}", status_code=500)
+        return Response(
+            content=f"服务器错误: {str(e)}",
+            status_code=500,
+            media_type="text/plain"
+        )
+
+@app.get("/healthcheck")
+async def healthcheck():
+    """健康检查端点"""
+    return {"status": "healthy"}
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)