plasma-workspace/libnotificationmanager/notifications.cpp

/*
 * Copyright 2019 Kai Uwe Broulik <kde@privat.broulik.de>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) version 3, or any
 * later version accepted by the membership of KDE e.V. (or its
 * successor approved by the membership of KDE e.V.), which shall
 * act as a proxy defined in Section 6 of version 3 of the license.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "notifications.h"

#include <KConcatenateRowsProxyModel>

#include "notificationmodel.h"
#include "notificationfilterproxymodel_p.h"
#include "notificationsortproxymodel_p.h"
#include "limitedrowcountproxymodel_p.h"

#include "jobsmodel.h"

#include "notification.h"

#include <QDebug>

using namespace NotificationManager;

class Q_DECL_HIDDEN Notifications::Private
{
public:
    explicit Private(Notifications *q);
    ~Private();

    void initModels();
    void updateCount();

    bool showJobs = false;

    Notifications::GroupMode groupMode = Notifications::GroupDisabled;

    int activeNotificationsCount = 0;
    int expiredNotificationsCount = 0;
    int activeJobsCount = 0;
    int jobsPercentage = 0;

    static uint notificationId(const QModelIndex &idx);
    static QString jobId(const QModelIndex &idx);

    // FIXME can we just use sharedptr?
    static int instanceCount;

    static NotificationModel *notificationsModel;

    static int jobsInstanceCount;
    static JobsModel *jobsModel;

    KConcatenateRowsProxyModel *notificationsAndJobsModel = nullptr;

    NotificationFilterProxyModel *filterModel = nullptr;
    NotificationSortProxyModel *sortModel = nullptr;
    LimitedRowCountProxyModel *limiterModel = nullptr;

private:
    Notifications *q;
};

int Notifications::Private::instanceCount = 0;
NotificationModel *Notifications::Private::notificationsModel = nullptr;

int Notifications::Private::jobsInstanceCount = 0;
JobsModel *Notifications::Private::jobsModel;

Notifications::Private::Private(Notifications *q)
    : q(q)
{
    ++instanceCount;
}

Notifications::Private::~Private()
{
    --instanceCount;

    if (!instanceCount) {
        delete notificationsModel;
        notificationsModel = nullptr;
    }

    --jobsInstanceCount;

    if (!jobsInstanceCount) {
        delete jobsModel;
        jobsModel = nullptr;
    }
}

void Notifications::Private::initModels()
{
    if (!notificationsModel) {
        notificationsModel = new NotificationModel();
    }

    // FIXME FIXME FIXME all of this leaks

    notificationsAndJobsModel = new KConcatenateRowsProxyModel();
    notificationsAndJobsModel->addSourceModel(notificationsModel);

    filterModel = new NotificationFilterProxyModel();
    connect(filterModel, &NotificationFilterProxyModel::urgenciesChanged, q, &Notifications::urgenciesChanged);
    connect(filterModel, &NotificationFilterProxyModel::showExpiredChanged, q, &Notifications::showExpiredChanged);
    connect(filterModel, &NotificationFilterProxyModel::showDismissedChanged, q, &Notifications::showDismissedChanged);

    filterModel->setSourceModel(notificationsAndJobsModel);

    sortModel = new NotificationSortProxyModel();
    sortModel->setSourceModel(filterModel);

    limiterModel = new LimitedRowCountProxyModel();
    connect(limiterModel, &LimitedRowCountProxyModel::limitChanged, q, &Notifications::limitChanged);
    limiterModel->setSourceModel(sortModel);

    q->setSourceModel(limiterModel);
}

void Notifications::Private::updateCount()
{
    int active = 0;
    int expired = 0;

    int jobs = 0;
    int totalPercentage = 0;

    for (int i = 0; i < q->rowCount(); ++i) {
        const QModelIndex idx = q->index(i, 0);

        // FIXME we don't show low urgency notifications in the popup
        // as such they never expire and keep the tray icon glowing indefinitely
        const bool isExpired = idx.data(Notifications::ExpiredRole).toBool();

        if (isExpired) {
            ++expired;
        }

        if (idx.data(Notifications::TypeRole) == Notifications::JobType) {
            if (idx.data(Notifications::JobStateRole) != Notifications::JobStateStopped) {
                ++jobs;

                totalPercentage += idx.data(Notifications::PercentageRole).toInt();
            }
        } else if (!isExpired) {
            ++active;
        }
    }

    if (activeNotificationsCount != active) {
        activeNotificationsCount = active;
        emit q->activeNotificationsCountChanged();
    }
    if (expiredNotificationsCount != expired) {
        expiredNotificationsCount = expired;
        emit q->expiredNotificationsCountChanged();
    }
    if (activeJobsCount != jobs) {
        activeJobsCount = jobs;
        emit q->activeJobsCountChanged();
    }

    const int percentage = (jobs > 0 ? totalPercentage / jobs : 0);
    if (jobsPercentage != percentage) {
        jobsPercentage = percentage;
        emit q->jobsPercentageChanged();
    }

    // TODO don't emit in dataChanged
    emit q->countChanged();
}

uint Notifications::Private::notificationId(const QModelIndex &idx)
{
    return idx.data(Notifications::IdRole).toUInt();
}

QString Notifications::Private::jobId(const QModelIndex &idx)
{
    return idx.data(Notifications::IdRole).toString();
}

Notifications::Notifications(QObject *parent)
    : QSortFilterProxyModel(parent)
    , d(new Private(this))
{
    d->initModels();

    connect(this, &QAbstractItemModel::rowsInserted, this, [this] {
        d->updateCount();
    });
    connect(this, &QAbstractItemModel::rowsRemoved, this, [this] {
        d->updateCount();
    });
    connect(this, &QAbstractItemModel::dataChanged, this, [this](const QModelIndex &topLeft, const QModelIndex &bottomRight, const QVector<int> &roles) {
        Q_UNUSED(topLeft);
        Q_UNUSED(bottomRight);
        if (roles.isEmpty()
                || roles.contains(Notifications::ExpiredRole)
                || roles.contains(Notifications::JobStateRole)
                || roles.contains(Notifications::PercentageRole)) {
            d->updateCount();
        }
    });
}

Notifications::~Notifications() = default;

// TODO why are we QQmlParserStatus if we don't use it?
void Notifications::classBegin()
{

}

void Notifications::componentComplete()
{

}

int Notifications::limit() const
{
    return d->limiterModel->limit();
}

void Notifications::setLimit(int limit)
{
    d->limiterModel->setLimit(limit);
}

bool Notifications::showExpired() const
{
    return d->filterModel->showExpired();
}

void Notifications::setShowExpired(bool show)
{
    d->filterModel->setShowExpired(show);
}

bool Notifications::showDismissed() const
{
    return d->filterModel->showDismissed();
}

void Notifications::setShowDismissed(bool show)
{
    d->filterModel->setShowDismissed(show);
}

bool Notifications::showSuppressed() const
{
    // TODO
    return true;
}

void Notifications::setShowSuppressed(bool show)
{
    // TODO
    Q_UNUSED(show);
}

bool Notifications::showJobs() const
{
    return d->showJobs;
}

void Notifications::setShowJobs(bool showJobs)
{
    if (d->showJobs == showJobs) {
        return;
    }

    if (showJobs) {
        ++d->jobsInstanceCount;
        if (!d->jobsModel) {
            d->jobsModel = new JobsModel();
        }
        d->notificationsAndJobsModel->addSourceModel(d->jobsModel);
    } else {
        d->notificationsAndJobsModel->removeSourceModel(d->jobsModel);
        --d->jobsInstanceCount;
        if (!d->jobsInstanceCount) {
            delete d->jobsModel;
            d->jobsModel = nullptr;
        }
    }
}

Notifications::Urgencies Notifications::urgencies() const
{
    return d->filterModel->urgencies();
}

void Notifications::setUrgencies(Urgencies urgencies)
{
    if (!urgencies) {
        // FIXME categorized logging
        qWarning("No urgencies set, no notifications will be added.");
    }
    d->filterModel->setUrgencies(urgencies);
}

Notifications::GroupMode Notifications::groupMode() const
{
    return d->groupMode;
}

void Notifications::setGroupMode(GroupMode groupMode)
{
    if (d->groupMode) {
        d->groupMode = groupMode;
        invalidate();
        emit groupModeChanged();
    }
}

int Notifications::count() const
{
    return rowCount(QModelIndex());
}

int Notifications::activeNotificationsCount() const
{
    return d->activeNotificationsCount;
}

int Notifications::expiredNotificationsCount() const
{
    return d->expiredNotificationsCount;
}

int Notifications::activeJobsCount() const
{
    return d->activeJobsCount;
}

int Notifications::jobsPercentage() const
{
    return d->jobsPercentage;
}

QModelIndex Notifications::makeModelIndex(int row, int childRow) const
{
    if (row < 0 || row >= rowCount()) {
        return QModelIndex();
    }

    if (childRow == -1) {
        return index(row, 0);
    } else {
        const QModelIndex &parent = index(row, 0);

        if (childRow < rowCount(parent)) {
            return parent.child(childRow, 0);
        }
    }

    return QModelIndex();
}

void Notifications::expire(const QModelIndex &idx)
{
    // TODO we could just call the NotificationServer directly?
    // FIXME just pass QModelIndex around

    switch (static_cast<Notifications::Type>(idx.data(Notifications::TypeRole).toInt())) {
    case Notifications::NotificationType:
        d->notificationsModel->expire(Private::notificationId(idx));
        break;
    case Notifications::JobType:
        d->jobsModel->expire(Private::jobId(idx));
        break;
    default:
        Q_UNREACHABLE();
    }
}

void Notifications::dismiss(const QModelIndex &idx)
{
    if (idx.data(Notifications::TypeRole).toInt() != Notifications::JobType) {
        qWarning() << "Can only dismiss jobs";
        return;
    }

    d->jobsModel->dismiss(Private::jobId(idx));
}

void Notifications::close(const QModelIndex &idx)
{
    switch (static_cast<Notifications::Type>(idx.data(Notifications::TypeRole).toInt())) {
    case Notifications::NotificationType:
        d->notificationsModel->close(Private::notificationId(idx));
        break;
    case Notifications::JobType:
        d->jobsModel->close(Private::jobId(idx));
        break;
    default:
        Q_UNREACHABLE();
    }
}

void Notifications::configure(const QModelIndex &idx)
{
    d->notificationsModel->configure(Private::notificationId(idx));
}

void Notifications::invokeDefaultAction(const QModelIndex &idx)
{
    d->notificationsModel->invokeDefaultAction(Private::notificationId(idx));
}

void Notifications::invokeAction(const QModelIndex &idx, const QString &actionId)
{
    d->notificationsModel->invoke(Private::notificationId(idx), actionId);
}

void Notifications::suspendJob(const QModelIndex &idx)
{
    d->jobsModel->suspend(Private::jobId(idx));
}

void Notifications::resumeJob(const QModelIndex &idx)
{
    d->jobsModel->resume(Private::jobId(idx));
}

void Notifications::killJob(const QModelIndex &idx)
{
    d->jobsModel->kill(Private::jobId(idx));
}

QVariant Notifications::data(const QModelIndex &index, int role) const
{
    return QSortFilterProxyModel::data(index, role);
}

bool Notifications::setData(const QModelIndex &index, const QVariant &value, int role)
{
    return QSortFilterProxyModel::setData(index, value, role);
}

bool Notifications::filterAcceptsRow(int source_row, const QModelIndex &source_parent) const
{
    return QSortFilterProxyModel::filterAcceptsRow(source_row, source_parent);
}

bool Notifications::lessThan(const QModelIndex &source_left, const QModelIndex &source_right) const
{
    return QSortFilterProxyModel::lessThan(source_left, source_right);
}

int Notifications::rowCount(const QModelIndex &parent) const
{
    return QSortFilterProxyModel::rowCount(parent);
}

QHash<int, QByteArray> Notifications::roleNames() const
{
    return QHash<int, QByteArray> {
        {IdRole, QByteArrayLiteral("notificationId")}, // id is QML-reserved
        {TypeRole, QByteArrayLiteral("type")},
        {CreatedRole, QByteArrayLiteral("created")},
        {UpdatedRole, QByteArrayLiteral("updated")},
        {SummaryRole, QByteArrayLiteral("summary")},
        {BodyRole, QByteArrayLiteral("body")},
        {IconNameRole, QByteArrayLiteral("iconName")},
        {ImageRole, QByteArrayLiteral("image")},
        {ApplicationNameRole, QByteArrayLiteral("applicationName")},
        {ApplicationIconNameRole, QByteArrayLiteral("applicationIconName")},

        {ActionNamesRole, QByteArrayLiteral("actionNames")},
        {ActionLabelsRole, QByteArrayLiteral("actionLabels")},
        {HasDefaultActionRole, QByteArrayLiteral("hasDefaultAction")},

        {UrlsRole, QByteArrayLiteral("urls")},
        {UrgencyRole, QByteArrayLiteral("urgency")},
        {TimeoutRole, QByteArrayLiteral("timeout")},

        {ConfigurableRole, QByteArrayLiteral("configurable")},
        {ConfigureActionLabelRole, QByteArrayLiteral("configureActionLabel")},

        {JobStateRole, QByteArrayLiteral("jobState")},
        {PercentageRole, QByteArrayLiteral("percentage")},
        {ErrorRole, QByteArrayLiteral("error")},
        {ErrorTextRole, QByteArrayLiteral("errorText")},
        {SuspendableRole, QByteArrayLiteral("suspendable")},
        {KillableRole, QByteArrayLiteral("killable")},
        {JobDetailsRole, QByteArrayLiteral("jobDetails")},

        {ExpiredRole, QByteArrayLiteral("expired")}
    };
}