41bbd8c182
Complete rewrite from task management to full-featured chat system: Core Features: - Real-time WebSocket messaging with connection management - Direct messages and group chats with admin controls - Message types: text, images, videos, audio, documents - Message status tracking: sent, delivered, read receipts - Typing indicators and user presence (online/offline) - Message replies, editing, and deletion Security & Encryption: - End-to-end encryption with RSA + AES hybrid approach - JWT authentication for API and WebSocket connections - Secure file storage with access control - Automatic RSA key pair generation per user Media & File Sharing: - Multi-format file upload (images, videos, audio, documents) - Automatic thumbnail generation for images/videos - File size validation and MIME type checking - Secure download endpoints with permission checks Notifications & Alerts: - Real-time WebSocket notifications - Push notifications via Firebase integration - @username mention alerts with notification history - Unread message and mention counting - Custom notification types (message, mention, group invite) Advanced Features: - Group chat management with roles (member, admin, owner) - User search and chat member management - Message pagination and chat history - Last seen timestamps and activity tracking - Comprehensive API documentation with WebSocket events Architecture: - Clean layered architecture with services, models, schemas - WebSocket connection manager for real-time features - Modular notification system with multiple channels - Comprehensive error handling and validation - Production-ready with Docker support Technologies: FastAPI, WebSocket, SQLAlchemy, SQLite, Cryptography, Firebase, Pillow
35 lines
1.5 KiB
Python
35 lines
1.5 KiB
Python
from sqlalchemy import Column, Integer, String, DateTime, Boolean, BigInteger, ForeignKey, Enum
|
|
from sqlalchemy.orm import relationship
|
|
from sqlalchemy.sql import func
|
|
from enum import Enum as PyEnum
|
|
from app.db.base import Base
|
|
|
|
class MediaType(PyEnum):
|
|
IMAGE = "image"
|
|
VIDEO = "video"
|
|
AUDIO = "audio"
|
|
DOCUMENT = "document"
|
|
OTHER = "other"
|
|
|
|
class Media(Base):
|
|
__tablename__ = "media"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
message_id = Column(Integer, ForeignKey("messages.id"), nullable=False)
|
|
uploader_id = Column(Integer, ForeignKey("users.id"), nullable=False)
|
|
filename = Column(String, nullable=False)
|
|
original_filename = Column(String, nullable=False)
|
|
file_path = Column(String, nullable=False)
|
|
file_size = Column(BigInteger, nullable=False) # Size in bytes
|
|
mime_type = Column(String, nullable=False)
|
|
media_type = Column(Enum(MediaType), nullable=False)
|
|
width = Column(Integer, nullable=True) # For images/videos
|
|
height = Column(Integer, nullable=True) # For images/videos
|
|
duration = Column(Integer, nullable=True) # For videos/audio in seconds
|
|
thumbnail_path = Column(String, nullable=True) # For videos/images
|
|
is_processed = Column(Boolean, default=False) # For media processing
|
|
created_at = Column(DateTime(timezone=True), server_default=func.now())
|
|
|
|
# Relationships
|
|
message = relationship("Message", back_populates="media_files")
|
|
uploader = relationship("User", back_populates="sent_media") |