Spaces:

bobocup
/

bbapi

Paused

App Files Files Community

bobocup commited on Dec 13, 2024

Commit

b0c9cde

verified ·

1 Parent(s): 5541149

Update app.py

Browse files

Files changed (1) hide show

app.py +34 -31

app.py CHANGED Viewed

@@ -21,61 +21,64 @@ app.add_middleware(
 @app.post("/{full_path:path}")
 async def proxy(request: Request, full_path: str = ""):
     try:
         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': '*/*',
-                'Host': 'beibeioo.top'
-            }
-        )
         content_type = response.headers.get('content-type', '')
-        logger.info(f"Request path: {full_path}")
-        logger.info(f"Response type: {content_type}")
-        # 处理不同类型的请求
-        if full_path.startswith('v1/'):
-            # API 请求
-            return Response(content=response.content, media_type=content_type)
-        elif full_path.startswith('assets/') or full_path.endswith(('.js', '.css', '.png', '.jpg', '.jpeg', '.gif')):
-            # 静态资源请求
             logger.info(f"Serving static file: {full_path}")
             return Response(
                 content=response.content,
-                media_type=content_type,
                 headers={'Access-Control-Allow-Origin': '*'}
             )
         elif 'text/html' in content_type:
-            # HTML 内容
             content = response.text
-            # 替换资源路径
             replacements = [
-                ('href="/', 'href="https://bobocup-api.hf.space/'),
-                ('src="/', 'src="https://bobocup-api.hf.space/'),
-                ('content="/', 'content="https://bobocup-api.hf.space/'),
-                ('url("/', 'url("https://bobocup-api.hf.space/'),
-                ('"/assets/', '"https://bobocup-api.hf.space/assets/'),
-                ('"/api/', '"https://bobocup-api.hf.space/api/'),
-                ('"/v1/', '"https://bobocup-api.hf.space/v1/')
             ]
             for old, new in replacements:
                 content = content.replace(old, new)
-            logger.info("HTML content processed")
             return HTMLResponse(
                 content=content,
-                headers={'Access-Control-Allow-Origin': '*'}
             )
         else:
-            # 其他类型的内容
             return Response(
                 content=response.content,
                 media_type=content_type,

 @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"Incoming request: {full_path}")
         logger.info(f"Forwarding to: {target_url}")
+        # 设置请求头
+        headers = {
+            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
+            'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8',
+            'Accept-Language': 'en-US,en;q=0.5',
+            'Connection': 'keep-alive',
+            'Host': 'beibeioo.top'
+        }
+        # 发送请求
+        response = requests.get(target_url, headers=headers)
         content_type = response.headers.get('content-type', '')
+        # 如果是静态资源请求
+        if full_path.endswith(('.js', '.css', '.png', '.jpg', '.jpeg', '.gif', '.ico')):
             logger.info(f"Serving static file: {full_path}")
             return Response(
                 content=response.content,
+                media_type=response.headers.get('content-type', 'application/octet-stream'),
                 headers={'Access-Control-Allow-Origin': '*'}
             )
+        # 如果是 HTML 内容
         elif 'text/html' in content_type:
             content = response.text
+            # 替换所有资源路径，包括 assets 目录
             replacements = [
+                ('"/assets/', '"assets/'),  # 改为相对路径
+                ('"/static/', '"static/'),  # 改为相对路径
+                ('"/logo.png', '"logo.png'),  # 改为相对路径
+                ('href="/', 'href="'),  # 改为相对路径
+                ('src="/', 'src="'),  # 改为相对路径
             ]
             for old, new in replacements:
                 content = content.replace(old, new)
+            # 添加基础路径
+            base_tag = f'<base href="{request.base_url}">'
+            if '<head>' in content:
+                content = content.replace('<head>', f'<head>{base_tag}')
             return HTMLResponse(
                 content=content,
+                headers={
+                    'Access-Control-Allow-Origin': '*',
+                    'Content-Security-Policy': "default-src * 'unsafe-inline' 'unsafe-eval'; img-src * data: blob:;"
+                }
             )
+        # 其他类型的内容
         else:
             return Response(
                 content=response.content,
                 media_type=content_type,