File indexing completed on 2024-06-02 05:33:01

 /*
  * SPDX-FileCopyrightText: 2014 Kevin Ottens <ervin@kde.org>
  * SPDX-License-Identifier: GPL-2.0-only OR GPL-3.0-only OR LicenseRef-KDE-Accepted-GPL
  */
 
 
 #include "projectpagemodel.h"
 
 #include <QMimeData>
 
 #include <KLocalizedString>
 
 #include "presentation/querytreemodel.h"
 
 using namespace Presentation;
 
 ProjectPageModel::ProjectPageModel(const Domain::Project::Ptr &project,
                                    const Domain::ProjectQueries::Ptr &projectQueries,
                                    const Domain::ProjectRepository::Ptr &projectRepository,
                                    const Domain::TaskQueries::Ptr &taskQueries,
                                    const Domain::TaskRepository::Ptr &taskRepository,
                                    QObject *parent)
     : PageModel(parent),
       m_projectQueries(projectQueries),
       m_projectRepository(projectRepository),
       m_project(project),
       m_taskQueries(taskQueries),
       m_taskRepository(taskRepository)
 {
 }
 
 Domain::Project::Ptr ProjectPageModel::project() const
 {
     return m_project;
 }
 
 Domain::Task::Ptr ProjectPageModel::addItem(const QString &title, const QModelIndex &parentIndex)
 {
     const auto parentData = parentIndex.data(QueryTreeModelBase::ObjectRole);
     const auto parentTask = parentData.value<Domain::Task::Ptr>();
 
     auto task = Domain::Task::Ptr::create();
     task->setTitle(title);
 
     const auto job = parentTask ? m_taskRepository->createChild(task, parentTask)
                    : m_taskRepository->createInProject(task, m_project);
     installHandler(job, i18n("Cannot add task %1 in project %2", title, m_project->name()));
 
     return task;
 }
 
 void ProjectPageModel::removeItem(const QModelIndex &index)
 {
     QVariant data = index.data(QueryTreeModelBase::ObjectRole);
     auto task = data.value<Domain::Task::Ptr>();
     Q_ASSERT(task);
     const auto job = m_taskRepository->remove(task);
     installHandler(job, i18n("Cannot remove task %1 from project %2", task->title(), m_project->name()));
 }
 
 void ProjectPageModel::promoteItem(const QModelIndex &index)
 {
     QVariant data = index.data(QueryTreeModelBase::ObjectRole);
     auto task = data.value<Domain::Task::Ptr>();
     Q_ASSERT(task);
     const auto job = m_taskRepository->promoteToProject(task);
     installHandler(job, i18n("Cannot promote task %1 to be a project", task->title()));
 }
 
 QAbstractItemModel *ProjectPageModel::createCentralListModel()
 {
     auto query = [this](const Domain::Task::Ptr &task) -> Domain::QueryResultInterface<Domain::Task::Ptr>::Ptr {
         if (!task)
             return m_projectQueries->findTopLevel(m_project);
         else
             return m_taskQueries->findChildren(task);
     };
 
     auto flags = [](const Domain::Task::Ptr &) {
         return Qt::ItemIsSelectable
              | Qt::ItemIsEnabled
              | Qt::ItemIsEditable
              | Qt::ItemIsDragEnabled
              | Qt::ItemIsUserCheckable
              | Qt::ItemIsDropEnabled;
     };
 
     auto data = [](const Domain::Task::Ptr &task, int role, int) -> QVariant {
         return defaultTaskData(task, role, TaskExtraDataPtr());
     };
 
     auto setData = [this](const Domain::Task::Ptr &task, const QVariant &value, int role) {
         if (role != Qt::EditRole && role != Qt::CheckStateRole) {
             return false;
         }
 
         const auto currentTitle = task->title();
         if (role == Qt::EditRole)
             task->setTitle(value.toString());
         else
             task->setDone(value.toInt() == Qt::Checked);
 
         const auto job = m_taskRepository->update(task);
         installHandler(job, i18n("Cannot modify task %1 in project %2", currentTitle, m_project->name()));
         return true;
     };
 
     auto drop = [this](const QMimeData *mimeData, Qt::DropAction, const Domain::Task::Ptr &parentTask) {
         if (!mimeData->hasFormat(QStringLiteral("application/x-zanshin-object")))
             return false;
 
         auto droppedTasks = mimeData->property("objects").value<Domain::Task::List>();
         if (droppedTasks.isEmpty())
             return false;
 
         using namespace std::placeholders;
         auto associate = std::function<KJob*(Domain::Task::Ptr)>();
         auto parentTitle = QString();
 
         if (parentTask) {
             associate = std::bind(&Domain::TaskRepository::associate, m_taskRepository, parentTask, _1);
             parentTitle = parentTask->title();
         } else {
             associate = std::bind(&Domain::ProjectRepository::associate, m_projectRepository, m_project, _1);
             parentTitle = m_project->name();
         }
 
         foreach(const Domain::Task::Ptr &childTask, droppedTasks) {
             const auto job = associate(childTask);
             installHandler(job, i18n("Cannot move task %1 as a sub-task of %2", childTask->title(), parentTitle));
         }
 
         return true;
     };
 
     auto drag = [](const Domain::Task::List &tasks) -> QMimeData* {
         if (tasks.isEmpty())
             return nullptr;
 
         auto data = new QMimeData;
         data->setData(QStringLiteral("application/x-zanshin-object"), "object");
         data->setProperty("objects", QVariant::fromValue(tasks));
         return data;
     };
 
     return new QueryTreeModel<Domain::Task::Ptr>(query, flags, data, setData, drop, drag, nullptr, this);
 }
 
 #include "moc_projectpagemodel.cpp"