#!/usr/bin/python3

from PyQt6.QtGui import QDragEnterEvent, QContextMenuEvent
from PyQt6.QtWidgets import QTabBar

from .tab_context_menu import PlaylistContextMenu


class PlaylistTabBar(QTabBar):
    def __init__(self, parent=None):
        super().__init__(parent)

        self.tab_widget = parent
        self.app = parent.library.app

        self.context_menu = PlaylistContextMenu(self)

        self.setAcceptDrops(True)

        self.tabBarClicked.connect(self.on_click)
        self.tabBarDoubleClicked.connect(self.on_doubleclick)

    def dragEnterEvent(self, event: QDragEnterEvent):
        index = self.tabAt(event.position().toPoint())

        self.tab_widget.setCurrentIndex(index)

    def on_click(self, index):
        playlist_view = self.tab_widget.widget(index)
        playlist = playlist_view.playlist

        self.app.gui.clicked_playlist = playlist

    def on_doubleclick(self, index):
        playlist_view = self.tab_widget.widget(index)
        playlist = playlist_view.playlist

        self.app.player.start_playlist(playlist)

    def contextMenuEvent(self, event: QContextMenuEvent, title=None):
        if title is None:
            pos = event.pos()

            index = self.tabAt(pos)
            title = self.tabButton(index, QTabBar.ButtonPosition.RightSide)

        self.context_menu.exec(event.globalPos(), title)