e21f5dfd50
matrix-nio's AsyncClient.set_presence() is called every 60s in a background task so Matrix clients see the bot as online. On shutdown the bot sets itself to offline. https://claude.ai/code/session_01DuzWyMMXvLMB4VxEwJyV4X
48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
import logging
|
|
from contextlib import asynccontextmanager
|
|
|
|
from fastapi import FastAPI
|
|
|
|
from app.database import close_db, get_thread_root, init_db, set_thread_root
|
|
from app.matrix import THREAD_ROOT_MESSAGES, close_client, send_thread_root, start_presence
|
|
from app.webhooks import router as webhook_router
|
|
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format="%(asctime)s %(levelname)s %(name)s: %(message)s",
|
|
)
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
log.info("Starting Melora")
|
|
await init_db()
|
|
await _ensure_thread_roots()
|
|
await start_presence()
|
|
log.info("Melora ready")
|
|
yield
|
|
log.info("Shutting down Melora")
|
|
await close_client()
|
|
await close_db()
|
|
|
|
|
|
async def _ensure_thread_roots() -> None:
|
|
for media_type in THREAD_ROOT_MESSAGES:
|
|
existing = await get_thread_root(media_type)
|
|
if existing:
|
|
log.info("Thread root for %s: %s", media_type, existing)
|
|
continue
|
|
log.info("Creating thread root for %s", media_type)
|
|
event_id = await send_thread_root(media_type)
|
|
await set_thread_root(media_type, event_id)
|
|
|
|
|
|
app = FastAPI(title="Melora", lifespan=lifespan)
|
|
app.include_router(webhook_router)
|
|
|
|
|
|
@app.get("/health")
|
|
async def health():
|
|
return {"status": "ok"}
|