"""Pydantic schemas for API requests and responses."""
from datetime import datetime
from typing import Optional
from pydantic import BaseModel, Field


class BibTeXVerifyRequest(BaseModel):
    """Request model for BibTeX verification."""

    bibtex_content: str = Field(
        ...,
        description="BibTeX content to verify",
        min_length=1,
        examples=[
            """@article{example2023,
  title={Example Paper Title},
  author={Smith, John and Doe, Jane},
  journal={Example Journal},
  year={2023}
}"""
        ],
    )


class EntryComparisonResponse(BaseModel):
    """Response model for a single entry comparison."""

    key: str = Field(..., description="BibTeX entry key")
    status: str = Field(..., description="Verification status: verified, warning, or error")
    is_match: bool = Field(..., description="Whether entry matches a database record")
    has_issues: bool = Field(..., description="Whether entry has metadata issues")
    source: Optional[str] = Field(None, description="Data source that verified the entry")
    confidence: float = Field(..., description="Confidence score (0-1)")
    title_match: bool = Field(..., description="Whether title matches")
    author_match: bool = Field(..., description="Whether authors match")
    year_match: bool = Field(..., description="Whether year matches")
    venue_match: Optional[bool] = Field(None, description="Whether venue matches")
    fetched_title: Optional[str] = Field(None, description="Title from database")
    fetched_authors: Optional[list[str]] = Field(None, description="Authors from database")
    fetched_year: Optional[str] = Field(None, description="Year from database")
    fetched_doi: Optional[str] = Field(None, description="DOI from database")
    fetched_url: Optional[str] = Field(None, description="URL from database")
    original_bibtex: str = Field(..., description="Original BibTeX entry")


class DuplicateGroupResponse(BaseModel):
    """Response model for duplicate entry groups."""

    entry_keys: list[str] = Field(..., description="Keys of duplicate entries")
    reason: str = Field(..., description="Reason for duplicate detection")


class BibTeXVerifyResponse(BaseModel):
    """Response model for BibTeX verification."""

    success: bool = Field(..., description="Whether verification completed successfully")
    message: str = Field(..., description="Status message")
    total_count: int = Field(..., description="Total number of entries")
    verified_count: int = Field(..., description="Number of verified entries")
    warning_count: int = Field(..., description="Number of entries with warnings")
    error_count: int = Field(..., description="Number of entries with errors")
    success_rate: float = Field(..., description="Success rate percentage")
    entries: list[EntryComparisonResponse] = Field(..., description="Verification results for each entry")
    duplicate_groups: list[DuplicateGroupResponse] = Field(
        default_factory=list, description="Groups of duplicate entries"
    )
    timestamp: datetime = Field(default_factory=datetime.utcnow, description="Verification timestamp")


class HealthResponse(BaseModel):
    """Response model for health check."""

    status: str = Field(..., description="Service status")
    version: str = Field(..., description="Application version")
    environment: str = Field(..., description="Runtime environment")
    timestamp: datetime = Field(default_factory=datetime.utcnow, description="Current timestamp")


class StatsResponse(BaseModel):
    """Response model for statistics."""

    cache_enabled: bool = Field(..., description="Whether cache is enabled")
    cache_size: int = Field(..., description="Current cache size")
    cache_max_size: int = Field(..., description="Maximum cache size")
    cache_ttl: int = Field(..., description="Cache TTL in seconds")


class ErrorResponse(BaseModel):
    """Response model for errors."""

    error: str = Field(..., description="Error type")
    message: str = Field(..., description="Error message")
    details: Optional[dict] = Field(None, description="Additional error details")
    timestamp: datetime = Field(default_factory=datetime.utcnow, description="Error timestamp")