gotify-tray-customized/gotify_tray/gui/widgets/MessageWidget.py

import os

from PyQt6 import QtCore, QtGui, QtWidgets

from ..models.MessagesModel import MessageItemDataRole, MessagesModelItem
from ..designs.widget_message import Ui_Form
from gotify_tray.database import Downloader
from gotify_tray.database import Settings
from gotify_tray.utils import convert_links, get_image
from gotify_tray.gui.themes import get_theme_file
from gotify_tray.gotify.models import GotifyMessageModel


settings = Settings("gotify-tray")


class MessageWidget(QtWidgets.QWidget, Ui_Form):
    deletion_requested = QtCore.pyqtSignal(MessagesModelItem)
    image_popup = QtCore.pyqtSignal(str, QtCore.QPoint)

    def __init__(
        self,
        app: QtWidgets.QApplication,
        parent: QtWidgets.QWidget,
        message_item: MessagesModelItem,
        icon: QtGui.QIcon = None,
    ):
        super(MessageWidget, self).__init__(parent)
        self.app = app
        self.parent = parent
        self.setupUi(self)
        self.setAutoFillBackground(True)
        self.message_item = message_item
        message: GotifyMessageModel = message_item.data(MessageItemDataRole.MessageRole)

        # Fonts
        self.set_fonts()

        # Display the message priority as a color
        self.set_priority_color(message.priority)

        # Display message contents
        self.label_title.setText(message.title)
        self.label_date.setText(message.date.strftime("%Y-%m-%d, %H:%M"))

        if markdown := (
            message.get("extras", {}).get("client::display", {}).get("contentType")
            == "text/markdown"
        ):
            self.label_message.setTextFormat(QtCore.Qt.TextFormat.MarkdownText)

        # If the message is only an image URL, then instead of showing the message,
        # download the image and show it in the message label
        if image_url := get_image(message.message):
            downloader = Downloader()
            filename = downloader.get_filename(image_url)
            self.set_message_image(filename)
        else:
            self.label_message.setText(convert_links(message.message))

        # Show the application icon
        if icon:
            image_size = settings.value("MessageWidget/image/size", type=int)
            pixmap = icon.pixmap(QtCore.QSize(image_size, image_size))
            self.label_image.setPixmap(pixmap)
        else:
            self.label_image.hide()

        # Set MessagesModelItem's size hint based on the size of this widget
        self.gridLayout_frame.setContentsMargins(0, 0, 5, 0)
        self.gridLayout.setContentsMargins(4, 5, 4, 0)
        self.adjustSize()
        size_hint = self.message_item.sizeHint()
        self.message_item.setSizeHint(QtCore.QSize(size_hint.width(), self.height()))

        self.set_icons()

        self.link_callbacks()

    def set_fonts(self):
        font_title = QtGui.QFont()
        font_date = QtGui.QFont()
        font_content = QtGui.QFont()

        if s := settings.value("MessageWidget/font/title", type=str):
            font_title.fromString(s)
        else:
            font_title.setBold(True)

        if s := settings.value("MessageWidget/font/date", type=str):
            font_date.fromString(s)
        else:
            font_date.setItalic(True)

        if s := settings.value("MessageWidget/font/message", type=str):
            font_content.fromString(s)

        self.label_title.setFont(font_title)
        self.label_date.setFont(font_date)
        self.label_message.setFont(font_content)

    def set_icons(self):
        self.pb_delete.setIcon(QtGui.QIcon(get_theme_file(self.app, "trashcan.svg")))
        self.pb_delete.setIconSize(QtCore.QSize(24, 24))

    def set_message_image(self, filename: str):
        pixmap = QtGui.QPixmap(filename)

        # Make sure the image fits within the listView
        W = settings.value("MessageWidget/content_image/W_percentage", type=float) * (
            self.parent.width() - self.label_image.width()
        )
        H = (
            settings.value("MessageWidget/content_image/H_percentage", type=float)
            * self.parent.height()
        )

        if pixmap.width() > W or pixmap.height() > H:
            pixmap = pixmap.scaled(
                QtCore.QSize(int(W), int(H)),
                aspectRatioMode=QtCore.Qt.AspectRatioMode.KeepAspectRatio,
                transformMode=QtCore.Qt.TransformationMode.SmoothTransformation,
            )

        self.label_message.setPixmap(pixmap)

    def set_priority_color(self, priority: int):
        if not settings.value("MessageWidget/priority_color", type=bool):
            self.label_priority.setFixedWidth(0) # set width to 0 instead of hiding, so we still get the content margins
            return

        if priority >= 4 and priority <= 7:
                self.label_priority.setStyleSheet("background-color: rgba(230, 126, 34, 0.7);")
        elif priority > 7:
            self.label_priority.setStyleSheet("background-color: #e74c3c;")

    def link_hovered_callback(self, link: str):
        if not settings.value("ImagePopup/enabled", type=bool):
            return

        qurl = QtCore.QUrl(link)
        _, ext = os.path.splitext(qurl.fileName())
        if ext in settings.value("ImagePopup/extensions", type=list):
            self.image_popup.emit(link, QtGui.QCursor.pos())

    def link_callbacks(self):
        self.pb_delete.clicked.connect(
            lambda: self.deletion_requested.emit(self.message_item)
        )
        self.label_message.linkHovered.connect(self.link_hovered_callback)