#!/usr/bin/env python3
"""
数据模型定义
Pydantic模型用于API请求和响应
"""

from typing import Dict, List, Optional, Union
from pydantic import BaseModel, EmailStr

# ============================================================================
# 请求模型
# ============================================================================

class EmailVerifyRequest(BaseModel):
    email: EmailStr

class EmailListRequest(BaseModel):
    email: EmailStr
    top: int = 5

class EmailDetailRequest(BaseModel):
    email: EmailStr
    message_id: str

class AccountCredentials(BaseModel):
    email: EmailStr
    password: str
    client_id: str
    refresh_token: str

class ImportAccountData(BaseModel):
    """单个导入账户数据模型"""
    email: str  # 暂时使用str而不EmailStr避免验证问题
    password: str = ""
    client_id: str = ""
    refresh_token: str

class ImportRequest(BaseModel):
    """批量导入请求模型"""
    accounts: List[ImportAccountData]
    merge_mode: str = "update"  # "update": 更新现有账户, "skip": 跳过重复账户, "replace": 替换所有数据

class ParsedImportRequest(BaseModel):
    """解析后的导入请求模型（包含解析统计信息）"""
    accounts: List[ImportAccountData]
    parsed_count: int
    error_count: int
    errors: List[str]
    merge_mode: str = "update"

class ImportResult(BaseModel):
    """导入结果模型"""
    success: bool
    total_count: int
    added_count: int
    updated_count: int
    skipped_count: int
    error_count: int
    details: List[Dict[str, str]]  # 详细信息
    message: str

class AdminTokenRequest(BaseModel):
    """管理令牌验证请求"""
    token: str

class DeleteAccountRequest(BaseModel):
    """删除账户请求"""
    email: EmailStr

class TempAccountRequest(BaseModel):
    """临时账户请求"""
    email: EmailStr
    password: str = ""
    client_id: str = ""
    refresh_token: str
    top: int = 5
    folder: str = "INBOX"

class TempMessageDetailRequest(BaseModel):
    """临时账户邮件详情请求"""
    email: EmailStr
    password: str = ""
    client_id: str = ""
    refresh_token: str
    message_id: str

class SystemConfigRequest(BaseModel):
    """系统配置请求"""
    email_limit: int = 5

class AccountTagRequest(BaseModel):
    """账户标签请求"""
    email: EmailStr
    tags: List[str] = []

class TestEmailRequest(BaseModel):
    """测试邮件请求模型"""
    email: EmailStr
    password: str = ""
    client_id: str = ""
    refresh_token: str = ""

# ============================================================================
# 响应模型
# ============================================================================

class ApiResponse(BaseModel):
    success: bool
    message: str = ""
    data: Optional[Union[Dict, List, str]] = None

class EmailMessage(BaseModel):
    """邮件消息模型"""
    id: str
    subject: str
    receivedDateTime: str
    sender: Dict
    from_: Optional[Dict] = None
    body: Dict
    bodyPreview: str = ""
    toRecipients: Optional[List[Dict]] = None
    
    class Config:
        fields = {'from_': 'from'}