from rest_framework import serializers
from .models import Activity, ActivitySignup, Topic, Reply, TopicMedia, Announcement
from shop.serializers import WeChatUserSerializer, ESP32ConfigSerializer, ServiceSerializer, VCCourseSerializer

class ActivitySerializer(serializers.ModelSerializer):
    display_banner_url = serializers.ReadOnlyField()
    
    class Meta:
        model = Activity
        fields = '__all__'

class ActivitySignupSerializer(serializers.ModelSerializer):
    class Meta:
        model = ActivitySignup
        fields = ['id', 'activity', 'user', 'signup_time', 'status']
        read_only_fields = ['signup_time', 'status']

class TopicMediaSerializer(serializers.ModelSerializer):
    url = serializers.SerializerMethodField()

    class Meta:
        model = TopicMedia
        fields = ['id', 'file', 'file_url', 'url', 'media_type', 'created_at']

    def get_url(self, obj):
        if obj.file:
            return obj.file.url
        return obj.file_url

class ReplySerializer(serializers.ModelSerializer):
    author_info = WeChatUserSerializer(source='author', read_only=True)
    media = TopicMediaSerializer(many=True, read_only=True)
    media_ids = serializers.ListField(
        child=serializers.IntegerField(), 
        write_only=True, 
        required=False
    )

    class Meta:
        model = Reply
        fields = ['id', 'topic', 'content', 'author', 'author_info', 'reply_to', 'media', 'created_at', 'media_ids']
        read_only_fields = ['author', 'created_at']

    def create(self, validated_data):
        media_ids = validated_data.pop('media_ids', [])
        reply = super().create(validated_data)
        if media_ids:
            TopicMedia.objects.filter(id__in=media_ids).update(reply=reply)
        return reply

class TopicSerializer(serializers.ModelSerializer):
    author_info = WeChatUserSerializer(source='author', read_only=True)
    replies = ReplySerializer(many=True, read_only=True)
    media = TopicMediaSerializer(many=True, read_only=True)
    is_verified_owner = serializers.BooleanField(read_only=True)
    
    product_info = ESP32ConfigSerializer(source='related_product', read_only=True)
    service_info = ServiceSerializer(source='related_service', read_only=True)
    course_info = VCCourseSerializer(source='related_course', read_only=True)

    media_ids = serializers.ListField(
        child=serializers.IntegerField(), 
        write_only=True, 
        required=False
    )

    class Meta:
        model = Topic
        fields = [
            'id', 'title', 'category', 'content', 'author', 'author_info', 
            'related_product', 'product_info', 
            'related_service', 'service_info',
            'related_course', 'course_info',
            'view_count', 'is_pinned', 'created_at', 'updated_at', 
            'is_verified_owner', 'replies', 'media', 'media_ids'
        ]
        read_only_fields = ['author', 'view_count', 'created_at', 'updated_at', 'is_verified_owner']

    def create(self, validated_data):
        media_ids = validated_data.pop('media_ids', [])
        topic = super().create(validated_data)
        if media_ids:
            TopicMedia.objects.filter(id__in=media_ids).update(topic=topic)
        return topic

class AnnouncementSerializer(serializers.ModelSerializer):
    display_image_url = serializers.ReadOnlyField()

    class Meta:
        model = Announcement
        fields = '__all__'