import uuid

from sqlalchemy import String, ForeignKey, UniqueConstraint
from sqlalchemy.orm import Mapped, mapped_column
from sqlalchemy.dialects.postgresql import UUID, ARRAY

from ..database import Base


class LearnableLanguageEntity(Base):
    __tablename__ = "learnable_languages"
    __table_args__ = (
        UniqueConstraint("user_id", "source_language", "target_language", name="uq_learnable_language_user_pair"),
    )

    id: Mapped[uuid.UUID] = mapped_column(
        UUID(as_uuid=True), primary_key=True, default=uuid.uuid4
    )
    user_id: Mapped[uuid.UUID] = mapped_column(
        UUID(as_uuid=True), ForeignKey("users.id"), nullable=False, index=True
    )
    source_language: Mapped[str] = mapped_column(String(10), nullable=False)
    target_language: Mapped[str] = mapped_column(String(10), nullable=False)
    proficiencies: Mapped[list[str]] = mapped_column(ARRAY(String(5)), nullable=False)