Private
Public Access
ksmithandClaude Sonnet 5
c78d7454b6
Messages can now carry an arbitrary file (MessageFile), parallel to the existing MessageImage feature rather than a refactor of it. Files serve with Content-Disposition: attachment to force a download and prevent an uploaded HTML/SVG from executing same-origin. No content-type allowlist, same 8MB cap as images for now (a separate size-limit redesign is tracked as its own issue). Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
37 lines
1.5 KiB
Python
37 lines
1.5 KiB
Python
import uuid
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import CheckConstraint, DateTime, ForeignKey, Text, func
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from app.models.base import Base
|
|
|
|
|
|
class Message(Base):
|
|
__tablename__ = "messages"
|
|
__table_args__ = (
|
|
CheckConstraint(
|
|
"content IS NOT NULL OR image_id IS NOT NULL OR file_id IS NOT NULL",
|
|
name="messages_content_or_attachment_required",
|
|
),
|
|
)
|
|
|
|
id: Mapped[uuid.UUID] = mapped_column(primary_key=True, default=uuid.uuid4)
|
|
room_id: Mapped[uuid.UUID] = mapped_column(ForeignKey("rooms.id"), index=True, nullable=False)
|
|
user_id: Mapped[uuid.UUID] = mapped_column(ForeignKey("users.id"), nullable=False)
|
|
# Nullable since Phase "image uploads": a message can be an image with
|
|
# no caption. The CheckConstraint above still requires at least one of
|
|
# content/image_id/file_id.
|
|
content: Mapped[str | None] = mapped_column(Text)
|
|
image_id: Mapped[uuid.UUID | None] = mapped_column(ForeignKey("message_images.id"))
|
|
file_id: Mapped[uuid.UUID | None] = mapped_column(ForeignKey("message_files.id"))
|
|
created_at: Mapped[datetime] = mapped_column(
|
|
DateTime(timezone=True), server_default=func.now(), index=True, nullable=False
|
|
)
|
|
edited_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
|
|
deleted_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
|
|
|
|
user = relationship("User")
|
|
image = relationship("MessageImage")
|
|
file = relationship("MessageFile")
|