feat: Initial commit - Task Reporter incident response system
Complete implementation of the production line incident response system (生產線異常即時反應系統) including: Backend (FastAPI): - User authentication with AD integration and session management - Chat room management (create, list, update, members, roles) - Real-time messaging via WebSocket (typing indicators, reactions) - File storage with MinIO (upload, download, image preview) Frontend (React + Vite): - Authentication flow with token management - Room list with filtering, search, and pagination - Real-time chat interface with WebSocket - File upload with drag-and-drop and image preview - Member management and room settings - Breadcrumb navigation - 53 unit tests (Vitest) Specifications: - authentication: AD auth, sessions, JWT tokens - chat-room: rooms, members, templates - realtime-messaging: WebSocket, messages, reactions - file-storage: MinIO integration, file management - frontend-core: React SPA structure 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
egg
1
app/modules/realtime/services/__init__.py
Normal file
1
app/modules/realtime/services/__init__.py
Normal file
@@ -0,0 +1 @@
|
||||
"""Service layer for realtime messaging"""
|
||||
406
app/modules/realtime/services/message_service.py
Normal file
406
app/modules/realtime/services/message_service.py
Normal file
@@ -0,0 +1,406 @@
|
||||
"""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.modules.realtime.models import Message, MessageType, MessageReaction, MessageEditHistory
|
||||
from app.modules.realtime.schemas import (
|
||||
MessageCreate,
|
||||
MessageResponse,
|
||||
MessageListResponse,
|
||||
ReactionSummary
|
||||
)
|
||||
|
||||
|
||||
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 (15 minutes)
|
||||
time_diff = datetime.utcnow() - message.created_at
|
||||
if time_diff > timedelta(minutes=15):
|
||||
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}
|
||||
Reference in New Issue
Block a user