"""
Device data models for the dashboard.
"""

from datetime import datetime
from enum import Enum
from typing import Optional

from pydantic import BaseModel, Field


class DeviceType(str, Enum):
    """Device connection type."""

    ADB = "adb"
    HDC = "hdc"
    IOS = "ios"


class DeviceStatus(str, Enum):
    """Device status."""

    ONLINE = "online"
    OFFLINE = "offline"
    BUSY = "busy"
    ERROR = "error"


class DeviceSchema(BaseModel):
    """Device schema for API responses."""

    device_id: str = Field(..., description="Unique device identifier")
    status: DeviceStatus = Field(default=DeviceStatus.OFFLINE, description="Device status")
    device_type: DeviceType = Field(..., description="Device connection type")
    model: Optional[str] = Field(None, description="Device model name")
    android_version: Optional[str] = Field(None, description="Android/iOS version")
    current_app: Optional[str] = Field(None, description="Currently active app")
    last_seen: datetime = Field(default_factory=datetime.now, description="Last connection time")
    is_connected: bool = Field(True, description="Whether device is connected")
    screenshot: Optional[str] = Field(None, description="Base64 encoded screenshot")

    class Config:
        json_schema_extra = {
            "example": {
                "device_id": "emulator-5554",
                "status": "online",
                "device_type": "adb",
                "model": "sdk_gphone64_x86_64",
                "android_version": "14",
                "current_app": "com.android.launcher3",
                "is_connected": True,
            }
        }


class DeviceInfo(BaseModel):
    """Extended device information."""

    device_id: str
    status: DeviceStatus
    device_type: DeviceType
    model: Optional[str] = None
    android_version: Optional[str] = None
    current_app: Optional[str] = None
    last_seen: datetime
    screenshot: Optional[str] = None
    is_connected: bool = True