from rest_framework import serializers from apps.books.models import Book, BookChapter, BookReadingProgress, Chapter, EBook, FontStyle, BackgroundColor, ReadingProgress, ReadingSettings, ReadingStatus, DownloadRecord class BookChapterSerializer(serializers.ModelSerializer): class Meta: model = BookChapter fields = ["id", "title", "index", "href", "children"] class EBookContentSerializer(serializers.Serializer): page = serializers.IntegerField() total_pages = serializers.IntegerField() content = serializers.CharField() chapter_title = serializers.CharField() format = serializers.CharField() class EBookTocSerializer(serializers.Serializer): chapters = serializers.ListField(child=BookChapterSerializer()) format = serializers.CharField() page_count = serializers.IntegerField() class BookListSerializer(serializers.ModelSerializer): reading_status_display = serializers.CharField(source="get_reading_status_display", read_only=True) class Meta: model = Book fields = ["id", "title", "author", "genre", "reading_status", "reading_status_display", "cover_image"] class BookDetailSerializer(serializers.ModelSerializer): reading_status_display = serializers.CharField(source="get_reading_status_display", read_only=True) class Meta: model = Book fields = ["id", "title", "author", "genre", "description", "reading_status", "reading_status_display", "cover_image", "total_pages", "created_at", "updated_at"] read_only_fields = ["id", "created_at", "updated_at"] class BookSerializer(serializers.ModelSerializer): class Meta: model = Book fields = ["id", "title", "author", "genre", "description", "reading_status", "cover_image", "total_pages", "created_at", "updated_at"] read_only_fields = ["id", "created_at", "updated_at"] class EBookListSerializer(serializers.ModelSerializer): filename = serializers.CharField(read_only=True) format = serializers.CharField(read_only=True) progress = serializers.SerializerMethodField() class Meta: model = EBook fields = ["id", "title", "author", "filename", "format", "page_count", "file_size", "cover_image", "created_at", "progress"] def get_progress(self, obj): try: return obj.reading_progress.current_position except ReadingProgress.DoesNotExist: return None class EBookDetailSerializer(serializers.ModelSerializer): filename = serializers.CharField(read_only=True) format = serializers.CharField(read_only=True) file_url = serializers.SerializerMethodField() progress = serializers.SerializerMethodField() class Meta: model = EBook fields = ["id", "title", "author", "filename", "format", "page_count", "file_size", "file_url", "cover_image", "created_at", "updated_at", "progress"] def get_file_url(self, obj): request = self.context.get("request") if request and obj.file: return request.build_absolute_uri(obj.file.url) return "" def get_progress(self, obj): try: rp = obj.reading_progress return {"current_position": rp.current_position, "last_page": rp.last_page} except ReadingProgress.DoesNotExist: return None class EBookUploadSerializer(serializers.ModelSerializer): class Meta: model = EBook fields = ["title", "author", "file", "cover_image"] extra_kwargs = {"title": {"required": True}, "file": {"required": True}} def validate_file(self, value): import os if value is None: return value ext = os.path.splitext(str(getattr(value, "name", "")))[1].lower() if ext not in (".epub", ".pdf"): raise serializers.ValidationError("Only EPUB and PDF files are supported.") return value def create(self, validated_data): validated_data["user"] = self.context["request"].user # Auto-detect format from file extension import os name = str(getattr(validated_data.get("file"), "name", "")) ext = os.path.splitext(name)[1].lower().lstrip(".") if ext: validated_data["format"] = ext return super().create(validated_data) class ReadingProgressSerializer(serializers.ModelSerializer): class Meta: model = ReadingProgress fields = ["current_position", "last_page", "device_id", "device_name", "version", "updated_at"] read_only_fields = ["version", "updated_at"] extra_kwargs = {"current_position": {"required": True, "min_value": 0.0, "max_value": 100.0}} def validate_current_position(self, value): if value < 0.0 or value > 100.0: raise serializers.ValidationError("Position must be between 0.0 and 100.0.") return value class ReadingSettingsSerializer(serializers.ModelSerializer): class Meta: model = ReadingSettings fields = ["font_size", "font_style", "background_color"] def validate_font_size(self, value): if value < 12 or value > 36: raise serializers.ValidationError("Font size must be between 12 and 36.") return value def validate_font_style(self, value): valid = [s.value for s in FontStyle] if value not in valid: raise serializers.ValidationError(f"Font style must be one of: {', '.join(valid)}") return value def validate_background_color(self, value): valid = [c.value for c in BackgroundColor] if value not in valid: raise serializers.ValidationError(f"Background color must be one of: {', '.join(valid)}") return value class DownloadRecordSerializer(serializers.ModelSerializer): ebook_id = serializers.IntegerField(source="ebook.id", read_only=True) ebook_title = serializers.CharField(source="ebook.title", read_only=True) author = serializers.CharField(source="ebook.author", read_only=True) filename = serializers.SerializerMethodField() cover_image = serializers.ImageField(source="ebook.cover_image", read_only=True) format = serializers.CharField(source="ebook.format", read_only=True) progress = serializers.SerializerMethodField() file_url = serializers.SerializerMethodField() class Meta: model = DownloadRecord fields = [ "id", "ebook_id", "ebook_title", "author", "filename", "file_url", "file_size", "cover_image", "format", "downloaded_at", "progress", ] def get_filename(self, obj): return obj.ebook.filename() def get_file_url(self, obj): request = self.context.get("request") if request and obj.ebook.file: return request.build_absolute_uri(obj.ebook.file.url) return "" def get_progress(self, obj): try: rp = obj.ebook.reading_progress return {"current_position": rp.current_position, "last_page": rp.last_page} except ReadingProgress.DoesNotExist: return None class StorageSummarySerializer(serializers.Serializer): total_downloads = serializers.IntegerField() total_size_bytes = serializers.IntegerField() ebooks = serializers.ListField(child=serializers.DictField()) class ChapterSummarySerializer(serializers.ModelSerializer): """Compact serializer for TOC listing — no content body.""" class Meta: model = Chapter fields = ["id", "book", "title", "number"] class ChapterDetailSerializer(serializers.ModelSerializer): """Full serializer with chapter content for reading view.""" class Meta: model = Chapter fields = ["id", "book", "title", "number", "content", "created_at", "updated_at"] read_only_fields = ["id", "created_at", "updated_at"] class BookReadingProgressSerializer(serializers.ModelSerializer): """Serialize chapter-based reading progress for a book.""" class Meta: model = BookReadingProgress fields = [ "id", "book", "current_chapter", "current_position", "percentage", "updated_at", ] read_only_fields = ["id", "updated_at"] def validate_percentage(self, value): if value < 0.0 or value > 100.0: raise serializers.ValidationError("Percentage must be between 0 and 100.") return value def validate_current_chapter(self, value): if value < 1: raise serializers.ValidationError("Chapter number must be positive.") return value