meeting-room/backend/users/views.py

from rest_framework import serializers, status, views
from rest_framework.response import Response
from django.contrib.auth import authenticate, get_user_model

User = get_user_model()


class LoginSerializer(serializers.Serializer):
    username = serializers.CharField()
    password = serializers.CharField()
    mode = serializers.ChoiceField(
        choices=['solo', 'team', 'agent_only'],
        default='solo',
        help_text='solo=单枪匹马，team=组队团战，agent_only=独当一面'
    )
    agent_ids = serializers.ListField(
        child=serializers.CharField(),
        required=False,
        default=list,
        help_text='选择的龙虾 ID 列表（team 或 agent_only 模式）'
    )


class LoginView(views.APIView):
    def post(self, request):
        serializer = LoginSerializer(data=request.data)
        if not serializer.is_valid():
            return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
        
        username = serializer.validated_data['username']
        password = serializer.validated_data['password']
        mode = serializer.validated_data.get('mode', 'solo')
        agent_ids = serializer.validated_data.get('agent_ids', [])
        
        user = authenticate(username=username, password=password)
        if not user:
            return Response(
                {'detail': '用户名或密码错误'},
                status=status.HTTP_401_UNAUTHORIZED
            )
        
        # 简单 Token（生产环境应该用 JWT）
        import uuid
        token = uuid.uuid4().hex
        
        # 构建会话信息
        sessions = []
        
        if mode in ['solo', 'team']:
            # 人类身份（单枪匹马 或 组队团战）
            sessions.append({
                'session_type': 'human',
                'nickname': user.username,
                'emoji': '👤',
                'user_id': user.id
            })
        
        if mode in ['team', 'agent_only']:
            # 龙虾身份（组队团战 或 独当一面）
            if not agent_ids:
                return Response(
                    {'error': '组队或独当一面模式需要选择至少一只龙虾'},
                    status=status.HTTP_400_BAD_REQUEST
                )
            
            # 添加所有选择的龙虾
            for agent_id in agent_ids:
                agent = user.get_linked_agent(agent_id)
                if agent:
                    sessions.append({
                        'session_type': 'agent',
                        'agent_id': agent['agent_id'],
                        'agent_name': agent['agent_name'],
                        'nickname': agent['agent_name'],
                        'emoji': agent.get('agent_emoji', '🤖'),
                        'instance_id': agent.get('instance_id')
                    })
                else:
                    return Response(
                        {'error': f'未找到绑定的龙虾：{agent_id}'},
                        status=status.HTTP_400_BAD_REQUEST
                    )
        
        # 模式名称映射
        mode_names = {
            'solo': '单枪匹马',
            'team': '组队团战',
            'agent_only': '独当一面'
        }
        
        return Response({
            'token': token,
            'user': {
                'id': user.id,
                'username': user.username,
                'email': user.email,
                'linked_agents': user.linked_agents
            },
            'sessions': sessions,
            'mode': mode,
            'mode_name': mode_names.get(mode, mode)
        })


class RegisterSerializer(serializers.Serializer):
    username = serializers.CharField()
    email = serializers.EmailField()
    password = serializers.CharField()


class RegisterView(views.APIView):
    def post(self, request):
        serializer = RegisterSerializer(data=request.data)
        if not serializer.is_valid():
            return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
        
        try:
            user = User.objects.create_user(
                username=serializer.validated_data['username'],
                email=serializer.validated_data['email'],
                password=serializer.validated_data['password']
            )
            
            import uuid
            token = uuid.uuid4().hex
            
            return Response({
                'token': token,
                'user': {
                    'id': user.id,
                    'username': user.username,
                    'email': user.email
                }
            }, status=status.HTTP_201_CREATED)
        except Exception as e:
            return Response(
                {'detail': str(e)},
                status=status.HTTP_400_BAD_REQUEST
            )


class LinkedAgentsView(views.APIView):
    """
    用户绑定的龙虾管理
    
    GET /api/v1/user/linked-agents/?username=xxx - 获取龙虾列表
    POST /api/v1/user/linked-agents/ - 添加龙虾
    DELETE /api/v1/user/linked-agents/{agent_id}/?username=xxx - 移除龙虾
    """
    def get(self, request):
        username = request.query_params.get('username')
        if not username:
            return Response({'error': '缺少 username'}, status=status.HTTP_400_BAD_REQUEST)
        
        try:
            user = User.objects.get(username=username)
            return Response({
                'linked_agents': user.linked_agents,
                'count': len(user.linked_agents)
            })
        except User.DoesNotExist:
            return Response({'error': '用户不存在'}, status=status.HTTP_404_NOT_FOUND)
    
    def post(self, request):
        username = request.data.get('username')
        agent_id = request.data.get('agent_id')
        agent_name = request.data.get('agent_name')
        agent_emoji = request.data.get('agent_emoji', '🤖')
        instance_id = request.data.get('instance_id')
        
        if not all([username, agent_id, agent_name]):
            return Response({'error': '缺少必要参数'}, status=status.HTTP_400_BAD_REQUEST)
        
        try:
            user = User.objects.get(username=username)
            user.add_linked_agent(agent_id, agent_name, agent_emoji, instance_id)
            return Response({
                'status': 'success',
                'linked_agents': user.linked_agents
            })
        except User.DoesNotExist:
            return Response({'error': '用户不存在'}, status=status.HTTP_404_NOT_FOUND)
    
    def delete(self, request, agent_id):
        username = request.query_params.get('username')
        if not username:
            return Response({'error': '缺少 username'}, status=status.HTTP_400_BAD_REQUEST)
        
        try:
            user = User.objects.get(username=username)
            user.remove_linked_agent(agent_id)
            return Response({
                'status': 'success',
                'linked_agents': user.linked_agents
            })
        except User.DoesNotExist:
            return Response({'error': '用户不存在'}, status=status.HTTP_404_NOT_FOUND)


class ScanLocalAgentsView(views.APIView):
    """
    扫描本机龙虾列表
    
    GET /api/v1/user/scan-local-agents/?instance_id=xxx&username=xxx - 扫描指定实例
    GET /api/v1/user/scan-local-agents/?username=xxx - 扫描所有实例
    """
    def get(self, request):
        instance_id = request.query_params.get('instance_id')
        username = request.query_params.get('username')
        
        # 获取用户绑定的龙虾信息（用于获取 agent_name 和 emoji）
        user_agents = {}
        if username:
            from django.contrib.auth import get_user_model
            User = get_user_model()
            try:
                user = User.objects.get(username=username)
                for agent in user.linked_agents:
                    user_agents[agent['agent_id']] = agent
            except User.DoesNotExist:
                pass
        
        if not instance_id:
            # 返回所有已注册实例的 Agent
            from instances.models import Instance
            instances = Instance.objects.filter(is_active=True)
            agents = []
            for inst in instances:
                for agent_id in inst.agent_ids:
                    agent_info = user_agents.get(agent_id, {})
                    agents.append({
                        'agent_id': agent_id,
                        'agent_name': agent_info.get('agent_name', agent_id),
                        'agent_emoji': agent_info.get('agent_emoji', '🤖'),
                        'instance_id': inst.instance_id,
                        'instance_name': inst.instance_name
                    })
        else:
            # 返回指定实例的 Agent
            from instances.models import Instance
            try:
                instance = Instance.objects.get(instance_id=instance_id, is_active=True)
                for agent_id in instance.agent_ids:
                    agent_info = user_agents.get(agent_id, {})
                    agents.append({
                        'agent_id': agent_id,
                        'agent_name': agent_info.get('agent_name', agent_id),
                        'agent_emoji': agent_info.get('agent_emoji', '🤖'),
                        'instance_id': instance.instance_id,
                        'instance_name': instance.instance_name
                    })
            except Instance.DoesNotExist:
                return Response({'error': '实例不存在'}, status=status.HTTP_404_NOT_FOUND)
        
        return Response({
            'agents': agents,
            'count': len(agents)
        })