jjlj

TUI/modalscreen/export.py

@@ -69,33 +69,34 @@ class ExportModal(ModalScreen):
         """
         ext_lower = ext.lower() if ext else ''
         
-        # Audio formats
-        audio_exts = {'.mp3', '.flac', '.wav', '.aac', '.ogg', '.m4a', '.wma', '.opus', '.mka'}
-        audio_formats = [("MKA", "mka"), ("MP3", "mp3"), ("M4A", "m4a"), ("FLAC", "flac"), ("WAV", "wav"), ("AAC", "aac"), ("OGG", "ogg"), ("Opus", "opus")]
+        from helper.utils_constant import mime_maps
         
-        # Video formats (can have audio too)
-        video_exts = {'.mp4', '.mkv', '.webm', '.avi', '.mov', '.flv', '.wmv', '.m4v', '.ts', '.mpg', '.mpeg'}
-        video_formats = [("MP4", "mp4"), ("MKV", "mkv"), ("WebM", "webm"), ("AVI", "avi"), ("MOV", "mov")]
+        found_type = "unknown"
         
-        # Image formats
-        image_exts = {'.jpg', '.jpeg', '.png', '.gif', '.webp', '.bmp', '.tiff', '.ico'}
-        image_formats = [("JPG", "jpg"), ("PNG", "png"), ("WebP", "webp"), ("GIF", "gif"), ("BMP", "bmp")]
+        # Find type based on extension
+        for category, formats in mime_maps.items():
+            for fmt_key, fmt_info in formats.items():
+                if fmt_info.get("ext") == ext_lower:
+                    found_type = category
+                    break
+            if found_type != "unknown":
+                break
         
-        # Document formats - no conversion for now
-        document_exts = {'.pdf', '.epub', '.txt', '.docx', '.doc', '.rtf', '.md', '.html', '.mobi', '.cbz', '.cbr'}
-        document_formats = []
+        # Build format options for the found type
+        format_options = []
         
-        if ext_lower in audio_exts:
-            return ('audio', audio_formats)
-        elif ext_lower in video_exts:
-            return ('video', video_formats)
-        elif ext_lower in image_exts:
-            return ('image', image_formats)
-        elif ext_lower in document_exts:
-            return ('document', document_formats)
-        else:
-            # Default to audio if unknown
-            return ('unknown', audio_formats)
+        # If unknown, fallback to audio (matching legacy behavior)
+        target_type = found_type if found_type in mime_maps else "audio"
+        
+        if target_type in mime_maps:
+            # Sort formats alphabetically
+            sorted_formats = sorted(mime_maps[target_type].items())
+            for fmt_key, fmt_info in sorted_formats:
+                label = fmt_key.upper()
+                value = fmt_key
+                format_options.append((label, value))
+                
+        return (target_type, format_options)
     
     def _get_library_options(self) -> list:
         """Get available library options from config.json."""