File indexing completed on 2024-04-21 05:31:09

 /*
     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);
     }
 }
 
 }
 }