mirror of
https://github.com/Lolendor/reVCDOS.git
synced 2026-10-10 11:53:11 +08:00
Alexey Olendor
89 lines
3.4 KiB
Python
89 lines
3.4 KiB
Python
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)
|