# (C) Copyright 2005-2022 Enthought, Inc., Austin, TX
# All rights reserved.
#
# This software is provided without warranty under the terms of the BSD
# license included in LICENSE.txt and may be redistributed only under
# the conditions described in the aforementioned license. The license
# is also available online at http://www.enthought.com/licenses/BSD.txt
#
# Thanks for using Enthought open source!
# (C) Copyright 2007 Riverbank Computing Limited
# This software is provided without warranty under the terms of the BSD license.
# However, when used with the GPL version of PyQt the additional terms described in the PyQt GPL exception also apply


from pyface.qt import QtCore, QtGui


from traits.api import Enum, provides, Str


from pyface.i_message_dialog import IMessageDialog, MMessageDialog
from .dialog import Dialog


# Map the ETS severity to the corresponding PyQt standard icon.
_SEVERITY_TO_ICON_MAP = {
    "information": QtGui.QMessageBox.Icon.Information,
    "warning": QtGui.QMessageBox.Icon.Warning,
    "error": QtGui.QMessageBox.Icon.Critical,
}

_TEXT_FORMAT_MAP = {
    "auto": QtCore.Qt.TextFormat.AutoText,
    "plain": QtCore.Qt.TextFormat.PlainText,
    "rich": QtCore.Qt.TextFormat.RichText,
}


@provides(IMessageDialog)
class MessageDialog(MMessageDialog, Dialog):
    """ The toolkit specific implementation of a MessageDialog.  See the
    IMessageDialog interface for the API documentation.
    """

    # 'IMessageDialog' interface -------------------------------------------

    message = Str()

    informative = Str()

    detail = Str()

    severity = Enum("information", "warning", "error")

    text_format = Enum("auto", "plain", "rich")

    # ------------------------------------------------------------------------
    # Protected 'IDialog' interface.
    # ------------------------------------------------------------------------

    def _create_contents(self, parent):
        # In PyQt this is a canned dialog.
        pass

    # ------------------------------------------------------------------------
    # Protected 'IWidget' interface.
    # ------------------------------------------------------------------------

    def _create_control(self, parent):
        # FIXME: should be possble to set ok_label, but not implemented
        message_box = QtGui.QMessageBox(
            _SEVERITY_TO_ICON_MAP[self.severity],
            self.title,
            self.message,
            QtGui.QMessageBox.StandardButton.Ok,
            parent,
        )
        message_box.setInformativeText(self.informative)
        message_box.setDetailedText(self.detail)
        message_box.setEscapeButton(QtGui.QMessageBox.StandardButton.Ok)
        message_box.setTextFormat(_TEXT_FORMAT_MAP[self.text_format])

        if self.size != (-1, -1):
            message_box.resize(*self.size)

        if self.position != (-1, -1):
            message_box.move(*self.position)

        return message_box
