a better main window

gotify_tray/gui/widgets/MainWindow.py

@@ -0,0 +1,154 @@
+from PyQt6 import QtCore, QtGui, QtWidgets
+from ..designs.widget_main import Ui_MainWindow
+from .StatusWidget import StatusWidget
+from ..models import (
+    ApplicationModel,
+    MessagesModel,
+    MessagesModelItem,
+)
+from . import MessageWidget
+
+from gotify_tray.__version__ import __title__
+from gotify_tray.database import Settings
+from gotify_tray.utils import get_abs_path
+
+
+settings = Settings("gotify-tray")
+
+
+class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
+    refresh = QtCore.pyqtSignal()
+    delete_all = QtCore.pyqtSignal(QtGui.QStandardItem)
+    delete_message = QtCore.pyqtSignal(MessagesModelItem)
+    application_selection_changed = QtCore.pyqtSignal(QtGui.QStandardItem)
+
+    def __init__(
+        self, application_model: ApplicationModel, messages_model: MessagesModel
+    ):
+        super(MainWindow, self).__init__()
+        self.setupUi(self)
+
+        self.setWindowTitle(__title__)
+
+        self.application_model = application_model
+        self.messages_model = messages_model
+
+        self.listView_applications.setModel(application_model)
+        self.listView_messages.setModel(messages_model)
+
+        # Do not expand the applications listview when resizing
+        self.splitter.setStretchFactor(0, 0)
+        self.splitter.setStretchFactor(1, 1)
+
+        self.status_widget = StatusWidget()
+        self.horizontalLayout.insertWidget(0, self.status_widget)
+
+        # Set button icons
+        self.pb_refresh.setIcon(
+            QtGui.QIcon(get_abs_path(f"gotify_tray/gui/images/refresh.svg"))
+        )
+        self.pb_delete_all.setIcon(
+            QtGui.QIcon(get_abs_path(f"gotify_tray/gui/images/trashcan.svg"))
+        )
+
+        # Resize the labels and icons
+        size = settings.value("MainWindow/label/size", type=int)
+        self.status_widget.setFixedSize(QtCore.QSize(size, size))
+
+        size = settings.value("MainWindow/button/size", type=int)
+        self.pb_refresh.setFixedSize(QtCore.QSize(size, size))
+        self.pb_delete_all.setFixedSize(QtCore.QSize(size, size))
+        self.pb_refresh.setIconSize(QtCore.QSize(0.7 * size, 0.7 * size))
+        self.pb_delete_all.setIconSize(QtCore.QSize(0.9 * size, 0.9 * size))
+
+        size = settings.value("MainWindow/application/icon/size", type=int)
+        self.listView_applications.setIconSize(QtCore.QSize(size, size))
+
+        font_title = QtGui.QFont()
+        font_title.fromString(settings.value("MainWindow/font/application", type=str))
+        self.label_application.setFont(font_title)
+
+        # Set tooltips
+        self.pb_refresh.setToolTip("Refresh")
+        self.pb_delete_all.setToolTip("Delete all messages")
+
+        self.restore_state()
+
+        self.link_callbacks()
+
+    def set_active(self):
+        self.status_widget.set_active()
+
+    def set_connecting(self):
+        self.status_widget.set_connecting()
+
+    def set_inactive(self):
+        self.status_widget.set_inactive()
+
+    def set_error(self):
+        self.status_widget.set_error()
+
+    def insert_message_widget(
+        self, message_item: MessagesModelItem, image_path: str = ""
+    ):
+        message_widget = MessageWidget(message_item, image_path=image_path)
+        self.listView_messages.setIndexWidget(
+            self.messages_model.indexFromItem(message_item), message_widget
+        )
+        message_widget.deletion_requested.connect(self.delete_message.emit)
+
+    def currentApplicationIndex(self) -> QtCore.QModelIndex:
+        return self.listView_applications.selectionModel().currentIndex()
+
+    def application_selection_changed_callback(
+        self, current: QtCore.QModelIndex, previous: QtCore.QModelIndex
+    ):
+        if item := self.application_model.itemFromIndex(current):
+            self.label_application.setText(item.text())
+            self.application_selection_changed.emit(item)
+
+    def delete_all_callback(self):
+        index = self.currentApplicationIndex()
+        if item := self.application_model.itemFromIndex(index):
+            self.delete_all.emit(item)
+
+    def disable_applications(self):
+        self.listView_applications.clearSelection()
+        self.listView_applications.setDisabled(True)
+
+    def enable_applications(self):
+        self.listView_applications.setEnabled(True)
+        self.listView_applications.setCurrentIndex(self.application_model.index(0, 0))
+
+    def bring_to_front(self):
+        self.ensurePolished()
+        self.setWindowState(
+            self.windowState() & ~QtCore.Qt.WindowState.WindowMinimized
+            | QtCore.Qt.WindowState.WindowActive
+        )
+        self.show()
+        self.activateWindow()
+
+    def link_callbacks(self):
+        self.pb_refresh.clicked.connect(self.refresh.emit)
+        self.pb_delete_all.clicked.connect(self.delete_all_callback)
+
+        self.listView_applications.selectionModel().currentChanged.connect(
+            self.application_selection_changed_callback
+        )
+
+    def store_state(self):
+        settings.setValue("MainWindow/geometry", self.saveGeometry())
+        settings.setValue("MainWindow/state", self.saveState())
+        settings.setValue("MainWindow/splitter", self.splitter.saveState())
+
+    def restore_state(self):
+        if geometry := settings.value("MainWindow/geometry", type=QtCore.QByteArray):
+            self.restoreGeometry(geometry)
+        if state := settings.value("MainWindow/state", type=QtCore.QByteArray):
+            self.restoreState(state)
+        if splitter := settings.value("MainWindow/splitter", type=QtCore.QByteArray):
+            self.splitter.restoreState(splitter)
+
+    def closeEvent(self, e: QtGui.QCloseEvent) -> None:
+        self.hide()