# Copyright (c) 2020 Ultimaker B.V.
# Cura is released under the terms of the LGPLv3 or higher.

from PyQt5.QtCore import Qt
from UM.Qt.ListModel import ListModel
from cura import ApplicationMetadata

from PyQt5.QtCore import pyqtSlot

from UM.Logger import Logger


class SubscribedPackagesModel(ListModel):
    def __init__(self, parent = None):
        super().__init__(parent)

        self._items = []
        self._metadata = None
        self._discrepancies = None
        self._sdk_version = ApplicationMetadata.CuraSDKVersion

        self.addRoleName(Qt.UserRole + 1, "name")
        self.addRoleName(Qt.UserRole + 2, "icon_url")
        self.addRoleName(Qt.UserRole + 3, "is_compatible")
        self.addRoleName(Qt.UserRole + 4, "is_dismissed")
        self.addRoleName(Qt.UserRole + 5, "package_id")


    def setMetadata(self, data):
        if self._metadata != data:
            self._metadata = data

    def addValue(self, discrepancy):
        if self._discrepancies != discrepancy:
            self._discrepancies = discrepancy

    def update(self):
        self._items.clear()

        for item in self._metadata:
            if item["package_id"] not in self._discrepancies:
                continue
            package = {"package_id": item["package_id"],
                       "name": item["display_name"],
                       "sdk_versions": item["sdk_versions"],
                       "is_dismissed": False
                       }
            if self._sdk_version not in item["sdk_versions"]:
                package.update({"is_compatible": False})
            else:
                package.update({"is_compatible": True})
            try:
                package.update({"icon_url": item["icon_url"]})
            except KeyError:  # There is no 'icon_url" in the response payload for this package
                package.update({"icon_url": ""})

            self._items.append(package)
        self.setItems(self._items)


    def hasCompatiblePackages(self) -> bool:
        has_compatible_items  = False
        for item in self._items:
            if item['is_compatible'] == True:
                has_compatible_items = True
        return has_compatible_items

    def hasIncompatiblePackages(self) -> bool:
        has_incompatible_items  = False
        for item in self._items:
            if item['is_compatible'] == False:
                has_incompatible_items = True
        return has_incompatible_items

    def setDismiss(self, package_id) -> None:
        package_id_in_list_of_items = self.find(key="package_id", value=package_id)
        if package_id_in_list_of_items != -1:
            self.setProperty(package_id_in_list_of_items, property="is_dismissed", value=True)
            Logger.debug("Package {} has been dismissed".format(package_id))

    def addDismissed(self, list_of_dismissed) -> None:
        for package in list_of_dismissed:
            item = self.find(key="package_id", value=package)
            if item != -1:
                self.setProperty(item, property="is_dismissed", value=True)