from datetime import datetime
from typing import Optional

from sqlalchemy import UniqueConstraint
from sqlmodel import Field, SQLModel


class Replica(SQLModel, table=True):
    __tablename__ = "replicas"  # type: ignore[assignment]

    id: Optional[int] = Field(default=None, primary_key=True)
    name: str
    url: str
    api_token: str  # Fernet-encrypted
    enabled: bool = True
    sync_interval_seconds: Optional[int] = None
    last_sync_ts: Optional[datetime] = None
    consecutive_failures: int = 0
    suspended_at: Optional[datetime] = None
    last_alert_at: Optional[datetime] = None
    promoted_from_master: Optional[bool] = Field(default=False)
    created_at: datetime = Field(default_factory=datetime.utcnow)


class SyncMap(SQLModel, table=True):
    __tablename__ = "sync_map"  # type: ignore[assignment]
    __table_args__ = (UniqueConstraint("replica_id", "master_doc_id"),)  # type: ignore[assignment]

    id: Optional[int] = Field(default=None, primary_key=True)
    replica_id: int = Field(foreign_key="replicas.id")
    master_doc_id: int
    replica_doc_id: Optional[int] = None
    task_id: Optional[str] = None
    last_synced: Optional[datetime] = None
    file_checksum: Optional[str] = None
    status: str = "pending"
    error_msg: Optional[str] = None
    retry_count: int = 0


class SyncRun(SQLModel, table=True):
    __tablename__ = "sync_runs"  # type: ignore[assignment]

    id: Optional[int] = Field(default=None, primary_key=True)
    replica_id: Optional[int] = Field(default=None, foreign_key="replicas.id")
    started_at: Optional[datetime] = None
    finished_at: Optional[datetime] = None
    triggered_by: Optional[str] = None
    docs_synced: int = 0
    docs_failed: int = 0
    timed_out: bool = False


class Log(SQLModel, table=True):
    __tablename__ = "logs"  # type: ignore[assignment]

    id: Optional[int] = Field(default=None, primary_key=True)
    run_id: Optional[int] = Field(default=None, foreign_key="sync_runs.id")
    replica_id: Optional[int] = Field(default=None, foreign_key="replicas.id")
    level: Optional[str] = None
    message: Optional[str] = None
    doc_id: Optional[int] = None
    created_at: datetime = Field(default_factory=datetime.utcnow)


class Setting(SQLModel, table=True):
    __tablename__ = "settings"  # type: ignore[assignment]

    key: str = Field(primary_key=True)
    value: Optional[str] = None