SelectionEncloser/ConfigurationPage/SelectionEncloserPage.py

# -*- coding: utf-8 -*-

# Copyright (c) 2014 - 2022 Detlev Offenbach <detlev@die-offenbachs.de>
#

"""
Module implementing Selection Encloser configuration page.
"""

import os

from PyQt6.QtCore import pyqtSlot, Qt
from PyQt6.QtWidgets import QTreeWidgetItem, QInputDialog, QLineEdit, QDialog

from eric7.EricWidgets.EricApplication import ericApp

from eric7.Preferences.ConfigurationPages.ConfigurationPageBase import ConfigurationPageBase
from .Ui_SelectionEncloserPage import Ui_SelectionEncloserPage

try:
    from eric7.EricGui import EricPixmapCache
except ImportError:
    from UI import PixmapCache as EricPixmapCache


class SelectionEncloserPage(ConfigurationPageBase, Ui_SelectionEncloserPage):
    """
    Class implementing Selection Encloser configuration page.
    """

    def __init__(self, plugin):
        """
        Constructor

        @param plugin reference to the plugin object
        @type SelectionEncloserPlugin
        """
        super().__init__()
        self.setupUi(self)
        self.setObjectName("SelectionEncloserPage")

        usesDarkPalette = ericApp().usesDarkPalette()
        iconSuffix = "dark" if usesDarkPalette else "light"

        self.editButton.setIcon(
            EricPixmapCache.getIcon(
                os.path.join(
                    "SelectionEncloser", "icons", "edit-{0}".format(iconSuffix)
                )
            )
        )
        self.addButton.setIcon(EricPixmapCache.getIcon("plus"))
        self.deleteButton.setIcon(EricPixmapCache.getIcon("minus"))
        self.upButton.setIcon(EricPixmapCache.getIcon("1uparrow"))
        self.downButton.setIcon(EricPixmapCache.getIcon("1downarrow"))
        self.addMenuButton.setIcon(
            EricPixmapCache.getIcon(
                os.path.join(
                    "SelectionEncloser", "icons", "topAdd-{0}".format(iconSuffix)
                )
            )
        )
        self.addSeparatorButton.setIcon(
            EricPixmapCache.getIcon(
                os.path.join(
                    "SelectionEncloser", "icons", "separatorAdd-{0}".format(iconSuffix)
                )
            )
        )

        self.editButton.setEnabled(False)
        self.addButton.setEnabled(False)
        self.addSeparatorButton.setEnabled(False)
        self.deleteButton.setEnabled(False)
        self.upButton.setEnabled(False)
        self.downButton.setEnabled(False)

        self.__plugin = plugin

        # set initial values
        hierarchy = self.__plugin.getPreferences("MenuHierarchy")
        for menuTitle, entries in hierarchy:
            if menuTitle == "--Separator--":
                menuTitle = self.tr("--Separator--")
            top = QTreeWidgetItem(self.menuTree, [menuTitle])
            for title, encString in entries:
                if title == "--Separator--":
                    title = self.tr("--Separator--")
                itm = QTreeWidgetItem(top, [title])
                itm.setData(0, Qt.ItemDataRole.UserRole, encString)
            top.setExpanded(True)

    def save(self):
        """
        Public slot to save the Selection Encloser configuration.
        """
        hierarchy = []
        for topIndex in range(self.menuTree.topLevelItemCount()):
            topItem = self.menuTree.topLevelItem(topIndex)
            menuTitle = topItem.text(0)
            if menuTitle == self.tr("--Separator--"):
                menuTitle = "--Separator--"
            topEntry = [menuTitle, []]
            for index in range(topItem.childCount()):
                itm = topItem.child(index)
                title = itm.text(0)
                if title == self.tr("--Separator--"):
                    title = "--Separator--"
                topEntry[1].append([title, itm.data(0, Qt.ItemDataRole.UserRole)])
            hierarchy.append(topEntry)
        self.__plugin.setPreferences("MenuHierarchy", hierarchy)

    @pyqtSlot()
    def on_addMenuButton_clicked(self):
        """
        Private slot to add a top level menu item.
        """
        menuTitle, ok = QInputDialog.getText(
            self,
            self.tr("Menu Title"),
            self.tr("Enter menu title:"),
            QLineEdit.EchoMode.Normal,
        )
        if ok and menuTitle:
            top = QTreeWidgetItem(self.menuTree, [menuTitle])
            top.setExpanded(True)

    @pyqtSlot()
    def on_addButton_clicked(self):
        """
        Private slot to add a menu entry.
        """
        from .SelectionEncloserEditDialog import SelectionEncloserEditDialog

        dlg = SelectionEncloserEditDialog(parent=self)
        if dlg.exec() == QDialog.DialogCode.Accepted:
            title, encString = dlg.getData()
            itm = QTreeWidgetItem(self.menuTree.selectedItems()[0], [title])
            itm.setData(0, Qt.ItemDataRole.UserRole, encString)

    @pyqtSlot()
    def on_addSeparatorButton_clicked(self):
        """
        Private slot to add a separator entry below the selected entry.
        """
        selItm = self.menuTree.selectedItems()[0]
        parent = selItm.parent()
        itm = QTreeWidgetItem([self.tr("--Separator--")])
        if parent is None:
            # top level item
            index = self.menuTree.indexOfTopLevelItem(selItm) + 1
            self.menuTree.insertTopLevelItem(index, itm)
        else:
            # sub item
            index = parent.indexOfChild(selItm) + 1
            parent.insertChild(index, itm)

    @pyqtSlot()
    def on_deleteButton_clicked(self):
        """
        Private slot to delete the selected entry.
        """
        itm = self.menuTree.selectedItems()[0]
        parent = itm.parent()
        if parent is None:
            index = self.menuTree.indexOfTopLevelItem(itm)
            self.menuTree.takeTopLevelItem(index)
        else:
            index = parent.indexOfChild(itm)
            parent.takeChild(index)
        del itm

    @pyqtSlot()
    def on_upButton_clicked(self):
        """
        Private slot to move an entry up.
        """
        self.__moveSelectedEntry(True)

    @pyqtSlot()
    def on_downButton_clicked(self):
        """
        Private slot to move an entry down.
        """
        self.__moveSelectedEntry(False)

    def __moveSelectedEntry(self, moveUp):
        """
        Private method to move the selected entry up or down.

        @param moveUp flag indicating to move the entry up
        @type bool
        """
        itm = self.menuTree.selectedItems()[0]
        parent = itm.parent()
        if parent is None:
            # top level item
            index = self.menuTree.indexOfTopLevelItem(itm)
            newIndex = index - 1 if moveUp else index + 1
            self.menuTree.takeTopLevelItem(index)
            self.menuTree.insertTopLevelItem(newIndex, itm)
            itm.setExpanded(True)
        else:
            # sub item
            index = parent.indexOfChild(itm)
            newIndex = index - 1 if moveUp else index + 1
            parent.takeChild(index)
            parent.insertChild(newIndex, itm)
        for sitm in self.menuTree.selectedItems():
            sitm.setSelected(False)
        itm.setSelected(True)

    @pyqtSlot()
    def on_editButton_clicked(self):
        """
        Private slot to edit the selected entry.
        """
        itm = self.menuTree.selectedItems()[0]
        parent = itm.parent()
        if parent is None:
            menuTitle, ok = QInputDialog.getText(
                self,
                self.tr("Menu Entry"),
                self.tr("Enter menu entry text:"),
                QLineEdit.EchoMode.Normal,
                itm.text(0),
            )
            if ok and menuTitle:
                itm.setText(0, menuTitle)
        else:
            from .SelectionEncloserEditDialog import SelectionEncloserEditDialog

            dlg = SelectionEncloserEditDialog(
                itm.text(0), itm.data(0, Qt.ItemDataRole.UserRole), self
            )
            if dlg.exec() == QDialog.DialogCode.Accepted:
                title, encString = dlg.getData()
                itm.setText(0, title)
                itm.setData(0, Qt.ItemDataRole.UserRole, encString)

    @pyqtSlot()
    def on_menuTree_itemSelectionChanged(self):
        """
        Private slot handling the selection of an item.
        """
        if len(self.menuTree.selectedItems()) == 0:
            self.addButton.setEnabled(False)
            self.addSeparatorButton.setEnabled(False)
            self.deleteButton.setEnabled(False)
            self.upButton.setEnabled(False)
            self.downButton.setEnabled(False)
            self.editButton.setEnabled(False)
        else:
            addEnable = True
            upEnable = True
            downEnable = True
            itm = self.menuTree.selectedItems()[0]
            parent = itm.parent()
            if parent is None:
                # top level item
                if self.menuTree.indexOfTopLevelItem(itm) == 0:
                    upEnable = False
                if (
                    self.menuTree.indexOfTopLevelItem(itm)
                    == self.menuTree.topLevelItemCount() - 1
                ):
                    downEnable = False
            else:
                # sub item
                if parent.indexOfChild(itm) == 0:
                    upEnable = False
                if parent.indexOfChild(itm) == parent.childCount() - 1:
                    downEnable = False
                addEnable = False
            self.addButton.setEnabled(addEnable)
            self.addSeparatorButton.setEnabled(True)
            self.deleteButton.setEnabled(True)
            self.upButton.setEnabled(upEnable)
            self.downButton.setEnabled(downEnable)
            self.editButton.setEnabled(itm.text(0) != self.tr("--Separator--"))