from sqlalchemy import Boolean, Column, DateTime, ForeignKey, Integer, JSON, String, Text
from sqlalchemy.orm import relationship
from sqlalchemy.sql import func

from app.db.base import Base


class DocumentReplicaLayoutVersion(Base):
    __tablename__ = "document_replica_layout_versions"

    id = Column(Integer, primary_key=True, index=True)
    document_id = Column(Integer, ForeignKey("documents.id"), nullable=False, index=True)

    version_number = Column(Integer, nullable=False)
    version_type = Column(String, nullable=False, default="heuristic")
    render_mode_source = Column(String, nullable=False, default="shared")
    is_current = Column(Boolean, nullable=False, default=True)

    created_by = Column(String, nullable=True)
    derived_from_text_version_id = Column(Integer, ForeignKey("text_versions.id"), nullable=True)
    derived_from_replica_layout_version_id = Column(Integer, ForeignKey("document_replica_layout_versions.id"), nullable=True)

    model_name = Column(String, nullable=True)
    model_version = Column(String, nullable=True)
    prompt_version = Column(String, nullable=True)

    quality_score = Column(String, nullable=True)
    quality_note = Column(Text, nullable=True)
    quality_flags = Column(JSON, nullable=True)
    inference_metadata_json = Column(JSON, nullable=True)
    layout_json = Column(JSON, nullable=False)

    created_at = Column(DateTime(timezone=True), server_default=func.now(), nullable=False)

    document = relationship("Document", back_populates="replica_layout_versions")
    outputs = relationship("DocumentReplicaOutput", back_populates="replica_layout_version", cascade="all, delete-orphan")
    parent_layout_version = relationship("DocumentReplicaLayoutVersion", remote_side=[id])