"""
智能体模块视图 - App端
"""
from rest_framework import viewsets, status
from rest_framework.permissions import IsAuthenticated

from utils.response import success, error
from utils.exceptions import ErrorCode
from apps.admins.authentication import AppJWTAuthentication
from .models import Spirit
from .serializers import (
    SpiritSerializer,
    SpiritListSerializer,
    CreateSpiritSerializer,
)


class SpiritViewSet(viewsets.ModelViewSet):
    """智能体视图集（App端）"""
    
    authentication_classes = [AppJWTAuthentication]
    permission_classes = [IsAuthenticated]
    
    def get_queryset(self):
        return Spirit.objects.filter(user=self.request.user)
    
    def get_serializer_class(self):
        if self.action == 'list':
            return SpiritListSerializer
        if self.action == 'create':
            return CreateSpiritSerializer
        return SpiritSerializer
    
    def list(self, request, *args, **kwargs):
        """
        获取我的智能体列表
        GET /api/v1/spirits
        """
        queryset = self.get_queryset()
        serializer = SpiritListSerializer(queryset, many=True)
        return success(data=serializer.data)
    
    def create(self, request, *args, **kwargs):
        """
        创建智能体
        POST /api/v1/spirits
        """
        serializer = CreateSpiritSerializer(data=request.data)
        if serializer.is_valid():
            spirit = serializer.save(user=request.user)
            return success(data=SpiritSerializer(spirit).data, message='创建成功')
        return error(message=str(serializer.errors))
    
    def retrieve(self, request, *args, **kwargs):
        """
        获取智能体详情
        GET /api/v1/spirits/{id}
        """
        try:
            instance = self.get_object()
            serializer = SpiritSerializer(instance)
            return success(data=serializer.data)
        except Spirit.DoesNotExist:
            return error(code=ErrorCode.SPIRIT_NOT_FOUND, message='智能体不存在')
    
    def update(self, request, *args, **kwargs):
        """
        更新智能体
        PUT /api/v1/spirits/{id}
        """
        instance = self.get_object()
        serializer = SpiritSerializer(instance, data=request.data, partial=True)
        if serializer.is_valid():
            serializer.save()
            return success(data=serializer.data, message='更新成功')
        return error(message=str(serializer.errors))
    
    def destroy(self, request, *args, **kwargs):
        """
        删除智能体
        DELETE /api/v1/spirits/{id}
        """
        instance = self.get_object()
        instance.delete()
        return success(message='删除成功')