from fastapi import FastAPI, Request, Response
from fastapi.responses import JSONResponse, PlainTextResponse
from starlette.middleware.base import BaseHTTPMiddleware
from .deps import get_geo
from .api import get_client_ip, router
from .config import settings
import uvicorn

app = FastAPI(title="IP Geo API")
app.include_router(router)


async def add_geo_headers(request, call_next):
    ip = get_client_ip(request)
    geo = get_geo()
    data = geo.lookup(ip)

    response: Response = await call_next(request)

    country = data.get("country", {}).get("name") if data.get("country") else None
    city = data.get("city")
    ip_val = data.get("ip")

    if ip_val and country:
        response.headers["X-IP-ADDRESS"] = ip_val
        response.headers["X-COUNTRY"] = country
        if city:
            response.headers["X-CITY"] = city

    return response


app.add_middleware(BaseHTTPMiddleware, dispatch=add_geo_headers)


@app.get("/favicon.ico")
async def favicon():
    return Response(status_code=204)


@app.api_route("/", methods=["GET", "HEAD"])
async def root(request: Request):
    ua = request.headers.get("user-agent", "").lower()
    ip = get_client_ip(request).strip()

    if any(x in ua for x in ["mozilla", "chrome", "safari", "edge", "firefox"]):
        if request.method == "HEAD":
            return Response(status_code=404)
        return JSONResponse({"detail": "Not Found"}, status_code=404)

    if request.method == "HEAD":
        return Response(status_code=200)
    return PlainTextResponse(ip + "\n")


if __name__ == "__main__":
    uvicorn.run(
        "app.main:app",
        host=settings.host,
        port=settings.port,
        log_level=settings.log_level,
        proxy_headers=True,
        forwarded_allow_ips="*",
        # access_log=True
    )