"""Persisted Forgejo mass import run summaries."""

from __future__ import annotations

from datetime import datetime
from uuid import UUID, uuid4

from sqlalchemy import Column, JSON
from sqlmodel import Field

from app.core.time import utcnow
from app.models.base import QueryModel

RUNTIME_ANNOTATION_TYPES = (datetime,)


class ForgejoImportRun(QueryModel, table=True):
    """Summary of one full or subset Forgejo issue import run."""

    __tablename__ = "forgejo_import_runs"  # pyright: ignore[reportAssignmentType]

    id: UUID = Field(default_factory=uuid4, primary_key=True)
    organization_id: UUID = Field(foreign_key="organizations.id", index=True)
    requested_by_user_id: UUID | None = Field(default=None, foreign_key="users.id")
    repository_ids: list[str] = Field(
        default_factory=list,
        sa_column=Column(JSON, nullable=False, server_default="[]"),
    )
    results: list[dict[str, object]] = Field(
        default_factory=list,
        sa_column=Column(JSON, nullable=False, server_default="[]"),
    )
    total_created: int = Field(default=0)
    total_updated: int = Field(default=0)
    total_stale_closed: int = Field(default=0)
    succeeded: int = Field(default=0)
    failed: int = Field(default=0)
    started_at: datetime = Field(default_factory=utcnow)
    finished_at: datetime | None = Field(default=None)
    duration_ms: int | None = Field(default=None)
    created_at: datetime = Field(default_factory=utcnow)