Spaces:

bobocup
/

bbapi

Paused

App Files Files Community

bobocup commited on Dec 13, 2024

Commit

23df053

verified ·

1 Parent(s): 4da8f98

Update app.py

Browse files

Files changed (1) hide show

app.py +77 -58

app.py CHANGED Viewed

@@ -1,104 +1,123 @@
 from fastapi import FastAPI, Request
 from fastapi.responses import Response, HTMLResponse
 import requests
 import logging
-import re
-# 设置日志
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 app = FastAPI()
-def replace_paths(content):
-    """替换所有资源路径"""
     replacements = [
-        # 基本替换
         ('href="/', 'href="http://beibeioo.top/'),
         ('src="/', 'src="http://beibeioo.top/'),
-        ("href='/", "href='http://beibeioo.top/"),
-        ("src='/", "src='http://beibeioo.top/"),
-        # 添加 manifest 和其他资源的替换
         ('content="/', 'content="http://beibeioo.top/'),
-        # 替换 JavaScript 中的路径
         ('url("/', 'url("http://beibeioo.top/'),
-        ("url('/", "url('http://beibeioo.top/"),
-        # API 路径替换
-        ('"/api/', '"http://beibeioo.top/api/'),
-        ("'/api/", "'http://beibeioo.top/api/"),
-        # 静态资源路径
-        ('"/static/', '"http://beibeioo.top/static/'),
-        ("'/static/", "'http://beibeioo.top/static/"),
-        # 添加 JSON 数据中的路径
-        (':"/', ':"http://beibeioo.top/'),
     ]
     for old, new in replacements:
         content = content.replace(old, new)
-    # 添加基础路径到 head
-    base_tag = '<base href="http://beibeioo.top/">'
-    if '<head>' in content:
-        content = content.replace('<head>', f'<head>{base_tag}')
     return content
 @app.get("/{full_path:path}")
 async def proxy(request: Request, full_path: str = ""):
     try:
         # 构建目标 URL
-        if full_path.startswith("api/"):
-            actual_path = full_path[4:]
-            target_url = f"http://beibeioo.top/{actual_path}"
-        else:
-            target_url = f"http://beibeioo.top/{full_path}"
         logger.info(f"Forwarding to: {target_url}")
         # 转发请求
-        response = requests.get(
-            target_url,
-            headers={
-                'User-Agent': 'Mozilla/5.0',
-                'Accept': '*/*',
-                'Accept-Encoding': 'gzip, deflate',
-                'Connection': 'keep-alive',
-                'Host': 'beibeioo.top'
-            }
         )
         content_type = response.headers.get('content-type', '')
-        logger.info(f"Response content type: {content_type}")
-        # 处理不同类型的响应
-        if 'text/html' in content_type or 'application/javascript' in content_type or 'text/css' in content_type:
-            content = response.text
-            content = replace_paths(content)
             return Response(
-                content=content,
                 media_type=content_type,
                 headers={
                     'Access-Control-Allow-Origin': '*',
-                    'Access-Control-Allow-Methods': 'GET, POST, OPTIONS',
                     'Access-Control-Allow-Headers': '*'
                 }
             )
-        else:
-            # 对于其他类型的内容（图片等），直接返回
-            return Response(
-                content=response.content,
-                media_type=content_type,
                 headers={
                     'Access-Control-Allow-Origin': '*',
-                    'Access-Control-Allow-Methods': 'GET, POST, OPTIONS',
                     'Access-Control-Allow-Headers': '*'
                 }
             )
     except Exception as e:
         logger.error(f"Error: {str(e)}")
-        return {"error": str(e)}
-@app.get("/healthz")
-async def health_check():
-    return {"status": "healthy"}

 from fastapi import FastAPI, Request
 from fastapi.responses import Response, HTMLResponse
+from fastapi.middleware.cors import CORSMiddleware
 import requests
 import logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 app = FastAPI()
+# 添加 CORS 中间件
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+def modify_html(content: str) -> str:
+    """修改 HTML 内容，处理所有资源路径"""
+    # 添加 base 标签
+    if '<head>' in content:
+        base_tag = '<base href="http://beibeioo.top/">'
+        content = content.replace('<head>', f'<head>{base_tag}')
+    # 替换所有资源路径
     replacements = [
         ('href="/', 'href="http://beibeioo.top/'),
         ('src="/', 'src="http://beibeioo.top/'),
         ('content="/', 'content="http://beibeioo.top/'),
         ('url("/', 'url("http://beibeioo.top/'),
+        ('data-src="/', 'data-src="http://beibeioo.top/'),
+        ('srcset="/', 'srcset="http://beibeioo.top/')
     ]
     for old, new in replacements:
         content = content.replace(old, new)
     return content
 @app.get("/{full_path:path}")
+@app.post("/{full_path:path}")
 async def proxy(request: Request, full_path: str = ""):
     try:
         # 构建目标 URL
+        target_url = f"http://beibeioo.top/{full_path}"
         logger.info(f"Forwarding to: {target_url}")
+        # 获取原始请求方法和头部
+        method = request.method
+        headers = {
+            'User-Agent': 'Mozilla/5.0',
+            'Accept': '*/*',
+            'Host': 'beibeioo.top'
+        }
+        # 获取请求体
+        body = await request.body() if method == "POST" else None
         # 转发请求
+        response = requests.request(
+            method=method,
+            url=target_url,
+            headers=headers,
+            data=body
         )
+        logger.info(f"Response status: {response.status_code}")
+        logger.info(f"Response type: {response.headers.get('content-type')}")
         content_type = response.headers.get('content-type', '')
+        # 如果是 API 请求，直接返回
+        if full_path.startswith('v1/'):
             return Response(
+                content=response.content,
+                status_code=response.status_code,
                 media_type=content_type,
                 headers={
                     'Access-Control-Allow-Origin': '*',
+                    'Access-Control-Allow-Methods': '*',
                     'Access-Control-Allow-Headers': '*'
                 }
             )
+        # 如果是 HTML 内容，需要修改资源路径
+        if 'text/html' in content_type:
+            modified_content = modify_html(response.text)
+            return HTMLResponse(
+                content=modified_content,
+                status_code=response.status_code,
                 headers={
                     'Access-Control-Allow-Origin': '*',
+                    'Access-Control-Allow-Methods': '*',
                     'Access-Control-Allow-Headers': '*'
                 }
             )
+        # 其他类型的内容直接返回
+        return Response(
+            content=response.content,
+            status_code=response.status_code,
+            media_type=content_type,
+            headers={
+                'Access-Control-Allow-Origin': '*',
+                'Access-Control-Allow-Methods': '*',
+                'Access-Control-Allow-Headers': '*'
+            }
+        )
     except Exception as e:
         logger.error(f"Error: {str(e)}")
+        return Response(
+            content=str(e),
+            status_code=500,
+            media_type='text/plain'
+        )
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)