from PyQt5.QtCore import *
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from ui_profileeditwnd import Ui_ProfileEditDialog
from commandeditwnd import CommandEditWnd
import json
from profileexecutor import *
import sys

class ProfileEditWnd(QDialog):
    def __init__(self, p_profile, p_parent = None):
        super().__init__(p_parent)
        self.m_profiles = []
        self.ui = Ui_ProfileEditDialog()
        self.ui.setupUi(self)
        self.m_profile = {}

        self.ui.cmdTable.setHorizontalHeaderLabels(('Spoken command', 'Actions'))
        self.ui.cmdTable.setSelectionBehavior(QAbstractItemView.SelectRows)
        self.ui.cmdTable.setSelectionMode(QAbstractItemView.SingleSelection)

        self.ui.cmdTable.horizontalHeader().setSectionResizeMode(0, QHeaderView.Interactive)
        self.ui.cmdTable.horizontalHeader().setSectionResizeMode(1, QHeaderView.Stretch)

        # define actions here
        self.ui.newCmd.clicked.connect(self.slotNewCmd)
        self.ui.editCmd.clicked.connect(self.slotEditCmd)
        self.ui.cmdTable.doubleClicked.connect(self.slotEditCmd)
        self.ui.deleteCmd.clicked.connect(self.slotDeleteCmd)
        self.ui.ok.clicked.connect(self.slotOK)
        self.ui.cancel.clicked.connect(self.slotCancel)

        if p_profile == None or p_profile == {}:
            self.ui.cmdTable.setRowCount(0)
            return

        self.ui.profileNameEdit.setText(p_profile['name'])
        w_commands = p_profile['commands']
        self.ui.cmdTable.setRowCount(len(w_commands))
        i = 0
        for w_command in w_commands:
            self.ui.cmdTable.setItem(i, 0, QTableWidgetItem(w_command['name']))
            w_text = json.dumps(w_command)
            w_item = QTableWidgetItem(w_text)
            w_item.setData(Qt.UserRole, json.dumps(w_command))
            self.ui.cmdTable.setItem(i, 1, w_item)
            i = i + 1

        QTimer.singleShot(100, self.ui.cmdTable.resizeRowsToContents)

    # def activateProfile(self, p_profile):
    #     if p_profile == None or p_profile == {}:
    #         self.ui.cmdTable.setRowCount(0)
    #         return
    #
    #     w_commands = p_profile['commands']
    #     self.ui.cmdTable.setRowCount(len(w_commands))
    #     i = 0
    #     for w_command in w_commands:
    #         self.ui.cmdTable.setItem(i, 0, QTableWidgetItem(w_command['name']))
    #         w_text = json.dumps(w_command)
    #         w_item = QTableWidgetItem(w_text)
    #         w_item.setData(Qt.UserRole, json.dumps(w_command))
    #         self.ui.cmdTable.setItem(i, 1, w_item)
    #         i = i + 1
    #     self.ui.cmdTable.resizeRowsToContents()

    # def updateActiveProfile(self):
    #     w_idx = self.m_curProfileIdx
    #     if w_idx < 0:
    #         return
    #
    #     w_profile = {}
    #     w_profile['name'] = self.ui.profileCbx.itemText(w_idx)
    #     w_commands = []
    #
    #     w_commandCnt = self.ui.cmdTable.rowCount()
    #     for w_i in range(w_commandCnt):
    #         w_jsonCommand = self.ui.cmdTable.item(w_i, 1).data(Qt.UserRole)
    #         w_command = json.loads(w_jsonCommand)
    #         w_commands.append(w_command)
    #     w_profile['commands'] = w_commands
    #     w_jsonProfile = json.dumps(w_profile)
    #     self.ui.profileCbx.setItemData(w_idx, w_jsonProfile)

    # def slotSelChanged(self, p_idx):
    #     self.updateActiveProfile()
    #     w_jsonProfile = self.ui.profileCbx.itemData(p_idx)
    #
    #     w_profile = {}
    #     if w_jsonProfile != None:
    #         w_profile = json.loads(w_jsonProfile)
    #
    #     self.activateProfile(w_profile)
    #
    #     self.m_curProfileIdx = p_idx

    # def importProfile(self, p_profile, p_update = True):
    #     if p_profile == None or p_profile == {}:
    #         return
    #
    #     w_profileCnt = self.ui.profileCbx.count()
    #     for w_idx in range(w_profileCnt):
    #         w_jsonProfile = self.ui.profileCbx.itemData(w_idx)
    #         if w_jsonProfile == None:
    #             continue
    #
    #         w_profile = json.loads(w_jsonProfile)
    #         if p_profile['name'] == w_profile['name']:
    #             if p_update:
    #                 w_jsonProfile = json.dumps(p_profile)
    #                 self.ui.profileCbx.setItemData(w_idx, w_jsonProfile)
    #                 return True
    #             return False
    #
    #     self.ui.profileCbx.addItem(p_profile['name'])
    #     w_jsonProfile = json.dumps(p_profile)
    #     self.ui.profileCbx.setItemData(w_profileCnt, w_jsonProfile)
    #     return True

    # def slotAddNewProfile(self):
    #     text, okPressed = QInputDialog.getText(self, "Get Profile Name", "Profile name:", QLineEdit.Normal, "")
    #     if okPressed and text != '':
    #         w_profile = {}
    #         w_profile['name'] = text
    #         w_profile['commands'] = []
    #         if self.importProfile(w_profile, False) == False:
    #             QMessageBox.critical(None, 'Error', 'Adding a new profile was failed')
    #             return
    #         self.ui.profileCbx.setCurrentIndex(self.ui.profileCbx.count()-1)

    # def slotRemoveProfile(self):
    #     w_curIdx = self.ui.profileCbx.currentIndex()
    #     if w_curIdx >= 0:
    #         self.ui.cmdTable.setRowCount(0)
    #         self.m_curProfileIdx = -1
    #         self.ui.profileCbx.removeItem(w_curIdx)
    #
    # def slotRenameProfile(self):
    #     w_curIdx = self.ui.profileCbx.currentIndex()
    #     if w_curIdx >= 0:
    #         text, okPressed = QInputDialog.getText(self, "Input Dialog", "Profile name:",
    #                                                QLineEdit.Normal, self.ui.profileCbx.itemText(w_curIdx))
    #         if okPressed and text != '':
    #             self.ui.profileCbx.setItemText(w_curIdx, text)
    #             w_jsonProfile = self.ui.profileCbx.itemData(w_curIdx)
    #             w_profile = json.loads(w_jsonProfile)
    #             w_profile['name'] = text
    #             w_jsonProfile = json.dumps(w_profile)
    #             self.ui.profileCbx.setItemData(w_curIdx, w_jsonProfile)

    def importCommand(self, p_command, p_update):
        w_commandCnt = self.ui.cmdTable.rowCount()
        for w_i in range(w_commandCnt):
            w_jsonCommand = self.ui.cmdTable.item(w_i, 1).data(Qt.UserRole)
            w_command = json.loads(w_jsonCommand)
            if w_command['name'] == p_command['name']:
                if p_update:
                    w_text = json.dumps(p_command)
                    w_item = QTableWidgetItem(w_text)
                    w_item.setData(Qt.UserRole, json.dumps(p_command))
                    self.ui.cmdTable.setItem(w_i, 1, w_item)
                    self.ui.cmdTable.resizeRowsToContents()
                    return True
                return False

        w_rowCnt = self.ui.cmdTable.rowCount()
        self.ui.cmdTable.setRowCount(w_rowCnt + 1)
        self.ui.cmdTable.setItem(w_rowCnt, 0, QTableWidgetItem(p_command['name']))
        w_text = json.dumps(p_command)
        w_item = QTableWidgetItem(w_text)
        w_item.setData(Qt.UserRole, json.dumps(p_command))
        self.ui.cmdTable.setItem(w_rowCnt, 1, w_item)
        self.ui.cmdTable.resizeRowsToContents()

    def slotNewCmd(self):
        w_cmdEditWnd = CommandEditWnd(None, self)
        if w_cmdEditWnd.exec_() == QDialog.Accepted:
            if self.importCommand(w_cmdEditWnd.m_command, False) == False:
                QMessageBox.critical(None, 'Error', 'Adding a new command was failed')
                return
            self.ui.cmdTable.selectRow(self.ui.cmdTable.rowCount() - 1)
            self.ui.cmdTable.setFocus()

    def slotEditCmd(self):
        w_modelIdxs = self.ui.cmdTable.selectionModel().selectedRows()
        if len(w_modelIdxs) == 0:
            return
        w_modelIdx = w_modelIdxs[0]
        w_jsonCommand = self.ui.cmdTable.item(w_modelIdx.row(), 1).data(Qt.UserRole)
        w_command = json.loads(w_jsonCommand)

        w_cmdEditWnd = CommandEditWnd(w_command, self)
        if w_cmdEditWnd.exec_() == QDialog.Accepted:
            self.importCommand(w_cmdEditWnd.m_command, True)
            self.ui.cmdTable.resizeRowsToContents()

    def slotDeleteCmd(self):
        w_modelIdxs = self.ui.cmdTable.selectionModel().selectedRows()
        i = 0
        for w_modelIdx in sorted(w_modelIdxs):
            self.ui.cmdTable.removeRow(w_modelIdx.row() - i)
            i = i + 1
        self.ui.cmdTable.setFocus()

    def slotOK(self):
        self.m_profile = {}
        self.m_profile['name'] = self.ui.profileNameEdit.text()
        w_commands = []

        w_commandCnt = self.ui.cmdTable.rowCount()
        for w_i in range(w_commandCnt):
            w_jsonCommand = self.ui.cmdTable.item(w_i, 1).data(Qt.UserRole)
            w_command = json.loads(w_jsonCommand)
            w_commands.append(w_command)
        self.m_profile['commands'] = w_commands

        super().accept()

    def slotCancel(self):
        super().reject()