latte-dock/app/view/tasksmodel.cpp

/*
    SPDX-FileCopyrightText: 2020 Michail Vourlakos <mvourlakos@gmail.com>
    SPDX-License-Identifier: GPL-2.0-or-later
*/

#include "tasksmodel.h"

// Qt
#include <QDebug>

// Plasma
#include <Plasma/Applet>
#include <PlasmaQuick/AppletQuickItem>

namespace Latte {
namespace ViewPart {

TasksModel::TasksModel(QObject *parent)
    : QAbstractListModel(parent)
{
}

int TasksModel::count() const
{
    return m_tasks.count();
}

int TasksModel::rowCount(const QModelIndex &parent) const
{
    return m_tasks.count();
}

QVariant TasksModel::data(const QModelIndex &index, int role) const
{
    bool rowIsValid = (index.row()>=0 && index.row()<m_tasks.count());
    if (!rowIsValid) {
        return QVariant();
    }

    if (role == Qt::UserRole) {
        return QVariant::fromValue(m_tasks[index.row()]);
    }

    return QVariant();
}


QHash<int, QByteArray> TasksModel::roleNames() const{
    QHash<int, QByteArray> roles;
    roles[Qt::UserRole] = "tasks";
    return roles;
}

void TasksModel::addTask(PlasmaQuick::AppletQuickItem *plasmoid)
{
    if (plasmoid && m_tasks.contains(plasmoid)) {
        return;
    }

    beginInsertRows(QModelIndex(), rowCount(), rowCount());
    m_tasks << plasmoid;
    endInsertRows();

    connect(plasmoid, &QObject::destroyed, this, [&, plasmoid](){
        removeTask(plasmoid);
    });

    connect(plasmoid->applet(), &Plasma::Applet::destroyedChanged, this, [&, plasmoid](const bool &destroyed){
        if (destroyed) {
            moveIntoWaitingTasks(plasmoid);
        } else {
            restoreFromWaitingTasks(plasmoid);
        }
    });

    emit countChanged();
}

void TasksModel::moveIntoWaitingTasks(PlasmaQuick::AppletQuickItem *plasmoid)
{
    if (plasmoid && !m_tasks.contains(plasmoid)) {
        return;
    }

    int tind = m_tasks.indexOf(plasmoid);

    if (tind >= 0) {
        beginRemoveRows(QModelIndex(), tind, tind);
        m_tasksWaiting << m_tasks.takeAt(tind);
        endRemoveRows();
        emit countChanged();
    }
}

void TasksModel::restoreFromWaitingTasks(PlasmaQuick::AppletQuickItem *plasmoid)
{
    if (plasmoid && !m_tasksWaiting.contains(plasmoid)) {
        return;
    }

    int tind = m_tasksWaiting.indexOf(plasmoid);

    if (tind >= 0) {
        beginInsertRows(QModelIndex(), rowCount(), rowCount());
        m_tasks << m_tasksWaiting.takeAt(tind);
        endInsertRows();
        emit countChanged();
    }
}

void TasksModel::removeTask(PlasmaQuick::AppletQuickItem *plasmoid)
{
    if (!plasmoid || (plasmoid && !m_tasks.contains(plasmoid) && !m_tasksWaiting.contains(plasmoid))) {
        return;
    }

    if (m_tasks.contains(plasmoid)) {
        int iex = m_tasks.indexOf(plasmoid);

        beginRemoveRows(QModelIndex(), iex, iex);
        m_tasks.removeAll(plasmoid);
        endRemoveRows();

        emit countChanged();
    } else if (m_tasksWaiting.contains(plasmoid)) {
        m_tasksWaiting.removeAll(plasmoid);
    }
}

}
}