import uuid
from sqlalchemy import Column, DateTime, Integer, String, Boolean, ForeignKey ,DateTime,JSON,Uuid,String
from sqlalchemy.orm import Mapped, relationship, declarative_base
import uuid
from sqlalchemy.sql import func

Base = declarative_base()



class users(Base):
    __tablename__ = 'users'

    id:Mapped[uuid.UUID] = Column(Uuid(as_uuid=True), primary_key=True, index=True, default=uuid.uuid4)
    username = Column(String, unique=True, index=True)
    email = Column(String, unique=True, index=True)
    hashed_password = Column(String)
    is_active = Column(Boolean, default=True)
    is_superuser = Column(Boolean, default=False)
    is_verified = Column(Boolean, default=False)
    created_at = Column(DateTime, default=func.now())
    updated_at = Column(DateTime, default=func.now(), onupdate=func.now())
    last_login = Column(DateTime, default=func.now(), onupdate=func.now())

    documents = relationship("documents", back_populates="user")
    conversations = relationship("conversations", back_populates="user")

class documents(Base):
    __tablename__ = 'documents'

    id:Mapped[uuid.UUID] = Column(Uuid(as_uuid=True), primary_key=True, index=True, default=uuid.uuid4)
    file_name = Column(String)
    file_type = Column(String)
    file_size = Column(Integer)
    filebase_key = Column(String, nullable=True, unique=True)
    filebase_url = Column(String, nullable=True)
    bucket_name = Column(String, nullable=False, default="pdf-doc-docx")
    chunks = Column(Integer)
    primary_language = Column(String)

    created_at = Column(DateTime, default=func.now())
    updated_at = Column(DateTime, default=func.now(), onupdate=func.now())

    user_id = Column(Uuid(as_uuid=True), ForeignKey('users.id'))
    user = relationship("users", back_populates="documents")


class conversations(Base):
    __tablename__ = 'conversations'

    id:Mapped[uuid.UUID] = Column(Uuid(as_uuid=True), primary_key=True, index=True, default=uuid.uuid4)
    conversation_uuid = Column(String,unique=True,index=True,nullable=False)
    user_id = Column(Uuid(as_uuid=True), ForeignKey('users.id'))
    meta_data = Column(JSON, nullable=True)

    user = relationship("users", back_populates="conversations")
    messages = relationship("messages", back_populates="conversation")


class messages(Base):
    __tablename__ = 'messages'

    id:Mapped[uuid.UUID] = Column(Uuid(as_uuid=True), primary_key=True, index=True, default=uuid.uuid4)
    conversation_id = Column(
        String,
        ForeignKey('conversations.conversation_uuid'),
        index=True,
        nullable=False
    )
    role = Column(String)
    content = Column(String)
    meta_data = Column(JSON, nullable=True)
    created_at = Column(DateTime, default=func.now())
    conversation = relationship("conversations", back_populates="messages")