feat: implement collaboration module

- Backend (FastAPI):
  - Task comments with nested replies and soft delete
  - @mention parsing with 10-mention limit per comment
  - Notification system with read/unread tracking
  - Blocker management with project owner notification
  - WebSocket endpoint with JWT auth and keepalive
  - User search API for @mention autocomplete
  - Alembic migration for 4 new tables

- Frontend (React + Vite):
  - Comments component with @mention autocomplete
  - NotificationBell with real-time WebSocket updates
  - BlockerDialog for task blocking workflow
  - NotificationContext for state management

- OpenSpec:
  - 4 requirements with scenarios defined
  - add-collaboration change archived

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
beabigegg
2025-12-29 20:45:07 +08:00
parent 61fe01cb6b
commit 3470428411
38 changed files with 3088 additions and 4 deletions

View File

@@ -0,0 +1,82 @@
import json
import asyncio
from typing import Dict, Set, Optional
from fastapi import WebSocket
from app.core.redis import get_redis_sync
class ConnectionManager:
"""Manager for WebSocket connections."""
def __init__(self):
# user_id -> set of WebSocket connections
self.active_connections: Dict[str, Set[WebSocket]] = {}
self._lock = asyncio.Lock()
async def connect(self, websocket: WebSocket, user_id: str):
"""Accept and track a new WebSocket connection."""
await websocket.accept()
async with self._lock:
if user_id not in self.active_connections:
self.active_connections[user_id] = set()
self.active_connections[user_id].add(websocket)
async def disconnect(self, websocket: WebSocket, user_id: str):
"""Remove a WebSocket connection."""
async with self._lock:
if user_id in self.active_connections:
self.active_connections[user_id].discard(websocket)
if not self.active_connections[user_id]:
del self.active_connections[user_id]
async def send_personal_message(self, message: dict, user_id: str):
"""Send a message to all connections of a specific user."""
if user_id in self.active_connections:
disconnected = set()
for connection in self.active_connections[user_id]:
try:
await connection.send_json(message)
except Exception:
disconnected.add(connection)
# Clean up disconnected connections
for conn in disconnected:
await self.disconnect(conn, user_id)
async def broadcast(self, message: dict):
"""Broadcast a message to all connected users."""
for user_id in list(self.active_connections.keys()):
await self.send_personal_message(message, user_id)
def is_connected(self, user_id: str) -> bool:
"""Check if a user has any active connections."""
return user_id in self.active_connections and len(self.active_connections[user_id]) > 0
# Global connection manager instance
manager = ConnectionManager()
async def publish_notification(user_id: str, notification_data: dict):
"""
Publish a notification to a user via WebSocket.
This can be called from anywhere in the application to send
real-time notifications to connected users.
"""
message = {
"type": "notification",
"data": notification_data,
}
await manager.send_personal_message(message, user_id)
async def publish_notification_count_update(user_id: str, unread_count: int):
"""
Publish an unread count update to a user.
"""
message = {
"type": "unread_count",
"data": {"unread_count": unread_count},
}
await manager.send_personal_message(message, user_id)