from fastapi import APIRouter, Depends, HTTPException
from sqlalchemy.orm import Session
from sqlalchemy import text
from app.api.db import get_db
from pydantic import BaseModel
from app.settings import WHITELIST_TABLE, WHITELIST_COLUMN

router = APIRouter()

# 请求模型
class EmailRequest(BaseModel):
    email: str

# 检查邮箱是否在白名单中
@router.post("/check-email")
async def check_email_in_whitelist(request: EmailRequest, db: Session = Depends(get_db)):
    try:
        # 构建查询语句
        query = text(f"SELECT COUNT(*) FROM {WHITELIST_TABLE} WHERE {WHITELIST_COLUMN} = :email")
        
        # 执行查询
        result = db.execute(query, {"email": request.email})
        count = result.scalar()
        
        if count > 0:
            return {"status": "success", "message": "继续操作"}
        else:
            raise HTTPException(status_code=403, detail="邮箱不在白名单中，请联系管理员")
            
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"服务器错误：{str(e)}")