meeting-room/backend/meetings/views.py

from rest_framework import viewsets, status, permissions
from rest_framework.decorators import action
from rest_framework.response import Response
from django.utils import timezone
from .models import Meeting, Participant, Message, MeetingMinutes
from .serializers import (
    MeetingSerializer, ParticipantSerializer, 
    MessageSerializer, InboxSerializer
)
from django.contrib.auth import get_user_model
from django.shortcuts import get_object_or_404
import uuid

User = get_user_model()


class MeetingViewSet(viewsets.ModelViewSet):
    """会议室视图集"""
    queryset = Meeting.objects.all()
    serializer_class = MeetingSerializer
    permission_classes = []  # 临时开放所有权限
    
    def get_queryset(self):
        # 简单返回所有会议
        return Meeting.objects.all().order_by('-created_at')
    
    def create(self, request, *args, **kwargs):
        """创建会议"""
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        
        # 临时：使用第一个用户作为 host
        from django.contrib.auth import get_user_model
        User = get_user_model()
        host = User.objects.first()
        
        meeting = serializer.save(host=host)
        
        # 创建主持人参会记录
        Participant.objects.create(
            meeting=meeting,
            user=host,
            agent_type='human',
            nickname=host.username,
            is_host=True
        )
        
        return Response(serializer.data, status=status.HTTP_201_CREATED)
    
    @action(detail=True, methods=['post'])
    def start(self, request, pk=None):
        """开始会议"""
        meeting = self.get_object()
        if meeting.host != request.user:
            return Response(
                {'error': '只有主持人可以开始会议'},
                status=status.HTTP_403_FORBIDDEN
            )
        
        meeting.status = 'active'
        meeting.started_at = timezone.now()
        meeting.save()
        
        return Response({'status': '会议已开始'})
    
    @action(detail=True, methods=['post'])
    def end(self, request, pk=None):
        """结束会议"""
        meeting = self.get_object()
        if meeting.host != request.user:
            return Response(
                {'error': '只有主持人可以结束会议'},
                status=status.HTTP_403_FORBIDDEN
            )
        
        meeting.status = 'ended'
        meeting.ended_at = timezone.now()
        meeting.save()
        
        return Response({'status': '会议已结束'})
    
    @action(detail=True, methods=['post'])
    def join(self, request, pk=None):
        """加入会议"""
        meeting = self.get_object()
        
        if meeting.status == 'ended':
            return Response(
                {'error': '会议已结束'},
                status=status.HTTP_400_BAD_REQUEST
            )
        
        invite_code = request.data.get('invite_code')
        if invite_code != meeting.invite_code:
            return Response(
                {'error': '邀请码错误'},
                status=status.HTTP_400_BAD_REQUEST
            )
        
        # 检查是否已加入
        existing = Participant.objects.filter(
            meeting=meeting,
            user=request.user,
            left_at__isnull=True
        ).first()
        
        if existing:
            return Response(ParticipantSerializer(existing).data)
        
        # 创建参会记录
        participant = Participant.objects.create(
            meeting=meeting,
            user=request.user,
            agent_type='human',
            nickname=request.user.username
        )
        
        return Response(ParticipantSerializer(participant).data, status=status.HTTP_201_CREATED)
    
    @action(detail=True, methods=['get'])
    def participants(self, request, pk=None):
        """获取参会者列表"""
        meeting = self.get_object()
        participants = meeting.participants.filter(left_at__isnull=True)
        serializer = ParticipantSerializer(participants, many=True)
        return Response(serializer.data)
    
    @action(detail=True, methods=['get'])
    def messages(self, request, pk=None):
        """获取消息（人类轮询）"""
        meeting = self.get_object()
        last_id = request.query_params.get('last_id', 0)
        
        messages = meeting.messages.filter(id__gt=last_id).select_related('sender')
        serializer = MessageSerializer(messages, many=True)
        
        return Response({'messages': serializer.data})
    
    @action(detail=True, methods=['post'])
    def send_message(self, request, pk=None):
        """发送消息"""
        meeting = self.get_object()
        
        # 获取或创建参会者（临时：使用第一个参会者或创建）
        participant = Participant.objects.filter(
            meeting=meeting,
            left_at__isnull=True
        ).first()
        
        if not participant:
            # 创建默认参会者
            host = meeting.host
            participant = Participant.objects.create(
                meeting=meeting,
                user=host,
                agent_type='human',
                nickname=host.username
            )
        
        content = request.data.get('content')
        if not content:
            return Response(
                {'error': '消息内容不能为空'},
                status=status.HTTP_400_BAD_REQUEST
            )
        
        message = Message.objects.create(
            meeting=meeting,
            sender=participant,
            content=content,
            is_broadcast=request.data.get('is_broadcast', True),
            requires_response=request.data.get('requires_response', False)
        )
        
        return Response(MessageSerializer(message).data, status=status.HTTP_201_CREATED)
    
    @action(detail=True, methods=['get'])
    def inbox(self, request, pk=None):
        """Agent 查阅信箱（自动加入会议如果还没加入）"""
        meeting = self.get_object()
        agent_id = request.query_params.get('agent_id')
        agent_name = request.query_params.get('agent_name', 'Agent')
        agent_emoji = request.query_params.get('agent_emoji', '🤖')
        
        if not agent_id:
            return Response(
                {'error': '缺少 agent_id 参数'},
                status=status.HTTP_400_BAD_REQUEST
            )
        
        # 找到或创建这个 Agent 的参会记录
        participant = Participant.objects.filter(
            meeting=meeting,
            agent_id=agent_id,
            left_at__isnull=True
        ).first()
        
        if not participant:
            # Agent 首次访问，自动加入会议
            participant = Participant.objects.create(
                meeting=meeting,
                agent_type='openclaw',
                agent_id=agent_id,
                agent_name=agent_name,
                agent_emoji=agent_emoji,
                nickname=f"{agent_emoji} {agent_name}"
            )
        
        # 获取发给这个 Agent 的消息（未读）
        messages = Message.objects.filter(
            meeting=meeting
        ).exclude(
            read_by=participant
        )
        
        # 标记为已读
        participant.read_messages.add(*messages)
        
        serializer = MessageSerializer(messages, many=True)
        
        return Response({
            'unread_count': messages.count(),
            'messages': serializer.data,
            'participant': ParticipantSerializer(participant).data
        })
    
    @action(detail=True, methods=['post'])
    def agent_reply(self, request, pk=None):
        """Agent 回复消息"""
        meeting = self.get_object()
        agent_id = request.data.get('agent_id')
        agent_name = request.data.get('agent_name', 'Agent')
        agent_emoji = request.data.get('agent_emoji', '🤖')
        content = request.data.get('content')
        in_reply_to = request.data.get('in_reply_to')
        
        if not agent_id:
            return Response(
                {'error': '缺少 agent_id 参数'},
                status=status.HTTP_400_BAD_REQUEST
            )
        
        if not content:
            return Response(
                {'error': '消息内容不能为空'},
                status=status.HTTP_400_BAD_REQUEST
            )
        
        # 找到或创建 Agent 参会记录
        participant = Participant.objects.filter(
            meeting=meeting,
            agent_id=agent_id,
            left_at__isnull=True
        ).first()
        
        if not participant:
            participant = Participant.objects.create(
                meeting=meeting,
                agent_type='openclaw',
                agent_id=agent_id,
                agent_name=agent_name,
                agent_emoji=agent_emoji,
                nickname=f"{agent_emoji} {agent_name}"
            )
        
        # 创建回复消息
        message = Message.objects.create(
            meeting=meeting,
            sender=participant,
            content=content,
            is_broadcast=request.data.get('is_broadcast', True),
            requires_response=request.data.get('requires_response', False),
            in_reply_to_id=in_reply_to
        )
        
        return Response(MessageSerializer(message).data, status=status.HTTP_201_CREATED)
    
    @action(detail=True, methods=['get'], url_path='minutes')
    def minutes(self, request, pk=None):
        """生成会议纪要"""
        meeting = self.get_object()
        
        try:
            from .utils import generate_meeting_minutes, export_minutes_to_markdown
            
            minutes = generate_meeting_minutes(str(meeting.id))
            output_format = request.query_params.get('output', 'json')
            
            if output_format == 'markdown':
                md_content = export_minutes_to_markdown(minutes)
                return Response({'markdown': md_content})
            else:
                return Response(minutes)
        except Exception as e:
            return Response(
                {'error': f'生成纪要失败：{str(e)}'},
                status=status.HTTP_500_INTERNAL_SERVER_ERROR
            )


class ParticipantViewSet(viewsets.ModelViewSet):
    """参会者视图集"""
    queryset = Participant.objects.all()
    serializer_class = ParticipantSerializer
    permission_classes = [permissions.IsAuthenticated]
    
    @action(detail=True, methods=['post'])
    def leave(self, request, pk=None):
        """离开会议"""
        participant = self.get_object()
        
        if participant.user != request.user:
            return Response(
                {'error': '无权操作'},
                status=status.HTTP_403_FORBIDDEN
            )
        
        participant.left_at = timezone.now()
        participant.save()
        
        return Response({'status': '已离开会议'})