from __future__ import annotations
import enum
import uuid
from datetime import datetime, timezone
from sqlalchemy import Boolean, DateTime, Enum, Integer, String
from sqlalchemy.orm import Mapped, mapped_column
from .base import Base


class ProbeType(str, enum.Enum):
    tcp = "tcp"
    icmp = "icmp"


class Host(Base):
    __tablename__ = "hosts"

    id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    name: Mapped[str] = mapped_column(String(128), nullable=False)
    address: Mapped[str] = mapped_column(String(255), nullable=False)
    probe_type: Mapped[ProbeType] = mapped_column(Enum(ProbeType), nullable=False, default=ProbeType.tcp)
    probe_port: Mapped[int] = mapped_column(Integer, nullable=False, default=80)
    ping_enabled: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True)
    dns_enabled: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
    dns_expected_ip: Mapped[str | None] = mapped_column(String(255), nullable=True)
    poll_interval_seconds: Mapped[int | None] = mapped_column(Integer, nullable=True)
    created_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), nullable=False, default=lambda: datetime.now(timezone.utc)
    )