app/projects/models/__init__.py

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.projects.models.rendering import ProjectRenderJob
from app.security.models import Base, utcnow


class Project(Base):
    """A durable workspace-owned container for future creative resources."""

    __tablename__ = "projects"
    __table_args__ = (
        CheckConstraint("status in ('active', 'archived')", name="ck_projects_status"),
        Index("ix_projects_workspace", "workspace_id"),
        Index("ix_projects_workspace_status", "workspace_id", "status"),
        Index("ix_projects_workspace_updated", "workspace_id", "updated_at"),
        Index("ix_projects_created_by", "created_by"),
    )

    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
    )
    created_by: Mapped[str] = mapped_column(
        String(36), ForeignKey("users.id", ondelete="RESTRICT"), nullable=False
    )
    name: Mapped[str] = mapped_column(String(200), nullable=False)
    description: Mapped[str | None] = mapped_column(Text)
    status: Mapped[str] = mapped_column(String(16), nullable=False, default="active")
    thumbnail_asset_id: Mapped[str | None] = mapped_column(
        String(36), ForeignKey("media_assets.id", ondelete="SET NULL")
    )
    brand_kit_id: Mapped[str | None] = mapped_column(
        String(36), ForeignKey("brand_kits.id", ondelete="RESTRICT")
    )
    metadata_json: Mapped[dict[str, object]] = mapped_column(
        "metadata", 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
    )
    archived_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))


class ProjectGenerationJob(Base):
    """Associates the existing durable generation job with one project."""

    __tablename__ = "project_generation_jobs"
    __table_args__ = (
        UniqueConstraint("generation_job_id", name="uq_project_generation_job"),
        Index(
            "ix_project_generation_jobs_workspace_project_created",
            "workspace_id",
            "project_id",
            "created_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
    )
    generation_job_id: Mapped[str] = mapped_column(
        String(36), ForeignKey("generation_jobs.id", ondelete="RESTRICT"), nullable=False
    )
    attached_by: Mapped[str] = mapped_column(
        String(36), ForeignKey("users.id", ondelete="RESTRICT"), nullable=False
    )
    created_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), nullable=False, default=utcnow
    )


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
    )

__all__ = [
    "Project",
    "ProjectGenerationJob",
    "ProjectEditorState",
    "ProjectRenderJob",
]