from __future__ import annotations

from datetime import datetime

from sqlalchemy import Boolean, DateTime, Integer, String
from sqlalchemy.orm import Mapped, mapped_column
from sqlalchemy.types import TypeDecorator

from admin_app.db.models.base import AdminTimestampedModel
from shared.contracts import StaffRole, normalize_staff_role

# Modelo da conta administrativa
# Ele representa o usuario interno do painel.


class StaffRoleType(TypeDecorator):
    impl = String(32)
    cache_ok = True

    @property
    def python_type(self):
        return StaffRole

    def process_bind_param(self, value, dialect):
        if value is None:
            return None
        return normalize_staff_role(value).value

    def process_result_value(self, value, dialect):
        if value is None:
            return None
        return normalize_staff_role(value)


class StaffAccount(AdminTimestampedModel):
    __tablename__ = "staff_accounts"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    email: Mapped[str] = mapped_column(String(255), unique=True, index=True, nullable=False)
    display_name: Mapped[str] = mapped_column(String(150), nullable=False)
    password_hash: Mapped[str] = mapped_column(String(255), nullable=False)
    role: Mapped[StaffRole] = mapped_column(
        StaffRoleType(),
        nullable=False,
        default=StaffRole.COLABORADOR,
    )
    is_active: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True)
    last_login_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)