From 7ac16031bb59d5362a46ccd8aacd859518ad67df Mon Sep 17 00:00:00 2001
From: Daniel O'Connell <github@ahiru.pl>
Date: Wed, 4 Jun 2025 02:52:42 +0200
Subject: [PATCH] better proxy

---
 docker-compose.yaml   | 6 +-----
 tools/simple_proxy.py | 6 +++++-
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/docker-compose.yaml b/docker-compose.yaml
index 4f62f0b..a42f1bc 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -162,13 +162,9 @@ services:
       dockerfile: docker/api/Dockerfile
     restart: unless-stopped
     networks: [kbnet]
-    depends_on: [api]
     environment:
       <<: *env
-      PROXY_EMAIL: "${PROXY_EMAIL}"
-      PROXY_PASSWORD: "${PROXY_PASSWORD}"
-      PROXY_REMOTE_SERVER: "http://api:8000"
-    command: ["python", "/app/tools/simple_proxy.py", "--remote-server", "http://api:8000", "--email", "${PROXY_EMAIL}", "--password", "${PROXY_PASSWORD}", "--port", "8001"]
+    command: ["python", "/app/tools/simple_proxy.py", "--remote-server", "${PROXY_REMOTE_SERVER:-http://api:8000}", "--email", "${PROXY_EMAIL}", "--password", "${PROXY_PASSWORD}", "--port", "8001"]
     volumes:
       - ./tools:/app/tools:ro
     ports:
diff --git a/tools/simple_proxy.py b/tools/simple_proxy.py
index 9c48bef..f5812d4 100644
--- a/tools/simple_proxy.py
+++ b/tools/simple_proxy.py
@@ -92,9 +92,13 @@ async def proxy_request(request: Request) -> Response:
             resp = Response(
                 content=response.content,
                 status_code=response.status_code,
-                headers=dict(response.headers),
+                headers={
+                    k: v.replace(state.remote_server, f"http://localhost:{state.port}")
+                    for k, v in response.headers.items()
+                },
                 media_type=response.headers.get("content-type"),
             )
+            print(resp.headers)
             return resp
 
         except httpx.RequestError as e: