License/backend/app/main.py

import asyncio
import logging
from contextlib import asynccontextmanager
from datetime import datetime, timezone

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware

from app.config import settings
from app.routers import domains, monitoring, license as license_router
from app.routers.license import public_router as public_license_router
from app.database import async_session
from app.services.license import _notify_on_expired, _notify_on_warning
from sqlalchemy import select, text
from app.models.license import SuperAdminLicense

logger = logging.getLogger(__name__)


async def _check_licenses():
    """定期检查 License：过期状态更新 + 7 天预警"""
    while True:
        try:
            async with async_session() as session:
                # 1. 检查已过期
                expired_result = await session.execute(
                    select(SuperAdminLicense).where(
                        SuperAdminLicense.status == "active",
                        SuperAdminLicense.expires_at <= text("NOW()"),
                    )
                )
                for lic in expired_result.scalars().all():
                    logger.info("检测到 License #%d 已过期，更新状态并发送预警短信", lic.id)
                    lic.status = "expired"
                    await session.commit()
                    await _notify_on_expired(session, lic)

                # 2. 检查剩余 7 天预警（未发送过的）
                from datetime import timedelta
                warning_result = await session.execute(
                    select(SuperAdminLicense).where(
                        SuperAdminLicense.status == "active",
                        SuperAdminLicense.warning_sent == False,
                        SuperAdminLicense.expires_at > text("NOW()"),
                        SuperAdminLicense.expires_at <= text("NOW() + INTERVAL '7 days'"),
                    )
                )
                for lic in warning_result.scalars().all():
                    days_left = (lic.expires_at - datetime.now(timezone.utc)).days
                    logger.info("检测到 License #%d 剩余 %d 天，发送预警短信", lic.id, days_left)
                    await _notify_on_warning(session, lic, days_left)
                    lic.warning_sent = True
                    await session.commit()
        except Exception:
            logger.exception("定时检查 License 异常")

        # 每 24 小时检查一次
        await asyncio.sleep(24 * 3600)


@asynccontextmanager
async def lifespan(app: FastAPI):
    # 启动时立即检查一次
    try:
        async with async_session() as session:
            # 1. 检查已过期
            expired_result = await session.execute(
                select(SuperAdminLicense).where(
                    SuperAdminLicense.status == "active",
                    SuperAdminLicense.expires_at <= text("NOW()"),
                )
            )
            for lic in expired_result.scalars().all():
                logger.info("启动时检测到 License #%d 已过期，更新状态并发送预警短信", lic.id)
                lic.status = "expired"
                await session.commit()
                await _notify_on_expired(session, lic)

            # 2. 检查剩余 7 天预警（未发送过的）
            from datetime import timedelta
            warning_result = await session.execute(
                select(SuperAdminLicense).where(
                    SuperAdminLicense.status == "active",
                    SuperAdminLicense.warning_sent == False,
                    SuperAdminLicense.expires_at > text("NOW()"),
                    SuperAdminLicense.expires_at <= text("NOW() + INTERVAL '7 days'"),
                )
            )
            for lic in warning_result.scalars().all():
                days_left = (lic.expires_at - datetime.now(timezone.utc)).days
                logger.info("启动时检测到 License #%d 剩余 %d 天，发送预警短信", lic.id, days_left)
                await _notify_on_warning(session, lic, days_left)
                lic.warning_sent = True
                await session.commit()

            logger.info("启动检查完成")
    except Exception:
        logger.exception("启动时检查 License 异常")

    # 启动后台定时任务
    task = asyncio.create_task(_check_licenses())
    logger.info("后台任务已启动：每 24 小时检查一次 License")
    yield
    task.cancel()
    try:
        await task
    except asyncio.CancelledError:
        pass


app = FastAPI(
    title="域名流水监控",
    version="0.1.0",
    debug=settings.debug,
    lifespan=lifespan,
)

# CORS 配置，允许前端 Vite 开发服务器访问
app.add_middleware(
    CORSMiddleware,
    allow_origins=["http://localhost:5173"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# 注册路由
app.include_router(domains.router)
app.include_router(monitoring.router)
app.include_router(license_router.router)
app.include_router(public_license_router)


@app.get("/health")
async def health():
    """健康检查接口"""
    return {"status": "ok"}