from fastapi.requests import Request

from config import conf
from core.app import app


@app.middleware("http")
async def prod_middleware(request: Request, call_next):
    if (
            request.url.hostname == "scm.haco.dc"
            or request.url.hostname == "scm.hacoupian.net"
            or request.url.hostname == "rc.hacoupian.net"
    ):
        conf.server.is_prod = True
    else:
        conf.server.is_prod = False
    response = await call_next(request)
    response.headers["is-host"] = request.url.hostname
    response.headers["is-prod"] = str(conf.server.is_prod)
    return response