"""SQLAlchemy models for the Unified Search plugin."""

from __future__ import annotations

import uuid
from datetime import datetime

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

from app.core.db import Base, TenantMixin


class SearchProviderRegistry(Base, TenantMixin):
    """Registry of active search providers per tenant."""

    __tablename__ = "unified_search_providers"
    __table_args__ = (
        Index("ix_usp_tenant", "tenant_id"),
    )

    id: Mapped[uuid.UUID] = mapped_column(
        PGUUID(as_uuid=True), primary_key=True, default=uuid.uuid4
    )
    entity_type: Mapped[str] = mapped_column(String(50), nullable=False)
    plugin_name: Mapped[str] = mapped_column(String(80), nullable=False)
    is_active: Mapped[bool] = mapped_column(
        nullable=False, default=True
    )
    config: Mapped[dict] = mapped_column(JSONB, nullable=False, default=dict)


class SearchIndexLog(Base, TenantMixin):
    """Log of indexing actions for audit and debugging."""

    __tablename__ = "unified_search_index_log"
    __table_args__ = (
        Index("ix_usil_tenant", "tenant_id"),
        Index("ix_usil_entity", "entity_type", "entity_id"),
    )

    id: Mapped[uuid.UUID] = mapped_column(
        PGUUID(as_uuid=True), primary_key=True, default=uuid.uuid4
    )
    entity_type: Mapped[str] = mapped_column(String(50), nullable=False)
    entity_id: Mapped[uuid.UUID] = mapped_column(PGUUID(as_uuid=True), nullable=False)
    action: Mapped[str] = mapped_column(String(20), nullable=False)
    status: Mapped[str] = mapped_column(String(20), nullable=False, default="pending")
    error_message: Mapped[str | None] = mapped_column(Text, nullable=True)