44 lines
1.6 KiB
Python
44 lines
1.6 KiB
Python
from contextlib import asynccontextmanager
|
|
|
|
from fastapi import FastAPI
|
|
|
|
from app.auth.router import router as auth_router
|
|
from app.chats.router import router as chats_router
|
|
from app.config.settings import settings
|
|
from app.database import models # noqa: F401
|
|
from app.database.base import Base
|
|
from app.database.session import engine
|
|
from app.media.router import router as media_router
|
|
from app.messages.router import router as messages_router
|
|
from app.notifications.router import router as notifications_router
|
|
from app.realtime.router import router as realtime_router
|
|
from app.realtime.service import realtime_gateway
|
|
from app.users.router import router as users_router
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(_app: FastAPI):
|
|
await realtime_gateway.start()
|
|
if settings.auto_create_tables:
|
|
async with engine.begin() as conn:
|
|
await conn.run_sync(Base.metadata.create_all)
|
|
yield
|
|
await realtime_gateway.stop()
|
|
|
|
|
|
app = FastAPI(title=settings.app_name, debug=settings.debug, lifespan=lifespan)
|
|
|
|
|
|
@app.get("/health", tags=["health"])
|
|
async def health() -> dict[str, str]:
|
|
return {"status": "ok"}
|
|
|
|
|
|
app.include_router(auth_router, prefix=settings.api_v1_prefix)
|
|
app.include_router(users_router, prefix=settings.api_v1_prefix)
|
|
app.include_router(chats_router, prefix=settings.api_v1_prefix)
|
|
app.include_router(messages_router, prefix=settings.api_v1_prefix)
|
|
app.include_router(media_router, prefix=settings.api_v1_prefix)
|
|
app.include_router(notifications_router, prefix=settings.api_v1_prefix)
|
|
app.include_router(realtime_router, prefix=settings.api_v1_prefix)
|