from sqlalchemy import Column, String, Integer, Boolean, DateTime, ForeignKey, Text, JSON, ARRAY
from sqlalchemy.orm import relationship
from sqlalchemy.dialects.postgresql import UUID
from src.config.database import Base
from sqlalchemy.sql import func
import uuid

class RAGDocument(Base):
    __tablename__ = 'rag_documents'

    id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4, nullable=False)
    document_type = Column(String(255), nullable=False)
    title = Column(String(255), nullable=False)
    payer_id = Column(UUID(as_uuid=True), nullable=True)
    payer_name = Column(String(255), nullable=True)
    specialty = Column(String(255), nullable=True)
    content = Column(Text, nullable=False)
    content_hash = Column(String(255), nullable=True)
    embedding_vector = Column(String(255), nullable=True)
    chunk_index = Column(Integer, nullable=True)
    parent_document_id = Column(UUID(as_uuid=True), nullable=True)
    source_url = Column(String(255), nullable=True)
    source_file_path = Column(String(255), nullable=True)
    effective_date = Column(DateTime, nullable=True)
    expiration_date = Column(DateTime, nullable=True)
    version = Column(String(255), nullable=True)
    is_stale = Column(Boolean, nullable=True)
    relevance_score = Column(String(255), nullable=True)
    usage_count = Column(Integer, nullable=True)
    last_used_at = Column(DateTime, nullable=True)
    doc_doc_doc_metadata = Column(JSON, nullable=True)
    tags = Column(JSON, nullable=True)
    
    payer_id = Column(UUID(as_uuid=True), ForeignKey('payers.id'), nullable=True)
    payer = relationship('Payer', back_populates='ragDocuments')
    
    parent_document_id = Column(UUID(as_uuid=True), ForeignKey('rag_documents.id'), nullable=True)
    parent_document = relationship('RAGDocument', remote_side=[id], back_populates='chunks')
    chunks = relationship('RAGDocument', back_populates='parent_document')
    
    uploaded_by_id = Column(UUID(as_uuid=True), ForeignKey('users.id'), nullable=True)
    uploader = relationship('User', back_populates='uploadedDocuments')


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

    def __repr__(self):
        return f'<RAGDocument(id={self.id})>'