92834dbe0e
## Database Migration (SQLite → MySQL) - Add Alembic migration framework - Add 'tr_' prefix to all tables to avoid conflicts in shared database - Remove SQLite support, use MySQL exclusively - Add pymysql driver dependency - Change ad_token column to Text type for long JWT tokens ## Unified Environment Configuration - Centralize all hardcoded settings to environment variables - Backend: Extend Settings class in app/core/config.py - Frontend: Use Vite environment variables (import.meta.env) - Docker: Move credentials to environment variables - Update .env.example files with comprehensive documentation ## Test Organization - Move root-level test files to tests/ directory: - test_chat_room.py → tests/test_chat_room.py - test_websocket.py → tests/test_websocket.py - test_realtime_implementation.py → tests/test_realtime_implementation.py - Fix path references in test_realtime_implementation.py Breaking Changes: - CORS now requires explicit origins (no more wildcard) - All database tables renamed with 'tr_' prefix - SQLite no longer supported 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
410 lines
11 KiB
Python
410 lines
11 KiB
Python
"""Message service layer for database operations"""
|
|
from sqlalchemy.orm import Session
|
|
from sqlalchemy import desc, and_, func
|
|
from typing import List, Optional, Dict, Any
|
|
from datetime import datetime, timedelta
|
|
import uuid
|
|
|
|
from app.core.config import get_settings
|
|
from app.modules.realtime.models import Message, MessageType, MessageReaction, MessageEditHistory
|
|
from app.modules.realtime.schemas import (
|
|
MessageCreate,
|
|
MessageResponse,
|
|
MessageListResponse,
|
|
ReactionSummary
|
|
)
|
|
|
|
settings = get_settings()
|
|
|
|
|
|
class MessageService:
|
|
"""Service for message operations"""
|
|
|
|
@staticmethod
|
|
def create_message(
|
|
db: Session,
|
|
room_id: str,
|
|
sender_id: str,
|
|
content: str,
|
|
message_type: MessageType = MessageType.TEXT,
|
|
metadata: Optional[Dict[str, Any]] = None
|
|
) -> Message:
|
|
"""
|
|
Create a new message
|
|
|
|
Args:
|
|
db: Database session
|
|
room_id: Room ID
|
|
sender_id: User ID who sent the message
|
|
content: Message content
|
|
message_type: Type of message
|
|
metadata: Optional metadata (mentions, file refs, etc.)
|
|
|
|
Returns:
|
|
Created Message object
|
|
"""
|
|
# Get next sequence number for this room
|
|
max_seq = db.query(func.max(Message.sequence_number)).filter(
|
|
Message.room_id == room_id
|
|
).scalar()
|
|
next_seq = (max_seq or 0) + 1
|
|
|
|
message = Message(
|
|
message_id=str(uuid.uuid4()),
|
|
room_id=room_id,
|
|
sender_id=sender_id,
|
|
content=content,
|
|
message_type=message_type,
|
|
message_metadata=metadata or {},
|
|
created_at=datetime.utcnow(),
|
|
sequence_number=next_seq
|
|
)
|
|
|
|
db.add(message)
|
|
db.commit()
|
|
db.refresh(message)
|
|
|
|
return message
|
|
|
|
@staticmethod
|
|
def get_message(db: Session, message_id: str) -> Optional[Message]:
|
|
"""
|
|
Get a message by ID
|
|
|
|
Args:
|
|
db: Database session
|
|
message_id: Message ID
|
|
|
|
Returns:
|
|
Message object or None
|
|
"""
|
|
return db.query(Message).filter(
|
|
Message.message_id == message_id,
|
|
Message.deleted_at.is_(None)
|
|
).first()
|
|
|
|
@staticmethod
|
|
def get_messages(
|
|
db: Session,
|
|
room_id: str,
|
|
limit: int = 50,
|
|
before_timestamp: Optional[datetime] = None,
|
|
offset: int = 0,
|
|
include_deleted: bool = False
|
|
) -> MessageListResponse:
|
|
"""
|
|
Get paginated messages for a room
|
|
|
|
Args:
|
|
db: Database session
|
|
room_id: Room ID
|
|
limit: Number of messages to return
|
|
before_timestamp: Get messages before this timestamp
|
|
offset: Offset for pagination
|
|
include_deleted: Include soft-deleted messages
|
|
|
|
Returns:
|
|
MessageListResponse with messages and pagination info
|
|
"""
|
|
query = db.query(Message).filter(Message.room_id == room_id)
|
|
|
|
if not include_deleted:
|
|
query = query.filter(Message.deleted_at.is_(None))
|
|
|
|
if before_timestamp:
|
|
query = query.filter(Message.created_at < before_timestamp)
|
|
|
|
# Get total count
|
|
total = query.count()
|
|
|
|
# Get messages in reverse chronological order
|
|
messages = query.order_by(desc(Message.created_at)).offset(offset).limit(limit).all()
|
|
|
|
# Get reaction counts for each message
|
|
message_responses = []
|
|
for msg in messages:
|
|
reaction_counts = MessageService._get_reaction_counts(db, msg.message_id)
|
|
msg_response = MessageResponse.from_orm(msg)
|
|
msg_response.reaction_counts = reaction_counts
|
|
message_responses.append(msg_response)
|
|
|
|
return MessageListResponse(
|
|
messages=message_responses,
|
|
total=total,
|
|
limit=limit,
|
|
offset=offset,
|
|
has_more=(offset + len(messages)) < total
|
|
)
|
|
|
|
@staticmethod
|
|
def edit_message(
|
|
db: Session,
|
|
message_id: str,
|
|
user_id: str,
|
|
new_content: str
|
|
) -> Optional[Message]:
|
|
"""
|
|
Edit a message (must be own message and within 15 minutes)
|
|
|
|
Args:
|
|
db: Database session
|
|
message_id: Message ID to edit
|
|
user_id: User ID making the edit
|
|
new_content: New message content
|
|
|
|
Returns:
|
|
Updated Message object or None if not allowed
|
|
"""
|
|
message = db.query(Message).filter(Message.message_id == message_id).first()
|
|
|
|
if not message:
|
|
return None
|
|
|
|
# Check permissions
|
|
if message.sender_id != user_id:
|
|
return None
|
|
|
|
# Check time limit (configurable via MESSAGE_EDIT_TIME_LIMIT_MINUTES)
|
|
time_diff = datetime.utcnow() - message.created_at
|
|
if time_diff > timedelta(minutes=settings.MESSAGE_EDIT_TIME_LIMIT_MINUTES):
|
|
return None
|
|
|
|
# Store original content in edit history
|
|
edit_history = MessageEditHistory(
|
|
message_id=message_id,
|
|
original_content=message.content,
|
|
edited_by=user_id,
|
|
edited_at=datetime.utcnow()
|
|
)
|
|
db.add(edit_history)
|
|
|
|
# Update message
|
|
message.content = new_content
|
|
message.edited_at = datetime.utcnow()
|
|
|
|
db.commit()
|
|
db.refresh(message)
|
|
|
|
return message
|
|
|
|
@staticmethod
|
|
def delete_message(
|
|
db: Session,
|
|
message_id: str,
|
|
user_id: str,
|
|
is_admin: bool = False
|
|
) -> Optional[Message]:
|
|
"""
|
|
Soft delete a message
|
|
|
|
Args:
|
|
db: Database session
|
|
message_id: Message ID to delete
|
|
user_id: User ID making the deletion
|
|
is_admin: Whether user is admin (can delete any message)
|
|
|
|
Returns:
|
|
Deleted Message object or None if not allowed
|
|
"""
|
|
message = db.query(Message).filter(Message.message_id == message_id).first()
|
|
|
|
if not message:
|
|
return None
|
|
|
|
# Check permissions (owner or admin)
|
|
if not is_admin and message.sender_id != user_id:
|
|
return None
|
|
|
|
# Soft delete
|
|
message.deleted_at = datetime.utcnow()
|
|
|
|
db.commit()
|
|
db.refresh(message)
|
|
|
|
return message
|
|
|
|
@staticmethod
|
|
def search_messages(
|
|
db: Session,
|
|
room_id: str,
|
|
query: str,
|
|
limit: int = 50,
|
|
offset: int = 0
|
|
) -> MessageListResponse:
|
|
"""
|
|
Search messages by content
|
|
|
|
Args:
|
|
db: Database session
|
|
room_id: Room ID to search in
|
|
query: Search query
|
|
limit: Number of results
|
|
offset: Offset for pagination
|
|
|
|
Returns:
|
|
MessageListResponse with search results
|
|
"""
|
|
# Simple LIKE search (for PostgreSQL, use full-text search)
|
|
search_filter = and_(
|
|
Message.room_id == room_id,
|
|
Message.deleted_at.is_(None),
|
|
Message.content.contains(query)
|
|
)
|
|
|
|
total = db.query(Message).filter(search_filter).count()
|
|
|
|
messages = (
|
|
db.query(Message)
|
|
.filter(search_filter)
|
|
.order_by(desc(Message.created_at))
|
|
.offset(offset)
|
|
.limit(limit)
|
|
.all()
|
|
)
|
|
|
|
message_responses = []
|
|
for msg in messages:
|
|
reaction_counts = MessageService._get_reaction_counts(db, msg.message_id)
|
|
msg_response = MessageResponse.from_orm(msg)
|
|
msg_response.reaction_counts = reaction_counts
|
|
message_responses.append(msg_response)
|
|
|
|
return MessageListResponse(
|
|
messages=message_responses,
|
|
total=total,
|
|
limit=limit,
|
|
offset=offset,
|
|
has_more=(offset + len(messages)) < total
|
|
)
|
|
|
|
@staticmethod
|
|
def add_reaction(
|
|
db: Session,
|
|
message_id: str,
|
|
user_id: str,
|
|
emoji: str
|
|
) -> Optional[MessageReaction]:
|
|
"""
|
|
Add a reaction to a message
|
|
|
|
Args:
|
|
db: Database session
|
|
message_id: Message ID
|
|
user_id: User ID adding reaction
|
|
emoji: Emoji character
|
|
|
|
Returns:
|
|
MessageReaction object or None if already exists
|
|
"""
|
|
# Check if reaction already exists
|
|
existing = db.query(MessageReaction).filter(
|
|
and_(
|
|
MessageReaction.message_id == message_id,
|
|
MessageReaction.user_id == user_id,
|
|
MessageReaction.emoji == emoji
|
|
)
|
|
).first()
|
|
|
|
if existing:
|
|
return existing
|
|
|
|
reaction = MessageReaction(
|
|
message_id=message_id,
|
|
user_id=user_id,
|
|
emoji=emoji,
|
|
created_at=datetime.utcnow()
|
|
)
|
|
|
|
db.add(reaction)
|
|
db.commit()
|
|
db.refresh(reaction)
|
|
|
|
return reaction
|
|
|
|
@staticmethod
|
|
def remove_reaction(
|
|
db: Session,
|
|
message_id: str,
|
|
user_id: str,
|
|
emoji: str
|
|
) -> bool:
|
|
"""
|
|
Remove a reaction from a message
|
|
|
|
Args:
|
|
db: Database session
|
|
message_id: Message ID
|
|
user_id: User ID removing reaction
|
|
emoji: Emoji character
|
|
|
|
Returns:
|
|
True if removed, False if not found
|
|
"""
|
|
reaction = db.query(MessageReaction).filter(
|
|
and_(
|
|
MessageReaction.message_id == message_id,
|
|
MessageReaction.user_id == user_id,
|
|
MessageReaction.emoji == emoji
|
|
)
|
|
).first()
|
|
|
|
if not reaction:
|
|
return False
|
|
|
|
db.delete(reaction)
|
|
db.commit()
|
|
|
|
return True
|
|
|
|
@staticmethod
|
|
def get_message_reactions(
|
|
db: Session,
|
|
message_id: str
|
|
) -> List[ReactionSummary]:
|
|
"""
|
|
Get aggregated reactions for a message
|
|
|
|
Args:
|
|
db: Database session
|
|
message_id: Message ID
|
|
|
|
Returns:
|
|
List of ReactionSummary objects
|
|
"""
|
|
reactions = db.query(MessageReaction).filter(
|
|
MessageReaction.message_id == message_id
|
|
).all()
|
|
|
|
# Group by emoji
|
|
reaction_map: Dict[str, List[str]] = {}
|
|
for reaction in reactions:
|
|
if reaction.emoji not in reaction_map:
|
|
reaction_map[reaction.emoji] = []
|
|
reaction_map[reaction.emoji].append(reaction.user_id)
|
|
|
|
return [
|
|
ReactionSummary(emoji=emoji, count=len(users), users=users)
|
|
for emoji, users in reaction_map.items()
|
|
]
|
|
|
|
@staticmethod
|
|
def _get_reaction_counts(db: Session, message_id: str) -> Dict[str, int]:
|
|
"""
|
|
Get reaction counts for a message
|
|
|
|
Args:
|
|
db: Database session
|
|
message_id: Message ID
|
|
|
|
Returns:
|
|
Dictionary of emoji -> count
|
|
"""
|
|
result = (
|
|
db.query(MessageReaction.emoji, func.count(MessageReaction.reaction_id))
|
|
.filter(MessageReaction.message_id == message_id)
|
|
.group_by(MessageReaction.emoji)
|
|
.all()
|
|
)
|
|
|
|
return {emoji: count for emoji, count in result}
|