from __future__ import annotations

from fastapi import APIRouter, Depends, HTTPException, Path
from sqlalchemy.ext.asyncio import AsyncSession

from app.dependencies import get_admin_user, get_db
from app.models.user import User
from app.models.virtual_portrait_v3.api_key_quota import VpV3ApiKeyQuota
from app.schemas.admin_api.vp_v3_quota import (
    VpV3QuotaConfigData,
    VpV3QuotaConfigResponse,
)
from app.services.api_v3 import key_service
from app.services.virtual_portrait_v3.quota_service import get_quota

router = APIRouter(prefix="/admin/api-keys", tags=["admin-vp-v3-quota"])


def _to_response(quota: VpV3ApiKeyQuota) -> VpV3QuotaConfigResponse:
    enabled = any([
        (quota.project_limit or 0) > 0,
        (quota.asset_limit or 0) > 0,
        (quota.storage_mb_limit or 0) > 0,
    ])
    return VpV3QuotaConfigResponse(
        api_key_id=quota.api_key_id,
        project_limit=int(quota.project_limit or 0),
        asset_limit=int(quota.asset_limit or 0),
        storage_mb_limit=int(quota.storage_mb_limit or 0),
        remark=quota.remark,
        project_used=int(quota.project_used or 0),
        asset_used=int(quota.asset_used or 0),
        storage_mb_used=float(quota.storage_mb_used or 0),
        enabled=enabled,
    )


@router.get(
    "/{key_id}/vp-v3-quota",
    response_model=VpV3QuotaConfigResponse,
    summary="获取 API Key 的虚拟素材库配额配置",
    description="返回指定 API Key 的虚拟素材库配额上限及当前使用量。不存在配额记录时自动创建默认 0 值。",
)
async def get_vp_v3_quota(
    key_id: str = Path(..., description="API Key ID"),
    admin: User = Depends(get_admin_user),
    db: AsyncSession = Depends(get_db),
) -> VpV3QuotaConfigResponse:
    key = await key_service.get_api_key(db, key_id)
    if not key:
        raise HTTPException(status_code=404, detail="API Key 不存在")
    quota = await get_quota(db, api_key_id=key_id, refresh=True)
    await db.commit()
    return _to_response(quota)


@router.post(
    "/{key_id}/vp-v3-quota",
    response_model=VpV3QuotaConfigResponse,
    summary="保存 API Key 的虚拟素材库配额配置",
    description="保存虚拟素材库配额（项目数/素材数/存储 MB），默认 0=不可使用该功能。保存后自动刷新已使用量。",
)
async def save_vp_v3_quota(
    payload: VpV3QuotaConfigData,
    key_id: str = Path(..., description="API Key ID"),
    admin: User = Depends(get_admin_user),
    db: AsyncSession = Depends(get_db),
) -> VpV3QuotaConfigResponse:
    key = await key_service.get_api_key(db, key_id)
    if not key:
        raise HTTPException(status_code=404, detail="API Key 不存在")
    quota = await get_quota(db, api_key_id=key_id, refresh=True)
    quota.project_limit = int(payload.project_limit or 0)
    quota.asset_limit = int(payload.asset_limit or 0)
    quota.storage_mb_limit = int(payload.storage_mb_limit or 0)
    quota.remark = payload.remark if payload.remark is not None else quota.remark
    await db.flush()
    await db.refresh(quota)
    await db.commit()
    return _to_response(quota)