"""Permission and ShareLink models for the Permissions plugin."""

from __future__ import annotations

import uuid
from datetime import datetime

from sqlalchemy import DateTime, Index, String, UniqueConstraint
from sqlalchemy.dialects.postgresql import UUID as PGUUID
from sqlalchemy.orm import Mapped, mapped_column

from app.core.db import Base, TenantMixin


class Permission(Base, TenantMixin):
    """File/folder permission — grants access_level to a user or group."""

    __tablename__ = "permissions"
    __table_args__ = (
        UniqueConstraint(
            "tenant_id",
            "file_id",
            "user_id",
            "access_level",
            name="uq_permissions_file_user_level",
        ),
        Index("ix_permissions_file", "tenant_id", "file_id"),
        Index("ix_permissions_user", "tenant_id", "user_id"),
    )

    id: Mapped[uuid.UUID] = mapped_column(
        PGUUID(as_uuid=True), primary_key=True, default=uuid.uuid4
    )
    file_id: Mapped[uuid.UUID] = mapped_column(PGUUID(as_uuid=True), nullable=False)
    user_id: Mapped[uuid.UUID] = mapped_column(PGUUID(as_uuid=True), nullable=False)
    group_id: Mapped[uuid.UUID | None] = mapped_column(PGUUID(as_uuid=True), nullable=True)
    access_level: Mapped[str] = mapped_column(String(10), nullable=False, default="read")


class ShareLink(Base, TenantMixin):
    """Public share link for a file — optional password and expiry."""

    __tablename__ = "share_links"
    __table_args__ = (
        Index("ix_share_links_token", "token", unique=True),
        Index("ix_share_links_file", "tenant_id", "file_id"),
    )

    id: Mapped[uuid.UUID] = mapped_column(
        PGUUID(as_uuid=True), primary_key=True, default=uuid.uuid4
    )
    file_id: Mapped[uuid.UUID] = mapped_column(PGUUID(as_uuid=True), nullable=False)
    token: Mapped[str] = mapped_column(String(64), nullable=False, unique=True)
    password_hash: Mapped[str | None] = mapped_column(String(255), nullable=True)
    expires_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
    access_level: Mapped[str] = mapped_column(String(10), nullable=False, default="download")
    created_by: Mapped[uuid.UUID | None] = mapped_column(PGUUID(as_uuid=True), nullable=True)
