matrixion/src/TrayIcon.h

// SPDX-FileCopyrightText: 2017 Konstantinos Sideris <siderisk@auth.gr>
// SPDX-FileCopyrightText: 2021 Nheko Contributors
//
// SPDX-License-Identifier: GPL-3.0-or-later

#pragma once

#include <QIcon>
#include <QIconEngine>
#include <QRect>
#include <QSystemTrayIcon>

class QAction;
class QPainter;

class MsgCountComposedIcon : public QIconEngine
{
public:
        MsgCountComposedIcon(const QString &filename);

        void paint(QPainter *p, const QRect &rect, QIcon::Mode mode, QIcon::State state) override;
        QIconEngine *clone() const override;
        QList<QSize> availableSizes(QIcon::Mode mode, QIcon::State state) const override;
        QPixmap pixmap(const QSize &size, QIcon::Mode mode, QIcon::State state) override;

        int msgCount = 0;

private:
        const int BubbleDiameter = 17;

        QIcon icon_;
};

class TrayIcon : public QSystemTrayIcon
{
        Q_OBJECT
public:
        TrayIcon(const QString &filename, QWidget *parent);

public slots:
        void setUnreadCount(int count);

private:
        QAction *viewAction_;
        QAction *quitAction_;

        MsgCountComposedIcon *icon_;
};