"""
Notification schemas for email and SMS notifications.
"""
from datetime import datetime
from typing import Optional
from pydantic import BaseModel, Field, EmailStr

from app.models.notification import (
    NotificationType,
    NotificationTemplate,
    NotificationStatus
)


class NotificationBase(BaseModel):
    """Base notification schema."""
    notification_type: NotificationType
    template: NotificationTemplate
    recipient_email: Optional[EmailStr] = None
    recipient_phone: Optional[str] = None
    recipient_name: Optional[str] = None
    subject: Optional[str] = None
    body: str


class NotificationCreate(NotificationBase):
    """Schema for creating a notification."""
    client_id: Optional[int] = None
    reservation_id: Optional[int] = None
    folio_id: Optional[int] = None


class NotificationResponse(NotificationBase):
    """Schema for notification response."""
    id: int
    status: NotificationStatus
    sent_at: Optional[datetime] = None
    error_message: Optional[str] = None
    retry_count: int
    client_id: Optional[int] = None
    reservation_id: Optional[int] = None
    folio_id: Optional[int] = None
    created_at: datetime
    updated_at: datetime

    class Config:
        from_attributes = True


class NotificationSettingBase(BaseModel):
    """Base notification settings schema."""
    smtp_host: Optional[str] = None
    smtp_port: int = 587
    smtp_username: Optional[str] = None
    smtp_password: Optional[str] = None
    smtp_use_tls: bool = True
    smtp_use_ssl: bool = False
    from_email: Optional[EmailStr] = None
    from_name: Optional[str] = None
    email_enabled: bool = True
    sms_enabled: bool = False
    auto_send_confirmation: bool = True
    auto_send_modification: bool = True
    auto_send_cancellation: bool = True
    auto_send_reminder: bool = True
    reminder_days_before: int = 1


class NotificationSettingCreate(NotificationSettingBase):
    """Schema for creating notification settings."""
    establishment_id: int


class NotificationSettingUpdate(BaseModel):
    """Schema for updating notification settings."""
    smtp_host: Optional[str] = None
    smtp_port: Optional[int] = None
    smtp_username: Optional[str] = None
    smtp_password: Optional[str] = None
    smtp_use_tls: Optional[bool] = None
    smtp_use_ssl: Optional[bool] = None
    from_email: Optional[EmailStr] = None
    from_name: Optional[str] = None
    email_enabled: Optional[bool] = None
    sms_enabled: Optional[bool] = None
    auto_send_confirmation: Optional[bool] = None
    auto_send_modification: Optional[bool] = None
    auto_send_cancellation: Optional[bool] = None
    auto_send_reminder: Optional[bool] = None
    reminder_days_before: Optional[int] = None


class NotificationSettingResponse(NotificationSettingBase):
    """Schema for notification settings response."""
    id: int
    establishment_id: int
    created_at: datetime
    updated_at: datetime

    class Config:
        from_attributes = True


class SendNotificationRequest(BaseModel):
    """Request to send a notification."""
    template: NotificationTemplate
    reservation_id: Optional[int] = None
    folio_id: Optional[int] = None
    client_id: Optional[int] = None
    recipient_email: Optional[EmailStr] = None
    recipient_phone: Optional[str] = None
    custom_subject: Optional[str] = None
    custom_body: Optional[str] = None


class TestEmailRequest(BaseModel):
    """Request to send a test email."""
    test_email: EmailStr = Field(..., description="Email address to send test to")