from datetime import datetime

from sqlalchemy import Boolean, DateTime, Integer, String, Text, UniqueConstraint
from sqlalchemy.dialects.mysql import JSON
from sqlalchemy.orm import Mapped, mapped_column

from app.db import Base
from app.time_utils import agora_local_banco


class KanbanColunaModel(Base):
    __tablename__ = "kanban_colunas"

    etapa_id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    slug: Mapped[str] = mapped_column(String(64), unique=True, index=True)
    titulo: Mapped[str] = mapped_column(String(255), default="")
    cor: Mapped[str] = mapped_column(String(16), default="#788394")
    ordem: Mapped[int] = mapped_column(Integer, default=0, index=True)
    eh_inicial: Mapped[bool] = mapped_column(Boolean, default=False)
    eh_final: Mapped[bool] = mapped_column(Boolean, default=False)
    ativo: Mapped[bool] = mapped_column(Boolean, default=True)
    criado_em: Mapped[datetime] = mapped_column(DateTime, default=agora_local_banco)
    atualizado_em: Mapped[datetime] = mapped_column(
        DateTime, default=agora_local_banco, onupdate=agora_local_banco
    )


class KanbanCardModel(Base):
    __tablename__ = "kanban_cards"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    card_id: Mapped[str] = mapped_column(String(64), unique=True, index=True)
    schoolname: Mapped[str] = mapped_column(String(64), index=True, default="")
    titulo: Mapped[str] = mapped_column(String(255), default="")
    id_avaliacao: Mapped[str] = mapped_column(String(64), index=True, default="")
    responsavel_entrega: Mapped[str] = mapped_column(String(255), default="")
    responsavel_correcao: Mapped[str] = mapped_column(String(255), default="")
    etapa: Mapped[str] = mapped_column(String(64), index=True, default="lote_recebido")
    nome_processamento: Mapped[str] = mapped_column(String(255), default="")
    total_cartoes: Mapped[int] = mapped_column(Integer, default=0)
    progresso: Mapped[int] = mapped_column(Integer, default=0)
    status_prazo: Mapped[str] = mapped_column(String(32), default="no_prazo")
    status_progresso: Mapped[str] = mapped_column(String(32), default="iniciado")
    data_entrega: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
    data_previsao_entrega: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
    unidade: Mapped[str] = mapped_column(String(128), default="")
    serie: Mapped[str] = mapped_column(String(512), default="")
    turma: Mapped[str] = mapped_column(String(255), default="")
    tipo_prova: Mapped[str] = mapped_column(String(255), default="")
    data_conclusao: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
    posicao: Mapped[int] = mapped_column(Integer, default=0)
    etapa_manual: Mapped[bool] = mapped_column(Boolean, default=False)
    historico_processamentos: Mapped[list] = mapped_column(JSON, default=list)
    criado_em: Mapped[datetime] = mapped_column(DateTime, default=agora_local_banco)
    atualizado_em: Mapped[datetime] = mapped_column(DateTime, default=agora_local_banco, onupdate=agora_local_banco)


class JobModel(Base):
    __tablename__ = "jobs"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    job_id: Mapped[str] = mapped_column(String(64), unique=True, index=True)
    status: Mapped[str] = mapped_column(String(32), default="aguardando")
    percentual: Mapped[int] = mapped_column(Integer, default=0)
    mensagem: Mapped[str] = mapped_column(Text, default="")
    total_arquivos: Mapped[int] = mapped_column(Integer, default=0)
    arquivo_atual: Mapped[str] = mapped_column(String(255), default="")
    nome_processamento: Mapped[str] = mapped_column(String(255), default="")
    erro: Mapped[str] = mapped_column(Text, default="")
    eventos: Mapped[list] = mapped_column(JSON, default=list)
    resultado: Mapped[dict] = mapped_column(JSON, default=dict)
    resumo: Mapped[dict] = mapped_column(JSON, default=dict)
    criado_em: Mapped[datetime] = mapped_column(DateTime, default=agora_local_banco)
    atualizado_em: Mapped[datetime] = mapped_column(DateTime, default=agora_local_banco, onupdate=agora_local_banco)


class LoteProcessamentoModel(Base):
    """Dados persistentes do lote; imagens ficam apenas no armazenamento temporário."""
    __tablename__ = "lotes_processamento"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    nome_processamento: Mapped[str] = mapped_column(String(255), unique=True, index=True)
    schoolname: Mapped[str] = mapped_column(String(64), index=True, default="")
    metadados: Mapped[dict] = mapped_column(JSON, default=dict)
    leituras: Mapped[dict] = mapped_column(JSON, default=dict)
    estado: Mapped[dict] = mapped_column(JSON, default=dict)
    criado_em: Mapped[datetime] = mapped_column(DateTime, default=agora_local_banco)
    atualizado_em: Mapped[datetime] = mapped_column(DateTime, default=agora_local_banco, onupdate=agora_local_banco)


class LoteLeituraModel(Base):
    """Uma leitura por cartão para não concentrar lotes grandes em um JSON."""
    __tablename__ = "lote_leituras"
    __table_args__ = (UniqueConstraint("lote_id", "chave", name="uq_lote_leituras_chave"),)

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    lote_id: Mapped[int] = mapped_column(Integer, index=True)
    chave: Mapped[str] = mapped_column(String(512))
    arquivo_original: Mapped[str] = mapped_column(String(512), default="")
    registro_academico: Mapped[str] = mapped_column(String(128), index=True, default="")
    codigo_barras: Mapped[str] = mapped_column(String(255), index=True, default="")
    respostas: Mapped[dict] = mapped_column(JSON, default=dict)
    erros: Mapped[list] = mapped_column(JSON, default=list)
    pontos_cantos: Mapped[dict] = mapped_column(JSON, default=dict)
    origem_cantos: Mapped[str] = mapped_column(String(16), default="")
    dados: Mapped[dict] = mapped_column(JSON, default=dict)


class LoteMarcacaoModel(Base):
    """Mapa de bolhas pesado, carregado somente na correção/visualização."""
    __tablename__ = "lote_marcacoes"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    leitura_id: Mapped[int] = mapped_column(Integer, unique=True, index=True)
    pontos_mapeados_zlib_v1: Mapped[str] = mapped_column(Text, default="")