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

import re
from typing import Dict, List, Optional, Union

from PyQt5.QtCore import Qt, pyqtProperty

from UM.Qt.ListModel import ListModel


##  Model that holds cura packages. By setting the filter property the instances held by this model can be changed.
class AuthorsModel(ListModel):
    def __init__(self, parent = None) -> None:
        super().__init__(parent)

        self._metadata = None  # type: Optional[List[Dict[str, Union[str, List[str], int]]]]

        self.addRoleName(Qt.UserRole + 1, "id")
        self.addRoleName(Qt.UserRole + 2, "name")
        self.addRoleName(Qt.UserRole + 3, "email")
        self.addRoleName(Qt.UserRole + 4, "website")
        self.addRoleName(Qt.UserRole + 5, "package_count")
        self.addRoleName(Qt.UserRole + 6, "package_types")
        self.addRoleName(Qt.UserRole + 7, "icon_url")
        self.addRoleName(Qt.UserRole + 8, "description")

        # List of filters for queries. The result is the union of the each list of results.
        self._filter = {}  # type: Dict[str, str]

    def setMetadata(self, data: List[Dict[str, Union[str, List[str], int]]]):
        if self._metadata != data:
            self._metadata = data
            self._update()

    def _update(self) -> None:
        items = []  # type: List[Dict[str, Union[str, List[str], int, None]]]
        if not self._metadata:
            self.setItems(items)
            return

        for author in self._metadata:
            items.append({
                "id":             author.get("author_id"),
                "name":           author.get("display_name"),
                "email":          author.get("email"),
                "website":        author.get("website"),
                "package_count":  author.get("package_count", 0),
                "package_types":  author.get("package_types", []),
                "icon_url":       author.get("icon_url"),
                "description":    "Material and quality profiles from {author_name}".format(author_name = author.get("display_name", ""))
            })

        # Filter on all the key-word arguments.
        for key, value in self._filter.items():
            if key == "package_types":
                key_filter = lambda item, value = value: value in item["package_types"]  # type: ignore
            elif "*" in value:
                key_filter = lambda item, key = key, value = value: self._matchRegExp(item, key, value)  # type: ignore
            else:
                key_filter = lambda item, key = key, value = value: self._matchString(item, key, value)  # type: ignore
            items = filter(key_filter, items)  # type: ignore

        # Execute all filters.
        filtered_items = list(items)

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

    ##  Set the filter of this model based on a string.
    #   \param filter_dict \type{Dict} Dictionary to do the filtering by.
    def setFilter(self, filter_dict: Dict[str, str]) -> None:
        if filter_dict != self._filter:
            self._filter = filter_dict
            self._update()

    @pyqtProperty("QVariantMap", fset = setFilter, constant = True)
    def filter(self) -> Dict[str, str]:
        return self._filter

    # Check to see if a container matches with a regular expression
    def _matchRegExp(self, metadata, property_name, value):
        if property_name not in metadata:
            return False
        value = re.escape(value) #Escape for regex patterns.
        value = "^" + value.replace("\\*", ".*") + "$" #Instead of (now escaped) asterisks, match on any string. Also add anchors for a complete match.
        if self._ignore_case:
            value_pattern = re.compile(value, re.IGNORECASE)
        else:
            value_pattern = re.compile(value)

        return value_pattern.match(str(metadata[property_name]))

    # Check to see if a container matches with a string
    def _matchString(self, metadata, property_name, value):
        if property_name not in metadata:
            return False
        return value.lower() == str(metadata[property_name]).lower()