from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
import httpx

# # Import bộ sinh header từ vnstock
# from vnstock.core.utils.user_agent import get_headers

# app = FastAPI(title="Proxy API với Header Browser động (vnstock)")

# class ProxyPayload(BaseModel):
#     url: str
#     method: str = "POST"
#     params: dict = None
#     headers: dict = None             # Header client truyền thêm (nếu có)
#     body: dict | str | None = None
#     data_source: str = "SSI"         # 'SSI', 'VND', 'VDS', v.v.
#     random_agent: bool = True
#     browser: str = "chrome"
#     platform: str = "windows"

# @app.post("/request")
# async def flexible_request(payload: ProxyPayload):
#     # Sinh header browser-like + data_source
#     headers = get_headers(
#         data_source=payload.data_source,
#         random_agent=payload.random_agent,
#         browser=payload.browser,
#         platform=payload.platform
#     )
#     # Merge thêm header từ client (nếu có), client truyền sẽ ưu tiên
#     if payload.headers:
#         headers.update(payload.headers)
#     try:
#         async with httpx.AsyncClient(timeout=15) as client:
#             response = await client.request(
#                 payload.method.upper(),
#                 payload.url,
#                 headers=headers,
#                 params=payload.params,
#                 json=payload.body if isinstance(payload.body, dict) else None,
#                 data=payload.body if isinstance(payload.body, str) else None,
#             )
#         try:
#             resp_json = response.json()
#         except Exception:
#             resp_json = {"error": "response is not JSON", "raw": response.text}
#         return {
#             "status_code": response.status_code,
#             "headers": dict(response.headers),
#             "data": resp_json
#         }
#     except httpx.RequestError as exc:
#         raise HTTPException(status_code=502, detail=f"Request error: {exc}")
#     except Exception as exc:
#         raise HTTPException(status_code=500, detail=f"Other error: {exc}")