from typing import Optional, TYPE_CHECKING

from PyQt5.QtCore import QObject, pyqtSlot, Qt, pyqtSignal, pyqtProperty

from UM.Qt.ListModel import ListModel

if TYPE_CHECKING:
    from cura.CuraApplication import CuraApplication


class DiscoveredUltimakerCloudPrintersModel(ListModel):
    DeviceKeyRole = Qt.UserRole + 1
    DeviceNameRole = Qt.UserRole + 2
    DeviceTypeRole = Qt.UserRole + 3
    DeviceFirmwareVersionRole = Qt.UserRole + 4

    cloudPrintersDetectedChanged = pyqtSignal()

    def __init__(self, application: "CuraApplication", parent: Optional["QObject"] = None) -> None:
        super().__init__(parent)

        self.addRoleName(self.DeviceKeyRole, "key")
        self.addRoleName(self.DeviceNameRole, "name")
        self.addRoleName(self.DeviceTypeRole, "machine_type")
        self.addRoleName(self.DeviceFirmwareVersionRole, "firmware_version")

        self._discovered_ultimaker_cloud_printers_list = []
        self._new_cloud_printers_detected = False
        self._application = application

    def addDiscoveredUltimakerCloudPrinters(self, new_devices) -> None:
        for device in new_devices:
            self._discovered_ultimaker_cloud_printers_list.append({
                "key": device.getId(),
                "name": device.name,
                "machine_type": device.printerTypeName,
                "firmware_version": device.firmwareVersion
            })
        self._update()

        # Inform whether new cloud printers have been detected. If they have, the welcome wizard can close.
        self._new_cloud_printers_detected = len(new_devices) > 0
        self.cloudPrintersDetectedChanged.emit()

    @pyqtSlot()
    def clear(self):
        self._discovered_ultimaker_cloud_printers_list = []
        self._update()
        self._new_cloud_printers_detected = False
        self.cloudPrintersDetectedChanged.emit()

    def _update(self):
        items = []

        for cloud_printer in self._discovered_ultimaker_cloud_printers_list:
            items.append(cloud_printer)

        # Execute all filters.
        filtered_items = list(items)

        filtered_items.sort(key = lambda k: k["name"])
        self.setItems(filtered_items)

    @pyqtProperty(bool, notify = cloudPrintersDetectedChanged)
    def newCloudPrintersDetected(self) -> bool:
        return self._new_cloud_printers_detected