from datetime import datetime

from core_db import AuditMixin, Base, EntityMixin, VersionMixin
from core_shared import JSONValue
from sqlalchemy import DateTime, String, Text
from sqlalchemy.dialects.sqlite import JSON
from sqlalchemy.orm import Mapped, mapped_column


class HumanTask(EntityMixin, AuditMixin, VersionMixin, Base):
    __tablename__ = "human_task"

    task_type: Mapped[str] = mapped_column(String(32), index=True)
    status: Mapped[str] = mapped_column(String(32), default="pending", index=True)
    title: Mapped[str] = mapped_column(String(128))
    description: Mapped[str | None] = mapped_column(Text, nullable=True)
    source_type: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)
    source_id: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)
    run_id: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True)
    node_run_id: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True)
    requested_by: Mapped[str | None] = mapped_column(String(36), nullable=True)
    assigned_to: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True)
    claimed_by: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True)
    request_payload_json: Mapped[dict[str, JSONValue]] = mapped_column(JSON, default=dict)
    response_payload_json: Mapped[dict[str, JSONValue] | None] = mapped_column(JSON, nullable=True)
    due_time: Mapped[datetime | None] = mapped_column(DateTime, nullable=True, index=True)
    claimed_time: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
    completed_time: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)