import os
from qtpy import QtCore, QtWidgets, QtGui

from openpype.lib import Logger

from openpype.tools.utils.constants import (
    LOCAL_PROVIDER_ROLE,
    REMOTE_PROVIDER_ROLE,
    LOCAL_PROGRESS_ROLE,
    REMOTE_PROGRESS_ROLE,
    LOCAL_DATE_ROLE,
    REMOTE_DATE_ROLE,
    LOCAL_FAILED_ROLE,
    REMOTE_FAILED_ROLE,
    EDIT_ICON_ROLE
)

log = Logger.get_logger("SyncServer")


class PriorityDelegate(QtWidgets.QStyledItemDelegate):
    """Creates editable line edit to set priority on representation"""
    def paint(self, painter, option, index):
        super(PriorityDelegate, self).paint(painter, option, index)

        if option.widget.selectionModel().isSelected(index) or \
                option.state & QtWidgets.QStyle.State_MouseOver:
            edit_icon = index.data(EDIT_ICON_ROLE)
            if not edit_icon:
                return

            state = QtGui.QIcon.On
            mode = QtGui.QIcon.Selected

            icon_side = 16
            icon_rect = QtCore.QRect(
                option.rect.left() + option.rect.width() - icon_side - 4,
                option.rect.top() + ((option.rect.height() - icon_side) / 2),
                icon_side,
                icon_side
            )

            edit_icon.paint(
                painter, icon_rect,
                QtCore.Qt.AlignRight, mode, state
            )

    def createEditor(self, parent, option, index):
        editor = PriorityLineEdit(
            parent,
            option.widget.selectionModel().selectedRows())
        editor.setFocus()
        return editor

    def setModelData(self, editor, model, index):
        for index in editor.selected_idxs:
            try:
                val = int(editor.text())
            except ValueError:
                val = model.sync_server.DEFAULT_PRIORITY
            model.set_priority_data(index, val)


class PriorityLineEdit(QtWidgets.QLineEdit):
    """Special LineEdit to consume Enter and store selected indexes"""
    def __init__(self, parent=None, selected_idxs=None):
        self.selected_idxs = selected_idxs
        super(PriorityLineEdit, self).__init__(parent)

    def keyPressEvent(self, event):
        result = super(PriorityLineEdit, self).keyPressEvent(event)
        if (
            event.key() in (QtCore.Qt.Key_Return, QtCore.Qt.Key_Enter)
        ):
            return event.accept()

        return result


class ImageDelegate(QtWidgets.QStyledItemDelegate):
    """
        Prints icon of site and progress of synchronization
    """

    def __init__(self, parent=None, side=None):
        super(ImageDelegate, self).__init__(parent)
        self.icons = {}
        self.side = side

    def paint(self, painter, option, index):
        super(ImageDelegate, self).paint(painter, option, index)
        option = QtWidgets.QStyleOptionViewItem(option)
        option.showDecorationSelected = True

        if not self.side:
            log.warning("No side provided, delegate won't work")
            return

        if self.side == 'local':
            provider = index.data(LOCAL_PROVIDER_ROLE)
            value = index.data(LOCAL_PROGRESS_ROLE)
            date_value = index.data(LOCAL_DATE_ROLE)
            is_failed = index.data(LOCAL_FAILED_ROLE)
        else:
            provider = index.data(REMOTE_PROVIDER_ROLE)
            value = index.data(REMOTE_PROGRESS_ROLE)
            date_value = index.data(REMOTE_DATE_ROLE)
            is_failed = index.data(REMOTE_FAILED_ROLE)

        if not self.icons.get(provider):
            resource_path = os.path.dirname(__file__)
            resource_path = os.path.join(resource_path, "..",
                                         "providers", "resources")
            pix_url = "{}/{}.png".format(resource_path, provider)
            pixmap = QtGui.QPixmap(pix_url)
            self.icons[provider] = pixmap
        else:
            pixmap = self.icons[provider]

        padding = 10
        point = QtCore.QPoint(option.rect.x() + padding,
                              option.rect.y() +
                              (option.rect.height() - pixmap.height()) / 2)
        painter.drawPixmap(point, pixmap)

        overlay_rect = option.rect.translated(0, 0)
        overlay_rect.setHeight(overlay_rect.height() * (1.0 - float(value)))
        painter.fillRect(overlay_rect,
                         QtGui.QBrush(QtGui.QColor(0, 0, 0, 100)))
        text_rect = option.rect.translated(10, 0)
        painter.drawText(text_rect,
                         QtCore.Qt.AlignCenter,
                         date_value)

        if is_failed:
            overlay_rect = option.rect.translated(0, 0)
            painter.fillRect(overlay_rect,
                             QtGui.QBrush(QtGui.QColor(255, 0, 0, 35)))