modcomms/backend/app/api/schemas.py

"""Pydantic schemas for API request/response validation."""
import uuid
from datetime import datetime
from typing import Optional

from pydantic import BaseModel


# Campaign schemas
class CampaignCreate(BaseModel):
    name: str
    workfront_id: Optional[str] = None
    client_lead: Optional[str] = None
    agency_lead: Optional[str] = None
    brand_guidelines: Optional[str] = None


class CampaignUpdate(BaseModel):
    name: Optional[str] = None
    workfront_id: Optional[str] = None
    client_lead: Optional[str] = None
    agency_lead: Optional[str] = None
    brand_guidelines: Optional[str] = None
    status: Optional[str] = None


class CampaignResponse(BaseModel):
    id: uuid.UUID
    name: str
    workfront_id: Optional[str]
    client_lead: Optional[str]
    agency_lead: Optional[str]
    brand_guidelines: Optional[str]
    status: str
    agency: Optional[str]
    created_at: datetime
    updated_at: datetime
    proofs: int = 0

    class Config:
        from_attributes = True


# Proof schemas
class ProofCreate(BaseModel):
    proof_name: str
    channel: Optional[str] = None
    sub_channel: Optional[str] = None
    proof_type: Optional[str] = None


class ProofVersionResponse(BaseModel):
    id: uuid.UUID
    version: int
    file_storage_key: Optional[str]
    thumbnail_url: Optional[str]
    agent_review: Optional[dict]
    overall_status: Optional[str]
    workfront_id: Optional[str]
    created_at: datetime

    class Config:
        from_attributes = True


class ProofResponse(BaseModel):
    id: uuid.UUID
    proof_name: str
    channel: Optional[str]
    sub_channel: Optional[str]
    proof_type: Optional[str]
    workfront_id: Optional[str]
    created_at: datetime
    versions: list[ProofVersionResponse] = []

    class Config:
        from_attributes = True


# Audit schemas
class FlaggedItemCreate(BaseModel):
    proof_version_id: uuid.UUID
    agent_flagged: str
    comments: Optional[str] = None


class FlaggedItemResponse(BaseModel):
    id: uuid.UUID
    proof_version_id: uuid.UUID
    agent_flagged: str
    comments: Optional[str]
    submitter_name: Optional[str]
    submitter_agency: Optional[str]
    campaign_name: Optional[str]
    proof_name: Optional[str]
    version: Optional[int]
    created_at: datetime

    class Config:
        from_attributes = True


class ResolvedItemCreate(BaseModel):
    proof_version_id: uuid.UUID
    agent: str
    issue: Optional[str] = None
    resolution: Optional[str] = None


class ResolvedItemResponse(BaseModel):
    id: uuid.UUID
    proof_version_id: uuid.UUID
    agent: str
    issue: Optional[str]
    resolution: Optional[str]
    submitter_name: Optional[str]
    submitter_agency: Optional[str]
    campaign_name: Optional[str]
    proof_name: Optional[str]
    version: Optional[int]
    created_at: datetime

    class Config:
        from_attributes = True


class ErrorItemResponse(BaseModel):
    id: uuid.UUID
    proof_version_id: uuid.UUID
    error_summary: Optional[str]
    campaign_name: Optional[str]
    proof_name: Optional[str]
    version: Optional[int]
    created_at: datetime

    class Config:
        from_attributes = True


# Analytics schemas
class AnalyticsResponse(BaseModel):
    total_reviews: int
    passed: int
    failed: int
    errors: int
    legal_review: int


# Dropdown options schemas
class DropdownOptionsResponse(BaseModel):
    campaigns: list[str]
    channels: dict[str, dict[str, list[str]]]
    brand_guidelines: list[str] = []


# Agency schemas
class AgencyResponse(BaseModel):
    id: uuid.UUID
    name: str

    class Config:
        from_attributes = True


# User schemas
class UserResponse(BaseModel):
    id: uuid.UUID
    email: str
    name: str
    role: str
    agency: Optional[str]
    created_at: datetime

    class Config:
        from_attributes = True