Commit

src/ui/dialogs/DownloadExtractedLinksDialog.py

@@ -0,0 +1,183 @@
+# --- Standard Library Imports ---
+from collections import defaultdict
+
+# --- PyQt5 Imports ---
+from PyQt5.QtCore import pyqtSignal, Qt
+from PyQt5.QtWidgets import (
+    QApplication, QDialog, QHBoxLayout, QLabel, QListWidget, QListWidgetItem,
+    QMessageBox, QPushButton, QVBoxLayout, QAbstractItemView
+)
+
+# --- Local Application Imports ---
+# This assumes the new project structure is in place.
+from ...i18n.translator import get_translation
+# get_app_icon_object is defined in the main window module in this refactoring plan.
+from ..main_window import get_app_icon_object
+
+
+class DownloadExtractedLinksDialog(QDialog):
+    """
+    A dialog to select and initiate the download for extracted, supported links
+    from external cloud services like Mega, Google Drive, and Dropbox.
+    """
+
+    # Signal emitted with a list of selected link information dictionaries
+    download_requested = pyqtSignal(list)
+
+    def __init__(self, links_data, parent_app, parent=None):
+        """
+        Initializes the dialog.
+
+        Args:
+            links_data (list): A list of dictionaries, each containing info about an extracted link.
+            parent_app (DownloaderApp): A reference to the main application window for theming and translations.
+            parent (QWidget, optional): The parent widget. Defaults to None.
+        """
+        super().__init__(parent)
+        self.links_data = links_data
+        self.parent_app = parent_app
+
+        # --- Basic Window Setup ---
+        app_icon = get_app_icon_object()
+        if not app_icon.isNull():
+            self.setWindowIcon(app_icon)
+
+        # Set window size dynamically based on the parent window's size
+        if parent:
+            parent_width = parent.width()
+            parent_height = parent.height()
+            # Use a scaling factor for different screen resolutions
+            screen_height = QApplication.primaryScreen().availableGeometry().height() if QApplication.primaryScreen() else 768
+            scale_factor = screen_height / 768.0
+
+            base_min_w, base_min_h = 500, 400
+            scaled_min_w = int(base_min_w * scale_factor)
+            scaled_min_h = int(base_min_h * scale_factor)
+
+            self.setMinimumSize(scaled_min_w, scaled_min_h)
+            self.resize(max(int(parent_width * 0.6 * scale_factor), scaled_min_w),
+                        max(int(parent_height * 0.7 * scale_factor), scaled_min_h))
+
+        # --- Initialize UI and Apply Theming ---
+        self._init_ui()
+        self._retranslate_ui()
+        self._apply_theme()
+
+    def _init_ui(self):
+        """Initializes all UI components and layouts for the dialog."""
+        layout = QVBoxLayout(self)
+
+        self.main_info_label = QLabel()
+        self.main_info_label.setAlignment(Qt.AlignHCenter | Qt.AlignTop)
+        self.main_info_label.setWordWrap(True)
+        layout.addWidget(self.main_info_label)
+
+        self.links_list_widget = QListWidget()
+        self.links_list_widget.setSelectionMode(QAbstractItemView.NoSelection)
+        self._populate_list()
+        layout.addWidget(self.links_list_widget)
+
+        # --- Control Buttons ---
+        button_layout = QHBoxLayout()
+        self.select_all_button = QPushButton()
+        self.select_all_button.clicked.connect(lambda: self._set_all_items_checked(Qt.Checked))
+        button_layout.addWidget(self.select_all_button)
+
+        self.deselect_all_button = QPushButton()
+        self.deselect_all_button.clicked.connect(lambda: self._set_all_items_checked(Qt.Unchecked))
+        button_layout.addWidget(self.deselect_all_button)
+        button_layout.addStretch()
+
+        self.download_button = QPushButton()
+        self.download_button.clicked.connect(self._handle_download_selected)
+        self.download_button.setDefault(True)
+        button_layout.addWidget(self.download_button)
+
+        self.cancel_button = QPushButton()
+        self.cancel_button.clicked.connect(self.reject)
+        button_layout.addWidget(self.cancel_button)
+        layout.addLayout(button_layout)
+
+    def _populate_list(self):
+        """Populates the list widget with the provided links, grouped by post title."""
+        grouped_links = defaultdict(list)
+        for link_info_item in self.links_data:
+            post_title_for_group = link_info_item.get('title', 'Untitled Post')
+            grouped_links[post_title_for_group].append(link_info_item)
+
+        sorted_post_titles = sorted(grouped_links.keys(), key=lambda x: x.lower())
+
+        for post_title_key in sorted_post_titles:
+            # Add a non-selectable header for each post
+            header_item = QListWidgetItem(f"{post_title_key}")
+            header_item.setFlags(Qt.NoItemFlags)
+            font = header_item.font()
+            font.setBold(True)
+            font.setPointSize(font.pointSize() + 1)
+            header_item.setFont(font)
+            self.links_list_widget.addItem(header_item)
+
+            # Add checkable items for each link within that post
+            for link_info_data in grouped_links[post_title_key]:
+                platform_display = link_info_data.get('platform', 'unknown').upper()
+                display_text = f"  [{platform_display}] {link_info_data['link_text']} ({link_info_data['url']})"
+                item = QListWidgetItem(display_text)
+                item.setData(Qt.UserRole, link_info_data)
+                item.setFlags(item.flags() | Qt.ItemIsUserCheckable)
+                item.setCheckState(Qt.Checked)
+                self.links_list_widget.addItem(item)
+
+    def _tr(self, key, default_text=""):
+        """Helper to get translation based on current app language."""
+        if callable(get_translation) and self.parent_app:
+            return get_translation(self.parent_app.current_selected_language, key, default_text)
+        return default_text
+
+    def _retranslate_ui(self):
+        """Sets the text for all translatable UI elements."""
+        self.setWindowTitle(self._tr("download_external_links_dialog_title", "Download Selected External Links"))
+        self.main_info_label.setText(self._tr("download_external_links_dialog_main_label", "Found {count} supported link(s)...").format(count=len(self.links_data)))
+        self.select_all_button.setText(self._tr("select_all_button_text", "Select All"))
+        self.deselect_all_button.setText(self._tr("deselect_all_button_text", "Deselect All"))
+        self.download_button.setText(self._tr("download_selected_button_text", "Download Selected"))
+        self.cancel_button.setText(self._tr("fav_posts_cancel_button", "Cancel"))
+
+    def _apply_theme(self):
+        """Applies the current theme from the parent application."""
+        is_dark_theme = self.parent() and hasattr(self.parent_app, 'current_theme') and self.parent_app.current_theme == "dark"
+
+        if is_dark_theme and hasattr(self.parent_app, 'get_dark_theme'):
+            self.setStyleSheet(self.parent_app.get_dark_theme())
+        
+        # Set header text color based on theme
+        header_color = Qt.cyan if is_dark_theme else Qt.blue
+        for i in range(self.links_list_widget.count()):
+            item = self.links_list_widget.item(i)
+            # Headers are not checkable
+            if not item.flags() & Qt.ItemIsUserCheckable:
+                item.setForeground(header_color)
+
+    def _set_all_items_checked(self, check_state):
+        """Sets the checked state for all checkable items in the list."""
+        for i in range(self.links_list_widget.count()):
+            item = self.links_list_widget.item(i)
+            if item.flags() & Qt.ItemIsUserCheckable:
+                item.setCheckState(check_state)
+
+    def _handle_download_selected(self):
+        """Gathers selected links and emits the download_requested signal."""
+        selected_links = []
+        for i in range(self.links_list_widget.count()):
+            item = self.links_list_widget.item(i)
+            if item.flags() & Qt.ItemIsUserCheckable and item.checkState() == Qt.Checked and item.data(Qt.UserRole) is not None:
+                selected_links.append(item.data(Qt.UserRole))
+
+        if selected_links:
+            self.download_requested.emit(selected_links)
+            self.accept()
+        else:
+            QMessageBox.information(
+                self,
+                self._tr("no_selection_title", "No Selection"),
+                self._tr("no_selection_message_links", "Please select at least one link to download.")
+            )