Commit

src/ui/classes/downloader_factory.py

@@ -0,0 +1,133 @@
+import re
+import requests
+from urllib.parse import urlparse
+
+from ...utils.network_utils import prepare_cookies_for_request
+from ...utils.file_utils import clean_folder_name
+from .allcomic_downloader_thread import AllcomicDownloadThread
+from .booru_downloader_thread import BooruDownloadThread
+from .bunkr_downloader_thread import BunkrDownloadThread
+from .discord_downloader_thread import DiscordDownloadThread
+from .drive_downloader_thread import DriveDownloadThread
+from .erome_downloader_thread import EromeDownloadThread
+from .external_link_downloader_thread import ExternalLinkDownloadThread
+from .fap_nation_downloader_thread import FapNationDownloadThread
+from .hentai2read_downloader_thread import Hentai2readDownloadThread
+from .mangadex_downloader_thread import MangaDexDownloadThread
+from .nhentai_downloader_thread import NhentaiDownloadThread
+from .pixeldrain_downloader_thread import PixeldrainDownloadThread
+from .saint2_downloader_thread import Saint2DownloadThread
+from .simp_city_downloader_thread import SimpCityDownloadThread
+from .toonily_downloader_thread import ToonilyDownloadThread
+from .rule34video_downloader_thread import Rule34VideoDownloadThread
+
+
+def create_downloader_thread(main_app, api_url, service, id1, id2, effective_output_dir_for_run):
+    """
+    Factory function to create and configure the correct QThread for a given URL.
+    Returns a configured QThread instance or None if no special handler is found.
+    """
+
+    # Handler for Booru sites (Danbooru, Gelbooru)
+    if service in ['danbooru', 'gelbooru']:
+        api_key = main_app.api_key_input.text().strip()
+        user_id = main_app.user_id_input.text().strip()
+        return BooruDownloadThread(
+            url=api_url, output_dir=effective_output_dir_for_run,
+            api_key=api_key, user_id=user_id, parent=main_app
+        )
+
+    # Handler for cloud storage sites (Mega, GDrive, etc.)
+    platform = None
+    if 'mega.nz' in api_url or 'mega.io' in api_url: platform = 'mega'
+    elif 'drive.google.com' in api_url: platform = 'gdrive'
+    elif 'dropbox.com' in api_url: platform = 'dropbox'
+    elif 'gofile.io' in api_url: platform = 'gofile'
+    if platform:
+        use_post_subfolder = main_app.use_subfolder_per_post_checkbox.isChecked()
+        return DriveDownloadThread(
+            api_url, effective_output_dir_for_run, platform, use_post_subfolder,
+            main_app.cancellation_event, main_app.pause_event, main_app.log_signal.emit
+        )
+
+    # Handler for Erome
+    if 'erome.com' in api_url:
+        return EromeDownloadThread(api_url, effective_output_dir_for_run, main_app)
+
+    # Handler for MangaDex
+    if 'mangadex.org' in api_url:
+        return MangaDexDownloadThread(api_url, effective_output_dir_for_run, main_app)
+
+    # Handler for Saint2
+    is_saint2_url = 'saint2.su' in api_url or 'saint2.pk' in api_url
+    if is_saint2_url and api_url.strip().lower() != 'saint2.su': # Exclude batch mode trigger
+        return Saint2DownloadThread(api_url, effective_output_dir_for_run, main_app)
+
+    # Handler for SimpCity
+    if service == 'simpcity':
+        cookies = prepare_cookies_for_request(
+            use_cookie_flag=True, cookie_text_input=main_app.cookie_text_input.text(),
+            selected_cookie_file_path=main_app.selected_cookie_filepath,
+            app_base_dir=main_app.app_base_dir, logger_func=main_app.log_signal.emit,
+            target_domain='simpcity.cr'
+        )
+        if not cookies:
+            # The main app will handle the error dialog
+            return "COOKIE_ERROR"
+        return SimpCityDownloadThread(api_url, id2, effective_output_dir_for_run, cookies, main_app)
+
+    if service == 'rule34video':
+        main_app.log_signal.emit("ℹ️ Rule34Video.com URL detected. Starting dedicated downloader.")
+        # id1 contains the video_id from extract_post_info
+        return Rule34VideoDownloadThread(api_url, effective_output_dir_for_run, main_app)
+
+    # Handler for official Discord URLs
+    if 'discord.com' in api_url and service == 'discord':
+        token = main_app.remove_from_filename_input.text().strip()
+        limit_text = main_app.discord_message_limit_input.text().strip()
+        message_limit = int(limit_text) if limit_text else None
+        mode = 'pdf' if main_app.discord_download_scope == 'messages' else 'files'
+        return DiscordDownloadThread(
+            mode=mode, session=requests.Session(), token=token, output_dir=effective_output_dir_for_run,
+            server_id=id1, channel_id=id2, url=api_url, app_base_dir=main_app.app_base_dir,
+            limit=message_limit, parent=main_app
+        )
+
+    # Handler for Allcomic/Allporncomic
+    if 'allcomic.com' in api_url or 'allporncomic.com' in api_url:
+        return AllcomicDownloadThread(api_url, effective_output_dir_for_run, main_app)
+
+    # Handler for Hentai2Read
+    if 'hentai2read.com' in api_url:
+        return Hentai2readDownloadThread(api_url, effective_output_dir_for_run, main_app)
+
+    # Handler for Fap-Nation
+    if 'fap-nation.com' in api_url or 'fap-nation.org' in api_url:
+        use_post_subfolder = main_app.use_subfolder_per_post_checkbox.isChecked()
+        return FapNationDownloadThread(
+            api_url, effective_output_dir_for_run, use_post_subfolder,
+            main_app.pause_event, main_app.cancellation_event, main_app.actual_gui_signals, main_app
+        )
+
+    # Handler for Pixeldrain
+    if 'pixeldrain.com' in api_url:
+        return PixeldrainDownloadThread(api_url, effective_output_dir_for_run, main_app)
+
+    # Handler for nHentai
+    if service == 'nhentai':
+        from ...core.nhentai_client import fetch_nhentai_gallery
+        gallery_data = fetch_nhentai_gallery(id1, main_app.log_signal.emit)
+        if not gallery_data:
+            return "FETCH_ERROR" # Sentinel value for fetch failure
+        return NhentaiDownloadThread(gallery_data, effective_output_dir_for_run, main_app)
+
+    # Handler for Toonily
+    if 'toonily.com' in api_url:
+        return ToonilyDownloadThread(api_url, effective_output_dir_for_run, main_app)
+
+    # Handler for Bunkr
+    if service == 'bunkr':
+        return BunkrDownloadThread(id1, effective_output_dir_for_run, main_app)
+
+    # If no special handler matched, return None
+    return None