from __future__ import annotations

import uuid
from typing import Any

from datetime import datetime

from sqlalchemy import BigInteger, Boolean, DateTime, ForeignKey, Index, Integer, JSON, String, Text, UniqueConstraint
from sqlalchemy.orm import Mapped, mapped_column

from govoplan_core.db.base import Base, TimestampMixin


def new_uuid() -> str:
    return str(uuid.uuid4())


class MailServerProfile(Base, TimestampMixin):
    __tablename__ = "mail_server_profiles"
    __table_args__ = (
        UniqueConstraint("tenant_id", "slug", name="uq_mail_server_profiles_tenant_slug"),
        Index("ix_mail_server_profiles_scope", "scope_type", "scope_id"),
    )

    id: Mapped[str] = mapped_column(String(36), primary_key=True, default=new_uuid)
    tenant_id: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True)
    scope_type: Mapped[str] = mapped_column(String(20), default="tenant", nullable=False, index=True)
    scope_id: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True)
    name: Mapped[str] = mapped_column(String(255), nullable=False)
    slug: Mapped[str] = mapped_column(String(100), nullable=False)
    description: Mapped[str | None] = mapped_column(Text)
    is_active: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False, index=True)
    smtp_config: Mapped[dict[str, Any]] = mapped_column(JSON, default=dict, nullable=False)
    smtp_username: Mapped[str | None] = mapped_column(String(320))
    smtp_password_encrypted: Mapped[str | None] = mapped_column(Text)
    smtp_transport_revision: Mapped[str] = mapped_column(String(36), default=new_uuid, nullable=False)
    imap_config: Mapped[dict[str, Any] | None] = mapped_column(JSON, nullable=True)
    imap_username: Mapped[str | None] = mapped_column(String(320))
    imap_password_encrypted: Mapped[str | None] = mapped_column(Text)
    imap_transport_revision: Mapped[str] = mapped_column(String(36), default=new_uuid, nullable=False)
    created_by_user_id: Mapped[str | None] = mapped_column(ForeignKey("access_users.id", ondelete="SET NULL"), nullable=True, index=True)
    updated_by_user_id: Mapped[str | None] = mapped_column(ForeignKey("access_users.id", ondelete="SET NULL"), nullable=True, index=True)


class MailProfilePolicy(Base, TimestampMixin):
    __tablename__ = "mail_profile_policies"
    __table_args__ = (
        UniqueConstraint("tenant_id", "scope_type", "scope_id", name="uq_mail_profile_policies_scope"),
        Index("ix_mail_profile_policies_scope", "scope_type", "scope_id"),
    )

    id: Mapped[str] = mapped_column(String(36), primary_key=True, default=new_uuid)
    tenant_id: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True)
    scope_type: Mapped[str] = mapped_column(String(20), nullable=False, index=True)
    scope_id: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True)
    policy: Mapped[dict[str, Any]] = mapped_column(JSON, default=dict, nullable=False)


class MailMailboxFolderIndex(Base, TimestampMixin):
    __tablename__ = "mail_mailbox_folder_index"
    __table_args__ = (
        UniqueConstraint("profile_id", "folder", name="uq_mail_mailbox_folder_index_profile_folder"),
        Index("ix_mail_mailbox_folder_index_tenant_profile", "tenant_id", "profile_id"),
    )

    id: Mapped[str] = mapped_column(String(36), primary_key=True, default=new_uuid)
    tenant_id: Mapped[str] = mapped_column(String(36), nullable=False, index=True)
    profile_id: Mapped[str] = mapped_column(ForeignKey("mail_server_profiles.id", ondelete="CASCADE"), nullable=False, index=True)
    folder: Mapped[str] = mapped_column(String(255), nullable=False, index=True)
    flags: Mapped[list[str]] = mapped_column(JSON, default=list, nullable=False)
    message_count: Mapped[int | None] = mapped_column(Integer, nullable=True)
    unseen_count: Mapped[int | None] = mapped_column(Integer, nullable=True)
    uidvalidity: Mapped[str | None] = mapped_column(String(255), nullable=True)
    indexed_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True, index=True)
    message_indexed_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True, index=True)


class MailMailboxMessageIndex(Base, TimestampMixin):
    __tablename__ = "mail_mailbox_message_index"
    __table_args__ = (
        UniqueConstraint("profile_id", "folder", "uid", name="uq_mail_mailbox_message_index_profile_folder_uid"),
        Index("ix_mail_mailbox_message_index_page", "tenant_id", "profile_id", "folder", "sort_position"),
    )

    id: Mapped[str] = mapped_column(String(36), primary_key=True, default=new_uuid)
    tenant_id: Mapped[str] = mapped_column(String(36), nullable=False, index=True)
    profile_id: Mapped[str] = mapped_column(ForeignKey("mail_server_profiles.id", ondelete="CASCADE"), nullable=False, index=True)
    folder: Mapped[str] = mapped_column(String(255), nullable=False, index=True)
    uid: Mapped[str] = mapped_column(String(255), nullable=False)
    uid_int: Mapped[int] = mapped_column(BigInteger, default=0, nullable=False, index=True)
    sort_position: Mapped[int] = mapped_column(BigInteger, default=0, nullable=False, index=True)
    subject: Mapped[str | None] = mapped_column(Text, nullable=True)
    from_header: Mapped[str | None] = mapped_column(Text, nullable=True)
    to_header: Mapped[str | None] = mapped_column(Text, nullable=True)
    cc_header: Mapped[str | None] = mapped_column(Text, nullable=True)
    date: Mapped[str | None] = mapped_column(String(255), nullable=True)
    message_id: Mapped[str | None] = mapped_column(Text, nullable=True)
    flags: Mapped[list[str]] = mapped_column(JSON, default=list, nullable=False)
    size_bytes: Mapped[int | None] = mapped_column(Integer, nullable=True)
    body_preview: Mapped[str | None] = mapped_column(Text, nullable=True)
    attachment_count: Mapped[int] = mapped_column(Integer, default=0, nullable=False)
    indexed_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, index=True)