From db169159e6009727a91f1042cfd12b5c9ebe5209 Mon Sep 17 00:00:00 2001
From: Anson <anson@ansonbiggs.com>
Date: Fri, 23 May 2025 15:31:59 -0600
Subject: [PATCH] Add Proxy Headers

---
 Dockerfile                 | 2 +-
 src/two_chainz/__init__.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index d39baaa..4bf00bf 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -22,4 +22,4 @@ COPY . /app
 RUN --mount=type=cache,target=/root/.cache/uv \
     uv sync --locked --no-dev
 
-CMD ["uv", "run", "fastapi", "run", "src/two_chainz", "--port", "80"]
+CMD ["uv", "run", "fastapi", "run", "src/two_chainz", "--port", "80", "--proxy-headers"]
diff --git a/src/two_chainz/__init__.py b/src/two_chainz/__init__.py
index 3e1ae66..8497b7f 100644
--- a/src/two_chainz/__init__.py
+++ b/src/two_chainz/__init__.py
@@ -10,7 +10,7 @@ from fastapi.middleware.trustedhost import TrustedHostMiddleware
 
 app = FastAPI()
 app.add_middleware(
-    TrustedHostMiddleware, allowed_hosts=["ansonbiggs.com", "*.ansonbiggs.com"]
+    TrustedHostMiddleware, allowed_hosts=["ansonbiggs.com", "*.ansonbiggs.com", "localhost"]
 )
 
 start_time = time.time()