from sqlalchemy.orm import Mapped, mapped_column
from sqlalchemy import String, Integer, Float, TIMESTAMP
from sqlalchemy.dialects.postgresql import UUID, JSONB, INET
from . import Base

class Cluster(Base):
    __tablename__ = "clusters"

    id: Mapped[int] = mapped_column(primary_key=True)
    uuid: Mapped[str] = mapped_column(UUID(as_uuid=False), unique=True)
    name: Mapped[str] = mapped_column(String(100), unique=True)
    type: Mapped[str] = mapped_column(String(50))
    node_count: Mapped[int] = mapped_column(Integer, default=0)
    health_status: Mapped[str] = mapped_column(String(20), default="unknown")
    cpu_avg: Mapped[float | None] = mapped_column(Float, nullable=True)
    memory_avg: Mapped[float | None] = mapped_column(Float, nullable=True)
    namenode_ip: Mapped[str | None] = mapped_column(INET, nullable=True)
    namenode_psw: Mapped[str | None] = mapped_column(String(255), nullable=True)
    rm_ip: Mapped[str | None] = mapped_column(INET, nullable=True)
    rm_psw: Mapped[str | None] = mapped_column(String(255), nullable=True)
    description: Mapped[str | None] = mapped_column(String, nullable=True)
    config_info: Mapped[dict | None] = mapped_column(JSONB, nullable=True)
    created_at: Mapped[str] = mapped_column(TIMESTAMP(timezone=True))
    updated_at: Mapped[str] = mapped_column(TIMESTAMP(timezone=True))

    def to_dict(self) -> dict:
        """将集群对象转换为可序列化字典。"""
        return {
            "id": self.id,
            "uuid": self.uuid,
            "name": self.name,
            "type": self.type,
            "node_count": self.node_count,
            "health_status": self.health_status,
            "cpu_avg": self.cpu_avg,
            "memory_avg": self.memory_avg,
            "namenode_ip": (str(self.namenode_ip) if self.namenode_ip else None),
            "namenode_psw": self.namenode_psw,
            "rm_ip": (str(self.rm_ip) if self.rm_ip else None),
            "rm_psw": self.rm_psw,
            "description": self.description,
            "config_info": self.config_info,
            "created_at": self.created_at.isoformat() if self.created_at else None,
            "updated_at": self.updated_at.isoformat() if self.updated_at else None,
        }