# modules/calendar/models.py
from sqlalchemy import Column, Integer, String, DateTime, ForeignKey, JSON, Boolean # Add Boolean
from sqlalchemy.orm import relationship
from core.database import Base

class CalendarEvent(Base):
    __tablename__ = "calendar_events"

    id = Column(Integer, primary_key=True)
    title = Column(String, nullable=False)
    description = Column(String)
    start = Column(DateTime, nullable=False)
    end = Column(DateTime)
    location = Column(String)
    all_day = Column(Boolean, default=False) # Add all_day column
    tags = Column(JSON)
    color = Column(String) # hex code for color
    user_id = Column(Integer, ForeignKey("users.id"), nullable=False)  # <-- Relationship

    # Bi-directional relationship (for eager loading)
    user = relationship("User", back_populates="calendar_events")