from PyQt5 import QtGui, QtCore, QtWidgets
from PyQt5.QtWidgets import QSizePolicy

from cdemu import CDEmuException
from .device_menu import QDeviceMenu
from mount import mount
from urllib.parse import unquote

class KCDEmuDevice(QtWidgets.QLabel):
    def __init__(self, device, width=32, height=32):
        super().__init__()
        self.setAcceptDrops(True)
        self.device = None
        self.empty_icon = ':/dropper_empty.png'
        self.full_icon = ':/dropper_full.png'
        self.refresh(device)
        self.setSizePolicy(QSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed))
        self.setAlignment(QtCore.Qt.AlignCenter)
        self.setMinimumSize(width, height)
        self.setBaseSize(width, height)
        self.setMaximumSize(width, height)
        self.ctx_menu = QDeviceMenu(device)

    def set_size(self, width, height):
        self.setMinimumSize(width, height)
        self.setBaseSize(width, height)
        self.setMaximumSize(width, height)
        self.adjustSize()

    def set_icons(self, empty_icon, full_icon):
        self.empty_icon = empty_icon
        self.full_icon = full_icon
        self.refresh()

    def refresh(self, device=None):
        if device is not None:
            if self.device and self.device != device:
                self.device.remove_handler('change')
            if self.device != device:
                self.device = device
                self.device.add_handler('change', self.refresh)
        self.setText(
            '<img src="{}"></img>'.format(
                self.full_icon if self.device.get_img() else self.empty_icon))
        if self.device.get_img():
            self.setToolTip('Click to unload "{}"'.format(self.device.get_img()))
        else:
            self.setToolTip('Click to load')

    def dragEnterEvent(self, event):
        if event.mimeData().hasFormat('text/plain'):
            text = event.mimeData().text()
            if text.startswith('file://'):
                event.acceptProposedAction()

    def dragLeaveEvent(self, event):
        pass

    def dropEvent(self, event):
        if event.mimeData().hasFormat('text/plain'):
            text = event.mimeData().text()
            if text.startswith('file://'):
                text = unquote(text.replace('file://', '').replace('\n', '').replace('\r', ''))
                self.device.load(text)

    def mousePressEvent(self, event):
        if event.button() == QtCore.Qt.RightButton:
            self.ctx_menu.popup(QtCore.QPoint(event.globalX(), event.globalY()))
        elif event.button() == QtCore.Qt.LeftButton:
            if self.device.get_img():
                try:
                    self.device.unload()
                except Exception as e:
                    QtWidgets.QMessageBox.critical(None, 'Error', str(e))
            else:
                fd = QtWidgets.QFileDialog(self)
                fd.setAcceptMode(QtWidgets.QFileDialog.AcceptOpen)
                if fd.exec() == QtWidgets.QDialog.Accepted:
                    self.device.load(fd.selectedFiles()[0])

class KCDEmuDropperWidget(QtWidgets.QWidget):
    changed = QtCore.pyqtSignal()

    def __init__(self, cdemu, config):
        """
        :type cdemu: cdemu.CDEmu
        :type config: config.config.ConfigFile
        """
        super().__init__(None, QtCore.Qt.FramelessWindowHint | QtCore.Qt.WindowStaysOnTopHint)
        self.config = config
        self.setWindowTitle('KCDEmu')
        self.setSizePolicy(QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Minimum))
        layout = QtWidgets.QHBoxLayout(self)
        layout.setSizeConstraint(QtWidgets.QLayout.SetFixedSize)
        layout.setContentsMargins(0,0,0,0)
        layout.setSpacing(0)
        self.setLayout(layout)
        self.cdemu = cdemu
        self.devices = []

        self.cdemu.add_handler('devices_changed', self.on_devices_changed)
        self.cdemu.add_handler('daemon_stopped', self.on_daemon_stopped)
        self.cdemu.add_handler('daemon_started', self.on_daemon_started)
        self.config.add_handler('changed', self.reload_config)
        self.reload_config()

        if self.cdemu.is_daemon_ready():
            self.on_devices_changed()

    def on_devices_changed(self, number=None):
        devices = self.cdemu.get_devices()
        while len(self.devices) > len(devices):
            dev = self.devices.pop()
            self.layout().removeWidget(dev)
            dev.hide()
        for index, device in enumerate(devices):
            if len(self.devices) <= index:
                icon_size = self.config.get_icon_size()
                dev = KCDEmuDevice(device, icon_size[0], icon_size[1])
                dev.set_icons(self.config.get_empty_icon(), self.config.get_full_icon())
                self.devices.append(dev)
                self.layout().addWidget(dev)
            else:
                self.devices[index].refresh(device)
        self.updateGeometry()

    def resizeEvent(self, event):
        print(self.sizeHint().width())
        self.changed.emit()

    def on_daemon_stopped(self):
        while len(self.devices) > 0:
            dev = self.devices.pop()
            self.layout().removeWidget(dev)

    def on_daemon_started(self):
        self.on_devices_changed()

    def reload_config(self):
        for i in range(self.layout().count()):
            widget = self.layout().itemAt(i).widget()
            if widget:
                icon_size = self.config.get_icon_size()
                widget.set_size(icon_size[0], icon_size[1])
                widget.set_icons(self.config.get_empty_icon(), self.config.get_full_icon())