server/database/models/event.py

"""
Event database model
"""

from sqlalchemy import Column, Integer, String, Text, DateTime, Date, ForeignKey, Boolean, JSON, Enum
from sqlalchemy.orm import relationship
from datetime import datetime, timezone
from .base import Base
from sqlalchemy import event
from sqlalchemy.orm import object_session
from sqlalchemy.exc import IntegrityError

import enum


class ExtendedPropertyScope(enum.Enum):
    private = "private"
    shared = "shared"

class EventTypeEnum(enum.Enum):
    birthday = "birthday"
    default = "default"
    focusTime = "focusTime"
    fromGmail = "fromGmail"
    outOfOffice = "outOfOffice"
    workingLocation = "workingLocation"


class ReminderMethodEnum(enum.Enum):
    email = "email"
    popup = "popup"

class RecurrenceFrequency(enum.Enum):
    """RRULE frequency values from RFC 5545"""
    SECONDLY = "SECONDLY"
    MINUTELY = "MINUTELY"
    HOURLY = "HOURLY"
    DAILY = "DAILY"
    WEEKLY = "WEEKLY"
    MONTHLY = "MONTHLY"
    YEARLY = "YEARLY"


class RecurrenceWeekday(enum.Enum):
    """Days of the week for RRULE BYDAY parameter"""
    MO = "MO"  # Monday
    TU = "TU"  # Tuesday
    WE = "WE"  # Wednesday
    TH = "TH"  # Thursday
    FR = "FR"  # Friday
    SA = "SA"  # Saturday
    SU = "SU"  # Sunday




class Attendees(Base):
    """Attendees database model"""

    __tablename__ = "attendees"

    attendees_id = Column(String(255), primary_key=True, nullable=False)
    comment = Column(String(255), nullable = True)
    displayName = Column(String(255), nullable = True)
    additionalGuests = Column(Integer, default = 0)
    optional = Column(Boolean, default=False, nullable = False)
    resource = Column(Boolean, default=False, nullable = False)
    responseStatus = Column(String(50), nullable = False, default="needsAction")
    event_id = Column(String(255), ForeignKey("events.event_id", ondelete="CASCADE"), nullable = False)
    user_id = Column(String(255), ForeignKey("users.user_id"), nullable=True)

    # Relationships
    event = relationship("Event", back_populates = "attendees")
    user = relationship("User", back_populates="attendees")



class Attachment(Base):
    """Attachment model for storing file URLs linked to events"""

    __tablename__ = "attachments"

    attachment_id = Column(String(255), primary_key=True, nullable=False)
    event_id = Column(String(255), ForeignKey("events.event_id", ondelete="CASCADE"), nullable=False)
    file_url = Column(String(2000), nullable=False)  # long enough for Drive links

    # Relationship
    event = relationship("Event", back_populates="attachments")

    def __repr__(self):
        return f"<Attachment(attachment_id='{self.attachment_id}', event_id='{self.event_id}', file_url='{self.file_url}')>"


class OfficeLocation(Base):

    __tablename__ = "office_locations"

    id = Column(String(255), primary_key=True, nullable=False)
    buildingId = Column(String(255), nullable = True)
    deskId = Column(String(255), nullable = True)
    floorId = Column(String(255), nullable = True)
    floorSectionId = Column(String(255), nullable = True)
    label = Column(String(255), nullable = False)

    workingLocation = relationship("WorkingLocationProperties", back_populates="officeLocation", uselist=False)



class WorkingLocationProperties(Base):
    """Working location properties model for events"""

    __tablename__ = "working_location_properties"

    working_location_id = Column(String(255), primary_key=True, nullable=False)
    event_id = Column(String(255), ForeignKey("events.event_id", ondelete="CASCADE"), nullable=False)
    type = Column(
        String(50),
        nullable=False,
        doc=(
            "Type of working location. Required. Values: "
            "'homeOffice', 'officeLocation', 'customLocation'"
        )
    )
    homeOffice = Column(
        JSON,
        nullable=True,
        doc="If present, specifies that the user is working at home."
    )
    customLocationLabel = Column(
        String(255),
        nullable=True,
        doc="Optional extra label for custom location additional information."
    )
    officeLocationId = Column(String(255), ForeignKey("office_locations.id"), unique=True, nullable=True)

    # Relationship
    officeLocation = relationship("OfficeLocation", back_populates="workingLocation", uselist=False)
    event = relationship("Event", back_populates="workingLocationProperties")

    def __repr__(self):
        return f"<WorkingLocationProperties(working_location_id='{self.working_location_id}', event_id='{self.event_id}', type='{self.type}')>"

class ConferenceData(Base):
    __tablename__ = "conference_data"

    id = Column(String(50), primary_key=True, nullable = False)
    event_id = Column(String(255), ForeignKey("events.event_id", ondelete="CASCADE"), nullable=False, unique=True)

    # Conference solution fields
    solution_type = Column(String(100), nullable=True)  # conferenceSolution.key.type
    solution_name = Column(String(255), nullable=True)  # conferenceSolution.name  
    solution_icon_uri = Column(String(500), nullable=True)  # conferenceSolution.iconUri
    
    # Create request fields
    request_id = Column(String(255), nullable=True)  # createRequest.requestId
    create_solution_type = Column(String(100), nullable=True)  # createRequest.conferenceSolution.key.type
    status_code = Column(String(50), nullable=True)  # createRequest.status.statusCode
    
    # Entry points (stored as JSON array for flexibility)
    entry_points = Column(JSON, nullable=True)  # entryPoints array
    
    # Additional conference fields
    notes = Column(Text, nullable=True)  # notes
    signature = Column(String(500), nullable=True)  # signature
    
    # Legacy field for backward compatibility
    meeting_uri = Column(String(500), nullable=True)  # Primary meeting URI
    label = Column(String(255), nullable=True)  # Deprecated - use solution_name

    # Relationships
    event = relationship("Event", back_populates="conferenceData")


class BirthdayProperties(Base):
    __tablename__ = "birthday_properties"

    id = Column(String(50), primary_key=True, nullable = False)
    event_id = Column(String(255), ForeignKey("events.event_id", ondelete="CASCADE"), nullable=False, unique=True)
    type = Column(String(50), nullable=False, default="birthday")  # Default enforced

    # Relationship back to event
    event = relationship("Event", back_populates="birthdayProperties")


class ExtendedProperty(Base):
    __tablename__ = "extended_properties"

    id = Column(String(255), primary_key=True, nullable=False)
    event_id = Column(String(255), ForeignKey("events.event_id", ondelete="CASCADE"), nullable=False)
    scope = Column(Enum(ExtendedPropertyScope), nullable=False)  # private or shared
    properties = Column(JSON, nullable=True)

    event = relationship("Event", back_populates="extendedProperties")



class Reminder(Base):
    __tablename__ = "reminders"

    id = Column(String(255), primary_key=True, nullable=False)
    event_id = Column(String(255), ForeignKey("events.event_id", ondelete="CASCADE"), nullable=False)

    method = Column(Enum(ReminderMethodEnum, name="reminder_method_enum"), nullable=False)
    minutes = Column(Integer, nullable=False, doc="Minutes before the event to trigger reminder")
    use_default = Column(Boolean, nullable=False, default=True)

    event = relationship("Event", back_populates="reminders")



class Event(Base):
    """Event database model"""
    
    __tablename__ = "events"
    
    event_id = Column(String(255), primary_key=True, nullable=False)
    calendar_id = Column(String(255), ForeignKey("calendars.calendar_id"), nullable=False)
    user_id = Column(String(255), ForeignKey("users.user_id"), nullable=False, index=True)
    
    # Organizer information stored separately from user_id
    organizer_id = Column(String(255), nullable=True, doc="Organizer user ID")
    organizer_email = Column(String(255), nullable=True, doc="Organizer email address")
    organizer_display_name = Column(String(255), nullable=True, doc="Organizer display name")
    organizer_self = Column(Boolean, default=False, doc="Whether the organizer is the current user")
    
    # Foreign key to recurring event (if this is an instance of a recurring event)
    recurring_event_id = Column(String(255), ForeignKey("recurring_events.recurring_event_id", ondelete="CASCADE"), nullable=True, index=True)
    
    summary = Column(String(255), nullable=True)
    description = Column(Text, nullable=True)
    location = Column(String(500), nullable=True)
    start_datetime = Column(DateTime, nullable=False)
    end_datetime = Column(DateTime, nullable=False)
    start_timezone = Column(String(100), nullable=True)
    end_timezone = Column(String(100), nullable=True)
    recurrence = Column(Text, nullable=True)  # RRULE format
    status = Column(String(50), nullable=False, default="confirmed")
    visibility = Column(String(50), nullable=False, default="default")
    color_id = Column(String(50), ForeignKey("colors.id", ondelete="SET NULL"), nullable = True)
    
    
    eventType = Column(
        Enum(EventTypeEnum, name="event_type_enum"),
        nullable=False,
        default=EventTypeEnum.default,
        doc=(
            "Specific type of the event. Cannot be modified after creation. "
            "Allowed values: 'birthday', 'default', 'focusTime', "
            "'fromGmail', 'outOfOffice', 'workingLocation'."
        ),
    )

    focusTimeProperties = Column(
        JSON,
        nullable=True,
        doc="Focus Time event data. Used if eventType is focusTime.",
    )
    guestsCanInviteOthers = Column(
        Boolean,
        nullable=False,
        default=True,
        doc="Whether attendees other than the organizer can invite others to the event.",
    )
    guestsCanModify = Column(
        Boolean,
        nullable=False,
        default=False,
        doc="Whether attendees other than the organizer can modify the event.",
    )
    guestsCanSeeOtherGuests = Column(
        Boolean,
        nullable=False,
        default=True,
        doc="Whether attendees other than the organizer can see who the event's attendees are.",
    )
    outOfOfficeProperties = Column(
        JSON,
        nullable=True,
        doc="Out of office event data. Used if eventType is outOfOffice.",
    )
    sequence = Column(
        Integer,
        nullable=True,
        doc="Sequence number as per iCalendar.",
    )
    source = Column(
        JSON,
        nullable=True,
        doc=(
            "Source information for the event. Contains 'title' and 'url' properties. "
            "URL scheme must be HTTP or HTTPS."
        ),
    )
    
    transparency = Column(
        String(50),
        nullable=True,
        default="opaque",
        doc="Whether event blocks time. Values: 'opaque', 'transparent'"
    )
    iCalUID = Column(
        String(255),
        nullable=True,
        doc="Event iCalendar UID for external integration"
    )
    privateCopy = Column(
        Boolean,
        nullable=True,
        default=False,
        doc="Whether this is a private copy of the event"
    )
    locked = Column(
        Boolean,
        nullable=True,
        default=False,
        doc="Whether the event is locked against changes"
    )
    hangoutLink = Column(
        String(500),
        nullable=True,
        doc="Hangout video call link"
    )
    
    # Original start time fields for recurring events and event tracking
    originalStartTime_date = Column(
        Date,
        nullable=True,
        doc="The date, in the format 'yyyy-mm-dd', if this is an all-day event."
    )
    originalStartTime_dateTime = Column(
        DateTime,
        nullable=True,
        doc="The time, as a combined date-time value (formatted according to RFC3339). A time zone offset is required unless a time zone is explicitly specified in timeZone."
    )
    originalStartTime_timeZone = Column(
        String(100),
        nullable=True,
        doc="The time zone in which the time is specified. (Formatted as an IANA Time Zone Database name, e.g. 'Europe/Zurich'.) For recurring events this field is required and specifies the time zone in which the recurrence is expanded. For single events this field is optional and indicates a custom time zone for the event start/end."
    )
    
    created_at = Column(DateTime, default=lambda: datetime.now(timezone.utc))
    updated_at = Column(DateTime, default=lambda: datetime.now(timezone.utc), onupdate=lambda: datetime.now(timezone.utc))
    
    # Relationships
    calendar = relationship("Calendar", back_populates="events")
    user = relationship("User")
    recurring_event = relationship("RecurringEvent", back_populates="event_instances")
    attendees = relationship("Attendees", back_populates = "event", cascade="all, delete-orphan")
    attachments = relationship("Attachment", back_populates="event", cascade="all, delete-orphan")
    conferenceData = relationship("ConferenceData", uselist=False, back_populates="event", cascade="all, delete-orphan")
    birthdayProperties = relationship("BirthdayProperties", uselist=False, back_populates="event", cascade="all, delete-orphan")
    color = relationship("Color", back_populates="events")
    extendedProperties = relationship("ExtendedProperty", back_populates="event", cascade="all, delete-orphan", lazy="joined")
    reminders = relationship("Reminder", back_populates="event", cascade="all, delete-orphan")
    workingLocationProperties = relationship("WorkingLocationProperties", uselist=False, back_populates="event", cascade="all, delete-orphan")
    
    def __repr__(self):
        return f"<Event(event_id='{self.event_id}', user_id='{self.user_id}', summary='{self.summary}', calendar_id='{self.calendar_id}')>"
    

class RecurringEvent(Base):
    """
    RecurringEvent database model
    """

    __tablename__ = "recurring_events"

    # Primary key
    recurring_event_id = Column(String(255), primary_key=True, nullable=False)
    
    # Original recurrence string for reference
    original_recurrence = Column(Text, nullable=True, doc="Original recurrence string array as provided by user")

    # One-to-many relationship with Event instances
    event_instances = relationship("Event", back_populates="recurring_event", cascade="all, delete-orphan")





@event.listens_for(Event, "before_update", propagate=True)
def prevent_event_type_update(mapper, connection, target):
    """Ensure eventType is immutable after creation"""
    session = object_session(target)

    if session:
        db_event_type = (
            session.query(Event.eventType)
            .filter(Event.event_id == target.event_id)
            .scalar()
        )

        if db_event_type:
            # Normalize both values to strings for comparison
            db_type_str = db_event_type.value if hasattr(db_event_type, 'value') else str(db_event_type)
            target_type_str = target.eventType.value if hasattr(target.eventType, 'value') else str(target.eventType)
            
            if db_type_str != target_type_str:
                raise IntegrityError(
                    None, None,
                    f"eventType cannot be modified once set (was '{db_type_str}', tried to change to '{target_type_str}')"
                )