File indexing completed on 2024-05-19 05:16:11

 /*
     SPDX-FileCopyrightText: 2010 Stephen Kelly <steveire@gmail.com>
 
     SPDX-License-Identifier: LGPL-2.0-or-later
 */
 
 #include "entitytreecreatejob.h"
 
 #include <Akonadi/CollectionCreateJob>
 #include <Akonadi/ItemCreateJob>
 
 #include <QVariant>
 
 #include "migration_debug.h"
 
 using namespace Akonadi;
 
 static const char collectionIdMappingProperty[] = "collectionIdMappingProperty";
 
 EntityTreeCreateJob::EntityTreeCreateJob(const QList<Akonadi::Collection::List> &collections, const Akonadi::Item::List &items, QObject *parent)
     : Akonadi::TransactionSequence(parent)
     , m_collections(collections)
     , m_items(items)
 {
 }
 
 void EntityTreeCreateJob::doStart()
 {
     if (!m_collections.isEmpty()) {
         createNextLevelOfCollections();
     }
     createReadyItems();
 }
 
 void EntityTreeCreateJob::createNextLevelOfCollections()
 {
     CollectionCreateJob *job = nullptr;
 
     const Collection::List colList = m_collections.takeFirst();
     for (const Collection &collection : colList) {
         ++m_pendingJobs;
         job = new CollectionCreateJob(collection, this);
         job->setProperty(collectionIdMappingProperty, collection.id());
         connect(job, &CollectionCreateJob::result, this, &EntityTreeCreateJob::collectionCreateJobDone);
     }
 }
 
 void EntityTreeCreateJob::createReadyItems()
 {
     Item::List::iterator it;
     for (it = m_items.begin(); it != m_items.end();) {
         Collection parentCollection = (*it).parentCollection();
         if (parentCollection.isValid()) {
             (void)new ItemCreateJob(*it, parentCollection, this);
             it = m_items.erase(it);
         } else {
             ++it;
         }
     }
     if (m_items.isEmpty() && m_collections.isEmpty()) {
         commit();
     }
 }
 
 void EntityTreeCreateJob::collectionCreateJobDone(KJob *job)
 {
     Q_ASSERT(m_pendingJobs > 0);
     --m_pendingJobs;
     auto collectionCreateJob = qobject_cast<CollectionCreateJob *>(job);
     Collection createdCollection = collectionCreateJob->collection();
 
     if (job->error()) {
         qCDebug(MIGRATION_LOG) << job->errorString();
         return;
     }
 
     const Collection::Id creationId = job->property(collectionIdMappingProperty).toLongLong();
 
     Item::List::iterator it;
     const Item::List::iterator end = m_items.end();
     for (it = m_items.begin(); it != end; ++it) {
         qCDebug(MIGRATION_LOG) << "updating items";
         if (it->parentCollection().id() == creationId) {
             it->setParentCollection(createdCollection);
         }
     }
 
     createReadyItems();
 
     if (!m_collections.isEmpty()) {
         Collection::List::iterator col_it;
         const Collection::List::iterator col_end = m_collections[0].end();
         for (col_it = m_collections[0].begin(); col_it != col_end; ++col_it) {
             if (col_it->parentCollection().id() == creationId) {
                 col_it->setParentCollection(createdCollection);
             }
         }
         if (m_pendingJobs == 0) {
             createNextLevelOfCollections();
         }
     }
 
     if (m_items.isEmpty() && m_collections.isEmpty()) {
         commit();
     }
 }
 
 #include "moc_entitytreecreatejob.cpp"