"""设备注册 / 心跳相关 schema(无障碍保护存活检测)。"""
from __future__ import annotations

from datetime import datetime

from pydantic import BaseModel, ConfigDict


class DeviceRegisterRequest(BaseModel):
    device_id: str
    registration_id: str | None = None
    platform: str = "android"
    app_version: str | None = None


class HeartbeatRequest(BaseModel):
    device_id: str
    source: str = "service"          # service | app
    accessibility_enabled: bool = True
    registration_id: str | None = None


class DeviceOut(BaseModel):
    model_config = ConfigDict(from_attributes=True)

    id: int
    device_id: str
    registration_id: str | None
    ever_protected: bool
    liveness_state: str
    last_heartbeat_at: datetime | None
    updated_at: datetime | None


class OkResponse(BaseModel):
    ok: bool = True