# #region TaskModels [C:1] [TYPE Module] [SEMANTICS sqlalchemy, task, model, schema, execution, task-record]
#
# @BRIEF Defines the database schema for task execution records.
# @LAYER     Domain
# @RELATION DEPENDS_ON -> [EXT:Library:sqlalchemy]
#
# @INVARIANT  All primary keys are UUID strings.

import uuid

from sqlalchemy import JSON, Column, DateTime, ForeignKey, Index, Integer, String, Text
from sqlalchemy.sql import func

from .mapping import Base


# #region TaskRecord [C:1] [TYPE Class]
# @BRIEF Represents a persistent record of a task execution.
class TaskRecord(Base):
    __tablename__ = "task_records"

    id = Column(String, primary_key=True, default=lambda: str(uuid.uuid4()))
    type = Column(String, nullable=False)  # e.g., "backup", "migration"
    status = Column(String, nullable=False)  # Enum: "PENDING", "RUNNING", "SUCCESS", "FAILED"
    environment_id = Column(String, ForeignKey("environments.id", ondelete="SET NULL"), nullable=True)
    started_at = Column(DateTime(timezone=True), nullable=True)
    finished_at = Column(DateTime(timezone=True), nullable=True)
    logs = Column(JSON, nullable=True)  # Store structured logs as JSON (legacy, kept for backward compatibility)
    error = Column(String, nullable=True)
    result = Column(JSON, nullable=True)
    created_at = Column(DateTime(timezone=True), server_default=func.now())
    params = Column(JSON, nullable=True)
# #endregion TaskRecord

# #region TaskLogRecord [C:3] [TYPE Class]
# @defgroup Models Module group.
# @BRIEF Represents a single persistent log entry for a task.
# @RELATION DEPENDS_ON -> TaskRecord
# @INVARIANT  Each log entry belongs to exactly one task.
#
# @TEST_CONTRACT  TaskLogCreate ->
# {
#   required_fields: {
#       task_id: str,
#       timestamp: datetime,
#       level: str,
#       source: str,
#       message: str
#   },
#   optional_fields: {
#       metadata_json: str,
#       id: int
#   },
#   invariants: [
#       "task_id matches an existing TaskRecord.id"
#   ]
# }
#
# @TEST_FIXTURE  basic_info_log ->
# {
#   task_id: "00000000-0000-0000-0000-000000000000",
#   timestamp: "2026-02-26T12:00:00Z",
#   level: "INFO",
#   source: "system",
#   message: "Task initialization complete"
# }
#
# @TEST_EDGE  missing_required_field ->
# {
#   timestamp: "2026-02-26T12:00:00Z",
#   level: "ERROR",
#   source: "system",
#   message: "Missing task_id"
# }
#
# @TEST_EDGE  invalid_type ->
# {
#   task_id: "00000000-0000-0000-0000-000000000000",
#   timestamp: "2026-02-26T12:00:00Z",
#   level: 500,
#   source: "system",
#   message: "Integer level"
# }
#
# @TEST_EDGE  empty_message ->
# {
#   task_id: "00000000-0000-0000-0000-000000000000",
#   timestamp: "2026-02-26T12:00:00Z",
#   level: "DEBUG",
#   source: "system",
#   message: ""
# }
#
# @TEST_INVARIANT  exact_one_task_association -> verifies: [basic_info_log, missing_required_field]
class TaskLogRecord(Base):
    __tablename__ = "task_logs"

    id = Column(Integer, primary_key=True, autoincrement=True)
    task_id = Column(String, ForeignKey("task_records.id", ondelete="CASCADE"), nullable=False, index=True)
    timestamp = Column(DateTime(timezone=True), nullable=False, index=True)
    level = Column(String(16), nullable=False)  # INFO, WARNING, ERROR, DEBUG
    source = Column(String(64), nullable=False, default="system")  # plugin, superset_api, git, etc.
    message = Column(Text, nullable=False)
    metadata_json = Column(Text, nullable=True)  # JSON string for additional metadata

    # Composite indexes for efficient filtering
    __table_args__ = (
        Index('ix_task_logs_task_timestamp', 'task_id', 'timestamp'),
        Index('ix_task_logs_task_level', 'task_id', 'level'),
        Index('ix_task_logs_task_source', 'task_id', 'source'),
    )
# #endregion TaskLogRecord

# #endregion TaskModels