from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from PyQt5.QtCore import *
from MainWindow import Ui_MainWindow

from solo_tool import SoloTool

POSITION_FACTOR = 100000
RATE_FACTOR = 10
UI_REFRESH_PERIOD_MS = 500

class PlaylistModel(QAbstractListModel):
    def __init__(self, soloTool, *args, **kwargs):
        super(PlaylistModel, self).__init__(*args, **kwargs)
        self.soloTool = soloTool

    def data(self, index, role):
        if role == Qt.DisplayRole:
            from pathlib import Path
            path = Path(self.soloTool.getSongs()[index.row()])
            return path.name

    def rowCount(self, index):
        return len(self.soloTool.getSongs())

class ABListModel(QAbstractListModel):
    def __init__(self, soloTool, *args, **kwargs):
        super(ABListModel, self).__init__(*args, **kwargs)
        self.soloTool = soloTool

    def data(self, index, role):
        if role == Qt.DisplayRole:
            ab = self.soloTool.getStoredAbLimits()[index.row()]
            return f"{ab[0]} - {ab[1]}"

    def rowCount(self, index):
        return len(self.soloTool.getStoredAbLimits())

class MainWindow(QMainWindow, Ui_MainWindow):
    def __init__(self, *args, **kwargs):
        super(MainWindow, self).__init__(*args, **kwargs)
        self.setupUi(self)
        
        self.timer = QTimer(self)
        self.timer.setInterval(UI_REFRESH_PERIOD_MS)
        self.timer.timeout.connect(self.timerCallback)

        self.soloTool = SoloTool()

        self.playlistModel = PlaylistModel(self.soloTool)
        self.songListView.setModel(self.playlistModel)
        self.songListView.selectionModel().selectionChanged.connect(self.playlistSelectionChanged)

        self.abListModel = ABListModel(self.soloTool)
        self.abListView.setModel(self.abListModel)
        self.abListView.selectionModel().selectionChanged.connect(self.abListSelectionChanged)

        self.songSlider.setMaximum(POSITION_FACTOR)
        self.songSlider.sliderPressed.connect(self.songSliderPressed)
        self.songSlider.sliderReleased.connect(self.songSliderReleased)

        self.aSlider.setMaximum(POSITION_FACTOR)
        self.aSlider.sliderReleased.connect(self.abSliderReleased)
        self.bSlider.setMaximum(POSITION_FACTOR)
        self.bSlider.sliderReleased.connect(self.abSliderReleased)

        self.rateSlider.setRange(int(0.5 * RATE_FACTOR), int(1.5 * RATE_FACTOR))
        self.rateSlider.setSingleStep(int(0.1 * RATE_FACTOR))
        self.rateSlider.setValue(int(1.0 * RATE_FACTOR))
        self.rateSlider.sliderReleased.connect(self.rateSliderReleased)

        self.playButton.pressed.connect(self.soloTool.play)
        self.pauseButton.pressed.connect(self.soloTool.pause)
        self.saveAbButton.pressed.connect(self.saveAbLimits)
        self.saveSessionButton.pressed.connect(self.saveSession)
        self.loadSessionButton.pressed.connect(self.loadSession)
        self.addSongButton.pressed.connect(self.addSong)
        self.abRepeatCheckBox.clicked.connect(self.toggleAbRepeat)
        #self.initMidiButton.pressed.connect()

        self.timer.start()
        self.show()

    def timerCallback(self):
        position = self.soloTool.getPlaybackPosition() * POSITION_FACTOR
        self.songSlider.setValue(int(position))
        self.soloTool.tick()

    def addSong(self):
        path, _ = QFileDialog.getOpenFileName(self, "Open file", "", "mp3 Audio (*.mp3);FLAC audio (*.flac);All files (*.*)")
        if path:
            self.soloTool.addSong(path)
            self.playlistModel.layoutChanged.emit()

    def saveAbLimits(self):
        a = self.aSlider.value() / float(POSITION_FACTOR)
        b = self.bSlider.value() / float(POSITION_FACTOR)
        self.soloTool.storeAbLimits(a, b)
        self.abListModel.layoutChanged.emit()

    def toggleAbRepeat(self):
        enable = self.abRepeatCheckBox.isChecked()
        self.soloTool.setAbLimitEnable(enable)
        
    def saveSession(self):
        path, _ = QFileDialog.getSaveFileName(self, "Open file", "", "session file (*.json)")
        if path:
            self.soloTool.saveSession(path)

    def loadSession(self):
        path, _ = QFileDialog.getOpenFileName(self, "Open file", "", "session file (*.json)")
        if path:
            self.soloTool.loadSession(path)
            self.playlistModel.layoutChanged.emit()
            self.abListModel.layoutChanged.emit()

    def songSliderPressed(self):
        self.timer.stop()

    def songSliderReleased(self):
        position = self.songSlider.value() / float(POSITION_FACTOR)
        self.soloTool.setPlaybackPosition(position)
        self.timer.start()

    def clearAbListSelection(self):
        i = self.abListView.selectionModel().currentIndex()
        self.abListView.selectionModel().select(i, QItemSelectionModel.Deselect)

    def abSliderReleased(self):
        a = self.aSlider.value() / float(POSITION_FACTOR)
        b = self.bSlider.value() / float(POSITION_FACTOR)
        self.soloTool.setAbLimits(a, b)
        self.clearAbListSelection()

    def rateSliderReleased(self):
        rate = self.rateSlider.value() / float(RATE_FACTOR)
        self.soloTool.setPlaybackRate(rate)

    def playlistSelectionChanged(self, i):
        index = i.indexes()[0].row()
        self.soloTool.setSong(index)
        self.abListModel.layoutChanged.emit()
        self.clearAbListSelection()

    def abListSelectionChanged(self, i):
        if i is not None and not i.isEmpty():
            index = i.indexes()[0].row()
            ab = self.soloTool.getStoredAbLimits()[index]
            self.soloTool.loadAbLimits(index)
            self.aSlider.setValue(int(ab[0] * POSITION_FACTOR))
            self.bSlider.setValue(int(ab[1] * POSITION_FACTOR))

if __name__ == '__main__':
    app = QApplication([])
    app.setApplicationName("Solo Tool")
    app.setStyle("Fusion")

    # Fusion dark palette from https://gist.github.com/QuantumCD/6245215.
    palette = QPalette()
    palette.setColor(QPalette.Window, QColor(53, 53, 53))
    palette.setColor(QPalette.WindowText, Qt.white)
    palette.setColor(QPalette.Base, QColor(25, 25, 25))
    palette.setColor(QPalette.AlternateBase, QColor(53, 53, 53))
    palette.setColor(QPalette.ToolTipBase, Qt.white)
    palette.setColor(QPalette.ToolTipText, Qt.white)
    palette.setColor(QPalette.Text, Qt.white)
    palette.setColor(QPalette.Button, QColor(53, 53, 53))
    palette.setColor(QPalette.ButtonText, Qt.white)
    palette.setColor(QPalette.BrightText, Qt.red)
    palette.setColor(QPalette.Link, QColor(42, 130, 218))
    palette.setColor(QPalette.Highlight, QColor(42, 130, 218))
    palette.setColor(QPalette.HighlightedText, Qt.black)
    app.setPalette(palette)
    app.setStyleSheet("QToolTip { color: #ffffff; background-color: #2a82da; border: 1px solid white; }")

    window = MainWindow()
    app.exec_()