v2

app.py

@@ -1,6 +1,7 @@
 # 导入FastAPI框架
 from fastapi import FastAPI
 from routes import v1
+from routes import v2
 # 导入数据库引擎
 from database import engine
 # 导入用户模型的Base类
@@ -13,6 +14,7 @@ Base.metadata.create_all(bind=engine)
 app = FastAPI()
 
 app.include_router(v1.router, prefix="/v1")
+app.include_router(v2.router, prefix="/v2")
 
 # 定义根路径的GET请求处理函数
 @app.get("/")

routes/v2.py

@@ -1 +1,66 @@
+import httpx
+from fastapi import APIRouter, HTTPException, Request
+import json
+
+router = APIRouter()
+
+@router.get("/")
+async def v2():
+    return {"message": "Hello World from v2"}
+
+@router.get("/{protocol}/{url:path}")
+async def proxy_request(protocol:str, url: str):
+    async with httpx.AsyncClient() as client:
+        try:
+            target_url = f"{protocol}://{url}"
+            response = await client.get(
+                target_url,
+                headers={},
+            )
+            response.raise_for_status()  # Raise HTTPError for bad responses (4xx or 5xx)
+            return response.content
+        except httpx.HTTPStatusError as e:
+            raise HTTPException(status_code=e.response.status_code, detail=str(e))
+        except httpx.RequestError as e:
+            raise HTTPException(status_code=500, detail=str(e))
+
+@router.post("/{protocol}/{url:path}")
+async def proxy_post(protocol:str, url: str, request: Request):
+    body = await request.body()
+    headers = dict(request.headers)  # 将请求头转换为字典
+    
+    # 移除 Content-Length 和 Host 头部
+    if 'content-length' in headers:
+        del headers['content-length']
+    if 'host' in headers:
+        del headers['host']
+        
+    headers['User-Agent']='curl/8.7.1'
+
+    dest_url = f"{protocol}://{url}"
+    
+    async with httpx.AsyncClient() as client:
+        try:
+            # 向目标 URL 发送 POST 请求
+            response = await client.post(dest_url, content=body, headers=headers)
+            
+            # 检查响应状态码
+            if response.status_code == 200:
+                # 检查响应内容类型是否为 JSON
+                if 'application/json' in response.headers.get('content-type', ''):
+                    return response.json()
+                else:
+                    return {"error": "Response is not in JSON format"}
+            else:
+                # 将错误响应转换为 JSON 格式并返回给客户端
+                try:
+                    error_data = response.json()
+                except ValueError:
+                    error_data = {"status_code": response.status_code, "detail": response.text}
+                return error_data
+        
+        except httpx.RequestError as e:
+            # 处理请求错误
+            raise HTTPException(status_code=500, detail=str(e))
+
 # routes for v2 API