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
168 lines
8.2 KiB
Python
168 lines
8.2 KiB
Python
"""Initial migration for chat system
|
|
|
|
Revision ID: 001
|
|
Revises:
|
|
Create Date: 2024-12-20 12:00:00.000000
|
|
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = '001'
|
|
down_revision = None
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# Create users table
|
|
op.create_table('users',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('username', sa.String(), nullable=False),
|
|
sa.Column('email', sa.String(), nullable=False),
|
|
sa.Column('hashed_password', sa.String(), nullable=False),
|
|
sa.Column('full_name', sa.String(), nullable=True),
|
|
sa.Column('avatar_url', sa.String(), nullable=True),
|
|
sa.Column('bio', sa.Text(), nullable=True),
|
|
sa.Column('is_active', sa.Boolean(), nullable=True),
|
|
sa.Column('is_online', sa.Boolean(), nullable=True),
|
|
sa.Column('last_seen', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('public_key', sa.Text(), nullable=True),
|
|
sa.Column('device_token', sa.String(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('(CURRENT_TIMESTAMP)'), nullable=True),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_users_email'), 'users', ['email'], unique=True)
|
|
op.create_index(op.f('ix_users_id'), 'users', ['id'], unique=False)
|
|
op.create_index(op.f('ix_users_username'), 'users', ['username'], unique=True)
|
|
|
|
# Create chats table
|
|
op.create_table('chats',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('name', sa.String(), nullable=True),
|
|
sa.Column('description', sa.Text(), nullable=True),
|
|
sa.Column('chat_type', sa.Enum('DIRECT', 'GROUP', name='chattype'), nullable=False),
|
|
sa.Column('avatar_url', sa.String(), nullable=True),
|
|
sa.Column('is_active', sa.Boolean(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('(CURRENT_TIMESTAMP)'), nullable=True),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_chats_id'), 'chats', ['id'], unique=False)
|
|
|
|
# Create chat_members table
|
|
op.create_table('chat_members',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('chat_id', sa.Integer(), nullable=False),
|
|
sa.Column('user_id', sa.Integer(), nullable=False),
|
|
sa.Column('role', sa.Enum('MEMBER', 'ADMIN', 'OWNER', name='memberrole'), nullable=True),
|
|
sa.Column('nickname', sa.String(), nullable=True),
|
|
sa.Column('is_muted', sa.Boolean(), nullable=True),
|
|
sa.Column('is_banned', sa.Boolean(), nullable=True),
|
|
sa.Column('joined_at', sa.DateTime(timezone=True), server_default=sa.text('(CURRENT_TIMESTAMP)'), nullable=True),
|
|
sa.Column('last_read_message_id', sa.Integer(), nullable=True),
|
|
sa.ForeignKeyConstraint(['chat_id'], ['chats.id'], ),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_chat_members_id'), 'chat_members', ['id'], unique=False)
|
|
|
|
# Create messages table
|
|
op.create_table('messages',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('chat_id', sa.Integer(), nullable=False),
|
|
sa.Column('sender_id', sa.Integer(), nullable=False),
|
|
sa.Column('reply_to_id', sa.Integer(), nullable=True),
|
|
sa.Column('content', sa.Text(), nullable=True),
|
|
sa.Column('content_type', sa.Enum('TEXT', 'IMAGE', 'VIDEO', 'AUDIO', 'FILE', 'SYSTEM', name='messagetype'), nullable=True),
|
|
sa.Column('status', sa.Enum('SENT', 'DELIVERED', 'READ', 'FAILED', name='messagestatus'), nullable=True),
|
|
sa.Column('is_edited', sa.Boolean(), nullable=True),
|
|
sa.Column('is_deleted', sa.Boolean(), nullable=True),
|
|
sa.Column('edited_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('(CURRENT_TIMESTAMP)'), nullable=True),
|
|
sa.ForeignKeyConstraint(['chat_id'], ['chats.id'], ),
|
|
sa.ForeignKeyConstraint(['reply_to_id'], ['messages.id'], ),
|
|
sa.ForeignKeyConstraint(['sender_id'], ['users.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_messages_id'), 'messages', ['id'], unique=False)
|
|
|
|
# Create media table
|
|
op.create_table('media',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('message_id', sa.Integer(), nullable=False),
|
|
sa.Column('uploader_id', sa.Integer(), nullable=False),
|
|
sa.Column('filename', sa.String(), nullable=False),
|
|
sa.Column('original_filename', sa.String(), nullable=False),
|
|
sa.Column('file_path', sa.String(), nullable=False),
|
|
sa.Column('file_size', sa.BigInteger(), nullable=False),
|
|
sa.Column('mime_type', sa.String(), nullable=False),
|
|
sa.Column('media_type', sa.Enum('IMAGE', 'VIDEO', 'AUDIO', 'DOCUMENT', 'OTHER', name='mediatype'), nullable=False),
|
|
sa.Column('width', sa.Integer(), nullable=True),
|
|
sa.Column('height', sa.Integer(), nullable=True),
|
|
sa.Column('duration', sa.Integer(), nullable=True),
|
|
sa.Column('thumbnail_path', sa.String(), nullable=True),
|
|
sa.Column('is_processed', sa.Boolean(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('(CURRENT_TIMESTAMP)'), nullable=True),
|
|
sa.ForeignKeyConstraint(['message_id'], ['messages.id'], ),
|
|
sa.ForeignKeyConstraint(['uploader_id'], ['users.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_media_id'), 'media', ['id'], unique=False)
|
|
|
|
# Create mentions table
|
|
op.create_table('mentions',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('message_id', sa.Integer(), nullable=False),
|
|
sa.Column('mentioned_user_id', sa.Integer(), nullable=False),
|
|
sa.Column('is_read', sa.Boolean(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('(CURRENT_TIMESTAMP)'), nullable=True),
|
|
sa.Column('read_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.ForeignKeyConstraint(['mentioned_user_id'], ['users.id'], ),
|
|
sa.ForeignKeyConstraint(['message_id'], ['messages.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_mentions_id'), 'mentions', ['id'], unique=False)
|
|
|
|
# Create notifications table
|
|
op.create_table('notifications',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('user_id', sa.Integer(), nullable=False),
|
|
sa.Column('chat_id', sa.Integer(), nullable=True),
|
|
sa.Column('message_id', sa.Integer(), nullable=True),
|
|
sa.Column('notification_type', sa.Enum('MESSAGE', 'MENTION', 'GROUP_INVITE', 'GROUP_JOIN', 'GROUP_LEAVE', name='notificationtype'), nullable=False),
|
|
sa.Column('title', sa.String(), nullable=False),
|
|
sa.Column('body', sa.Text(), nullable=False),
|
|
sa.Column('data', sa.Text(), nullable=True),
|
|
sa.Column('is_read', sa.Boolean(), nullable=True),
|
|
sa.Column('is_sent', sa.Boolean(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('(CURRENT_TIMESTAMP)'), nullable=True),
|
|
sa.Column('read_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.ForeignKeyConstraint(['chat_id'], ['chats.id'], ),
|
|
sa.ForeignKeyConstraint(['message_id'], ['messages.id'], ),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_notifications_id'), 'notifications', ['id'], unique=False)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_index(op.f('ix_notifications_id'), table_name='notifications')
|
|
op.drop_table('notifications')
|
|
op.drop_index(op.f('ix_mentions_id'), table_name='mentions')
|
|
op.drop_table('mentions')
|
|
op.drop_index(op.f('ix_media_id'), table_name='media')
|
|
op.drop_table('media')
|
|
op.drop_index(op.f('ix_messages_id'), table_name='messages')
|
|
op.drop_table('messages')
|
|
op.drop_index(op.f('ix_chat_members_id'), table_name='chat_members')
|
|
op.drop_table('chat_members')
|
|
op.drop_index(op.f('ix_chats_id'), table_name='chats')
|
|
op.drop_table('chats')
|
|
op.drop_index(op.f('ix_users_username'), table_name='users')
|
|
op.drop_index(op.f('ix_users_id'), table_name='users')
|
|
op.drop_index(op.f('ix_users_email'), table_name='users')
|
|
op.drop_table('users') |