#  -*- coding: utf-8 -*-
# *****************************************************************************
# Copyright (c) 2015-2016 by the authors, see LICENSE
#
# This program is free software; you can redistribute it and/or modify it under
# the terms of the GNU General Public License as published by the Free Software
# Foundation; either version 2 of the License, or (at your option) any later
# version.
#
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
# details.
#
# You should have received a copy of the GNU General Public License along with
# this program; if not, write to the Free Software Foundation, Inc.,
# 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
#
# Module authors:
#   Alexander Lenz <alexander.lenz@frm2.tum.de>
#
# *****************************************************************************

import pprint
import json

from PyQt4.QtGui import QWidget, QTextCursor, QFont, QFontMetrics, QLabel, QPushButton, QLineEdit, QMessageBox, QCheckBox, QSizePolicy
from PyQt4.QtCore import pyqtSignature as qtsig, Qt, pyqtSignal

from secop.gui.util import loadUi
from secop.protocol.errors import SECOPError
from secop.datatypes import StringType, EnumType


class NodeCtrl(QWidget):

    def __init__(self, node, parent=None):
        super(NodeCtrl, self).__init__(parent)
        loadUi(self, 'nodectrl.ui')

        self._node = node

        self.contactPointLabel.setText(self._node.contactPoint)
        self.equipmentIdLabel.setText(self._node.equipmentId)
        self.protocolVersionLabel.setText(self._node.protocolVersion)
        self._clearLog()

        # now populate modules tab
        self._init_modules_tab()

    @qtsig('')
    def on_sendPushButton_clicked(self):
        msg = self.msgLineEdit.text().strip()

        if not msg:
            return

        self._addLogEntry(
            '<span style="font-weight:bold">Request:</span> '
            '%s:' % msg,
            raw=True)
        #        msg = msg.split(' ', 2)
        try:
            reply = self._node.syncCommunicate(*self._node.decode_message(msg))
            if msg == 'describe':
                _, eid, stuff = self._node.decode_message(reply)
                reply = '%s %s %s' % (_, eid, json.dumps(
                    stuff, indent=2, separators=(',', ':'), sort_keys=True))
                self._addLogEntry(reply, newline=True, pretty=False)
            else:
                self._addLogEntry(reply, newline=True, pretty=True)
        except SECOPError as e:
            self._addLogEntry(
                'error %s %s' % (e.name, json.dumps(e.args)),
                newline=True,
                pretty=True,
                error=True)

    @qtsig('')
    def on_clearPushButton_clicked(self):
        self._clearLog()

    def _clearLog(self):
        self.logTextBrowser.clear()

        self._addLogEntry('SECoP Communication Shell')
        self._addLogEntry('=========================')
        self._addLogEntry('', newline=True)

    def _addLogEntry(self,
                     msg,
                     newline=False,
                     pretty=False,
                     raw=False,
                     error=False):
        if pretty:
            msg = pprint.pformat(msg, width=self._getLogWidth())
            msg = msg[1:-1]

        if not raw:
            if error:
                msg = '<div style="color:#FF0000"><b><pre>%s</pre></b></div>' % Qt.escape(
                    str(msg)).replace('\n', '<br />')
            else:
                msg = '<pre>%s</pre>' % Qt.escape(str(msg)).replace('\n',
                                                                    '<br />')

        content = ''
        if self.logTextBrowser.toPlainText():
            content = self.logTextBrowser.toHtml()
        content += msg

        if newline:
            content += '<br />'

        self.logTextBrowser.setHtml(content)
        self.logTextBrowser.moveCursor(QTextCursor.End)

    def _getLogWidth(self):
        fontMetrics = QFontMetrics(QFont('Monospace'))
        # calculate max avail characters by using an a (which is possible
        # due to monospace)
        result = self.logTextBrowser.width() / fontMetrics.width('a')
        return result

    def _init_modules_tab(self):
        self._moduleWidgets = []
        layout = self.scrollAreaWidgetContents.layout()
        labelfont = self.font()
        labelfont.setBold(True)
        row = 0
        for modname in sorted(self._node.modules):
            modprops = self._node.getModuleProperties(modname)
            baseclass = modprops['interface']
            description = modprops['interface']
            unit = self._node.getProperties(modname, 'value').get('unit', '')

            if unit:
                labelstr = '%s (%s):' % (modname, unit)
            else:
                labelstr = '%s:' % (modname,)
            label = QLabel(labelstr)
            label.setFont(labelfont)

            if baseclass == 'Driveable':
                widget = DriveableWidget(self._node, modname, self)
            elif baseclass == 'Readable':
                widget = ReadableWidget(self._node, modname, self)
            else:
                widget = QLabel('Unsupported Interfaceclass %r' % baseclass)

            if description:
                widget.setToolTip(description)

            layout.addWidget(label, row, 0)
            layout.addWidget(widget, row, 1)

            row += 1
            self._moduleWidgets.extend((label, widget))


class ReadableWidget(QWidget):

    def __init__(self, node, module, parent=None):
        super(ReadableWidget, self).__init__(parent)
        self._node = node
        self._module = module

        params = self._node.getProperties(self._module, 'value')
        datatype = params.get('datatype', StringType())
        self._is_enum = isinstance(datatype, EnumType)

        loadUi(self, 'modulebuttons.ui')

        # populate comboBox, keeping a mapping of Qt-index to EnumValue
        if self._is_enum:
            self._map = {}  # maps QT-idx to name/value
            self._revmap = {}  # maps value/name to QT-idx
            for idx, (val, name) in enumerate(
                    sorted(datatype.entries.items())):
                self._map[idx] = (name, val)
                self._revmap[name] = idx
                self._revmap[val] = idx
                self.targetComboBox.addItem(name, val)

        self._init_status_widgets()
        self._init_current_widgets()
        self._init_target_widgets()

        self._node.newData.connect(self._updateValue)

    def _get(self, pname, fallback=Ellipsis):
        params = self._node.queryCache(self._module)
        if pname in params:
            return params[pname].value
        try:
            return self._node.getParameter(self._module, pname)
        except Exception:
            self.log.exception()
            if fallback is not Ellipsis:
                return fallback
            raise

    def _init_status_widgets(self):
        self.update_status(self._get('status', (999, '<not supported>')))
        # XXX: also connect update_status signal to LineEdit ??

    def update_status(self, status, qualifiers={}):
        self.statusLineEdit.setText(str(status))
        # may change meaning of cmdPushButton

    def _init_current_widgets(self):
        self.update_current(self._get('value', ''))

    def update_current(self, value, qualifiers={}):
        self.currentLineEdit.setText(str(value))

    def _init_target_widgets(self):
        #  Readable has no target: disable widgets
        self.targetLineEdit.setHidden(True)
        self.targetComboBox.setHidden(True)
        self.cmdPushButton.setHidden(True)

    def update_target(self, target, qualifiers={}):
        pass

    def target_go(self, target):
        try:
            self._node.setParameter(self._module, 'target', target)
        except Exception as e:
            QMessageBox.warning(self.parent(), 'Operation failed', str(e))

    def _updateValue(self, module, parameter, value):
        if module != self._module:
            return
        if parameter == 'status':
            self.update_status(*value)
        elif parameter == 'value':
            self.update_current(*value)
        elif parameter == 'target':
            self.update_target(*value)


class DriveableWidget(ReadableWidget):

    def _init_target_widgets(self):
        params = self._node.getProperties(self._module, 'target')
        if self._is_enum:
            # EnumType: disable Linedit
            self.targetLineEdit.setHidden(True)
        else:
            # normal types: disable Combobox
            self.targetComboBox.setHidden(True)
            target = self._get('target', None)
            if target:
                if isinstance(target, list) and isinstance(target[1], dict):
                    self.update_target(target[0])
                else:
                    self.update_target(target)

    def update_current(self, value, qualifiers={}):
        if self._is_enum:
            self.currentLineEdit.setText(self._map[self._revmap[value]][0])
        else:
            self.currentLineEdit.setText(str(value))

    def update_target(self, target, qualifiers={}):
        if self._is_enum:
            # update selected item
            if target in self._revmap:
                self.targetComboBox.setCurrentIndex(self._revmap[target])
            else:
                print(
                    "%s: Got invalid target value %r!" %
                    (self._module, target))
        else:
            self.targetLineEdit.setText(str(target))

    def on_cmdPushButton_clicked(self, toggle=False):
        if toggled:
            return
        if self._is_enum:
            self.on_targetComboBox_activated()
        else:
            self.on_targetLineEdit_returnPressed()

    def on_targetLineEdit_returnPressed(self):
        self.target_go(self.targetLineEdit.text())

    def on_targetComboBox_activated(self, stuff=''):
        if isinstance(stuff, (str, unicode)):
            return
        self.target_go(self._map[self.targetComboBox.currentIndex()][0])