from sqlalchemy import Column, Integer, String, Text, ForeignKey, DateTime
from sqlalchemy.sql import func
from .database import Base


class User(Base):
    __tablename__ = "users"
    id = Column(Integer, primary_key=True, index=True)
    email = Column(String, unique=True, index=True)
    hashed_password = Column(String)
    mobile = Column(String, nullable=True)
    name = Column(String, nullable=True)
    dob = Column(String, nullable=True)
    preparing_for = Column(String, nullable=True)
    # Additional preferences/fields
    exam = Column(String, nullable=True)
    subjects = Column(Text, nullable=True)  # comma-separated subjects
    # Password reset support
    password_reset_token = Column(String, nullable=True, index=True)
    password_reset_expires = Column(DateTime(timezone=True), nullable=True)


class VideoUpload(Base):
    __tablename__ = "video_uploads"
    id = Column(Integer, primary_key=True, index=True)
    user_id = Column(Integer, ForeignKey("users.id"))
    video_url = Column(Text)
    pdf_url = Column(Text)
    status = Column(String, default="pending")  # pending, processing, completed
    created_at = Column(DateTime(timezone=True), server_default=func.now())
    updated_at = Column(DateTime(timezone=True), onupdate=func.now())


class Feed(Base):
    __tablename__ = "feeds"
    id = Column(Integer, primary_key=True, index=True)
    user_id = Column(Integer, ForeignKey("users.id"), index=True, nullable=False)
    title = Column(String, nullable=False)
    content = Column(Text, nullable=False)
    media_url = Column(Text, nullable=True)
    tags = Column(Text, nullable=True)  # comma-separated list
    category = Column(String, nullable=True)
    source = Column(String, nullable=True)
    relevance_score = Column(Integer, default=0)  # 0-100
    likes = Column(Integer, default=0)
    comments_count = Column(Integer, default=0)
    shares = Column(Integer, default=0)
    created_at = Column(DateTime(timezone=True), server_default=func.now())
    updated_at = Column(DateTime(timezone=True), onupdate=func.now())