from pydantic import BaseModel


class PlanResponse(BaseModel):
    id: str
    name: str
    display_name: str
    price: float
    currency: str
    features: list[str]
    max_conversations: int | None = None
    max_chapters: int | None = None
    max_words: int | None = None
    is_popular: bool = False


class PlanUsageResponse(BaseModel):
    conversations: int
    chapters: int
    max_conversations: int | None = None
    max_chapters: int | None = None


class CurrentPlanResponse(BaseModel):
    plan_id: str
    plan_name: str
    subscription_type: str
    expires_at: str | None = None
    features: list[str]
    usage: PlanUsageResponse