Update app.py

app.py

@@ -1,7 +1,11 @@
 from fastapi import FastAPI, Request
-from fastapi.responses import Response
+from fastapi.responses import Response, HTMLResponse
 import requests
-from urllib.parse import urljoin
+import logging
+
+# 设置日志
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 app = FastAPI()
 
@@ -15,44 +19,67 @@ async def proxy(request: Request, full_path: str = ""):
         else:
             target_url = f"http://beibeioo.top/{full_path}"
             
-        print(f"Forwarding to: {target_url}")  # 调试日志
+        logger.info(f"Forwarding to: {target_url}")
         
         # 转发请求
         response = requests.get(
             target_url,
             headers={
                 'User-Agent': 'Mozilla/5.0',
-                'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
-                'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8',
+                'Accept': '*/*',
+                'Accept-Encoding': 'gzip, deflate',
+                'Connection': 'keep-alive',
                 'Host': 'beibeioo.top'
             }
         )
         
-        content_type = response.headers.get('content-type', 'text/html')
-        content = response.content
+        # 获取内容类型
+        content_type = response.headers.get('content-type', '')
+        logger.info(f"Response content type: {content_type}")
         
-        # 如果是 HTML 内容，替换所有资源路径
+        # 获取响应内容
         if 'text/html' in content_type:
             content = response.text
-            # 替换相对路径为绝对路径
-            content = content.replace('href="/', 'href="https://beibeioo.top/')
-            content = content.replace('src="/', 'src="https://beibeioo.top/')
-            content = content.replace("href='/", "href='https://beibeioo.top/")
-            content = content.replace("src='/", "src='https://beibeioo.top/")
-            content = bytes(content, 'utf-8')
-        
-        return Response(
-            content=content,
-            status_code=response.status_code,
-            media_type=content_type,
-            headers={
-                'Access-Control-Allow-Origin': '*',
-                'Access-Control-Allow-Methods': 'GET, POST, OPTIONS',
-                'Access-Control-Allow-Headers': '*'
-            }
-        )
+            logger.info(f"Response length: {len(content)}")
+            logger.info(f"First 200 chars: {content[:200]}")  # 打印前200个字符用于调试
+            
+            # 替换所有相对路径为绝对路径
+            content = content.replace('href="/', f'href="http://beibeioo.top/')
+            content = content.replace('src="/', f'src="http://beibeioo.top/')
+            content = content.replace("href='/", f"href='http://beibeioo.top/")
+            content = content.replace("src='/", f"src='http://beibeioo.top/")
+            
+            # 如果是API响应，可能需要特殊处理
+            if not content.strip():
+                return Response(
+                    content=response.content,
+                    media_type='application/json',
+                    headers={'Access-Control-Allow-Origin': '*'}
+                )
+            
+            return HTMLResponse(
+                content=content,
+                status_code=response.status_code,
+                headers={
+                    'Access-Control-Allow-Origin': '*',
+                    'Access-Control-Allow-Methods': 'GET, POST, OPTIONS',
+                    'Access-Control-Allow-Headers': '*'
+                }
+            )
+        else:
+            # 对于非HTML内容，直接返回
+            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:
-        print(f"Error: {str(e)}")  # 调试日志
+        logger.error(f"Error: {str(e)}")
         return {"error": str(e)}
 
 @app.get("/healthz")