"""用户模型"""
from __future__ import annotations

from .base import Base
from .enums import UserRole, UserStatus
from .types import JSONB
from sqlalchemy.orm import Mapped, mapped_column, relationship
from sqlalchemy import BigInteger, VARCHAR, TEXT, Integer, Boolean
from typing import List, Optional, TYPE_CHECKING, Any

if TYPE_CHECKING:
    from .user_skill_binding import UserSkillBinding
    from .user_project import UserProject
    from .user_workload import UserWorkload


class User(Base):
    __tablename__ = "users"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    username: Mapped[str] = mapped_column(VARCHAR(100), nullable=False, unique=True)
    email: Mapped[str] = mapped_column(VARCHAR(255), nullable=False, unique=True)
    password_hash: Mapped[str] = mapped_column(VARCHAR(255), nullable=False)
    name: Mapped[str] = mapped_column(VARCHAR(100), nullable=False)
    role: Mapped[str] = mapped_column(VARCHAR(50), nullable=False)
    assigned_domains: Mapped[Any] = mapped_column(JSONB, nullable=True, default=list)
    workload: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
    status: Mapped[str] = mapped_column(VARCHAR(50), nullable=False, default="active")
    is_active: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True)
    avatar_url: Mapped[Optional[str]] = mapped_column(VARCHAR(500), nullable=True)
    bio: Mapped[Optional[str]] = mapped_column(TEXT, nullable=True)

    skill_bindings: Mapped[List["UserSkillBinding"]] = relationship(
        "UserSkillBinding", back_populates="user", cascade="all, delete-orphan"
    )
    workload_rel: Mapped[Optional["UserWorkload"]] = relationship(
        "UserWorkload", back_populates="user", uselist=False, cascade="all, delete-orphan"
    )
    project_links: Mapped[List["UserProject"]] = relationship(
        "UserProject", back_populates="user", cascade="all, delete-orphan"
    )

    def to_dict(self) -> dict:
        return {
            "id": str(self.id),
            "username": self.username,
            "name": self.name,
            "email": self.email,
            "role": self.role,
            "assigned_domains": self.assigned_domains or [],
            "workload": self.workload,
            "status": self.status,
            "is_active": self.is_active,
            "created_at": self.created_at.isoformat() if self.created_at else None,
            "updated_at": self.updated_at.isoformat() if self.updated_at else None,
        }