from datetime import datetime, timezone
from typing import TYPE_CHECKING

from sqlalchemy import DateTime, ForeignKey, Index, String, Text
from sqlalchemy.orm import Mapped, mapped_column, relationship

from backend.models.database import Base

if TYPE_CHECKING:
    from backend.models.check_in_task import CheckInTask


class CheckInRecord(Base):
    """打卡记录模型"""

    __tablename__ = "check_in_records"

    id: Mapped[int] = mapped_column(primary_key=True, index=True, autoincrement=True)
    task_id: Mapped[int] = mapped_column(
        ForeignKey("check_in_tasks.id", ondelete="CASCADE"),
        index=True,
        comment="任务 ID",
    )
    status: Mapped[str] = mapped_column(
        String(20),
        index=True,
        comment="状态: success/failure/out_of_time/unknown/pending",
    )
    response_text: Mapped[str] = mapped_column(Text, default="", comment="响应文本")
    error_message: Mapped[str] = mapped_column(Text, default="", comment="错误信息")
    location: Mapped[str] = mapped_column(Text, default="{}", comment="位置信息 JSON")
    trigger_type: Mapped[str] = mapped_column(
        String(50), default="scheduled", comment="触发类型: scheduled/manual/admin"
    )
    check_in_time: Mapped[datetime] = mapped_column(
        DateTime(timezone=True),
        default=lambda: datetime.now(timezone.utc),
        index=True,
        comment="打卡时间（UTC）",
    )

    # 关联任务
    task: Mapped["CheckInTask"] = relationship(back_populates="check_in_records")

    # 添加复合索引：加速常见查询
    __table_args__ = (
        Index("ix_record_task_time", "task_id", "check_in_time"),  # 获取任务的打卡记录(按时间排序)
        Index("ix_record_status_time", "status", "check_in_time"),  # 按状态和时间查询
    )

    def __repr__(self):
        return f"<CheckInRecord(id={self.id}, task_id={self.task_id}, status={self.status})>"