from datetime import datetime, timezone
from typing import ClassVar, Optional
from uuid import UUID, uuid4

from sqlalchemy import Column, Text
from sqlalchemy import Enum as SAEnum
from sqlmodel import Field, SQLModel

from domain.tasks.tasks import Source


class TaskRow(SQLModel, table=True):
    __tablename__: ClassVar[str] = "tasks"  # pyright: ignore[reportIncompatibleVariableOverride]

    id: UUID = Field(default_factory=uuid4, primary_key=True, index=True)
    task_source: str
    job_started: Optional[datetime] = None
    job_completed: Optional[datetime] = None
    status: str = Field(default="waiting")
    service: Optional[str] = None
    # FE-owned (Drizzle) TEXT column holding the task's request as a JSON
    # string. Declared here so the backend can read it (e.g. the Bulk Document
    # Download route reads its property selection from it, ADR-0060) and so the
    # test schema builds the column; prod owns the real column via FE migrations.
    inputs: Optional[str] = Field(default=None, sa_column=Column(Text, nullable=True))
    updated_at: datetime = Field(
        default_factory=lambda: datetime.now(timezone.utc)
    )

    source: Optional[Source] = Field(
        default=None,
        sa_column=Column(
            SAEnum(
                Source,
                name="source",
                values_callable=lambda cls: [m.value for m in cls],  # pyright: ignore[reportUnknownLambdaType, reportUnknownMemberType, reportUnknownVariableType]
            ),
            nullable=True,
        ),
    )
    source_id: Optional[str] = None