main.py

import asyncio
import logging
from contextlib import asynccontextmanager
from datetime import date

from slowapi import Limiter, _rate_limit_exceeded_handler
from slowapi.util import get_remote_address
from slowapi.errors import RateLimitExceeded


def _get_real_ip(request):
    # Cloudflare passes real IP in CF-Connecting-IP
    return request.headers.get("CF-Connecting-IP") or get_remote_address(request)

limiter = Limiter(key_func=_get_real_ip)

logging.basicConfig(
    level=logging.INFO,
    format="%(levelname)s:     %(name)s - %(message)s",
)

from fastapi import FastAPI
from fastapi.requests import Request
from fastapi.responses import HTMLResponse, FileResponse
from fastapi.staticfiles import StaticFiles
from fastapi.templating import Jinja2Templates

from config import VAPID_PUBLIC_KEY
from db.database import init_db
from routers import rates, alerts
from services.monitor import poll_loop


@asynccontextmanager
async def lifespan(app: FastAPI):
    await init_db()
    task = asyncio.create_task(poll_loop())
    yield
    task.cancel()


app = FastAPI(lifespan=lifespan)
app.state.limiter = limiter
app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler)
app.mount("/static", StaticFiles(directory="static"), name="static")
templates = Jinja2Templates(directory="templates")

app.include_router(rates.router)
app.include_router(alerts.router)


@app.get("/sw.js")
async def service_worker():
    return FileResponse("static/sw.js", media_type="application/javascript")


@app.get("/", response_class=HTMLResponse)
async def index(request: Request):
    return templates.TemplateResponse(request, "index.html", {
        "vapid_public_key": VAPID_PUBLIC_KEY,
        "today": date.today().isoformat(),
    })