"""
Pydantic schemas for Site model.

Request and response schemas for client physical locations.
"""

from datetime import datetime
from typing import Optional
from uuid import UUID

from pydantic import BaseModel, Field


class SiteBase(BaseModel):
    """Base schema with shared Site fields."""

    client_id: UUID = Field(..., description="Reference to the client this site belongs to")
    name: str = Field(..., description="Site name (e.g., 'Main Office', 'SLC - Salt Lake City')")
    network_subnet: Optional[str] = Field(None, description="Network subnet for the site (e.g., '172.16.9.0/24')")
    vpn_required: bool = Field(False, description="Whether VPN is required to access this site")
    vpn_subnet: Optional[str] = Field(None, description="VPN subnet if applicable (e.g., '192.168.1.0/24')")
    gateway_ip: Optional[str] = Field(None, description="Gateway IP address (IPv4 or IPv6)")
    dns_servers: Optional[str] = Field(None, description="JSON array of DNS server addresses")
    notes: Optional[str] = Field(None, description="Additional notes about the site")


class SiteCreate(SiteBase):
    """Schema for creating a new Site."""
    pass


class SiteUpdate(BaseModel):
    """Schema for updating an existing Site. All fields are optional."""

    client_id: Optional[UUID] = Field(None, description="Reference to the client this site belongs to")
    name: Optional[str] = Field(None, description="Site name (e.g., 'Main Office', 'SLC - Salt Lake City')")
    network_subnet: Optional[str] = Field(None, description="Network subnet for the site (e.g., '172.16.9.0/24')")
    vpn_required: Optional[bool] = Field(None, description="Whether VPN is required to access this site")
    vpn_subnet: Optional[str] = Field(None, description="VPN subnet if applicable (e.g., '192.168.1.0/24')")
    gateway_ip: Optional[str] = Field(None, description="Gateway IP address (IPv4 or IPv6)")
    dns_servers: Optional[str] = Field(None, description="JSON array of DNS server addresses")
    notes: Optional[str] = Field(None, description="Additional notes about the site")


class SiteResponse(SiteBase):
    """Schema for Site responses with ID and timestamps."""

    id: UUID = Field(..., description="Unique identifier for the site")
    created_at: datetime = Field(..., description="Timestamp when the site was created")
    updated_at: datetime = Field(..., description="Timestamp when the site was last updated")

    model_config = {"from_attributes": True}
