models.py

from __future__ import annotations

from enum import Enum
from typing import Optional, List, Any
from pydantic import BaseModel, Field


class Status(str, Enum):
    ACTIVE = "ACTIVE"
    RUNNING = "RUNNING"
    COMPLETED = "COMPLETED"
    NULL = "NULL"


class RiskLevel(str, Enum):
    SAFE = "안전"
    CAUTION = "주의"
    DANGER = "위험"


class AccountStatus(str, Enum):
    NORMAL = "정상"
    SUSPICIOUS = "의심"
    REPORTED = "신고"
    UNKNOWN = "UNKNOWN"


class AmountTier(str, Enum):
    SMALL = "소액"
    MEDIUM = "중액"
    LARGE = "고액"


class Envelope(BaseModel):
    """모든 API 응답의 공통 래핑 구조"""
    data: Optional[Any] = None
    error: Optional[str] = None
    request_id: str
    timestamp: str


# ---------- FN-001 ----------
class UploadResponse(BaseModel):
    image_id: str
    preview_url: str
    status: Status


# ---------- FN-002 ----------
class AnalyzeResponse(BaseModel):
    image_id: str
    risk_level: Optional[RiskLevel] = None
    risk_score: Optional[float] = None
    keywords: List[str] = Field(default_factory=list)
    mode: str = "FULL"  # FULL | BASIC_ACCOUNT_ONLY
    status: Status


# ---------- FN-003 ----------
class AccountVerifyRequest(BaseModel):
    bank_code: str
    account_number: str
    holder_name: Optional[str] = None


class AccountVerifyResponse(BaseModel):
    account_status: AccountStatus
    report_count: int = 0
    mode: str = "REALTIME"  # REALTIME | DELAYED
    status: Status


# ---------- FN-004 ----------
class AmountRequest(BaseModel):
    amount: float


class AmountResponse(BaseModel):
    transfer_id: str
    amount: float
    amount_tier: Optional[AmountTier] = None
    status: Status


# ---------- FN-005 ----------
class TemplateResponse(BaseModel):
    template_type: AmountTier
    ui_components: List[str]
    confirm_steps: int


# ---------- FN-006 ----------
class GuidanceResponse(BaseModel):
    transfer_id: str
    summary: str
    share_link: str
    status: Status