"""Manifest domain — SQLAlchemy models."""
from __future__ import annotations

import enum
from datetime import datetime
from typing import Optional
from uuid import UUID, uuid4

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

from src.core.database import Base

import sqlalchemy as sa


class ManifestRunStatus(str, enum.Enum):
    active = "active"
    completed = "completed"
    failed = "failed"
    expired = "expired"


MANIFEST_RUN_STATUS_ENUM = sa.Enum(
    "active", "completed", "failed", "expired",
    name="p2hs_manifest_run_status",
    create_type=False,
)


class ManifestRun(Base):
    """Short-lived codegen run session with a scoped token.

    token_hash stores SHA-256 of the plain token (returned once at creation).
    The run is expired after TTL or on explicit complete/fail.
    """

    __tablename__ = "manifest_runs"

    id: Mapped[UUID] = mapped_column(PGUUID(as_uuid=True), primary_key=True, default=uuid4)
    project_id: Mapped[UUID] = mapped_column(
        PGUUID(as_uuid=True),
        ForeignKey("projects.id", ondelete="CASCADE"),
        nullable=False,
    )
    initiated_by: Mapped[Optional[UUID]] = mapped_column(
        PGUUID(as_uuid=True),
        ForeignKey("users.id", ondelete="SET NULL"),
        nullable=True,
    )
    status: Mapped[str] = mapped_column(
        MANIFEST_RUN_STATUS_ENUM,
        nullable=False,
        server_default="active",
    )
    requested_scopes: Mapped[Optional[dict]] = mapped_column(JSONB, nullable=True)
    token_hash: Mapped[str] = mapped_column(String(64), nullable=False, unique=True)
    expires_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False)
    completed_at: Mapped[Optional[datetime]] = mapped_column(DateTime(timezone=True), nullable=True)
    failed_reason: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
    manifest_snapshot_hash: Mapped[Optional[str]] = mapped_column(String(64), nullable=True)
    created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())