console/calendarjanitor/backuper.cpp

   1 /*
   2   Copyright (c) 2013 Sérgio Martins <iamsergio@gmail.com>
   3
   4   This program is free software; you can redistribute it and/or modify
   5   it under the terms of the GNU General Public License as published by
   6   the Free Software Foundation; either version 2 of the License, or
   7   (at your option) any later version.
   8
   9   This program is distributed in the hope that it will be useful,
  10   but WITHOUT ANY WARRANTY; without even the implied warranty of
  11   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12   GNU General Public License for more details.
  13
  14   You should have received a copy of the GNU General Public License along
  15   with this program; if not, write to the Free Software Foundation, Inc.,
  16   51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  17
  18   As a special exception, permission is given to link this program
  19   with any edition of Qt, and distribute the resulting executable,
  20   without including the source code for Qt in the source distribution.
  21 */
  22
  23 #include "backuper.h"
  24
  25 #include <calendarsupport/utils.h>
  26
  27 #include <KCalCore/Incidence>
  28 #include <KCalCore/FileStorage>
  29
  30 #include <Akonadi/CollectionFetchJob>
  31 #include <Akonadi/CollectionFetchScope>
  32 #include <Akonadi/ItemFetchJob>
  33 #include <Akonadi/ItemFetchScope>
  34
  35 #include <KLocale>
  36 #include <KJob>
  37
  38 #include <QCoreApplication>
  39
  40 static void print(const QString &message)
  41 {
  42     QTextStream out(stdout);
  43     out << message << "\n";
  44 }
  45
  46 void Backuper::emitFinished(bool success, const QString &message)
  47 {
  48     if (success) {
  49         print(QLatin1Char('\n') + i18np("Backup was successful. %1 incidence was saved.", "Backup was successful. %1 incidences were saved.", m_calendar->incidences().count()));
  50     } else {
  51         print(message);
  52     }
  53
  54     m_calendar.clear();
  55
  56     emit finished(success, message);
  57     qApp->exit(success ? 0 : -1); // TODO: If we move this class to kdepimlibs, remove this
  58 }
  59
  60 Backuper::Backuper(QObject *parent) : QObject(parent), m_backupInProgress(false)
  61 {
  62 }
  63
  64 void Backuper::backup(const QString &filename, const QList<Akonadi::Entity::Id> &collectionIds)
  65 {
  66     if (filename.isEmpty()) {
  67         emitFinished(false, i18n("File is empty."));
  68         return;
  69     }
  70
  71     if (m_backupInProgress) {
  72         emitFinished(false, i18n("A backup is already in progress."));
  73         return;
  74     }
  75     print(i18n("Backing up your calendar data..."));
  76     m_calendar = KCalCore::MemoryCalendar::Ptr(new KCalCore::MemoryCalendar(KDateTime::LocalZone));
  77     m_requestedCollectionIds = collectionIds;
  78     m_backupInProgress = true;
  79     m_filename = filename;
  80
  81     Akonadi::CollectionFetchJob *job = new Akonadi::CollectionFetchJob(Akonadi::Collection::root(),
  82                                                                        Akonadi::CollectionFetchJob::Recursive);
  83
  84     job->fetchScope().setContentMimeTypes(KCalCore::Incidence::mimeTypes());
  85     connect(job, SIGNAL(result(KJob*)), SLOT(onCollectionsFetched(KJob*)));
  86     job->start();
  87 }
  88
  89 void Backuper::onCollectionsFetched(KJob *job)
  90 {
  91     if (job->error() == 0) {
  92         QSet<QString> mimeTypeSet = KCalCore::Incidence::mimeTypes().toSet();
  93         Akonadi::CollectionFetchJob *cfj = qobject_cast<Akonadi::CollectionFetchJob*>(job);
  94         foreach(const Akonadi::Collection &collection, cfj->collections()) {
  95             if (!m_requestedCollectionIds.isEmpty() && !m_requestedCollectionIds.contains(collection.id()))
  96                 continue;
  97             if (!mimeTypeSet.intersect(collection.contentMimeTypes().toSet()).isEmpty()) {
  98                 m_collections << collection;
  99                 loadCollection(collection);
 100             }
 101         }
 102
 103         if (m_collections.isEmpty()) {
 104             emitFinished(false, i18n("No data to backup."));
 105         }
 106     } else {
 107         kError() << job->errorString();
 108         m_backupInProgress = false;
 109         emitFinished(false, job->errorString());
 110     }
 111 }
 112
 113 void Backuper::loadCollection(const Akonadi::Collection &collection)
 114 {
 115     print(i18n("Processing collection %1 (id=%2)...", collection.displayName(), collection.id()));
 116     Akonadi::ItemFetchJob *ifj = new Akonadi::ItemFetchJob(collection, this);
 117     ifj->setProperty("collectionId", collection.id());
 118     ifj->fetchScope().fetchFullPayload(true);
 119     connect(ifj, SIGNAL(result(KJob*)), SLOT(onCollectionLoaded(KJob*)));
 120     m_pendingCollections << collection.id();
 121 }
 122
 123 void Backuper::onCollectionLoaded(KJob *job)
 124 {
 125     if (job->error()) {
 126         m_backupInProgress = false;
 127         m_calendar.clear();
 128         emitFinished(false, job->errorString());
 129     } else {
 130         Akonadi::ItemFetchJob *ifj = qobject_cast<Akonadi::ItemFetchJob *>(job);
 131         Akonadi::Collection::Id id = ifj->property("collectionId").toInt();
 132         Q_ASSERT(id != -1);
 133         Akonadi::Item::List items = ifj->items();
 134         m_pendingCollections.removeAll(id);
 135
 136         foreach (const Akonadi::Item &item, items) {
 137             KCalCore::Incidence::Ptr incidence = CalendarSupport::incidence(item);
 138             Q_ASSERT(incidence);
 139             m_calendar->addIncidence(incidence);
 140         }
 141
 142         if (m_pendingCollections.isEmpty()) { // We're done
 143             KCalCore::FileStorage storage(m_calendar, m_filename);
 144             bool success = storage.save();
 145             QString message = success ? QString() : i18n("An error occurred");
 146             emitFinished(success, message);
 147         }
 148     }
 149 }