Merge pull request #78 from TukaTek/feature/middleware/TUK-112

Feature/middleware/tuk 112

Author: AnkitTukatek

backend/alembic/versions/0a98909f2757_enable_encrypted_fields.py

@@ -5,7 +5,6 @@
 Create Date: 2024-05-05 19:30:34.317972
 
 """
-
 from alembic import op
 import sqlalchemy as sa
 from sqlalchemy.sql import table

backend/middleware/Dockerfile

@@ -0,0 +1,17 @@
+# Use an official Python runtime as a parent image
+FROM python:3.11-slim
+
+# Set working directory inside the container
+WORKDIR /app
+
+# Copy requirements file and install dependencies
+RUN pip install fastapi uvicorn httpx
+
+# Copy the application code
+COPY . /app
+
+# Expose the port the app runs on
+EXPOSE 8000
+
+# Start the FastAPI app with uvicorn
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]

backend/middleware/main.py

@@ -0,0 +1,137 @@
+# import logging
+# from fastapi import FastAPI, Request, Response, HTTPException
+# from fastapi.responses import RedirectResponse, JSONResponse
+# import os
+
+# # Configure logging
+# logging.basicConfig(level=logging.INFO)
+# logger = logging.getLogger(__name__)
+
+# app = FastAPI()
+
+# MIDDLEWARE_BACKEND_URL = os.getenv("MIDDLEWARE_BACKEND_URL")  # e.g. http://middleware_endpoint_handler:5000
+# API_SERVER_URL = os.getenv("API_SERVER_URL")  # e.g. http://api_server_gpt_5:8080
+
+# @app.get("/health")
+# async def health_check():
+#     return {"status": "healthy", "service": "middleware_route_filter"}
+
+# @app.middleware("http")
+# async def route_redirect(request: Request, call_next):
+#     try:
+#         path = request.url.path
+#         query = request.url.query
+
+#         # Block docs endpoints to avoid exposing backend docs
+#         if path in ["/docs", "/openapi.json", "/api/docs", "/api/openapi.json"]:
+#             raise HTTPException(status_code=404, detail="Not found")
+
+#         # Allow health endpoint locally
+#         if path == "/health":
+#             return await call_next(request)
+        
+#         logger.info(f"Raw incoming path: {path}")
+
+
+#         # Decide target backend URL based on path
+#         if path.startswith("/middleware"):
+#             target_url = f"{MIDDLEWARE_BACKEND_URL}{path[len('/middleware'):] or '/'}"
+#         elif path.startswith("/api"):
+#             target_url = f"{API_SERVER_URL}{path}"
+#         else:
+#             target_url = f"{API_SERVER_URL}{path}"
+
+#         # Append query parameters if present
+#         if query:
+#             target_url = f"{target_url}?{query}"
+
+#         logger.info(f"Redirecting {path} to {target_url}")
+
+#         # Redirect client to backend server URL
+#         return RedirectResponse(url=target_url)
+
+#     except HTTPException as http_exc:
+#         logger.warning(f"HTTPException: {http_exc.detail} for path {request.url.path}")
+#         return JSONResponse(status_code=http_exc.status_code, content={"detail": http_exc.detail})
+
+#     except Exception as exc:
+#         logger.error(f"Unhandled error: {exc}")
+#         return JSONResponse(status_code=500, content={"detail": "Internal server error"})
+
+import os
+import logging
+from fastapi import FastAPI, Request, HTTPException
+from fastapi.responses import JSONResponse, Response
+import httpx
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+app = FastAPI()
+
+# Set these as ENV vars in your docker-compose for the middleware container!
+MIDDLEWARE_BACKEND_URL = os.getenv("MIDDLEWARE_BACKEND_URL", "http://middleware_endpoint_handler:5000")
+API_SERVER_URL = os.getenv("API_SERVER_URL", "http://api_server_gpt_5:8080")
+
+@app.get("/health")
+async def health_check():
+    return {"status": "healthy", "service": "middleware_route_filter"}
+
+@app.middleware("http")
+async def proxy_middleware(request: Request, call_next):
+    try:
+        path = request.url.path
+        query = request.url.query
+
+        # Block docs endpoints to avoid exposing backend docs
+        if path in ["/docs", "/openapi.json", "/api/docs", "/api/openapi.json"]:
+            raise HTTPException(status_code=404, detail="Not found")
+
+        if path == "/health":
+            return await call_next(request)
+
+        logger.info(f"Raw incoming path: {path}")
+
+        # Decide target backend URL based on path
+        if path.startswith("/middleware"):
+            target_url = f"{MIDDLEWARE_BACKEND_URL}{path[len('/middleware'):] or '/'}"
+        else:
+            target_url = f"{API_SERVER_URL}{path}"
+
+        if query:
+            target_url = f"{target_url}?{query}"
+
+        logger.info(f"Proxying {path} to {target_url}")
+
+        # Prepare headers (remove host header to avoid conflicts)
+        headers = dict(request.headers)
+        headers.pop("host", None)
+
+        # Proxy the request using httpx
+        async with httpx.AsyncClient() as client:
+            proxied_response = await client.request(
+                method=request.method,
+                url=target_url,
+                headers=headers,
+                content=await request.body(),
+                timeout=None  # Consider setting a reasonable timeout
+            )
+
+        # Copy headers except for transfer-encoding, content-encoding, connection
+        excluded_headers = {"content-encoding", "transfer-encoding", "connection"}
+        response_headers = {k: v for k, v in proxied_response.headers.items() if k.lower() not in excluded_headers}
+
+        return Response(
+            content=proxied_response.content,
+            status_code=proxied_response.status_code,
+            headers=response_headers,
+            media_type=proxied_response.headers.get("content-type")
+        )
+
+    except HTTPException as http_exc:
+        logger.warning(f"HTTPException: {http_exc.detail} for path {request.url.path}")
+        return JSONResponse(status_code=http_exc.status_code, content={"detail": http_exc.detail})
+    except Exception as exc:
+        logger.error(f"Unhandled error: {exc}")
+        return JSONResponse(status_code=500, content={"detail": "Internal server error"})