Initial draft

src/main.py

@@ -0,0 +1,147 @@
+import asyncio
+from contextlib import asynccontextmanager
+from datetime import UTC, datetime
+
+import uvicorn
+from fastapi import Depends, FastAPI, HTTPException, Request
+from seadex import EntryNotFoundError, SeaDexEntry
+
+from db_module import (
+	close_connection_pool,
+	create_connection_pool,
+	get_db,
+	start_healthcheck_thread,
+)
+from metrics_server import REQUEST_COUNTER, metrics_api
+from secret_manager import cleanup_secret_files
+from send_notification import send_notification
+from simple_logger_handler import LOG_LEVEL, setup_logger
+from uvicorn_logger_config import LOGGING_CONFIG
+
+logger = setup_logger(__name__)
+
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+	logger.info("[App] Starting application...")
+
+	logger.info("[DB] Creating MySQL connection pool...")
+	create_connection_pool("royalRoadPool")
+
+	start_healthcheck_thread()
+	logger.info("[DB] MySQL healthcheck thread started.")
+
+	SECRET_PATHS = frozenset({
+		"/etc/secrets/api_key",
+		"/etc/secrets/db_username",
+		"/etc/secrets/db_password"
+	})
+	cleanup_secret_files(SECRET_PATHS)
+
+	yield
+	logger.info("[App] Closing MySQL connection pool...")
+	close_connection_pool()
+	logger.info("[App] Shutdown complete.")
+
+
+
+api = FastAPI(
+		title="Service SeaDex",
+		description="Queries SeaDex for new torrent information",
+		version="1.0.0",
+		lifespan=lifespan
+)
+
+@api.middleware("http")
+async def prometheus_middleware(request: Request, call_next):
+	logger.debug(f"[Metrics] Incoming request: {request.method} {request.url.path}")
+	status = 500
+	try:
+		response = await call_next(request)
+		status = response.status_code
+		logger.debug(f"[Metrics] Request processed with status {status}")
+	except Exception:
+		logger.exception("[Metrics] Exception occurred:")
+		raise
+	finally:
+		REQUEST_COUNTER.labels(request.method, request.url.path, status).inc()
+		logger.debug(f"[Metrics] Counter incremented for {request.method} {request.url.path} [{status}]")
+	return response
+
+
+def to_iso(dt: datetime) -> str:
+	return dt.astimezone(UTC).isoformat(timespec="microseconds")
+
+
+def check_for_update(seadex: SeaDexEntry, anilist_id: int) -> str | None:
+	try:
+		return to_iso(seadex.from_id(anilist_id).updated_at)
+	except EntryNotFoundError:
+		return None
+
+
+@api.post("/update")
+def check_seadex(db=Depends(get_db)):
+	results = {"checked": 0, "new": 0, "errors": 0}
+
+	with db.cursor() as cursor:
+		cursor.execute(
+			"SELECT id, anilist_id, anime_name, last_updated "
+			"FROM seadex_state WHERE active = 1"
+		)
+		anime = cursor.fetchall()
+
+	with SeaDexEntry() as seadex:
+		for row_id, anilist_id, anime_name, last_updated in anime:
+			results["checked"] += 1
+			try:
+				current = check_for_update(seadex, anilist_id)
+
+				if current is None:
+					logger.debug(f"[FastAPI] No entry yet for {anime_name} on SeaDex")
+					continue
+
+				if current != last_updated:
+					logger.info(
+						f"[FastAPI] {anime_name} updated, Anilist ID: {anilist_id}"
+					)
+					send_notification(anime_name, f"https://seadex.moe/{anilist_id}/")
+
+					with db.cursor() as cursor:
+						cursor.execute(
+							"UPDATE seadex_state SET last_updated = %s WHERE id = %s",
+							(current, row_id),
+						)
+					db.commit()
+
+					logger.debug(
+						f"[FastAPI] Notified and updated db for id={row_id}, "
+						f"Anilist ID={anilist_id}"
+					)
+					results["new"] += 1
+
+			except Exception:
+				db.rollback()
+				results["errors"] += 1
+				logger.exception(f"[FastAPI] Failed for id={row_id} Anilist={anilist_id}",)
+
+	if results["errors"]:
+		logger.warning(f"[FastAPI] SeaDex check finished with errors: {results}")
+		raise HTTPException(status_code=500, detail=results)
+
+	logger.info(f"[FastAPI] SeaDex check complete: {results}")
+	return results
+
+async def start_servers():
+	logger.info("[Server] Starting main API (port 5000) and metrics server (port 9000)...")
+	config_main = uvicorn.Config(api, host="0.0.0.0", port=5000, log_level=LOG_LEVEL.lower(), log_config=LOGGING_CONFIG)
+	config_metrics = uvicorn.Config(metrics_api, host="0.0.0.0", port=9000, log_level=LOG_LEVEL.lower(), log_config=LOGGING_CONFIG)
+
+	server_main = uvicorn.Server(config_main)
+	server_metrics = uvicorn.Server(config_metrics)
+
+	await asyncio.gather(server_main.serve(), server_metrics.serve())
+	logger.info("[Server] Both servers have stopped.")
+
+if __name__ == "__main__":
+	asyncio.run(start_servers())