from PyQt5 import QtWidgets
from PyQt5.QtCore import Qt
from PyQt5.QtGui import QStandardItem, QStandardItemModel

from ArtNet.gui.dialogs.presence_selection_dialog.presence_selection_dialog import Ui_presence_selection_dialog


class PresenceSelectDialog(QtWidgets.QDialog):

    def __init__(self, parent=None):
        super().__init__(parent)
        self.parent = parent
        self.ui = Ui_presence_selection_dialog()
        self.ui.setupUi(self)

        self.__curr_presence: list = None
        self.__curr_search_result: list = None

        self.ui.presence_name_search_line.textChanged.connect(self.on_search_change)
        self.ui.presence_domain_search_line.textChanged.connect(self.on_search_change)

    def set_search_result_list(self, result: list):
        self.__curr_search_result = result
        item_model = QStandardItemModel(self.ui.presence_search_list)

        for presence in result:
            item = QStandardItem(presence)
            item.setFlags(Qt.ItemIsUserCheckable | Qt.ItemIsEnabled)
            item.setData(Qt.Unchecked, Qt.CheckStateRole)

            if self.__curr_presence is not None:
                curr_presence = self.__curr_presence[0] + ":" + self.__curr_presence[1]
                if presence == curr_presence:
                    item.setCheckState(Qt.Checked)

            item_model.appendRow(item)

        item_model.itemChanged.connect(self.on_search_presence_item_changed)
        self.ui.presence_search_list.setModel(item_model)

    def on_search_change(self):
        name = self.ui.presence_name_search_line.text()
        domain = self.ui.presence_domain_search_line.text()

        if len(name) == 0 and len(domain) == 0:  # nothing to search for
            self.set_search_result_list([])
            return
        presences = self.parent.parent.get_authors(name, domain)

        result = []
        for ID, desc in presences:
            result.append(ID+":"+desc)

        self.set_search_result_list(result)

    def on_search_presence_item_changed(self, item: QStandardItem):
        name, domain = item.text().split(":")
        if item.checkState() == Qt.Checked:
            self.__curr_presence = self.parent.parent.get_presence(name, domain)
            if self.__curr_presence is not None:
                self.__curr_presence = self.__curr_presence[0]  # unpack from list
        elif item.checkState() == Qt.Unchecked:
            self.__curr_presence = None
        self.set_search_result_list(self.__curr_search_result)

    def exec_(self) -> list:
        if super(PresenceSelectDialog, self).exec_() == QtWidgets.QDialog.Rejected:
            return None

        return self.__curr_presence