import os
from fastapi import FastAPI, Request, HTTPException
from fastapi.responses import StreamingResponse, FileResponse, Response
from fastapi.staticfiles import StaticFiles
from starlette.background import BackgroundTask
import httpx

app = FastAPI()

VCSKY_BASE_URL = "https://cdn.dos.zone/vcsky/"
BR_BASE_URL = "https://br.cdn.dos.zone/vcsky/"

def request_to_url(request: Request, path: str, base_url = VCSKY_BASE_URL):
    query_string = str(request.url.query) if request.url.query else ""
    url = f"{base_url}{path}"
    if query_string:
        url = f"{url}?{query_string}"
    return url

async def _proxy_request(request: Request, url: str):
    client = httpx.AsyncClient(timeout=None)
    headers = {k: v for k, v in request.headers.items() if k.lower() not in ["host", "content-length"]}
    
    req = client.build_request(request.method, url, headers=headers)
    r = await client.send(req, stream=True)
    
    excluded_headers = {"content-length", "transfer-encoding", "connection", "keep-alive", "upgrade", "content-encoding", "x-content-encoding"}
    response_headers = {k: v for k, v in r.headers.items() if k.lower() not in excluded_headers}
    
    response_headers["Cross-Origin-Opener-Policy"] = "same-origin"
    response_headers["Cross-Origin-Embedder-Policy"] = "require-corp"

    return StreamingResponse(
        r.aiter_bytes(),
        status_code=r.status_code,
        headers=response_headers,
        background=BackgroundTask(client.aclose)
    )

@app.api_route("/vcsky/{path:path}", methods=["GET", "POST", "PUT", "DELETE", "PATCH", "HEAD", "OPTIONS"])
async def vc_sky_proxy(request: Request, path: str):
    return await _proxy_request(request, request_to_url(request, path))

@app.api_route("/vcbr/{path:path}", methods=["GET", "POST", "PUT", "DELETE", "PATCH", "HEAD", "OPTIONS"])
async def vc_br_proxy(request: Request, path: str):
    return await _proxy_request(request, request_to_url(request, path, BR_BASE_URL))

### local use ###

# app.mount("/vcsky", StaticFiles(directory="vcsky"), name="vcsky") # audio/, data/, models/, anim/
# @app.get("/vcbr/{file_path:path}") # vc-sky-en-v6.data.br, vc-sky-en-v6.wasm.br, vc-sky-ru-v6.data.br, vc-sky-ru-v6.wasm.br
# async def serve_vcbr(file_path: str):
#     file_location = os.path.join("vcbr", file_path)
#     if not os.path.isfile(file_location):
#         raise HTTPException(status_code=404, detail="File not found")
    
#     headers = {
#         "Cross-Origin-Opener-Policy": "same-origin",
#         "Cross-Origin-Embedder-Policy": "require-corp"
#     }
    
#     media_type = "application/octet-stream"
#     if file_path.endswith(".wasm.br"):
#         media_type = "application/wasm"
#         headers["Content-Encoding"] = "br"
#     elif file_path.endswith(".data.br"):
#         media_type = "application/octet-stream"
#         headers["Content-Encoding"] = "br"
#     elif file_path.endswith(".wasm"):
#         media_type = "application/wasm"
    
#     return FileResponse(file_location, media_type=media_type, headers=headers)

###############

@app.get("/")
async def read_index():
    return FileResponse("dist/index.html", headers={
        "Cross-Origin-Opener-Policy": "same-origin",
        "Cross-Origin-Embedder-Policy": "require-corp"
    })

app.mount("/", StaticFiles(directory="dist"), name="root")

if __name__ == "__main__":
    import uvicorn
    print("Starting server on http://localhost:8000")
    uvicorn.run(app, host="0.0.0.0", port=8000)