from datetime import datetime

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

from app.db import Base


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=datetime.utcnow)
    atualizado_em: Mapped[datetime] = mapped_column(
        DateTime, default=datetime.utcnow, onupdate=datetime.utcnow
    )


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)
    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=datetime.utcnow)
    atualizado_em: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)


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=datetime.utcnow)
    atualizado_em: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)