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
13
app/modules/chat_room/services/__init__.py
Normal file
13
app/modules/chat_room/services/__init__.py
Normal file
@@ -0,0 +1,13 @@
|
||||
"""Chat room services
|
||||
|
||||
Business logic for room management operations
|
||||
"""
|
||||
from app.modules.chat_room.services.room_service import room_service
|
||||
from app.modules.chat_room.services.membership_service import membership_service
|
||||
from app.modules.chat_room.services.template_service import template_service
|
||||
|
||||
__all__ = [
|
||||
"room_service",
|
||||
"membership_service",
|
||||
"template_service"
|
||||
]
|
||||
345
app/modules/chat_room/services/membership_service.py
Normal file
345
app/modules/chat_room/services/membership_service.py
Normal file
@@ -0,0 +1,345 @@
|
||||
"""Membership service for managing room members
|
||||
|
||||
Handles business logic for room membership operations
|
||||
"""
|
||||
from sqlalchemy.orm import Session
|
||||
from sqlalchemy import and_
|
||||
from typing import List, Optional
|
||||
from datetime import datetime
|
||||
|
||||
from app.modules.chat_room.models import RoomMember, IncidentRoom, MemberRole
|
||||
|
||||
|
||||
class MembershipService:
|
||||
"""Service for room membership operations"""
|
||||
|
||||
# System admin email (hardcoded as per requirement)
|
||||
SYSTEM_ADMIN_EMAIL = "ymirliu@panjit.com.tw"
|
||||
|
||||
def add_member(
|
||||
self,
|
||||
db: Session,
|
||||
room_id: str,
|
||||
user_id: str,
|
||||
role: MemberRole,
|
||||
added_by: str
|
||||
) -> Optional[RoomMember]:
|
||||
"""Add a member to a room
|
||||
|
||||
Args:
|
||||
db: Database session
|
||||
room_id: Room ID
|
||||
user_id: User to add
|
||||
role: Role to assign
|
||||
added_by: User adding the member
|
||||
|
||||
Returns:
|
||||
Created member or None if already exists
|
||||
"""
|
||||
# Check if member already exists (active)
|
||||
existing = db.query(RoomMember).filter(
|
||||
and_(
|
||||
RoomMember.room_id == room_id,
|
||||
RoomMember.user_id == user_id,
|
||||
RoomMember.removed_at.is_(None)
|
||||
)
|
||||
).first()
|
||||
|
||||
if existing:
|
||||
return None
|
||||
|
||||
# Create new member
|
||||
member = RoomMember(
|
||||
room_id=room_id,
|
||||
user_id=user_id,
|
||||
role=role,
|
||||
added_by=added_by,
|
||||
added_at=datetime.utcnow()
|
||||
)
|
||||
db.add(member)
|
||||
|
||||
# Update member count
|
||||
self._update_member_count(db, room_id)
|
||||
|
||||
db.commit()
|
||||
db.refresh(member)
|
||||
return member
|
||||
|
||||
def remove_member(
|
||||
self,
|
||||
db: Session,
|
||||
room_id: str,
|
||||
user_id: str
|
||||
) -> bool:
|
||||
"""Remove a member from a room (soft delete)
|
||||
|
||||
Args:
|
||||
db: Database session
|
||||
room_id: Room ID
|
||||
user_id: User to remove
|
||||
|
||||
Returns:
|
||||
True if removed, False if not found
|
||||
"""
|
||||
member = db.query(RoomMember).filter(
|
||||
and_(
|
||||
RoomMember.room_id == room_id,
|
||||
RoomMember.user_id == user_id,
|
||||
RoomMember.removed_at.is_(None)
|
||||
)
|
||||
).first()
|
||||
|
||||
if not member:
|
||||
return False
|
||||
|
||||
# Soft delete
|
||||
member.removed_at = datetime.utcnow()
|
||||
|
||||
# Update member count
|
||||
self._update_member_count(db, room_id)
|
||||
|
||||
db.commit()
|
||||
return True
|
||||
|
||||
def update_member_role(
|
||||
self,
|
||||
db: Session,
|
||||
room_id: str,
|
||||
user_id: str,
|
||||
new_role: MemberRole
|
||||
) -> Optional[RoomMember]:
|
||||
"""Update a member's role
|
||||
|
||||
Args:
|
||||
db: Database session
|
||||
room_id: Room ID
|
||||
user_id: User ID
|
||||
new_role: New role
|
||||
|
||||
Returns:
|
||||
Updated member or None if not found
|
||||
"""
|
||||
member = db.query(RoomMember).filter(
|
||||
and_(
|
||||
RoomMember.room_id == room_id,
|
||||
RoomMember.user_id == user_id,
|
||||
RoomMember.removed_at.is_(None)
|
||||
)
|
||||
).first()
|
||||
|
||||
if not member:
|
||||
return None
|
||||
|
||||
member.role = new_role
|
||||
db.commit()
|
||||
db.refresh(member)
|
||||
return member
|
||||
|
||||
def transfer_ownership(
|
||||
self,
|
||||
db: Session,
|
||||
room_id: str,
|
||||
current_owner_id: str,
|
||||
new_owner_id: str
|
||||
) -> bool:
|
||||
"""Transfer room ownership to another member
|
||||
|
||||
Args:
|
||||
db: Database session
|
||||
room_id: Room ID
|
||||
current_owner_id: Current owner's user ID
|
||||
new_owner_id: New owner's user ID
|
||||
|
||||
Returns:
|
||||
True if successful, False otherwise
|
||||
"""
|
||||
# Verify new owner is a member
|
||||
new_owner = db.query(RoomMember).filter(
|
||||
and_(
|
||||
RoomMember.room_id == room_id,
|
||||
RoomMember.user_id == new_owner_id,
|
||||
RoomMember.removed_at.is_(None)
|
||||
)
|
||||
).first()
|
||||
|
||||
if not new_owner:
|
||||
return False
|
||||
|
||||
# Get current owner
|
||||
current_owner = db.query(RoomMember).filter(
|
||||
and_(
|
||||
RoomMember.room_id == room_id,
|
||||
RoomMember.user_id == current_owner_id,
|
||||
RoomMember.role == MemberRole.OWNER,
|
||||
RoomMember.removed_at.is_(None)
|
||||
)
|
||||
).first()
|
||||
|
||||
if not current_owner:
|
||||
return False
|
||||
|
||||
# Transfer ownership
|
||||
new_owner.role = MemberRole.OWNER
|
||||
current_owner.role = MemberRole.EDITOR
|
||||
|
||||
# Update room ownership transfer tracking
|
||||
room = db.query(IncidentRoom).filter(
|
||||
IncidentRoom.room_id == room_id
|
||||
).first()
|
||||
|
||||
if room:
|
||||
room.ownership_transferred_at = datetime.utcnow()
|
||||
room.ownership_transferred_by = current_owner_id
|
||||
room.last_updated_at = datetime.utcnow()
|
||||
room.last_activity_at = datetime.utcnow()
|
||||
|
||||
db.commit()
|
||||
return True
|
||||
|
||||
def get_room_members(
|
||||
self,
|
||||
db: Session,
|
||||
room_id: str
|
||||
) -> List[RoomMember]:
|
||||
"""Get all active members of a room
|
||||
|
||||
Args:
|
||||
db: Database session
|
||||
room_id: Room ID
|
||||
|
||||
Returns:
|
||||
List of active members
|
||||
"""
|
||||
return db.query(RoomMember).filter(
|
||||
and_(
|
||||
RoomMember.room_id == room_id,
|
||||
RoomMember.removed_at.is_(None)
|
||||
)
|
||||
).all()
|
||||
|
||||
def get_user_rooms(
|
||||
self,
|
||||
db: Session,
|
||||
user_id: str
|
||||
) -> List[IncidentRoom]:
|
||||
"""Get all rooms where user is a member
|
||||
|
||||
Args:
|
||||
db: Database session
|
||||
user_id: User ID
|
||||
|
||||
Returns:
|
||||
List of rooms
|
||||
"""
|
||||
return db.query(IncidentRoom).join(RoomMember).filter(
|
||||
and_(
|
||||
RoomMember.user_id == user_id,
|
||||
RoomMember.removed_at.is_(None)
|
||||
)
|
||||
).all()
|
||||
|
||||
def get_user_role_in_room(
|
||||
self,
|
||||
db: Session,
|
||||
room_id: str,
|
||||
user_id: str
|
||||
) -> Optional[MemberRole]:
|
||||
"""Get user's role in a specific room
|
||||
|
||||
Args:
|
||||
db: Database session
|
||||
room_id: Room ID
|
||||
user_id: User ID
|
||||
|
||||
Returns:
|
||||
User's role or None if not a member
|
||||
"""
|
||||
member = db.query(RoomMember).filter(
|
||||
and_(
|
||||
RoomMember.room_id == room_id,
|
||||
RoomMember.user_id == user_id,
|
||||
RoomMember.removed_at.is_(None)
|
||||
)
|
||||
).first()
|
||||
|
||||
return member.role if member else None
|
||||
|
||||
def check_user_permission(
|
||||
self,
|
||||
db: Session,
|
||||
room_id: str,
|
||||
user_id: str,
|
||||
permission: str
|
||||
) -> bool:
|
||||
"""Check if user has specific permission in room
|
||||
|
||||
Args:
|
||||
db: Database session
|
||||
room_id: Room ID
|
||||
user_id: User ID
|
||||
permission: Permission to check
|
||||
|
||||
Returns:
|
||||
True if user has permission, False otherwise
|
||||
"""
|
||||
# Check if user is system admin
|
||||
if self.is_system_admin(user_id):
|
||||
return True
|
||||
|
||||
# Get user role
|
||||
role = self.get_user_role_in_room(db, room_id, user_id)
|
||||
|
||||
if not role:
|
||||
return False
|
||||
|
||||
# Permission matrix
|
||||
permissions = {
|
||||
MemberRole.OWNER: [
|
||||
"read", "write", "manage_members", "transfer_ownership",
|
||||
"update_status", "delete", "update_metadata"
|
||||
],
|
||||
MemberRole.EDITOR: [
|
||||
"read", "write", "add_viewer"
|
||||
],
|
||||
MemberRole.VIEWER: [
|
||||
"read"
|
||||
]
|
||||
}
|
||||
|
||||
return permission in permissions.get(role, [])
|
||||
|
||||
def is_system_admin(self, user_email: str) -> bool:
|
||||
"""Check if user is system administrator
|
||||
|
||||
Args:
|
||||
user_email: User's email
|
||||
|
||||
Returns:
|
||||
True if system admin, False otherwise
|
||||
"""
|
||||
return user_email == self.SYSTEM_ADMIN_EMAIL
|
||||
|
||||
def _update_member_count(self, db: Session, room_id: str) -> None:
|
||||
"""Update room's member count
|
||||
|
||||
Args:
|
||||
db: Database session
|
||||
room_id: Room ID
|
||||
"""
|
||||
count = db.query(RoomMember).filter(
|
||||
and_(
|
||||
RoomMember.room_id == room_id,
|
||||
RoomMember.removed_at.is_(None)
|
||||
)
|
||||
).count()
|
||||
|
||||
room = db.query(IncidentRoom).filter(
|
||||
IncidentRoom.room_id == room_id
|
||||
).first()
|
||||
|
||||
if room:
|
||||
room.member_count = count
|
||||
|
||||
|
||||
# Create singleton instance
|
||||
membership_service = MembershipService()
|
||||
386
app/modules/chat_room/services/room_service.py
Normal file
386
app/modules/chat_room/services/room_service.py
Normal file
@@ -0,0 +1,386 @@
|
||||
"""Room service for managing incident rooms
|
||||
|
||||
Handles business logic for room CRUD operations
|
||||
"""
|
||||
from sqlalchemy.orm import Session
|
||||
from sqlalchemy import or_, and_, func
|
||||
from typing import List, Optional, Dict
|
||||
from datetime import datetime
|
||||
import uuid
|
||||
|
||||
from app.modules.chat_room.models import IncidentRoom, RoomMember, RoomStatus, MemberRole
|
||||
from app.modules.chat_room.schemas import CreateRoomRequest, UpdateRoomRequest, RoomFilterParams
|
||||
|
||||
|
||||
class RoomService:
|
||||
"""Service for room management operations"""
|
||||
|
||||
def create_room(
|
||||
self,
|
||||
db: Session,
|
||||
user_id: str,
|
||||
room_data: CreateRoomRequest
|
||||
) -> IncidentRoom:
|
||||
"""Create a new incident room
|
||||
|
||||
Args:
|
||||
db: Database session
|
||||
user_id: ID of user creating the room
|
||||
room_data: Room creation data
|
||||
|
||||
Returns:
|
||||
Created room instance
|
||||
"""
|
||||
# Create room
|
||||
room = IncidentRoom(
|
||||
room_id=str(uuid.uuid4()),
|
||||
title=room_data.title,
|
||||
incident_type=room_data.incident_type,
|
||||
severity=room_data.severity,
|
||||
location=room_data.location,
|
||||
description=room_data.description,
|
||||
status=RoomStatus.ACTIVE,
|
||||
created_by=user_id,
|
||||
created_at=datetime.utcnow(),
|
||||
last_activity_at=datetime.utcnow(),
|
||||
last_updated_at=datetime.utcnow(),
|
||||
member_count=1
|
||||
)
|
||||
db.add(room)
|
||||
|
||||
# Add creator as owner
|
||||
owner = RoomMember(
|
||||
room_id=room.room_id,
|
||||
user_id=user_id,
|
||||
role=MemberRole.OWNER,
|
||||
added_by=user_id,
|
||||
added_at=datetime.utcnow()
|
||||
)
|
||||
db.add(owner)
|
||||
|
||||
db.commit()
|
||||
db.refresh(room)
|
||||
return room
|
||||
|
||||
def get_room(
|
||||
self,
|
||||
db: Session,
|
||||
room_id: str,
|
||||
user_id: str,
|
||||
is_admin: bool = False
|
||||
) -> Optional[IncidentRoom]:
|
||||
"""Get room details
|
||||
|
||||
Args:
|
||||
db: Database session
|
||||
room_id: Room ID
|
||||
user_id: User requesting access
|
||||
is_admin: Whether user is system admin
|
||||
|
||||
Returns:
|
||||
Room instance if user has access, None otherwise
|
||||
"""
|
||||
room = db.query(IncidentRoom).filter(
|
||||
IncidentRoom.room_id == room_id
|
||||
).first()
|
||||
|
||||
if not room:
|
||||
return None
|
||||
|
||||
# Check access: admin or member
|
||||
if not is_admin:
|
||||
member = db.query(RoomMember).filter(
|
||||
and_(
|
||||
RoomMember.room_id == room_id,
|
||||
RoomMember.user_id == user_id,
|
||||
RoomMember.removed_at.is_(None)
|
||||
)
|
||||
).first()
|
||||
|
||||
if not member:
|
||||
return None
|
||||
|
||||
return room
|
||||
|
||||
def list_user_rooms(
|
||||
self,
|
||||
db: Session,
|
||||
user_id: str,
|
||||
filters: RoomFilterParams,
|
||||
is_admin: bool = False
|
||||
) -> List[IncidentRoom]:
|
||||
"""List rooms accessible to user with filters
|
||||
|
||||
Args:
|
||||
db: Database session
|
||||
user_id: User ID
|
||||
filters: Filter parameters
|
||||
is_admin: Whether user is system admin
|
||||
|
||||
Returns:
|
||||
List of accessible rooms
|
||||
"""
|
||||
query = db.query(IncidentRoom)
|
||||
|
||||
# Access control: admin sees all, others see only their rooms
|
||||
if not is_admin or not filters.all:
|
||||
# Join with room_members to filter by membership
|
||||
query = query.join(RoomMember).filter(
|
||||
and_(
|
||||
RoomMember.user_id == user_id,
|
||||
RoomMember.removed_at.is_(None)
|
||||
)
|
||||
)
|
||||
|
||||
# Apply filters
|
||||
if filters.status:
|
||||
query = query.filter(IncidentRoom.status == filters.status)
|
||||
|
||||
if filters.incident_type:
|
||||
query = query.filter(IncidentRoom.incident_type == filters.incident_type)
|
||||
|
||||
if filters.severity:
|
||||
query = query.filter(IncidentRoom.severity == filters.severity)
|
||||
|
||||
if filters.created_after:
|
||||
query = query.filter(IncidentRoom.created_at >= filters.created_after)
|
||||
|
||||
if filters.created_before:
|
||||
query = query.filter(IncidentRoom.created_at <= filters.created_before)
|
||||
|
||||
if filters.search:
|
||||
search_term = f"%{filters.search}%"
|
||||
query = query.filter(
|
||||
or_(
|
||||
IncidentRoom.title.ilike(search_term),
|
||||
IncidentRoom.description.ilike(search_term)
|
||||
)
|
||||
)
|
||||
|
||||
# Order by last activity (most recent first)
|
||||
query = query.order_by(IncidentRoom.last_activity_at.desc())
|
||||
|
||||
# Apply pagination
|
||||
total = query.count()
|
||||
rooms = query.offset(filters.offset).limit(filters.limit).all()
|
||||
|
||||
return rooms, total
|
||||
|
||||
def update_room(
|
||||
self,
|
||||
db: Session,
|
||||
room_id: str,
|
||||
updates: UpdateRoomRequest
|
||||
) -> Optional[IncidentRoom]:
|
||||
"""Update room metadata
|
||||
|
||||
Args:
|
||||
db: Database session
|
||||
room_id: Room ID
|
||||
updates: Update data
|
||||
|
||||
Returns:
|
||||
Updated room or None if not found
|
||||
"""
|
||||
room = db.query(IncidentRoom).filter(
|
||||
IncidentRoom.room_id == room_id
|
||||
).first()
|
||||
|
||||
if not room:
|
||||
return None
|
||||
|
||||
# Apply updates
|
||||
if updates.title is not None:
|
||||
room.title = updates.title
|
||||
|
||||
if updates.severity is not None:
|
||||
room.severity = updates.severity
|
||||
|
||||
if updates.location is not None:
|
||||
room.location = updates.location
|
||||
|
||||
if updates.description is not None:
|
||||
room.description = updates.description
|
||||
|
||||
if updates.resolution_notes is not None:
|
||||
room.resolution_notes = updates.resolution_notes
|
||||
|
||||
# Handle status transitions
|
||||
if updates.status is not None:
|
||||
if not self._validate_status_transition(room.status, updates.status):
|
||||
raise ValueError(f"Invalid status transition from {room.status} to {updates.status}")
|
||||
|
||||
room.status = updates.status
|
||||
|
||||
# Update timestamps based on status
|
||||
if updates.status == RoomStatus.RESOLVED:
|
||||
room.resolved_at = datetime.utcnow()
|
||||
elif updates.status == RoomStatus.ARCHIVED:
|
||||
room.archived_at = datetime.utcnow()
|
||||
|
||||
# Update activity timestamps
|
||||
room.last_updated_at = datetime.utcnow()
|
||||
room.last_activity_at = datetime.utcnow()
|
||||
|
||||
db.commit()
|
||||
db.refresh(room)
|
||||
return room
|
||||
|
||||
def change_room_status(
|
||||
self,
|
||||
db: Session,
|
||||
room_id: str,
|
||||
new_status: RoomStatus
|
||||
) -> Optional[IncidentRoom]:
|
||||
"""Change room status with validation
|
||||
|
||||
Args:
|
||||
db: Database session
|
||||
room_id: Room ID
|
||||
new_status: New status
|
||||
|
||||
Returns:
|
||||
Updated room or None
|
||||
"""
|
||||
room = db.query(IncidentRoom).filter(
|
||||
IncidentRoom.room_id == room_id
|
||||
).first()
|
||||
|
||||
if not room:
|
||||
return None
|
||||
|
||||
if not self._validate_status_transition(room.status, new_status):
|
||||
raise ValueError(f"Invalid status transition from {room.status} to {new_status}")
|
||||
|
||||
room.status = new_status
|
||||
|
||||
# Update timestamps
|
||||
if new_status == RoomStatus.RESOLVED:
|
||||
room.resolved_at = datetime.utcnow()
|
||||
elif new_status == RoomStatus.ARCHIVED:
|
||||
room.archived_at = datetime.utcnow()
|
||||
|
||||
room.last_updated_at = datetime.utcnow()
|
||||
room.last_activity_at = datetime.utcnow()
|
||||
|
||||
db.commit()
|
||||
db.refresh(room)
|
||||
return room
|
||||
|
||||
def search_rooms(
|
||||
self,
|
||||
db: Session,
|
||||
user_id: str,
|
||||
search_term: str,
|
||||
is_admin: bool = False
|
||||
) -> List[IncidentRoom]:
|
||||
"""Search rooms by title or description
|
||||
|
||||
Args:
|
||||
db: Database session
|
||||
user_id: User ID
|
||||
search_term: Search string
|
||||
is_admin: Whether user is system admin
|
||||
|
||||
Returns:
|
||||
List of matching rooms
|
||||
"""
|
||||
query = db.query(IncidentRoom)
|
||||
|
||||
# Access control
|
||||
if not is_admin:
|
||||
query = query.join(RoomMember).filter(
|
||||
and_(
|
||||
RoomMember.user_id == user_id,
|
||||
RoomMember.removed_at.is_(None)
|
||||
)
|
||||
)
|
||||
|
||||
# Search filter
|
||||
search_pattern = f"%{search_term}%"
|
||||
query = query.filter(
|
||||
or_(
|
||||
IncidentRoom.title.ilike(search_pattern),
|
||||
IncidentRoom.description.ilike(search_pattern)
|
||||
)
|
||||
)
|
||||
|
||||
return query.order_by(IncidentRoom.last_activity_at.desc()).all()
|
||||
|
||||
def delete_room(
|
||||
self,
|
||||
db: Session,
|
||||
room_id: str
|
||||
) -> bool:
|
||||
"""Soft delete a room (archive it)
|
||||
|
||||
Args:
|
||||
db: Database session
|
||||
room_id: Room ID
|
||||
|
||||
Returns:
|
||||
True if deleted, False if not found
|
||||
"""
|
||||
room = db.query(IncidentRoom).filter(
|
||||
IncidentRoom.room_id == room_id
|
||||
).first()
|
||||
|
||||
if not room:
|
||||
return False
|
||||
|
||||
room.status = RoomStatus.ARCHIVED
|
||||
room.archived_at = datetime.utcnow()
|
||||
room.last_updated_at = datetime.utcnow()
|
||||
|
||||
db.commit()
|
||||
return True
|
||||
|
||||
def _validate_status_transition(
|
||||
self,
|
||||
current_status: RoomStatus,
|
||||
new_status: RoomStatus
|
||||
) -> bool:
|
||||
"""Validate status transition
|
||||
|
||||
Valid transitions:
|
||||
- active -> resolved
|
||||
- resolved -> archived
|
||||
- active -> archived (allowed but not recommended)
|
||||
|
||||
Args:
|
||||
current_status: Current status
|
||||
new_status: New status
|
||||
|
||||
Returns:
|
||||
True if valid, False otherwise
|
||||
"""
|
||||
valid_transitions = {
|
||||
RoomStatus.ACTIVE: [RoomStatus.RESOLVED, RoomStatus.ARCHIVED],
|
||||
RoomStatus.RESOLVED: [RoomStatus.ARCHIVED],
|
||||
RoomStatus.ARCHIVED: [] # No transitions from archived
|
||||
}
|
||||
|
||||
return new_status in valid_transitions.get(current_status, [])
|
||||
|
||||
def update_room_activity(
|
||||
self,
|
||||
db: Session,
|
||||
room_id: str
|
||||
) -> None:
|
||||
"""Update room's last activity timestamp
|
||||
|
||||
Args:
|
||||
db: Database session
|
||||
room_id: Room ID
|
||||
"""
|
||||
room = db.query(IncidentRoom).filter(
|
||||
IncidentRoom.room_id == room_id
|
||||
).first()
|
||||
|
||||
if room:
|
||||
room.last_activity_at = datetime.utcnow()
|
||||
db.commit()
|
||||
|
||||
|
||||
# Create singleton instance
|
||||
room_service = RoomService()
|
||||
179
app/modules/chat_room/services/template_service.py
Normal file
179
app/modules/chat_room/services/template_service.py
Normal file
@@ -0,0 +1,179 @@
|
||||
"""Template service for room templates
|
||||
|
||||
Handles business logic for room template operations
|
||||
"""
|
||||
from sqlalchemy.orm import Session
|
||||
from typing import List, Optional
|
||||
import json
|
||||
from datetime import datetime
|
||||
|
||||
from app.modules.chat_room.models import RoomTemplate, IncidentRoom, RoomMember, IncidentType, SeverityLevel, MemberRole
|
||||
from app.modules.chat_room.services.room_service import room_service
|
||||
from app.modules.chat_room.services.membership_service import membership_service
|
||||
|
||||
|
||||
class TemplateService:
|
||||
"""Service for room template operations"""
|
||||
|
||||
def get_templates(self, db: Session) -> List[RoomTemplate]:
|
||||
"""Get all available templates
|
||||
|
||||
Args:
|
||||
db: Database session
|
||||
|
||||
Returns:
|
||||
List of templates
|
||||
"""
|
||||
return db.query(RoomTemplate).all()
|
||||
|
||||
def get_template_by_name(
|
||||
self,
|
||||
db: Session,
|
||||
template_name: str
|
||||
) -> Optional[RoomTemplate]:
|
||||
"""Get template by name
|
||||
|
||||
Args:
|
||||
db: Database session
|
||||
template_name: Template name
|
||||
|
||||
Returns:
|
||||
Template or None if not found
|
||||
"""
|
||||
return db.query(RoomTemplate).filter(
|
||||
RoomTemplate.name == template_name
|
||||
).first()
|
||||
|
||||
def create_room_from_template(
|
||||
self,
|
||||
db: Session,
|
||||
template_id: int,
|
||||
user_id: str,
|
||||
title: str,
|
||||
location: Optional[str] = None,
|
||||
description: Optional[str] = None
|
||||
) -> Optional[IncidentRoom]:
|
||||
"""Create a room from a template
|
||||
|
||||
Args:
|
||||
db: Database session
|
||||
template_id: Template ID
|
||||
user_id: User creating the room
|
||||
title: Room title
|
||||
location: Optional location override
|
||||
description: Optional description override
|
||||
|
||||
Returns:
|
||||
Created room or None if template not found
|
||||
"""
|
||||
# Get template
|
||||
template = db.query(RoomTemplate).filter(
|
||||
RoomTemplate.template_id == template_id
|
||||
).first()
|
||||
|
||||
if not template:
|
||||
return None
|
||||
|
||||
# Create room with template defaults
|
||||
room = IncidentRoom(
|
||||
title=title,
|
||||
incident_type=template.incident_type,
|
||||
severity=template.default_severity,
|
||||
location=location,
|
||||
description=description or template.description,
|
||||
created_by=user_id,
|
||||
status="active",
|
||||
created_at=datetime.utcnow(),
|
||||
last_activity_at=datetime.utcnow(),
|
||||
last_updated_at=datetime.utcnow(),
|
||||
member_count=1
|
||||
)
|
||||
db.add(room)
|
||||
db.flush() # Get room_id
|
||||
|
||||
# Add creator as owner
|
||||
owner = RoomMember(
|
||||
room_id=room.room_id,
|
||||
user_id=user_id,
|
||||
role=MemberRole.OWNER,
|
||||
added_by=user_id,
|
||||
added_at=datetime.utcnow()
|
||||
)
|
||||
db.add(owner)
|
||||
|
||||
# Add default members from template
|
||||
if template.default_members:
|
||||
try:
|
||||
default_members = json.loads(template.default_members)
|
||||
for member_config in default_members:
|
||||
if member_config.get("user_id") != user_id: # Don't duplicate owner
|
||||
member = RoomMember(
|
||||
room_id=room.room_id,
|
||||
user_id=member_config["user_id"],
|
||||
role=member_config.get("role", MemberRole.VIEWER),
|
||||
added_by=user_id,
|
||||
added_at=datetime.utcnow()
|
||||
)
|
||||
db.add(member)
|
||||
room.member_count += 1
|
||||
except (json.JSONDecodeError, KeyError):
|
||||
# Invalid template configuration, skip default members
|
||||
pass
|
||||
|
||||
db.commit()
|
||||
db.refresh(room)
|
||||
return room
|
||||
|
||||
def initialize_default_templates(self, db: Session) -> None:
|
||||
"""Initialize default templates if none exist
|
||||
|
||||
Args:
|
||||
db: Database session
|
||||
"""
|
||||
# Check if templates already exist
|
||||
existing = db.query(RoomTemplate).count()
|
||||
if existing > 0:
|
||||
return
|
||||
|
||||
# Create default templates
|
||||
templates = [
|
||||
RoomTemplate(
|
||||
name="equipment_failure",
|
||||
description="Equipment failure incident requiring immediate attention",
|
||||
incident_type=IncidentType.EQUIPMENT_FAILURE,
|
||||
default_severity=SeverityLevel.HIGH,
|
||||
default_members=json.dumps([
|
||||
{"user_id": "maintenance_team@panjit.com.tw", "role": "editor"},
|
||||
{"user_id": "engineering@panjit.com.tw", "role": "viewer"}
|
||||
])
|
||||
),
|
||||
RoomTemplate(
|
||||
name="material_shortage",
|
||||
description="Material shortage affecting production",
|
||||
incident_type=IncidentType.MATERIAL_SHORTAGE,
|
||||
default_severity=SeverityLevel.MEDIUM,
|
||||
default_members=json.dumps([
|
||||
{"user_id": "procurement@panjit.com.tw", "role": "editor"},
|
||||
{"user_id": "logistics@panjit.com.tw", "role": "editor"}
|
||||
])
|
||||
),
|
||||
RoomTemplate(
|
||||
name="quality_issue",
|
||||
description="Quality control issue requiring investigation",
|
||||
incident_type=IncidentType.QUALITY_ISSUE,
|
||||
default_severity=SeverityLevel.HIGH,
|
||||
default_members=json.dumps([
|
||||
{"user_id": "quality_team@panjit.com.tw", "role": "editor"},
|
||||
{"user_id": "production_manager@panjit.com.tw", "role": "viewer"}
|
||||
])
|
||||
)
|
||||
]
|
||||
|
||||
for template in templates:
|
||||
db.add(template)
|
||||
|
||||
db.commit()
|
||||
|
||||
|
||||
# Create singleton instance
|
||||
template_service = TemplateService()
|
||||
Reference in New Issue
Block a user