path: root/tidal_dl_ng/gui/playlist_membership_mixin.py

"""Playlist Membership Mixin for MainWindow.

Integrates the playlist membership management system into the main window,
handling initialization, signal connections, and UI updates.
"""

from PySide6 import QtCore
from tidalapi import Track

from tidal_dl_ng.gui.dialog_playlist_manager import PlaylistManagerDialog
from tidal_dl_ng.gui.playlist_membership import (
    PlaylistCellState,
    PlaylistColumnDelegate,
    PlaylistContextLoader,
    ThreadSafePlaylistCache,
)
from tidal_dl_ng.logger import logger_gui
from tidal_dl_ng.model.gui_data import StatusbarMessage


class PlaylistMembershipMixin:
    """Mixin for playlist membership management integration into MainWindow."""

    # Type hints
    playlist_cache: ThreadSafePlaylistCache
    playlist_loader: "PlaylistContextLoader | None"
    playlist_column_delegate: PlaylistColumnDelegate

    # Signals for playlist events
    s_playlist_cache_ready: QtCore.Signal = QtCore.Signal(dict)
    s_playlist_loader_error: QtCore.Signal = QtCore.Signal(str)
    s_playlist_loader_progress: QtCore.Signal = QtCore.Signal(int, int)

    def init_playlist_membership_manager(self) -> None:
        """Initialize the playlist membership management system.

        Sets up:
        - Thread-safe cache for track→playlist mappings
        - Background worker for loading playlists
        - Custom delegate for the playlists column
        - Signal connections
        """

        # 1. Create thread-safe cache
        self.playlist_cache = ThreadSafePlaylistCache()

        # 2. Store worker reference (will be recreated on each results change)
        self.playlist_loader: PlaylistContextLoader | None = None

        # 3. Create custom delegate for playlists column
        self.playlist_column_delegate = PlaylistColumnDelegate(parent=self.tr_results)
        self.playlist_column_delegate.set_cache(self.playlist_cache)
        self.playlist_column_delegate.set_obj_column_index(1)  # hidden obj column
        self.playlist_column_delegate.button_clicked.connect(self.on_playlist_column_button_clicked)

        # 5. Attach delegate to the results table (column 9 = playlists)
        # Note: Column indices: 0=index, 1=obj, 2=artist, 3=title, 4=album,
        #       5=duration, 6=quality, 7=date, 8=downloaded, 9=playlists

        # Diagnostic: check if column 9 exists
        if hasattr(self, "model_tr_results") and self.model_tr_results:
            col_count = self.model_tr_results.columnCount()
            if col_count <= 9:
                logger_gui.error(f"❌ Column 9 does NOT exist! Model only has {col_count} columns")

        self.tr_results.setItemDelegateForColumn(9, self.playlist_column_delegate)

        # Configure column width and appearance
        self.tr_results.setColumnWidth(9, 60)

        # Make sure column is visible
        if self.tr_results.isColumnHidden(9):
            self.tr_results.setColumnHidden(9, False)

        self.tr_results.setColumnHidden(1, True)  # Hide obj column

        # 6. Connect model signals to trigger preloading
        if hasattr(self, "model_tr_results") and self.model_tr_results:
            self.model_tr_results.modelReset.connect(self.on_results_layout_changed)
            self.model_tr_results.layoutChanged.connect(self.on_results_layout_changed)

            # Initialize states for any existing rows
            existing_rows = self.model_tr_results.rowCount()
            if existing_rows > 0:
                for r in range(existing_rows):
                    self.playlist_column_delegate.set_cell_state(r, PlaylistCellState.READY)
        else:
            logger_gui.error("❌ model_tr_results not found - signal not connected!")

        # 7. Launch initial playlist loading immediately
        self._load_playlists()

    def connect_playlist_signals(self) -> None:
        """Connect playlist-related signals (called from main signal setup)."""
        # Signal connections already done in init_playlist_membership_manager
        pass

    def _load_playlists(self) -> None:
        """Load user playlists in background.

        Creates a new worker and launches it in the threadpool.
        """
        # Purge ENTIRE cache (tracks + metadata) to ensure first-load starts clean
        # et éviter que des métadonnées obsolètes n'influencent l'affichage.
        if hasattr(self, "playlist_cache") and self.playlist_cache:
            self.playlist_cache.clear()

        # Indiquer explicitement au délégué que le cache n'est pas prêt
        # afin d'afficher le spinner au lieu des compteurs.
        if hasattr(self, "playlist_column_delegate") and self.playlist_column_delegate:
            self.playlist_column_delegate.set_cache_ready(False)

        self.playlist_loader = PlaylistContextLoader(
            session=self.tidal.session,
            user_id=str(self.tidal.session.user.id),
            max_workers=5,
        )

        # Connect signals for this worker instance
        self.playlist_loader.signals.started.connect(self.on_playlist_loader_started)
        self.playlist_loader.signals.cache_ready.connect(self.on_playlist_cache_ready)
        self.playlist_loader.signals.metadata_ready.connect(self.on_playlist_metadata_ready)
        self.playlist_loader.signals.error.connect(self.on_playlist_loader_error)
        self.playlist_loader.signals.progress.connect(self.on_playlist_loader_progress)
        self.playlist_loader.signals.finished.connect(self.on_playlist_loader_finished)

        # Start loading playlists in background
        self.threadpool.start(self.playlist_loader)

    def on_results_layout_changed(self) -> None:
        """Handle when new results are displayed in the main table."""
        # Initialize READY states for all visible rows
        try:
            rows = self.model_tr_results.rowCount()
            for r in range(rows):
                self.playlist_column_delegate.set_cell_state(r, PlaylistCellState.READY)
        except Exception as e:
            logger_gui.warning(f"Failed to initialize cell states: {e}")

        # Force repaint to show buttons (if cache is already ready)
        if self.playlist_column_delegate._cache_ready:
            try:
                self.tr_results.viewport().update()
                self.tr_results.viewport().repaint()
            except RuntimeError:
                pass

    def on_playlist_loader_started(self) -> None:
        """Called when playlist loader starts."""
        pass  # Silent

    def on_playlist_cache_ready(self, cache: dict) -> None:
        """Called when playlist cache is ready.

        Args:
            cache: Dict[track_id, Set[playlist_id]] from worker
        """

        # Update cache WITHOUT clearing (metadata was already loaded)
        # NOTE: We don't call clear() because metadata_ready signal was already
        # emitted and stored. We only update track→playlist mapping.
        self.playlist_cache.update_from_dict(cache)

        # Initialize cell states for all current rows as READY
        try:
            rows = self.model_tr_results.rowCount()
            for r in range(rows):
                self.playlist_column_delegate.set_cell_state(r, PlaylistCellState.READY)
        except Exception as e:
            logger_gui.warning(f"Failed to initialize cell states: {e}")

        # Notify delegate that cache is ready (stops animation timer)
        self.playlist_column_delegate.set_cache_ready(True)

        # Force complete repaint of the table to show buttons instead of spinners
        self.tr_results.viewport().update()
        self.tr_results.viewport().repaint()

    def on_playlist_metadata_ready(self, metadata: dict) -> None:
        """Receive playlist metadata and store it in the cache.

        Args:
            metadata: Dict[playlist_id, {name: str, item_count: int}]
        """
        try:
            for pid, info in metadata.items():
                name = str(info.get("name", pid))
                count = int(info.get("item_count", 0))
                self.playlist_cache.set_playlist_metadata(pid, name, count)
            logger_gui.debug(f"📝 Stored metadata for {len(metadata)} playlists")
        except Exception as e:
            logger_gui.warning(f"Failed to store playlist metadata: {e}")

    def on_playlist_loader_error(self, error_msg: str) -> None:
        """Handle playlist loader error.

        Args:
            error_msg: Error description
        """
        logger_gui.warning(f"Playlist loader error: {error_msg}")

        # Show user-friendly notification
        self.s_statusbar_message.emit(
            StatusbarMessage(
                message=f"Erreur: {error_msg}",
                timeout=5000,
            )
        )

    def on_playlist_loader_progress(self, current: int, total: int) -> None:
        """Handle playlist loader progress.

        Args:
            current: Number of playlists processed
            total: Total number of playlists
        """
        pass  # Silent - no need to spam logs

    def on_playlist_loader_finished(self) -> None:
        """Handle playlist loader finished (success or error)."""
        pass  # Silent

    def on_playlist_column_button_clicked(self, index: QtCore.QModelIndex) -> None:
        """Handle click on playlist column button.

        Opens the playlist manager dialog for the clicked track.

        Args:
            index: QModelIndex of the cell clicked
        """
        # Map proxy index to source index if using proxy model
        if isinstance(self.tr_results.model(), type(self.proxy_tr_results)):
            source_index = self.proxy_tr_results.mapToSource(index)
        else:
            source_index = index

        # Get the track object from the model
        obj_item = self.model_tr_results.item(source_index.row(), 1)  # Column 1 = obj
        if not obj_item:
            logger_gui.warning("Failed to get track object from model")
            return

        track = obj_item.data(QtCore.Qt.ItemDataRole.UserRole)
        if not isinstance(track, Track):
            logger_gui.warning(f"Cell does not contain Track object: {type(track)}")
            return

        # Create and show the dialog
        dialog = PlaylistManagerDialog(
            track=track,
            cache=self.playlist_cache,
            session=self.tidal.session,
            threadpool=self.threadpool,
            parent=self,
        )

        # Connect signals for tracking changes
        dialog.playlist_added.connect(self.on_track_added_to_playlist)
        dialog.playlist_removed.connect(self.on_track_removed_from_playlist)

        # Show dialog modally
        dialog.exec()

    def on_track_added_to_playlist(self, track_id: str, playlist_id: str) -> None:
        """Handle track added to playlist.

        Args:
            track_id: Track UUID
            playlist_id: Playlist UUID
        """
        pass  # Silent - update handled by dialog

    def on_track_removed_from_playlist(self, track_id: str, playlist_id: str) -> None:
        """Handle track removed from playlist.

        Args:
            track_id: Track UUID
            playlist_id: Playlist UUID
        """
        pass  # Silent - update handled by dialog