hms-licht-ton/backend/app/main.py

"""FastAPI application assembly with CORS, routers, and APScheduler."""
import logging
from contextlib import asynccontextmanager
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from apscheduler.schedulers.asyncio import AsyncIOScheduler

from app.config import get_settings
from app.database import engine, init_db, async_session
from app.cache import cache
from app.routers import equipment, health, admin, rental_requests, contact
from app.services.sync_service import SyncService
from app.services.email_service import EmailService

logger = logging.getLogger(__name__)
settings = get_settings()

scheduler = AsyncIOScheduler()


async def run_equipment_sync() -> None:
    """Scheduled job: run equipment sync every 6 hours."""
    logger.info("Starting scheduled equipment sync")
    async with async_session() as db:
        sync_service = SyncService(db)
        result = await sync_service.run_sync()
        logger.info("Scheduled sync complete: %s", result)


async def run_email_retry() -> None:
    """Scheduled job: retry failed emails every 15 minutes."""
    logger.info("Starting scheduled email retry")
    async with async_session() as db:
        sent = await EmailService.retry_failed_emails(db)
        logger.info("Email retry complete: %d emails sent", sent)


@asynccontextmanager
async def lifespan(app: FastAPI):
    """Application lifespan: init DB, start scheduler, connect cache."""
    # Create tables (for dev/testing; production uses Alembic)
    await init_db()

    # Connect Redis cache
    await cache.connect()

    # Start APScheduler
    scheduler.add_job(
        run_equipment_sync,
        trigger="cron",
        hour="*/6",
        id="equipment_sync",
        replace_existing=True,
    )
    scheduler.add_job(
        run_email_retry,
        trigger="cron",
        minute="*/15",
        id="email_retry",
        replace_existing=True,
    )
    scheduler.start()
    logger.info("APScheduler started with equipment_sync (every 6h) and email_retry (every 15min)")

    yield

    # Shutdown
    scheduler.shutdown(wait=False)
    await cache.disconnect()
    await engine.dispose()


app = FastAPI(
    title="HMS Licht & Ton API",
    version="1.0.0",
    lifespan=lifespan,
)

# CORS
origins = [o.strip() for o in settings.cors_origins.split(",") if o.strip()]
app.add_middleware(
    CORSMiddleware,
    allow_origins=origins,
    allow_methods=["GET", "POST"],
    allow_headers=["*"],
    allow_credentials=True,
)

# Routers
app.include_router(equipment.router)
app.include_router(health.router)
app.include_router(admin.router)
app.include_router(rental_requests.router)
app.include_router(contact.router)