from __future__ import annotations

from datetime import datetime
from uuid import uuid4

from sqlalchemy import (
    JSON,
    CheckConstraint,
    DateTime,
    ForeignKey,
    Index,
    Integer,
    String,
    Text,
    UniqueConstraint,
)
from sqlalchemy.orm import Mapped, mapped_column

from app.security.models import Base, utcnow
from app.projects.models.project import Project, ProjectGenerationJob


class ProjectEditorState(Base):
    """The one authoritative current editor document for a project."""
    __tablename__ = "project_editor_states"
    __table_args__ = (
        UniqueConstraint("project_id", name="uq_project_editor_state_project"),
        Index("ix_project_editor_states_workspace_project", "workspace_id", "project_id"),
    )

    id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid4()))
    workspace_id: Mapped[str] = mapped_column(
        String(36), ForeignKey("workspaces.id", ondelete="RESTRICT"), nullable=False
    )
    project_id: Mapped[str] = mapped_column(
        String(36), ForeignKey("projects.id", ondelete="RESTRICT"), nullable=False
    )
    revision: Mapped[int] = mapped_column(Integer, nullable=False, default=1)
    schema_version: Mapped[int] = mapped_column(Integer, nullable=False)
    state_json: Mapped[dict[str, object]] = mapped_column("state", JSON, nullable=False, default=dict)
    created_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), nullable=False, default=utcnow
    )
    updated_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), nullable=False, default=utcnow, onupdate=utcnow
    )
    updated_by: Mapped[str] = mapped_column(
        String(36), ForeignKey("users.id", ondelete="RESTRICT"), nullable=False
    )


class ProjectRenderJob(Base):
    """A durable render of one immutable editor revision snapshot."""
    __tablename__ = "project_render_jobs"
    __table_args__ = (
        UniqueConstraint(
            "project_id", "editor_revision", "idempotency_key",
            name="uq_project_render_idempotency",
        ),
        CheckConstraint(
            "status in ('queued', 'processing', 'completed', 'failed', 'cancelling', 'cancelled')",
            name="ck_project_render_status",
        ),
        Index("ix_project_render_jobs_workspace_status", "workspace_id", "status"),
        Index("ix_project_render_jobs_project_created", "project_id", "created_at"),
        Index("ix_project_render_jobs_dispatch", "status", "next_attempt_at"),
    )

    id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid4()))
    workspace_id: Mapped[str] = mapped_column(
        String(36), ForeignKey("workspaces.id", ondelete="RESTRICT"), nullable=False
    )
    project_id: Mapped[str] = mapped_column(
        String(36), ForeignKey("projects.id", ondelete="RESTRICT"), nullable=False
    )
    editor_revision: Mapped[int] = mapped_column(Integer, nullable=False)
    editor_schema_version: Mapped[int] = mapped_column(Integer, nullable=False)
    editor_state_json: Mapped[dict[str, object]] = mapped_column("editor_state", JSON, nullable=False)
    render_settings_json: Mapped[dict[str, object]] = mapped_column("render_settings", JSON, nullable=False)
    request_fingerprint: Mapped[str] = mapped_column(String(64), nullable=False)
    idempotency_key: Mapped[str] = mapped_column(String(255), nullable=False)
    requested_by: Mapped[str] = mapped_column(
        String(36), ForeignKey("users.id", ondelete="RESTRICT"), nullable=False
    )
    status: Mapped[str] = mapped_column(String(32), nullable=False, default="queued")
    attempt_count: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
    max_attempts: Mapped[int] = mapped_column(Integer, nullable=False, default=3)
    next_attempt_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
    output_asset_id: Mapped[str | None] = mapped_column(
        String(36), ForeignKey("media_assets.id", ondelete="RESTRICT")
    )
    error_code: Mapped[str | None] = mapped_column(String(100))
    error_message: Mapped[str | None] = mapped_column(Text)
    created_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), nullable=False, default=utcnow
    )
    started_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
    completed_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
    cancelled_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
    updated_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), nullable=False, default=utcnow, onupdate=utcnow
    )


from app.projects.models.collaboration import (  # noqa: E402
    ApprovalRequest,
    ApprovalWorkflow,
    CollaborationActivity,
    Invitation,
    ProjectCollaborator,
    ReviewComment,
    Team,
    TeamMember,
)
from app.projects.models.notifications import NotificationPreference  # noqa: E402

__all__ = [
    "ApprovalRequest",
    "ApprovalWorkflow",
    "CollaborationActivity",
    "Invitation",
    "NotificationPreference",
    "Project",
    "ProjectCollaborator",
    "ProjectEditorState",
    "ProjectGenerationJob",
    "ProjectRenderJob",
    "ReviewComment",
    "Team",
    "TeamMember",
]