g0v0-server/main.py

from __future__ import annotations

import asyncio
from contextlib import asynccontextmanager
from datetime import datetime
import os

from app.config import settings
from app.dependencies.database import engine, redis_client
from app.dependencies.fetcher import get_fetcher
from app.dependencies.scheduler import start_scheduler, stop_scheduler
from app.log import logger
from app.router import (
    api_v1_router,
    api_v2_router,
    auth_router,
    chat_router,
    fetcher_router,
    file_router,
    private_router,
    redirect_api_router,
    signalr_router,
)
from app.router.redirect import redirect_router
from app.scheduler.cache_scheduler import start_cache_scheduler, stop_cache_scheduler
from app.service.beatmap_download_service import download_service
from app.service.calculate_all_user_rank import calculate_user_rank
from app.service.create_banchobot import create_banchobot
from app.service.daily_challenge import daily_challenge_job, process_daily_challenge_top
from app.service.geoip_scheduler import schedule_geoip_updates
from app.service.init_geoip import init_geoip
from app.service.load_achievements import load_achievements
from app.service.osu_rx_statistics import create_rx_statistics
from app.service.recalculate import recalculate
from app.service.redis_message_system import redis_message_system

# 检查 New Relic 配置文件是否存在，如果存在则初始化 New Relic
newrelic_config_path = os.path.join(os.path.dirname(__file__), "newrelic.ini")
if os.path.exists(newrelic_config_path):
    try:
        import newrelic.agent

        environment = os.environ.get(
            "NEW_RELIC_ENVIRONMENT",
            "production" if not settings.debug else "development",
        )

        newrelic.agent.initialize(newrelic_config_path, environment)
        logger.info(f"[NewRelic] Enabled, environment: {environment}")
    except ImportError:
        logger.warning(
            "[NewRelic] Config file found but 'newrelic' package is not installed"
        )
    except Exception as e:
        logger.error(f"[NewRelic] Initialization failed: {e}")
else:
    logger.info("[NewRelic] No newrelic.ini config file found, skipping initialization")


from fastapi import FastAPI, HTTPException, Request
from fastapi.exceptions import RequestValidationError
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import JSONResponse
import sentry_sdk


@asynccontextmanager
async def lifespan(app: FastAPI):
    # on startup
    await get_fetcher()  # 初始化 fetcher
    await init_geoip()  # 初始化 GeoIP 数据库
    await create_rx_statistics()
    await calculate_user_rank(True)
    start_scheduler()
    schedule_geoip_updates()  # 调度 GeoIP 定时更新任务
    await daily_challenge_job()
    await process_daily_challenge_top()
    await create_banchobot()
    await download_service.start_health_check()  # 启动下载服务健康检查
    await start_cache_scheduler()  # 启动缓存调度器
    redis_message_system.start()  # 启动 Redis 消息系统
    load_achievements()
    # on shutdown
    yield
    stop_scheduler()
    redis_message_system.stop()  # 停止 Redis 消息系统
    await stop_cache_scheduler()  # 停止缓存调度器
    await download_service.stop_health_check()  # 停止下载服务健康检查
    await engine.dispose()
    await redis_client.aclose()


desc = (
    "osu! API 模拟服务器，支持 osu! API v1, v2 和 osu!lazer 的绝大部分功能。\n\n"
    "官方文档：[osu!web 文档](https://osu.ppy.sh/docs/index.html)\n\n"
    "V1 API 文档：[osu-api](https://github.com/ppy/osu-api/wiki)"
)

if settings.sentry_dsn is not None:
    sentry_sdk.init(
        dsn=str(settings.sentry_dsn),
        send_default_pii=False,
        environment="production" if not settings.debug else "development",
    )

app = FastAPI(
    title="g0v0-server",
    version="0.1.0",
    lifespan=lifespan,
    description=desc,
)

app.include_router(api_v2_router)
app.include_router(api_v1_router)
app.include_router(chat_router)
app.include_router(redirect_api_router)
app.include_router(signalr_router)
app.include_router(fetcher_router)
app.include_router(file_router)
app.include_router(auth_router)
app.include_router(private_router)

# CORS 配置
origins = []
for url in [*settings.cors_urls, settings.server_url]:
    origins.append(str(url))
    origins.append(str(url).removesuffix("/"))
if settings.frontend_url:
    origins.append(str(settings.frontend_url))
    origins.append(str(settings.frontend_url).removesuffix("/"))
app.add_middleware(
    CORSMiddleware,
    allow_origins=origins,
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

if settings.frontend_url is not None:
    app.include_router(redirect_router)


@app.get("/", include_in_schema=False)
async def root():
    """根端点"""
    return {"message": "osu! API 模拟服务器正在运行"}


@app.get("/health", include_in_schema=False)
async def health_check():
    """健康检查端点"""
    return {"status": "ok", "timestamp": datetime.utcnow().isoformat()}


@app.exception_handler(RequestValidationError)
async def validation_exception_handler(request: Request, exc: RequestValidationError):
    return JSONResponse(
        status_code=422,
        content={
            "error": exc.errors(),
        },
    )


@app.exception_handler(HTTPException)
async def http_exception_handler(requst: Request, exc: HTTPException):
    return JSONResponse(status_code=exc.status_code, content={"error": exc.detail})


if settings.secret_key == "your_jwt_secret_here":
    logger.warning(
        "jwt_secret_key is unset. Your server is unsafe. "
        "Use this command to generate: openssl rand -hex 32"
    )
if settings.osu_web_client_secret == "your_osu_web_client_secret_here":
    logger.warning(
        "osu_web_client_secret is unset. Your server is unsafe. "
        "Use this command to generate: openssl rand -hex 40"
    )

if __name__ == "__main__":
    import uvicorn

    if os.environ.get("RECALCULATE", "false").lower() == "true":
        asyncio.run(recalculate())
        exit()

    uvicorn.run(
        "main:app",
        host=settings.host,
        port=settings.port,
        reload=settings.debug,
        log_config=None,  # 禁用uvicorn默认日志配置
        access_log=True,  # 启用访问日志
    )