from sqlalchemy import Column, Integer, String, ForeignKey, TIMESTAMP, Text, JSON
from passlib.context import CryptContext
from sqlalchemy.orm import sessionmaker, relationship
from datetime import datetime
from .database import Base

pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")

class User(Base):
    __tablename__ = "users"
    
    user_id = Column(Integer, primary_key=True, index=True)
    name = Column(String(50), nullable=False)
    email = Column(String(50), unique=True, nullable=False)
    password_hash = Column(String(255), nullable=False)
    token = Column(String, unique=True, index=True)
    user_code = Column(String(25), nullable=True)
    org_id = Column(String(5), nullable=True)
    org_name = Column(String(255), nullable=True)
    plan = Column(String(5), nullable=True)
    allocated_storage = Column(Integer, nullable=True)
    folder_limit = Column(Integer, nullable=True)
    created_at = Column(TIMESTAMP, default=datetime.utcnow)
    updated_at = Column(TIMESTAMP, default=datetime.utcnow, onupdate=datetime.utcnow)

    def verify_password(self, password: str):
        return pwd_context.verify(password, self.password_hash)

    @staticmethod
    def hash_password(password: str):
        return pwd_context.hash(password)

# Category Model
class Folder(Base):
    __tablename__ = "folders"
    
    folder_id = Column(Integer, primary_key=True, index=True)
    name = Column(String(255), nullable=False)
    parent_id = Column(Integer, nullable=True)
    user_id = Column(Integer, nullable=True)
    shared = Column(String(25), default='No')
    sanitized_name = Column(String(50), nullable=True)
    created_at = Column(TIMESTAMP, default=datetime.utcnow)
    updated_at = Column(TIMESTAMP, default=datetime.utcnow, onupdate=datetime.utcnow)
    
    # parent = relationship("Folder", remote_side=[id], backref="subfolders")
    documents = relationship("Documents", back_populates="folder")

# Document model
class Documents(Base):
    __tablename__ = "documents"
    
    doc_id = Column(Integer, primary_key=True, index=True)
    folder_id = Column(Integer, ForeignKey("folders.folder_id"), nullable=False)
    document_uuid = Column(String, nullable=True)
    document_type = Column(String(255), nullable=True)
    doc_name = Column(String)
    sanitized_name = Column(String, nullable=True)
    doc_url = Column(String)
    doc_metadata = Column(JSON, nullable=True)
    doc_status = Column(String(15),nullable=True)
    task_id = Column(String, nullable=True)
    created_at = Column(TIMESTAMP, default=datetime.utcnow)
    updated_at = Column(TIMESTAMP, default=datetime.utcnow, onupdate=datetime.utcnow)
    folder = relationship('Folder', back_populates='documents')

class SharedFolder(Base):
    __tablename__ = "shared_folder"

    shared_folder_id = Column(Integer, primary_key=True, index=True)
    email = Column(String(255), nullable=True)
    organization = Column(String(255), nullable=True)
    folder_id = Column(Integer, ForeignKey("folders.folder_id"), nullable=False)
    created_at = Column(TIMESTAMP, default=datetime.utcnow)
    updated_at = Column(TIMESTAMP, default=datetime.utcnow, onupdate=datetime.utcnow)

class ChatHistory(Base):
    __tablename__ = "chat_history"

    chat_history_id = Column(Integer, primary_key=True, index=True)
    user_id = Column(Integer, ForeignKey("users.user_id"), nullable=False)
    doc_id = Column(Integer, nullable=True)
    document_uuid = Column(String, nullable=False)
    query = Column(Text, nullable=True)
    search_results = Column(Text, nullable=True)
    citation = Column(Text, nullable=True)
    query_prompt = Column(Text, nullable=True)
    query_response = Column(Text, nullable=True)
    created_at = Column(TIMESTAMP, default=datetime.utcnow)
    updated_at = Column(TIMESTAMP, default=datetime.utcnow, onupdate=datetime.utcnow)