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

from typing import TYPE_CHECKING

from UM.Settings.ContainerRegistry import ContainerRegistry
from UM.Settings.Interfaces import ContainerInterface
from cura.Machines.ContainerNode import ContainerNode
from cura.Machines.QualityNode import QualityNode

if TYPE_CHECKING:
    from typing import Dict
    from cura.Machines.VariantNode import VariantNode

##  Represents a material in the container tree.
#
#   Its subcontainers are quality profiles.
class MaterialNode(ContainerNode):
    def __init__(self, container_id, variant: "VariantNode") -> None:
        super().__init__(container_id)
        self.variant = variant
        self.qualities = {}  # type: Dict[str, QualityNode] # Mapping container IDs to quality profiles.
        container_registry = ContainerRegistry.getInstance()
        my_metadata = container_registry.findContainersMetadata(id = container_id)[0]
        self.base_file = my_metadata["base_file"]
        container_registry.containerAdded.connect(self._qualityAdded)
        self._loadAll()

    def _loadAll(self) -> None:
        container_registry = ContainerRegistry.getInstance()
        # Find all quality profiles that fit on this material.
        if not self.variant.machine.has_machine_quality:  # Need to find the global qualities.
            qualities = container_registry.findInstanceContainersMetadata(type = "quality", definition = "fdmprinter")
        else:
            # Need to find the qualities that specify a material profile with the same material type.
            my_metadata = container_registry.findInstanceContainersMetadata(id = self.container_id)[0]
            my_material_type = my_metadata.get("material")
            qualities = []
            for material_metadata in container_registry.findInstanceContainersMetadata(type = "material", material = my_material_type):
                qualities.extend(container_registry.findInstanceContainersMetadata(type = "quality", definition = self.variant.machine.quality_definition, variant = self.variant.variant_name, material = material_metadata["id"]))
            if not qualities:  # No quality profiles found. Go by GUID then.
                my_guid = my_metadata.get("material")
                for material_metadata in container_registry.findInstanceContainersMetadata(type = "material", guid = my_guid):
                    qualities.extend(container_registry.findInstanceContainersMetadata(type = "quality", definition = self.variant.machine.quality_definition, variant = self.variant.variant_name, material = material_metadata["id"]))

        for quality in qualities:
            quality_id = quality["id"]
            if quality_id not in self.qualities:
                self.qualities[quality_id] = QualityNode(quality_id, parent = self)

    def _qualityAdded(self, container: ContainerInterface) -> None:
        if container.getMetaDataEntry("type") != "quality":
            return  # Not interested.
        if not self.variant.machine.has_machine_quality:
            if container.getMetaDataEntry("definition") != "fdmprinter":
                return  # Only want global qualities.
        else:
            if container.getMetaDataEntry("definition") != self.variant.machine.quality_definition:
                return  # Doesn't match the machine.
            if container.getMetaDataEntry("variant") != self.variant.variant_name:
                return  # Doesn't match the variant.
            # Detect if we're falling back to matching via GUID.
            # If so, we might need to erase the current list and put just this one in (i.e. no longer use the fallback).
            container_registry = ContainerRegistry.getInstance()
            my_metadata = container_registry.findInstanceContainersMetadata(id = self.container_id)[0]
            my_material_type = my_metadata.get("material")
            allowed_material_ids = {metadata["id"] for metadata in container_registry.findInstanceContainersMetadata(type = "material", material = my_material_type)}
            is_fallback_guid = len(self.qualities) == 0 or next(iter(self.qualities.values())).getMetaDataEntry("material") not in allowed_material_ids  # Select any quality profile; if the material is not matching by material type, we've been falling back to GUID.

            if is_fallback_guid and container.getMetaDataEntry("material") in allowed_material_ids:  # So far we needed the fallback, but no longer!
                self.qualities.clear()
            else:
                if not is_fallback_guid:
                    if container.getMetaDataEntry("material") not in allowed_material_ids:
                        return  # Doesn't match the material type.
                else:
                    my_material_guid = my_metadata["guid"]
                    allowed_material_ids = {metadata["id"] for metadata in container_registry.findInstanceContainersMetadata(type = "material", guid = my_material_guid)}
                    if container.getMetaDataEntry("material") not in allowed_material_ids:
                        return  # Doesn't match the material GUID.

        quality_id = container.getId()
        self.qualities[quality_id] = QualityNode(quality_id, parent = self)