spacebar/SplatNet-Backend
spacebar/SplatNet-Backend
Watch
13
0
Fork
You've already forked SplatNet-Backend
 1
Code Issues 1 Pull requests 5 Projects Releases Packages Wiki Activity

feat: forward /api/v1/post to /post in judd

Browse source
This commit is contained in:
kittentm 2026-03-30 21:37:00 +02:00
commit 3b40d565a4
No known key found for this signature in database
GPG key ID: 520EDAA0C8CDE295
2 changed files with 44 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

71
main.py
View file

@ -119,36 +119,49 @@ app.add_middleware(
@app.middleware("http")
async def proxy_fallback(request: Request, call_next):
response = await call_next(request)
path = request.url.path
if path == "/api/v1/post":
path = "/post"
url = f"http://127.0.0.1:{settings.judd_port}{path}"
body = await request.body()
async with httpx.AsyncClient() as client:
try:
judd_resp = await client.request(
request.method,
url,
headers={k: v for k, v in request.headers.items() if k.lower() != "host"},
params=request.query_params,
content=body,
)
response = Response(
content=judd_resp.content,
status_code=judd_resp.status_code,
headers=dict(judd_resp.headers),
)
except httpx.RequestError:
response = await call_next(request)
else:
response = await call_next(request)
if response.status_code == 404:
url = f"http://127.0.0.1:{settings.judd_port}{path}"
body = await request.body()
async with httpx.AsyncClient() as client:
try:
judd_resp = await client.request(
request.method,
url,
headers={k: v for k, v in request.headers.items() if k.lower() != "host"},
params=request.query_params,
content=body,
)
response = Response(
content=judd_resp.content,
status_code=judd_resp.status_code,
headers=dict(judd_resp.headers),
)
except httpx.RequestError:
pass
if response.status_code != 404:
return response
url = f"http://127.0.0.1:{settings.judd_port}{request.url.path}"
body = await request.body()
async with httpx.AsyncClient() as client:
try:
judd_resp = await client.request(
request.method,
url,
headers={k: v for k, v in request.headers.items() if k.lower() != "host"},
params=request.query_params,
content=body,
)
except httpx.RequestError:
return response
return Response(
content=judd_resp.content,
status_code=judd_resp.status_code,
headers=dict(judd_resp.headers),
)
@app.middleware("http")
async def force_cors_on_errors(request: Request, call_next):
response = await call_next(request)
origin = request.headers.get("origin")
if origin == settings.frontend_url:
response.headers["Access-Control-Allow-Origin"] = origin