pesterchum/toast.py

"""FIXME: not sure how this works exactly but it seems kinda broken!!"""
import os

# import time
import inspect
import logging

try:
    from PyQt6 import QtCore, QtGui, QtWidgets
except ImportError:
    print("PyQt5 fallback (toast.py)")
    from PyQt5 import QtCore, QtGui, QtWidgets

import ostools

_datadir = ostools.getDataDir()
PchumLog = logging.getLogger("pchumLogger")

# try:
#    import pynotify
# except:
#    pynotify = None

# Pynotify is broken.
pynotify = None


class DefaultToast:
    def __init__(self, machine, title, msg, icon, parent=None):
        self.machine = machine
        self.title = title
        self.msg = msg
        self.icon = icon

    def show(self):
        print(self.title, self.msg, self.icon)
        self.done()

    def done(self):
        t = self.machine.toasts[0]
        if t.title == self.title and t.msg == self.msg and t.icon == self.icon:
            self.machine.toasts.pop(0)
            self.machine.displaying = False
            PchumLog.info("Done")


class ToastMachine:
    class __Toast__:
        def __init__(self, machine, title, msg, time=3000, icon="", importance=0):
            self.machine = machine
            self.title = title
            self.msg = msg
            self.time = time
            if icon:
                icon = os.path.abspath(icon)
            self.icon = icon
            self.importance = importance
            if inspect.ismethod(self.title) or inspect.isfunction(self.title):
                self.title = self.title()

        def titleM(self, title=None):
            if title:
                self.title = title
                if inspect.ismethod(self.title) or inspect.isfunction(self.title):
                    self.title = self.title()
            else:
                return self.title

        def msgM(self, msg=None):
            if msg:
                self.msg = msg
            else:
                return self.msg

        def timeM(self, time=None):
            if time:
                self.time = time
            else:
                return self.time

        def iconM(self, icon=None):
            if icon:
                self.icon = icon
            else:
                return self.icon

        def importanceM(self, importance=None):
            if importance:
                self.importance = importance
            else:
                return self.importance

        def show(self):
            if self.machine.on:
                # Use libnotify's queue if using libnotify
                if self.machine.type in ("libnotify", "twmn"):
                    self.realShow()
                elif self.machine.toasts:
                    self.machine.toasts.append(self)
                else:
                    self.machine.toasts.append(self)
                    self.realShow()

        def realShow(self):
            self.machine.displaying = True
            t = None
            for k, v in self.machine.types.items():
                if self.machine.type == k:
                    try:
                        args = inspect.getfullargspec(v.__init__).args
                    except:
                        args = []

                    extras = {}
                    if "parent" in args:
                        extras["parent"] = self.machine.parent
                    if "time" in args:
                        extras["time"] = self.time
                    if k in ("libnotify", "twmn"):
                        t = v(self.title, self.msg, self.icon, **extras)
                    else:
                        t = v(self.machine, self.title, self.msg, self.icon, **extras)
                    # Use libnotify's urgency setting
                    if k == "libnotify":
                        if self.importance < 0:
                            t.set_urgency(pynotify.URGENCY_CRITICAL)
                        elif self.importance == 0:
                            t.set_urgency(pynotify.URGENCY_NORMAL)
                        elif self.importance > 0:
                            t.set_urgency(pynotify.URGENCY_LOW)
                    break
            if not t:
                if "default" in self.machine.types:
                    if (
                        "parent"
                        in inspect.getfullargspec(
                            self.machine.types["default"].__init__
                        ).args
                    ):
                        t = self.machine.types["default"](
                            self.machine,
                            self.title,
                            self.msg,
                            self.icon,
                            self.machine.parent,
                        )
                    else:
                        t = self.machine.types["default"](
                            self.machine, self.title, self.msg, self.icon
                        )
                else:
                    t = DefaultToast(self.machine, self.title, self.msg, self.icon)
            t.show()

    def __init__(
        self,
        parent,
        name,
        on=True,
        type="default",
        types=(
            {"default": DefaultToast, "libnotify": pynotify.Notification}
            if pynotify
            else {"default": DefaultToast}
        ),
        extras={},
    ):
        self.parent = parent
        self.name = name
        self.on = on
        types.update(extras)
        self.types = types
        self.type = "default"
        self.quit = False
        self.displaying = False

        self.setCurrentType(type)

        self.toasts = []

    def Toast(self, title, msg, icon="", time=3000):
        return self.__Toast__(self, title, msg, time=time, icon=icon)

    def setEnabled(self, on):
        self.on = on is True

    def currentType(self):
        return self.type

    def availableTypes(self):
        return sorted(self.types.keys())

    def setCurrentType(self, type):
        if type in self.types:
            if type == "libnotify":
                if not pynotify or not pynotify.init("ToastMachine"):
                    PchumLog.info("Problem initilizing pynotify")
                    return
                    # self.type = type = "default"
            elif type == "twmn":
                import pytwmn

                try:
                    pytwmn.init()
                except Exception:
                    PchumLog.exception("Problem initilizing pytwmn.")
                    return
                    # self.type = type = "default"
            self.type = type

    def appName(self):
        if inspect.ismethod(self.name) or inspect.isfunction(self.name):
            return self.name()
        else:
            return self.name

    def showNext(self):
        if not self.displaying and self.toasts:
            self.toasts.sort(key=lambda x: x.importance)
            self.toasts[0].realShow()

    def showAll(self):
        while self.toasts:
            self.showNext()

    def run(self):
        while not self.quit:
            if self.on and self.toasts:
                self.showNext()


class PesterToast(QtWidgets.QWidget, DefaultToast):
    def __init__(self, machine, title, msg, icon, time=3000, parent=None):
        # FIXME: Not sure how this works exactly either xd, can't we init the parents seperately?
        kwds = {
            "parent": parent,
            "machine": machine,
            "title": title,
            "msg": msg,
            "icon": icon,
        }
        super().__init__(**kwds)

        self.machine = machine
        self.time = time

        if ostools.isWin32():
            self.setWindowFlags(QtCore.Qt.WindowType.ToolTip)
        else:
            self.setWindowFlags(
                QtCore.Qt.WindowType.WindowStaysOnTopHint
                | QtCore.Qt.WindowType.X11BypassWindowManagerHint
                | QtCore.Qt.WindowType.ToolTip
            )

        self.m_animation = QtCore.QParallelAnimationGroup()
        anim = QtCore.QPropertyAnimation(self)
        anim.setTargetObject(self)
        self.m_animation.addAnimation(anim)
        anim.setEasingCurve(QtCore.QEasingCurve.Type.OutBounce)
        anim.setDuration(1000)
        anim.finished.connect(self.reverseTrigger)

        self.m_animation.setDirection(QtCore.QAbstractAnimation.Direction.Forward)

        self.title = QtWidgets.QLabel(title, self)
        self.msg = QtWidgets.QLabel(msg, self)
        self.content = msg
        if icon:
            self.icon = QtWidgets.QLabel("")
            iconPixmap = QtGui.QPixmap(icon).scaledToWidth(30)
            self.icon.setPixmap(iconPixmap)
        # else:
        #    self.icon.setPixmap(QtGui.QPixmap(30, 30))
        #    self.icon.pixmap().fill(QtGui.QColor(0,0,0,0))

        layout_0 = QtWidgets.QVBoxLayout()
        layout_0.setContentsMargins(0, 0, 0, 0)

        if self.icon:
            layout_1 = QtWidgets.QGridLayout()
            layout_1.addWidget(self.icon, 0, 0, 1, 1)
            layout_1.addWidget(self.title, 0, 1, 1, 7)
            layout_1.setAlignment(self.msg, QtCore.Qt.AlignmentFlag.AlignTop)
            layout_0.addLayout(layout_1)
        else:
            layout_0.addWidget(self.title)
        layout_0.addWidget(self.msg)

        self.setMaximumWidth(self.parent().theme["toasts/width"])
        self.msg.setMaximumWidth(self.parent().theme["toasts/width"])
        self.title.setMinimumHeight(self.parent().theme["toasts/title/minimumheight"])

        self.setLayout(layout_0)

        self.setGeometry(
            0,
            0,
            self.parent().theme["toasts/width"],
            self.parent().theme["toasts/height"],
        )
        self.setStyleSheet(self.parent().theme["toasts/style"])
        self.title.setStyleSheet(self.parent().theme["toasts/title/style"])
        if self.icon:
            self.icon.setStyleSheet(self.parent().theme["toasts/icon/style"])
        self.msg.setStyleSheet(self.parent().theme["toasts/content/style"])
        self.layout().setSpacing(0)

        self.msg.setText(
            PesterToast.wrapText(
                self.msg.font(),
                self.msg.text(),
                self.parent().theme["toasts/width"],
                self.parent().theme["toasts/content/style"],
            )
        )

        screens = QtWidgets.QApplication.screens()
        screen = screens[0]  #  Should be the main one right???
        # This 100% doesn't work with multiple screens.
        p = screen.availableGeometry().bottomRight()
        o = screen.geometry().bottomRight()
        anim.setStartValue(p.y() - o.y())
        anim.setEndValue(100)
        anim.valueChanged[QtCore.QVariant].connect(self.updateBottomLeftAnimation)

        self.byebye = False

    @QtCore.pyqtSlot()
    def show(self):
        self.m_animation.start()

    @QtCore.pyqtSlot()
    def done(self):
        QtWidgets.QWidget.hide(self)
        t = self.machine.toasts[0]
        if t.title == self.title.text() and t.msg == self.content:
            self.machine.toasts.pop(0)
            self.machine.displaying = False
        if self.machine.on:
            self.machine.showNext()
        del self

    @QtCore.pyqtSlot()
    def reverseTrigger(self):
        if self.time >= 0:
            QtCore.QTimer.singleShot(self.time, self.reverseStart)

    @QtCore.pyqtSlot()
    def reverseStart(self):
        if not self.byebye:
            self.byebye = True
            anim = self.m_animation.animationAt(0)
            self.m_animation.setDirection(QtCore.QAbstractAnimation.Direction.Backward)
            anim.setEasingCurve(QtCore.QEasingCurve.Type.InCubic)
            anim.finished.disconnect(self.reverseTrigger)
            anim.finished.connect(self.done)
            self.m_animation.start()

    @QtCore.pyqtSlot(QtCore.QVariant)
    def updateBottomLeftAnimation(self, value):
        # p = QtWidgets.QApplication.desktop().availableGeometry(self).bottomRight()
        screens = QtWidgets.QApplication.screens()
        screen = screens[0]  # Main window?
        p = screen.availableGeometry().bottomRight()
        val = (self.height()) / 100
        # Does type casting this to an int have any negative consequences?
        self.move(int(p.x() - self.width()), int(p.y() - (value * val) + 1))
        self.layout().setSpacing(0)
        QtWidgets.QWidget.show(self)

    def mousePressEvent(self, event):
        if event.button() == QtCore.Qt.MouseButton.RightButton:
            self.reverseStart()
        elif event.button() == QtCore.Qt.MouseButton.LeftButton:
            pass

    @staticmethod
    def wrapText(font, text, maxwidth, css=""):
        ret = []
        metric = QtGui.QFontMetrics(font)
        if "padding" in css:
            if css[css.find("padding") + 7] != "-":
                colon = css.find(":", css.find("padding"))
                semicolon = css.find(";", css.find("padding"))
                if semicolon < 0:
                    stuff = css[colon + 1 :]
                else:
                    stuff = css[colon + 1 : semicolon]
                stuff = stuff.replace("px", "").lstrip().rstrip()
                stuff = stuff.split(" ")
                if len(stuff) == 1:
                    maxwidth -= int(stuff[0]) * 2
                elif len(stuff) == 2:
                    maxwidth -= int(stuff[1]) * 2
                elif len(stuff) == 3:
                    maxwidth -= int(stuff[1]) * 2
                elif len(stuff) == 4:
                    maxwidth -= int(stuff[1]) + int(stuff[3])
            else:
                if "padding-left" in css:
                    colon = css.find(":", css.find("padding-left"))
                    semicolon = css.find(";", css.find("padding-left"))
                    if semicolon < 0:
                        stuff = css[colon + 1 :]
                    else:
                        stuff = css[colon + 1 : semicolon]
                    stuff = stuff.replace("px", "").lstrip().rstrip()
                    if stuff.isdigit():
                        maxwidth -= int(stuff)
                if "padding-right" in css:
                    colon = css.find(":", css.find("padding-right"))
                    semicolon = css.find(";", css.find("padding-right"))
                    if semicolon < 0:
                        stuff = css[colon + 1 :]
                    else:
                        stuff = css[colon + 1 : semicolon]
                    stuff = stuff.replace("px", "").lstrip().rstrip()
                    if stuff.isdigit():
                        maxwidth -= int(stuff)

        if metric.horizontalAdvance(text) < maxwidth:
            return text
        while metric.horizontalAdvance(text) > maxwidth:
            lastspace = text.find(" ")
            curspace = lastspace
            while metric.horizontalAdvance(text, curspace) < maxwidth:
                lastspace = curspace
                curspace = text.find(" ", lastspace + 1)
                if curspace == -1:
                    break
            if (metric.horizontalAdvance(text[:lastspace]) > maxwidth) or len(
                text[:lastspace]
            ) < 1:
                for i in range(len(text)):
                    if metric.horizontalAdvance(text[:i]) > maxwidth:
                        lastspace = i - 1
                        break
            ret.append(text[:lastspace])
            text = text[lastspace + 1 :]
        ret.append(text)
        return "\n".join(ret)


class PesterToastMachine(ToastMachine, QtCore.QObject):
    def __init__(
        self,
        parent,
        name,
        on=True,
        type="default",
        types=(
            {"default": DefaultToast, "libnotify": pynotify.Notification}
            if pynotify
            else {"default": DefaultToast}
        ),
        extras={},
    ):
        ToastMachine.__init__(self, parent, name, on, type, types, extras)
        QtCore.QObject.__init__(self, parent)

    def setEnabled(self, on):
        oldon = self.on
        ToastMachine.setEnabled(self, on)
        if oldon != self.on:
            self.parent.config.set("notify", self.on)
            if self.on:
                self.timer.start()
            else:
                self.timer.stop()

    def setCurrentType(self, type):
        oldtype = self.type
        ToastMachine.setCurrentType(self, type)
        if oldtype != self.type:
            self.parent.config.set("notifyType", self.type)

    @QtCore.pyqtSlot()
    def showNext(self):
        ToastMachine.showNext(self)

    def run(self):
        pass
        # ~ self.timer = QtCore.QTimer(self)
        # ~ self.timer.setInterval(1000)
        # ~ self.connect(self.timer, QtCore.SIGNAL('timeout()'),
        # ~ self, QtCore.SLOT('showNext()'))
        # ~ if self.on:
        # ~     self.timer.start()