58 lines
1.5 KiB
Python
58 lines
1.5 KiB
Python
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, geo_headers
|
|
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)
|
|
|
|
for k, v in geo_headers(data).items():
|
|
response.headers[k] = v
|
|
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
|
|
)
|