"""
Pydantic schemas for Infrastructure model.

Request and response schemas for infrastructure assets including servers,
network devices, workstations, and other IT infrastructure.
"""

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

from pydantic import BaseModel, Field


class InfrastructureBase(BaseModel):
    """Base schema with shared Infrastructure fields."""

    client_id: Optional[str] = Field(None, description="Reference to the client")
    site_id: Optional[str] = Field(None, description="Reference to the site this infrastructure is located at")
    asset_type: str = Field(..., description="Type: physical_server, virtual_machine, container, network_device, nas_storage, workstation, firewall, domain_controller")
    hostname: str = Field(..., description="Hostname of the infrastructure")
    ip_address: Optional[str] = Field(None, description="IP address (IPv4 or IPv6)")
    mac_address: Optional[str] = Field(None, description="MAC address")
    os: Optional[str] = Field(None, description="Operating system name (e.g., 'Ubuntu 22.04', 'Windows Server 2022')")
    os_version: Optional[str] = Field(None, description="Operating system version (e.g., '6.22', '2008 R2', '22.04')")
    role_description: Optional[str] = Field(None, description="Description of the infrastructure's role")
    parent_host_id: Optional[str] = Field(None, description="Reference to parent host for VMs/containers")
    status: str = Field("active", description="Status: active, migration_source, migration_destination, decommissioned")
    environmental_notes: Optional[str] = Field(None, description="Special environmental constraints or notes")
    powershell_version: Optional[str] = Field(None, description="PowerShell version (e.g., '2.0', '5.1', '7.4')")
    shell_type: Optional[str] = Field(None, description="Shell type: bash, cmd, powershell, sh")
    package_manager: Optional[str] = Field(None, description="Package manager: apt, yum, chocolatey, none")
    has_gui: bool = Field(True, description="Whether the system has a GUI")
    limitations: Optional[str] = Field(None, description='JSON array of limitations (e.g., ["no_ps7", "smb1_only", "dos_6.22_commands"])')
    notes: Optional[str] = Field(None, description="Additional notes")


class InfrastructureCreate(InfrastructureBase):
    """Schema for creating a new Infrastructure item."""
    pass


class InfrastructureUpdate(BaseModel):
    """Schema for updating an existing Infrastructure item. All fields are optional."""

    client_id: Optional[str] = Field(None, description="Reference to the client")
    site_id: Optional[str] = Field(None, description="Reference to the site this infrastructure is located at")
    asset_type: Optional[str] = Field(None, description="Type: physical_server, virtual_machine, container, network_device, nas_storage, workstation, firewall, domain_controller")
    hostname: Optional[str] = Field(None, description="Hostname of the infrastructure")
    ip_address: Optional[str] = Field(None, description="IP address (IPv4 or IPv6)")
    mac_address: Optional[str] = Field(None, description="MAC address")
    os: Optional[str] = Field(None, description="Operating system name (e.g., 'Ubuntu 22.04', 'Windows Server 2022')")
    os_version: Optional[str] = Field(None, description="Operating system version (e.g., '6.22', '2008 R2', '22.04')")
    role_description: Optional[str] = Field(None, description="Description of the infrastructure's role")
    parent_host_id: Optional[str] = Field(None, description="Reference to parent host for VMs/containers")
    status: Optional[str] = Field(None, description="Status: active, migration_source, migration_destination, decommissioned")
    environmental_notes: Optional[str] = Field(None, description="Special environmental constraints or notes")
    powershell_version: Optional[str] = Field(None, description="PowerShell version (e.g., '2.0', '5.1', '7.4')")
    shell_type: Optional[str] = Field(None, description="Shell type: bash, cmd, powershell, sh")
    package_manager: Optional[str] = Field(None, description="Package manager: apt, yum, chocolatey, none")
    has_gui: Optional[bool] = Field(None, description="Whether the system has a GUI")
    limitations: Optional[str] = Field(None, description='JSON array of limitations (e.g., ["no_ps7", "smb1_only", "dos_6.22_commands"])')
    notes: Optional[str] = Field(None, description="Additional notes")


class InfrastructureResponse(InfrastructureBase):
    """Schema for Infrastructure responses with ID and timestamps."""

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

    model_config = {"from_attributes": True}
