Update app.py

app.py

@@ -6,7 +6,6 @@ from typing import Optional
 
 app = FastAPI()
 
-# 添加 CORS 支持
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -15,10 +14,8 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# 目标网站
 TARGET_URL = "http://beibeioo.top"
 
-# 创建一个全局的 httpx 客户端
 async def get_http_client():
     return httpx.AsyncClient(
         timeout=30.0,
@@ -26,35 +23,22 @@ async def get_http_client():
         http2=True
     )
 
-# 处理 /api 路径
-@app.api_route("/api", methods=["GET", "POST", "PUT", "DELETE", "OPTIONS", "HEAD", "PATCH"])
-async def api_root(request: Request):
-    return await proxy_handler("", request)  # 传空字符串，直接访问根路径
-
-# 处理 /api/xxx 路径
-@app.api_route("/api/{path:path}", methods=["GET", "POST", "PUT", "DELETE", "OPTIONS", "HEAD", "PATCH"])
-async def api_proxy(path: str, request: Request):
-    return await proxy_handler(path, request)
-
-# 处理其他所有路径
-@app.api_route("/{path:path}", methods=["GET", "POST", "PUT", "DELETE", "OPTIONS", "HEAD", "PATCH"])
-async def general_proxy(path: str, request: Request):
-    return await proxy_handler(path, request)
-
-# 代理处理函数
 async def proxy_handler(path: str, request: Request):
     try:
-        # 构建目标URL
-        url = f"{TARGET_URL}/{path}"
+        # 移除开头和结尾的斜杠
+        path = path.strip('/')
+        url = f"{TARGET_URL}/{path}" if path else TARGET_URL
         
-        # 获取并处理请求头
         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), f"{TARGET_URL}/")
+        
         params = dict(request.query_params)
         body = await request.body()
         
@@ -67,46 +51,57 @@ async def proxy_handler(path: str, request: Request):
                 content=body if body else None,
             )
             
-            # 处理响应头
             response_headers = dict(response.headers)
             response_headers.pop('transfer-encoding', None)
             response_headers.pop('content-encoding', None)
             response_headers.pop('content-length', 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('/'))
+            # 修改响应中的所有目标URL引用
+            content = response.content
+            if response.headers.get('content-type', '').startswith(('text/html', 'text/css', 'application/javascript')):
+                content = content.replace(
+                    TARGET_URL.encode(),
+                    str(request.base_url).rstrip('/').encode()
+                )
             
             return Response(
-                content=response.content,
+                content=content,
                 status_code=response.status_code,
                 headers=response_headers,
                 media_type=response.headers.get('content-type')
             )
             
     except httpx.TimeoutException:
-        return Response(
-            content="请求超时",
-            status_code=504,
-            media_type="text/plain"
-        )
+        return Response(content="请求超时", status_code=504)
     except httpx.RequestError:
-        return Response(
-            content="无法连接到目标服务器",
-            status_code=502,
-            media_type="text/plain"
-        )
+        return Response(content="无法连接到目标服务器", status_code=502)
     except Exception as e:
-        return Response(
-            content=f"服务器错误: {str(e)}",
-            status_code=500,
-            media_type="text/plain"
-        )
+        return Response(content=f"服务器错误: {str(e)}", status_code=500)
+
+# 处理 /api 路径（映射到根路径）
+@app.api_route("/api", methods=["GET", "POST", "PUT", "DELETE", "OPTIONS", "HEAD", "PATCH"])
+async def api_root(request: Request):
+    return await proxy_handler("", request)
+
+# 处理 /api/ 路径（也映射到根路径）
+@app.api_route("/api/", methods=["GET", "POST", "PUT", "DELETE", "OPTIONS", "HEAD", "PATCH"])
+async def api_root_slash(request: Request):
+    return await proxy_handler("", request)
+
+# 处理 /api/xxx 路径
+@app.api_route("/api/{path:path}", methods=["GET", "POST", "PUT", "DELETE", "OPTIONS", "HEAD", "PATCH"])
+async def api_proxy(path: str, request: Request):
+    return await proxy_handler(path, request)
+
+# 处理其他路径
+@app.api_route("/{path:path}", methods=["GET", "POST", "PUT", "DELETE", "OPTIONS", "HEAD", "PATCH"])
+async def general_proxy(path: str, request: Request):
+    if path == "api":
+        return await proxy_handler("", request)
+    return await proxy_handler(path, request)
 
 @app.get("/healthcheck")
 async def healthcheck():
-    """健康检查端点"""
     return {"status": "healthy"}
 
 if __name__ == "__main__":